Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
Dunstack_test.cpp52 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
143 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
178 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Dlog_softmax_test.cpp52 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
161 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Donehot_builder_test.cpp53 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
161 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
196 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Dflatten_test.cpp53 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveParamsTensor() local
162 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
198 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Dscale_builder_test.cpp54 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
179 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, nullptr, OH_NN_SCA… variable
223 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, axistDim, nullptr, OH_NN_SCALE… variable
310 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCA… variable
Dl2_normalize_test.cpp55 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetAxis() local
182 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable
299 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Dquant_dtype_cast_builder_test.cpp76 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
241 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT32, m_paramDim, variable
358 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
Dcrop_test.cpp70 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxis() local
194 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_FLOAT32, m_axisDim, variable
Dsqueeze_builder_test.cpp44 std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
201 …std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZ… variable
Dstack_builder_test.cpp56 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
219 …std::shared_ptr<NNTensor> axisTensor =TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_… variable
Dsoftmax_builder_test.cpp44 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
238 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTM… variable
Dunsqueeze_builder_test.cpp44 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
218 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQU… variable
Dtopk_builder_test.cpp55 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
247 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K… variable
Dall_test.cpp67 std::shared_ptr<NNTensor> axisTensor; in SetInputTensor() local
Dsplit_builder_test.cpp84 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveAxisTensor() local
301 …std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT… variable
Dstrided_slice_builder_test.cpp90 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveNewAxisTensor() local
455 std::shared_ptr<NNTensor> axisTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, variable