Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dconv2d_builder.h26 class Conv2DBuilder : public OpsBuilder {
28 typedef OH_NN_ReturnCode (Conv2DBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
30 Conv2DBuilder();
31 ~Conv2DBuilder() override;
64 {OH_NN_CONV2D_STRIDES, &Conv2DBuilder::SetStrides},
65 {OH_NN_CONV2D_PAD, &Conv2DBuilder::SetPad},
66 {OH_NN_CONV2D_DILATION, &Conv2DBuilder::SetDilation},
67 {OH_NN_CONV2D_PAD_MODE, &Conv2DBuilder::SetPad},
68 {OH_NN_CONV2D_ACTIVATION_TYPE, &Conv2DBuilder::SetActavitation},
69 {OH_NN_CONV2D_GROUP, &Conv2DBuilder::SetGroup}
H A Dconv2d_builder.cpp39 Conv2DBuilder::Conv2DBuilder() {} in Conv2DBuilder() function in OHOS::NeuralNetworkRuntime::Ops::Conv2DBuilder
41 Conv2DBuilder::~Conv2DBuilder() {} in ~Conv2DBuilder()
59 OH_NN_ReturnCode Conv2DBuilder::SetChannel(const std::vector<uint32_t>& inputsIndex, in SetChannel()
75 void Conv2DBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize()
86 OH_NN_ReturnCode Conv2DBuilder::SetStrides(const std::shared_ptr<NNTensor>& tensor) in SetStrides()
107 OH_NN_ReturnCode Conv2DBuilder::SetDilation(const std::shared_ptr<NNTensor>& tensor) in SetDilation()
128 OH_NN_ReturnCode Conv2DBuilder::SetPad(const std::shared_ptr<NNTensor>& tensor) in SetPad()
173 OH_NN_ReturnCode Conv2DBuilder::SetGroup(const std::shared_ptr<NNTensor>& tensor) in SetGroup()
226 OH_NN_ReturnCode Conv2DBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
276 LiteGraphPrimitvePtr Conv2DBuilder::GetPrimitive() in GetPrimitive()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dconv2d_pad_test.cpp38 Conv2DBuilder m_builder;
H A Dconv2d_padmode_test.cpp38 Conv2DBuilder m_builder;