Home
last modified time | relevance | path

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

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/fusion/
Dconv2d_transpose_fusion.cc52 void Conv2dTransposeFusion::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2dTransposeFusion
Davg_pool_fusion.cc34 void AvgPoolFusion::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::AvgPoolFusion
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dpool_grad.cc71 void PoolGrad::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::PoolGrad
Dmax_pool_grad_grad.cc45 void MaxPoolGradGrad::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPoolGradGrad
Dmax_pool_grad_with_argmax.cc61 void MaxPoolGradWithArgmax::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPoolGradWithArgmax
Dde_conv2d_grad_filter.cc68 void DeConv2DGradFilter::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::DeConv2DGradFilter
Dmax_pool3d_grad_with_argmax.cc57 void MaxPool3DGradWithArgmax::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPool3DGradWithArgmax
Dgroup_conv2d_grad_input.cc68 void GroupConv2DGradInput::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::GroupConv2DGradInput
Ddeformable_offsets_grad.cc97 void DeformableOffsetsGrad::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::DeformableOffsetsGrad
Dmax_pool_grad_with_argmax_v2.cc40 void MaxPoolGradWithArgmaxV2::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPoolGradWithArgmaxV2
Dconv3d_backprop_input.cc82 void Conv3DBackpropInput::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv3DBackpropInput
Dconv3d_backprop_filter.cc81 void Conv3DBackpropFilter::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv3DBackpropFilter
Dconv2d_backprop_filter.cc172 void Conv2DBackpropFilter::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2DBackpropFilter
Dconv2d_backprop_input.cc178 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-src/source/mindspore/core/ops/
Dadder.cc58 void Adder::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Adder
Dextract_volume_patches.cc148 void ExtractVolumePatches::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::ExtractVolumePatches
Dconv2d_transpose.cc67 void Conv2DTranspose::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2DTranspose
Dmax_pool_with_argmax_v2.cc41 void MaxPoolWithArgmaxV2::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPoolWithArgmaxV2
Dmax_pool_with_argmax.cc77 void MaxPoolWithArgmax::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPoolWithArgmax
Dmax_pool3d_with_argmax.cc62 void MaxPool3DWithArgmax::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPool3DWithArgmax
Dmax_pool.cc201 void MaxPool::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPool
Ddeformable_offsets.cc216 void DeformableOffsets::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::DeformableOffsets
Dmax_pool3d.cc68 void MaxPool3D::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::MaxPool3D
Dconv2d.cc321 void Conv2D::set_kernel_size(const std::vector<int64_t> &kernel_size) { in set_kernel_size() function in mindspore::ops::Conv2D
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc183 void AvgPoolGrad::set_kernel_size(const std::vector<int64_t> &kernel_size) { (void)this… in set_kernel_size() function in mindspore::ops::AvgPoolGrad
202 void AvgPool::set_kernel_size(const std::vector<int64_t> &kernel_size) { (void)this->Ad… in set_kernel_size() function in mindspore::ops::AvgPool
1213 void MaxPoolGradWithIndices::set_kernel_size(const std::vector<int64_t> &kernel_size) {… in set_kernel_size() function in mindspore::ops::MaxPoolGradWithIndices
1240 void MaxPoolGradWithMask::set_kernel_size(const std::vector<int64_t> &kernel_size) { (v… in set_kernel_size() function in mindspore::ops::MaxPoolGradWithMask
1267 void MaxPoolWithIndices::set_kernel_size(const std::vector<int64_t> &kernel_size) { (vo… in set_kernel_size() function in mindspore::ops::MaxPoolWithIndices
1294 void MaxPoolWithMask::set_kernel_size(const std::vector<int64_t> &kernel_size) { (void)… in set_kernel_size() function in mindspore::ops::MaxPoolWithMask

12