Searched refs:SetArgmaxOutMaxValue (Results 1 – 1 of 1) sorted by relevance
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | argmax_test.cpp | 38 void SetArgmaxOutMaxValue(OH_NN_DataType dataType,
85 void ArgMaxBuilderTest::SetArgmaxOutMaxValue(OH_NN_DataType dataType, in SetArgmaxOutMaxValue() function in OHOS::NeuralNetworkRuntime::UnitTest::ArgMaxBuilderTest
109 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
131 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
155 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
178 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
201 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
224 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
248 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
272 SetArgmaxOutMaxValue(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_ARG_MAX_OUT_MAX_VALUE);
[all …]
|