Home
last modified time | relevance | path

Searched refs:SaveAxisTensor (Results 1 - 8 of 8) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dunsqueeze_builder_test.cpp33 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 Dstack_builder_test.cpp33 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 Dsoftmax_builder_test.cpp33 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 Dsplit_builder_test.cpp33 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 Dsqueeze_builder_test.cpp33 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 Dquant_dtype_cast_builder_test.cpp37 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 Dtopk_builder_test.cpp35 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 Dscale_builder_test.cpp33 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...]

Completed in 5 milliseconds