Home
last modified time | relevance | path

Searched refs:DepthwiseConv2DNativeBuilder (Results 1 – 4 of 4) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Ddepthwise_conv2d_native_builder.h26 class DepthwiseConv2DNativeBuilder : public OpsBuilder {
28 …typedef OH_NN_ReturnCode (DepthwiseConv2DNativeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&…
30 DepthwiseConv2DNativeBuilder();
31 ~DepthwiseConv2DNativeBuilder() override;
60 {OH_NN_DEPTHWISE_CONV2D_NATIVE_STRIDES, &DepthwiseConv2DNativeBuilder::SetStrides},
61 {OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD, &DepthwiseConv2DNativeBuilder::SetPadModeOrPaddings},
62 {OH_NN_DEPTHWISE_CONV2D_NATIVE_DILATION, &DepthwiseConv2DNativeBuilder::SetDilation},
63 … {OH_NN_DEPTHWISE_CONV2D_NATIVE_PAD_MODE, &DepthwiseConv2DNativeBuilder::SetPadModeOrPaddings},
64 … {OH_NN_DEPTHWISE_CONV2D_NATIVE_ACTIVATION_TYPE, &DepthwiseConv2DNativeBuilder::SetActivation}
H A Ddepthwise_conv2d_native_builder.cpp40 DepthwiseConv2DNativeBuilder::DepthwiseConv2DNativeBuilder() {} in DepthwiseConv2DNativeBuilder() function in OHOS::NeuralNetworkRuntime::Ops::DepthwiseConv2DNativeBuilder
42 DepthwiseConv2DNativeBuilder::~DepthwiseConv2DNativeBuilder() {} in ~DepthwiseConv2DNativeBuilder()
44 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetIsPadMode(const std::shared_ptr<NNTensor>& tensor, in SetIsPadMode()
58 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetActivation(const std::shared_ptr<NNTensor>& tenso… in SetActivation()
87 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsInd… in SetKernelSize()
104 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetStrides(const std::shared_ptr<NNTensor>& tensor) in SetStrides()
123 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetDilation(const std::shared_ptr<NNTensor>& tensor) in SetDilation()
184 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::SetInputAndOutput( in SetInputAndOutput()
200 OH_NN_ReturnCode DepthwiseConv2DNativeBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
255 LiteGraphPrimitvePtr DepthwiseConv2DNativeBuilder::GetPrimitive() in GetPrimitive()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddepthwise_conv2d_native_padmode_test.cpp38 DepthwiseConv2DNativeBuilder m_builder;
H A Ddepthwise_conv2d_native_pad_test.cpp38 DepthwiseConv2DNativeBuilder m_builder;