Home
last modified time | relevance | path

Searched refs:axisTensor (Results 1 – 16 of 16) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dscale_builder_test.cpp54 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()
181 axisTensor->SetBuffer(&axisValue, sizeof(axisValue));
182 m_allTensors.emplace_back(axisTensor);
186 axisTensor->SetBuffer(nullptr, 0);
225 axisTensor->SetBuffer(axisValue, 2 * sizeof(int64_t));
226 m_allTensors.emplace_back(axisTensor);
232 axisTensor->SetBuffer(nullptr, 0);
311 m_allTensors.emplace_back(axisTensor);
[all …]
H A Donehot_builder_test.cpp53 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, variable
164 axisTensor->SetBuffer(&axisValue, sizeof(axisValue));
165 m_allTensors.emplace_back(axisTensor);
169 axisTensor->SetBuffer(nullptr, 0);
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
198 m_allTensors.emplace_back(axisTensor);
H A Dunstack_test.cpp52 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, variable
146 axisTensor->SetBuffer(axisValue, sizeof(axisValue));
147 m_allTensors.emplace_back(axisTensor);
178 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
180 m_allTensors.emplace_back(axisTensor);
H A Dflatten_test.cpp53 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, variable
166 axisTensor->SetBuffer(axisValue, sizeof(float));
167 m_allTensors.emplace_back(axisTensor);
198 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
200 m_allTensors.emplace_back(axisTensor);
H A Dlog_softmax_test.cpp52 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, variable
164 axisTensor->SetBuffer(axisValue, sizeof(float));
165 m_allTensors.emplace_back(axisTensor);
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
198 m_allTensors.emplace_back(axisTensor);
H A Dsqueeze_builder_test.cpp44 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_SQUEEZ… variable
202 axisTensor->SetBuffer(nullptr, 0);
203 m_allTensors.emplace_back(axisTensor);
H A Dquant_dtype_cast_builder_test.cpp76 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, variable
244 axisTensor->SetBuffer(&axisValue, sizeof(axisValue));
245 m_allTensors.emplace_back(axisTensor);
249 axisTensor->SetBuffer(nullptr, 0);
358 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
360 m_allTensors.emplace_back(axisTensor);
H A Dstack_builder_test.cpp56 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_… variable
220 axisTensor->SetBuffer(nullptr, 0);
221 m_allTensors.emplace_back(axisTensor);
H A Dunsqueeze_builder_test.cpp44 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_UNSQU… variable
219 axisTensor->SetBuffer(nullptr, 0);
220 m_allTensors.emplace_back(axisTensor);
H A Dl2_normalize_test.cpp55 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, variable
185 axisTensor->SetBuffer(axisValue, sizeof(float));
186 m_allTensors.emplace_back(axisTensor);
299 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
301 m_allTensors.emplace_back(axisTensor);
H A Dsoftmax_builder_test.cpp44 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_SOFTM… variable
239 axisTensor->SetBuffer(nullptr, 0);
240 m_allTensors.emplace_back(axisTensor);
H A Dcrop_test.cpp70 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, variable
197 axisTensor->SetBuffer(axisValue, sizeof(float));
198 m_allTensors.emplace_back(axisTensor);
H A Dtopk_builder_test.cpp55 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… variable
248 axisTensor->SetBuffer(nullptr, 0);
249 m_allTensors.emplace_back(axisTensor);
H A Dsplit_builder_test.cpp84 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… variable
302 axisTensor->SetBuffer(nullptr, 0);
303 m_allTensors.emplace_back(axisTensor);
H A Dall_test.cpp67 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 Dstrided_slice_builder_test.cpp90 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, variable
457 axisTensor->SetBuffer(nullptr, 0);
458 m_allTensors.emplace_back(axisTensor);