Home
last modified time | relevance | path

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

/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, in HWTEST_F() local
177 keepDimsTensor->SetBuffer(keepDimsValue, sizeof(float)); in HWTEST_F()
178 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
209 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
211 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
H A Dreduce_mean_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_prod_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_min_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_max_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_sum_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_all_builder_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]
H A Dreduce_l2_test.cpp57 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetKeepDims() local
60 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, in HWTEST_F() local
197 keepDimsTensor->SetBuffer(&keepDimsValue, sizeof(keepDimsValue)); in HWTEST_F()
198 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
204 keepDimsTensor->SetBuffer(nullptr, 0); in HWTEST_F()
265 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, m_paramDims, in HWTEST_F() local
268 keepDimsTensor->SetBuffer(keepDimsValue, 2 * sizeof(bool)); in HWTEST_F()
269 m_allTensors.emplace_back(keepDimsTensor); in HWTEST_F()
389 std::shared_ptr<NNTensor> keepDimsTensor = TransToNNTensor(OH_NN_BOOL, HWTEST_F() local
[all...]

Completed in 12 milliseconds