| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | unstack_test.cpp | 52 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
|
| H A D | log_softmax_test.cpp | 52 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
|
| H A D | flatten_test.cpp | 53 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
|
| H A D | onehot_builder_test.cpp | 53 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
|
| H A D | scale_builder_test.cpp | 54 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
|
| H A D | l2_normalize_test.cpp | 55 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
|
| H A D | quant_dtype_cast_builder_test.cpp | 76 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
|
| H A D | crop_test.cpp | 70 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
|
| H A D | squeeze_builder_test.cpp | 44 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
|
| H A D | stack_builder_test.cpp | 56 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
|
| H A D | softmax_builder_test.cpp | 44 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
|
| H A D | unsqueeze_builder_test.cpp | 44 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
|
| H A D | topk_builder_test.cpp | 55 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
|
| H A D | all_test.cpp | 67 std::shared_ptr<NNTensor> axisTensor; in SetInputTensor() local
|
| H A D | split_builder_test.cpp | 84 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
|
| H A D | strided_slice_builder_test.cpp | 90 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
|