Home
last modified time | relevance | path

Searched refs:m_kernelSize (Results 1 – 10 of 10) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Ddepthwise_conv2d_native_builder.cpp98 m_kernelSize.clear(); in SetKernelSize()
99 m_kernelSize.emplace_back(weightShape[HEIGHT_IN_WEIGHT]); in SetKernelSize()
100 m_kernelSize.emplace_back(weightShape[WIDTH_IN_WEIGHT]); in SetKernelSize()
262 auto primitive = MindIR_Conv2DFusion_CreatePrimitive(m_kernelSize, m_strides, in GetPrimitive()
H A Dconv2d_builder.cpp81 m_kernelSize.clear(); in SetKernelSize()
82 m_kernelSize.emplace_back(weightShape[KERNEL_HEIGHT_INDEX]); in SetKernelSize()
83 m_kernelSize.emplace_back(weightShape[KERNEL_WEIGHT_INDEX]); in SetKernelSize()
283 auto primitive = MindIR_Conv2DFusion_CreatePrimitive(m_kernelSize, m_strides, in GetPrimitive()
H A Dconv2d_transpose_builder.cpp74 m_kernelSize.clear(); in SetKernelSize()
75 m_kernelSize.emplace_back(weightShape[KERNEL_HEIGHT_INDEX]); in SetKernelSize()
76 m_kernelSize.emplace_back(weightShape[KERNEL_WEIGHT_INDEX]); in SetKernelSize()
293 void* primitive = MindIR_Conv2dTransposeFusion_CreatePrimitive(m_kernelSize, in GetPrimitive()
H A Dmaxpool_builder.cpp49 void* primitive = MindIR_MaxPoolFusion_CreatePrimitive(m_kernelSize, m_strides, m_pad, in GetPrimitive()
H A Davgpool_builder.cpp50 …void* primitive = mindspore::lite::MindIR_AvgPoolFusion_CreatePrimitive(m_kernelSize, m_strides, m… in GetPrimitive()
H A Ddepthwise_conv2d_native_builder.h53 std::vector<int64_t> m_kernelSize; variable
H A Dconv2d_builder.h57 std::vector<int64_t> m_kernelSize; variable
H A Dconv2d_transpose_builder.h56 std::vector<int64_t> m_kernelSize; variable
H A Dpooling_builder.h50 std::vector<int64_t> m_kernelSize;
H A Dpooling_builder.cpp112 m_kernelSize.assign(pKernelSize, pKernelSize + kernelSize); in SetKernel()