Home
last modified time | relevance | path

Searched refs:SetUseAxis (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dfullconnection_with_axis_test.cpp37 void SetUseAxis(OH_NN_DataType dataType,
98 void FullConnectionAxisBuilderTest::SetUseAxis(OH_NN_DataType dataType, in SetUseAxis() function in OHOS::NeuralNetworkRuntime::UnitTest::FullConnectionAxisBuilderTest
134 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
154 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
176 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
197 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
221 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
246 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
291 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
323 SetUseAxis(OH_NN_BOOL, m_param_dim, nullptr, OH_NN_FULL_CONNECTION_USE_AXIS);
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dfullconnection_builder.h45 OH_NN_ReturnCode SetUseAxis(const std::shared_ptr<NNTensor>& tensor);
57 {OH_NN_FULL_CONNECTION_USE_AXIS, &FullConnectionBuilder::SetUseAxis},
H A Dfullconnection_builder.cpp79 OH_NN_ReturnCode FullConnectionBuilder::SetUseAxis(const std::shared_ptr<NNTensor>& tensor) in SetUseAxis() function in OHOS::NeuralNetworkRuntime::Ops::FullConnectionBuilder