Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dconv2d_transpose_builder.h26 class Conv2DTransposeBuilder : public OpsBuilder {
28 typedef OH_NN_ReturnCode (Conv2DTransposeBuilder::*FuncPtr)(const std::shared_ptr<NNTensor>&);
30 Conv2DTransposeBuilder();
31 ~Conv2DTransposeBuilder() override;
64 {OH_NN_CONV2D_TRANSPOSE_STRIDES, &Conv2DTransposeBuilder::SetStrides},
65 {OH_NN_CONV2D_TRANSPOSE_PAD, &Conv2DTransposeBuilder::SetPad},
66 {OH_NN_CONV2D_TRANSPOSE_DILATION, &Conv2DTransposeBuilder::SetDilation},
67 {OH_NN_CONV2D_TRANSPOSE_OUTPUT_PADDINGS, &Conv2DTransposeBuilder::SetOutPadding},
68 {OH_NN_CONV2D_TRANSPOSE_PAD_MODE, &Conv2DTransposeBuilder::SetPad},
69 {OH_NN_CONV2D_TRANSPOSE_ACTIVATION_TYPE, &Conv2DTransposeBuilder::SetActivation},
[all …]
H A Dconv2d_transpose_builder.cpp39 Conv2DTransposeBuilder::Conv2DTransposeBuilder() {} in Conv2DTransposeBuilder() function in OHOS::NeuralNetworkRuntime::Ops::Conv2DTransposeBuilder
41 Conv2DTransposeBuilder::~Conv2DTransposeBuilder() {} in ~Conv2DTransposeBuilder()
43 OH_NN_ReturnCode Conv2DTransposeBuilder::SetInput(const std::vector<uint32_t>& inputsIndex, in SetInput()
69 void Conv2DTransposeBuilder::SetKernelSize(const std::vector<uint32_t>& inputsIndex, in SetKernelSize()
79 OH_NN_ReturnCode Conv2DTransposeBuilder::SetStrides(const std::shared_ptr<NNTensor>& tensor) in SetStrides()
100 OH_NN_ReturnCode Conv2DTransposeBuilder::SetDilation(const std::shared_ptr<NNTensor>& tensor) in SetDilation()
121 OH_NN_ReturnCode Conv2DTransposeBuilder::SetPad(const std::shared_ptr<NNTensor>& tensor) in SetPad()
166 OH_NN_ReturnCode Conv2DTransposeBuilder::SetGroup(const std::shared_ptr<NNTensor>& tensor) in SetGroup()
240 OH_NN_ReturnCode Conv2DTransposeBuilder::Build(const std::vector<uint32_t>& paramsIndex, in Build()
286 LiteGraphPrimitvePtr Conv2DTransposeBuilder::GetPrimitive() in GetPrimitive()
[all …]
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dconv2d_transpose_pad_test.cpp40 Conv2DTransposeBuilder m_builder;
H A Dconv2d_tranpose_padmode_test.cpp40 Conv2DTransposeBuilder m_builder;