Home
last modified time | relevance | path

Searched refs:keepDimsTensor (Results 1 – 8 of 8) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dall_test.cpp53 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
56 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(int64_t)); in SaveParamsTensor()
57 m_allTensors.emplace_back(keepDimsTensor); in SaveParamsTensor()
174 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
177 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(float));
178 m_allTensors.emplace_back(keepDimsTensor);
209 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
211 m_allTensors.emplace_back(keepDimsTensor);
H A Dreduce_all_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
[all …]
H A Dreduce_l2_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
[all …]
H A Dreduce_max_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
[all …]
H A Dreduce_mean_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
[all …]
H A Dreduce_min_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
[all …]
H A Dreduce_prod_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
[all …]
H A Dreduce_sum_builder_test.cpp60 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(bool)); in SetKeepDims()
61 m_allTensors.emplace_back(keepDimsTensor); in SetKeepDims()
194 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, variable
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue));
198 m_allTensors.emplace_back(keepDimsTensor);
204 keepDimsTensor->SetBuffer(nullptr, 0);
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool));
269 m_allTensors.emplace_back(keepDimsTensor);
275 keepDimsTensor->SetBuffer(nullptr, 0);
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, variable
[all …]