Home
last modified time | relevance | path

Searched refs:get_pad_list (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc39 return ori_conv_prim->get_pad_list(); in GetSplitPadList()
150 …stride_h * split_axis_dim - ori_conv_prim->get_pad_list()[kPadUp] + ori_conv_prim->get_kernel_size… in CalSplitInShape()
152 tmp = stride_h * split_axis_dim - ori_conv_prim->get_pad_list()[kPadDown] + in CalSplitInShape()
164 tmp = stride_h * split_axis_reduce_dim - ori_conv_prim->get_pad_list()[kPadDown] - in CalSplitInShape()
165 ori_conv_prim->get_pad_list()[kPadUp] + ori_conv_prim->get_kernel_size()[kIndexH]; in CalSplitInShape()
167 tmp = stride_h * split_axis_reduce_dim - ori_conv_prim->get_pad_list()[kPadUp] + in CalSplitInShape()
202 new_prim->set_pad_list(ori_conv_prim->get_pad_list()); in CopyConvPrim()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dconv2d_info.cc172 MS_CHECK_GE(conv_prim->get_pad_list().size(), 1, nullptr); in CreateOutputsOfSplit()
175 … conv_prim->get_pad_list().at(kPadUp), conv_prim->get_stride().at(kIndexH))) { in CreateOutputsOfSplit()
296 prim->set_pad_list(conv_prim->get_pad_list()); in ConstructOutputCNodes()
302 auto pad = prim->get_pad_list(); in ConstructOutputCNodes()
307 auto pad = prim->get_pad_list(); in ConstructOutputCNodes()
Ddepthwise_conv2d_info.cc223 auto pad = conv_prim->get_pad_list(); in AdJustConvPrim()
228 auto pad = conv_prim->get_pad_list(); in AdJustConvPrim()
367 depth_wise_conv_prim->get_pad_list().at(kPadUp), in CreateOutputsOfSplit()
/third_party/mindspore/mindspore/core/ops/fusion/
Dconv2d_fusion.h83 std::vector<int64_t> get_pad_list() const;
Dconv2d_fusion.cc54 std::vector<int64_t> Conv2DFusion::get_pad_list() const { in get_pad_list() function in mindspore::ops::Conv2DFusion
/third_party/mindspore/mindspore/core/ops/grad/
Dpooling_grad.h54 std::vector<int64_t> get_pad_list() const;
Dde_conv2d_grad_filter.h56 std::vector<int64_t> get_pad_list() const;
Dconv2d_backprop_filter.h60 std::vector<int64_t> get_pad_list() const;
Dgroup_conv2d_grad_input.h59 std::vector<int64_t> get_pad_list() const;
Dconv2d_backprop_input.h61 std::vector<int64_t> get_pad_list() const;
Dpooling_grad.cc77 std::vector<int64_t> PoolingGrad::get_pad_list() const { in get_pad_list() function in mindspore::ops::PoolingGrad
Dde_conv2d_grad_filter.cc87 std::vector<int64_t> DeConv2DGradFilter::get_pad_list() const { in get_pad_list() function in mindspore::ops::DeConv2DGradFilter
Dgroup_conv2d_grad_input.cc88 std::vector<int64_t> GroupConv2DGradInput::get_pad_list() const { in get_pad_list() function in mindspore::ops::GroupConv2DGradInput
Dconv2d_backprop_filter.cc177 std::vector<int64_t> Conv2DBackpropFilter::get_pad_list() const { in get_pad_list() function in mindspore::ops::Conv2DBackpropFilter
Dconv2d_backprop_input.cc296 std::vector<int64_t> Conv2DBackpropInput::get_pad_list() const { in get_pad_list() function in mindspore::ops::Conv2DBackpropInput
/third_party/mindspore/mindspore/core/ops/
Dadder.h129 std::vector<int64_t> get_pad_list() const;
Dconv2d_transpose.h115 std::vector<int64_t> get_pad_list() const;
Dadder.cc78 std::vector<int64_t> Adder::get_pad_list() const { in get_pad_list() function in mindspore::ops::Adder
Dconv2d_transpose.cc168 std::vector<int64_t> Conv2DTranspose::get_pad_list() const { in get_pad_list() function in mindspore::ops::Conv2DTranspose