Home
last modified time | relevance | path

Searched defs:set_kernel_size (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/core/ops/fusion/
Dconv2d_transpose_fusion.cc42 void Conv2dTransposeFusion::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2dTransposeFusion
/third_party/mindspore/mindspore/core/ops/grad/
Dpool_grad.cc57 void PoolGrad::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::PoolGrad
Dde_conv2d_grad_filter.cc62 void DeConv2DGradFilter::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::DeConv2DGradFilter
Dgroup_conv2d_grad_input.cc63 void GroupConv2DGradInput::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::GroupConv2DGradInput
Dconv2d_backprop_filter.cc152 void Conv2DBackpropFilter::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2DBackpropFilter
Dconv2d_backprop_input.cc194 void Conv2DBackpropInput::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2DBackpropInput
/third_party/mindspore/mindspore/core/ops/
Dadder.cc50 void Adder::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Adder
Davg_pool.cc35 void AvgPool::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::AvgPool
Dconv2d_transpose.cc55 void Conv2DTranspose::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2DTranspose
Dmax_pool.cc35 void MaxPool::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPool
Dconv2d.cc269 void Conv2D::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2D