Home
last modified time | relevance | path

Searched refs:SaveParamsTensor (Results 1 – 12 of 12) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dadd_test.cpp32 void SaveParamsTensor(const std::vector<uint32_t>& m_param, OH_NN_DataType dataType,
49 void AddFusionBuilderTest::SaveParamsTensor(const std::vector<uint32_t>& m_param, OH_NN_DataType da… in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::AddFusionBuilderTest
69 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
83 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
102 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
120 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
138 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
156 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
247 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
268 SaveParamsTensor(m_param, OH_NN_INT8, m_param_dim, nullptr, OH_NN_ADD_ACTIVATIONTYPE);
H A Dall_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void AllBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::AllBuilderTest
81 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
96 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
116 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
134 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
193 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
226 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ALL_KEEP_DIMS);
H A Dassert_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void AssertBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::AssertBuilderTest
81 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
96 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
116 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
134 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
193 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
226 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ASSERT_SUMMARIZE);
H A Ddiv_test.cpp32 void SaveParamsTensor(const std::vector<uint32_t>& m_params, OH_NN_DataType dataType,
49 void DivFusionTest::SaveParamsTensor(const std::vector<uint32_t>& m_params, OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::DivFusionTest
69 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
82 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
100 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
117 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
134 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
151 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
259 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
280 SaveParamsTensor(m_params, OH_NN_INT8, m_param_dim, nullptr, OH_NN_DIV_ACTIVATIONTYPE);
H A Dbroadcast_to_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void BroadcastToBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::BroadcastToBuilderTest
70 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BROADCAST_TO_SHAPE);
85 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BROADCAST_TO_SHAPE);
105 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BROADCAST_TO_SHAPE);
123 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BROADCAST_TO_SHAPE);
183 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
216 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_BROADCAST_TO_SHAPE);
H A Dflatten_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void FlattenBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::FlattenBuilderTest
69 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
84 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
104 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
122 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
182 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
215 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_FLATTEN_AXIS);
H A Dinstance_norm_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
52 void InstanceNormBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::InstanceNormBuilderTest
87 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_INSTANCE_NORM_EPSILON);
102 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_INSTANCE_NORM_EPSILON);
122 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_INSTANCE_NORM_EPSILON);
140 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_INSTANCE_NORM_EPSILON);
200 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
233 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_INSTANCE_NORM_EPSILON);
H A Dlog_softmax_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
49 void LogSoftmaxBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::LogSoftmaxBuilderTest
68 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LOG_SOFTMAX_AXIS);
83 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LOG_SOFTMAX_AXIS);
103 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LOG_SOFTMAX_AXIS);
121 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LOG_SOFTMAX_AXIS);
180 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
213 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LOG_SOFTMAX_AXIS);
H A Donehot_builder_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void OneHotBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::OneHotBuilderTest
69 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
84 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
104 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
122 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
181 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
213 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_ONE_HOT_AXIS);
H A Dbatchnorm_builder_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void BatchNormBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::BatchNormBuilderTest
69 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
84 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
104 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
122 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
203 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_MATMUL_TRANSPOSE_A);
235 SaveParamsTensor(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_BATCH_NORM_EPSILON);
H A Dunstack_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
49 void UnstackBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::UnstackBuilderTest
68 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_UNSTACK_AXIS);
83 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_UNSTACK_AXIS);
103 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_UNSTACK_AXIS);
162 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
195 SaveParamsTensor(OH_NN_INT64, m_paramDim, nullptr, OH_NN_UNSTACK_AXIS);
H A Dmul_builder_test.cpp33 void SaveParamsTensor(OH_NN_DataType dataType,
50 void MulBuilderTest::SaveParamsTensor(OH_NN_DataType dataType, in SaveParamsTensor() function in OHOS::NeuralNetworkRuntime::UnitTest::MulBuilderTest
69 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
84 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
104 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
122 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);
229 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MATMUL_ACTIVATION_TYPE);
262 SaveParamsTensor(OH_NN_INT8, m_paramDim, nullptr, OH_NN_MUL_ACTIVATION_TYPE);