| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | onehot_builder_test.cpp | 53 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
56 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveParamsTensor()
57 m_allTensors.emplace_back(axisTensor); in SaveParamsTensor()
161 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, in HWTEST_F() local
164 axisTensor->SetBuffer(&axisValue, sizeof(axisValue)); in HWTEST_F()
165 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
169 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
198 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | flatten_test.cpp | 53 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SaveParamsTensor() local
56 axisTensor->SetBuffer(axisValue, sizeof(int64_t));
in SaveParamsTensor()
57 m_allTensors.emplace_back(axisTensor);
in SaveParamsTensor()
162 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
166 axisTensor->SetBuffer(axisValue, sizeof(float));
in HWTEST_F()
167 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
198 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
200 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
|
| H A D | log_softmax_test.cpp | 52 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SaveParamsTensor() local
55 axisTensor->SetBuffer(axisValue, sizeof(int64_t));
in SaveParamsTensor()
56 m_allTensors.emplace_back(axisTensor);
in SaveParamsTensor()
161 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
164 axisTensor->SetBuffer(axisValue, sizeof(float));
in HWTEST_F()
165 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
198 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
|
| H A D | unstack_test.cpp | 52 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SaveParamsTensor() local
55 axisTensor->SetBuffer(axisValue, sizeof(int64_t));
in SaveParamsTensor()
56 m_allTensors.emplace_back(axisTensor);
in SaveParamsTensor()
143 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
146 axisTensor->SetBuffer(axisValue, sizeof(axisValue));
in HWTEST_F()
147 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
178 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
180 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
|
| H A D | scale_builder_test.cpp | 54 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
57 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
58 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
179 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() local
181 axisTensor->SetBuffer(&axisValue, sizeof(axisValue)); in HWTEST_F()
182 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
186 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
223 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, axistDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() local
225 axisTensor->SetBuffer(axisValue, 2 * sizeof(int64_t)); in HWTEST_F()
226 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
310 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); HWTEST_F() local
[all...] |
| H A D | quant_dtype_cast_builder_test.cpp | 76 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
79 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
80 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
241 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, in HWTEST_F() local
244 axisTensor->SetBuffer(&axisValue, sizeof(axisValue)); in HWTEST_F()
245 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
249 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
358 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
360 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | unsqueeze_builder_test.cpp | 44 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
46 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
47 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
218 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() local
219 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
220 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | stack_builder_test.cpp | 56 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
59 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
60 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
219 std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() local
220 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
221 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | softmax_builder_test.cpp | 44 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
47 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
48 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
238 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() local
239 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
240 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | squeeze_builder_test.cpp | 44 std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
47 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
48 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
201 std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() local
202 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
203 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | l2_normalize_test.cpp | 55 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SetAxis() local
58 axisTensor->SetBuffer(axisValue, sizeof(int64_t));
in SetAxis()
59 m_allTensors.emplace_back(axisTensor);
in SetAxis()
182 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim,
in HWTEST_F() local
185 axisTensor->SetBuffer(axisValue, sizeof(float));
in HWTEST_F()
186 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
299 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim,
in HWTEST_F() local
301 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
|
| H A D | topk_builder_test.cpp | 55 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
58 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
59 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
247 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() local
248 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
249 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | split_builder_test.cpp | 84 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
87 axisTensor->SetBuffer(axisValue, sizeof(int64_t)); in SaveAxisTensor()
88 m_allTensors.emplace_back(axisTensor); in SaveAxisTensor()
301 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() local
302 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
303 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|
| H A D | crop_test.cpp | 70 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type);
in SaveAxis() local
72 axisTensor->SetBuffer(axisValue, sizeof(int64_t));
in SaveAxis()
73 m_allTensors.emplace_back(axisTensor);
in SaveAxis()
194 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_axisDim,
in HWTEST_F() local
197 axisTensor->SetBuffer(axisValue, sizeof(float));
in HWTEST_F()
198 m_allTensors.emplace_back(axisTensor);
in HWTEST_F()
|
| H A D | all_test.cpp | 67 std::shared_ptr<NNTensor> axisTensor;
in SetInputTensor() local
68 axisTensor = TransToNNTensor(OH_NN_INT32, m_dim, nullptr, OH_NN_TENSOR);
in SetInputTensor()
69 m_allTensors.emplace_back(axisTensor);
in SetInputTensor()
|
| H A D | strided_slice_builder_test.cpp | 90 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveNewAxisTensor() local
93 axisTensor->SetBuffer(newAxisMaskValue, sizeof(int64_t)); in SaveNewAxisTensor()
94 m_allTensors.emplace_back(axisTensor); in SaveNewAxisTensor()
455 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, in HWTEST_F() local
457 axisTensor->SetBuffer(nullptr, 0); in HWTEST_F()
458 m_allTensors.emplace_back(axisTensor); in HWTEST_F()
|