/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | unsqueeze_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 41 void UnsqueezeBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::UnsqueezeBuilderTest 75 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 93 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 112 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 130 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 163 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 181 SaveAxisTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 199 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_UNSQUEEZE_AXIS); in HWTEST_F() 238 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | stack_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 53 void StackBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::StackBuilderTest 76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 164 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 182 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 200 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_STACK_AXIS); in HWTEST_F() 239 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | softmax_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 41 void SoftmaxBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::SoftmaxBuilderTest 76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 163 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 202 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 220 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SOFTMAX_AXIS); in HWTEST_F() 258 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | split_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 81 void SplitBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::SplitBuilderTest 105 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 124 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 145 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 166 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 200 SaveAxisTensor(OH_NN_INT8, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 220 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 240 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SPLIT_AXIS); in HWTEST_F() 260 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | squeeze_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 41 void SqueezeBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::SqueezeBuilderTest 76 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 94 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 113 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 131 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 164 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 182 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_SQUEEZE_AXIS); in HWTEST_F() 221 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 251 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | quant_dtype_cast_builder_test.cpp | 37 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 73 void QuantDTypeCastBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::QuantDTypeCastBuilderTest 94 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 111 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 133 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 153 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 199 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 222 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 264 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_QUANT_DTYPE_CAST_AXIS); in HWTEST_F() 282 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | topk_builder_test.cpp | 35 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 52 void TopKBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::TopKBuilderTest 87 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 107 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 128 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 148 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 183 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 203 SaveAxisTensor(OH_NN_INT32, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 226 SaveAxisTensor(OH_NN_INT64, paramDim, nullptr, OH_NN_TOP_K_AXIS); in HWTEST_F() 269 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |
H A D | scale_builder_test.cpp | 33 void SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, 51 void ScaleBuilderTest::SaveAxisTensor(OH_NN_DataType dataType, const std::vector<int32_t> &dim, in SaveAxisTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::ScaleBuilderTest 80 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 96 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 117 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 136 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 199 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 245 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 269 SaveAxisTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_SCALE_AXIS); in HWTEST_F() 291 SaveAxisTensor(OH_NN_INT6 in HWTEST_F() [all...] |