Home
last modified time | relevance | path

Searched defs:pad_list (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Davgpool_3d_grad_fusion.cc41 … std::vector<int64_t> *pad_list, std::vector<int64_t> *origin_input_shape, bool *ceil_mode, in GetAttrs()
82 const std::vector<int64_t> &pad_list) { in IsVectorImpl()
99 bool IsZeroPads(const std::vector<int64_t> &pad_list) { in IsZeroPads()
103 AnfNodePtr ConstructFilter(const FuncGraphPtr &func_graph, const std::vector<int64_t> &pad_list, in… in ConstructFilter()
123 … const std::vector<int64_t> &strides, const std::vector<int64_t> &pad_list, in ConstructMultiplier()
205 std::vector<int64_t> pad_list; in Process() local
Davgpool_3d_fusion.cc108 void GetAttrs(const AnfNodePtr &node, std::vector<int64_t> *pad_list, bool *count_include_pad, bool… in GetAttrs()
126 …sVectorImpl(int64_t fh, int64_t fw, int64_t kh, int64_t kw, const std::vector<int64_t> &pad_list) { in IsVectorImpl()
136 bool IsZeroPads(const std::vector<int64_t> &pad_list) { in IsZeroPads()
140 AnfNodePtr ConstructFilter(const FuncGraphPtr &func_graph, const std::vector<int64_t> &pad_list, in… in ConstructFilter()
160 … int64_t sd, int64_t sh, int64_t sw, const std::vector<int64_t> &pad_list, in ConstructMultiplier()
286 std::vector<int64_t> pad_list; in Process() local
/third_party/mindspore/mindspore/core/ops/fusion/
Dconv2d_fusion.cc27 … const std::vector<int64_t> &pad_list, const ActivationType &activation_type) { in Init()
42 void Conv2DFusion::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dadder_fusion.cc24 … const std::vector<int64_t> &pad_list, const std::vector<int64_t> &dilation, const int64_t group, in Init()
Dconv2d_backprop_filter_fusion.cc25 … const PadMode &pad_mode, const std::vector<int64_t> &pad_list, const int64_t mode, in Init()
Dconv2d_backprop_input_fusion.cc28 … int64_t group, const Format &format, const std::vector<int64_t> &pad_list, in Init()
Dconv2d_transpose_fusion.cc25 … int64_t group, const Format &format, const std::vector<int64_t> &pad_list, in Init()
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc35 int64_t *pad_mode, std::vector<int64_t> *pad_list, bool *ceil_mode) { in GetAttrs()
60 const std::vector<int64_t> &pad_list, bool ceil_mode) { in GetOutputShape()
94 const std::vector<int64_t> &padding, std::vector<int64_t> *pad_list) { in GetPadsByPadding()
130 std::vector<int64_t> pad_list; in InferShape() local
Dadder.cc23 … const PadMode &pad_mode, const std::vector<int64_t> &stride, const std::vector<int64_t> &pad_list, in Init()
76 void Adder::set_pad_list(const std::vector<int64_t> &pad_list) { (void)this->AddAttr(kPadList, Make… in set_pad_list()
Dconv2d_transpose.cc31 const Format &format, const std::vector<int64_t> &pad_list) { in Init()
112 void Conv2DTranspose::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dconv2d.cc72 void Conv2DPadFunction(std::vector<int64_t> *output_hw, std::vector<int64_t> *pad_list, const int64… in Conv2DPadFunction()
209 std::vector<int64_t> pad_list; in Conv2dInferShape() local
/third_party/mindspore/mindspore/core/ops/grad/
Dpooling_grad.cc24 … const std::vector<int64_t> &pad_list, const RoundMode &round_mode, const Format &format, in Init()
73 void PoolingGrad::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dde_conv2d_grad_filter.cc26 … const std::vector<int64_t> &pad_list, const std::vector<int64_t> &stride, in Init()
83 void DeConv2DGradFilter::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dconv2d_backprop_input.cc45 ShapeVector pad_list = {0, 0, 0, 0}; in SetPadList() local
176 const Format &format, const std::vector<int64_t> &pad_list) { in Init()
238 void Conv2DBackpropInput::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dgroup_conv2d_grad_input.cc25 … const std::vector<int64_t> &pad_list, const std::vector<int64_t> &stride, in Init()
84 void GroupConv2DGradInput::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
Dconv2d_backprop_filter.cc124 … const PadMode &pad_mode, const std::vector<int64_t> &pad_list, const int64_t mode, in Init()
173 void Conv2DBackpropFilter::set_pad_list(const std::vector<int64_t> &pad_list) { in set_pad_list()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dadder_populate.cc42 auto pad_list = value->pad_list(); in PopulateAdderParameter() local
Dconv2d_populate.cc42 auto pad_list = value->pad_list(); in PopulateConvParameter() local
Ddeconv2d_populate.cc43 auto pad_list = value->pad_list(); in PopulateDeconvParameter() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dconv3d_transpose_gpu_kernel.h142 std::vector<int> pad_list; in Init() local
373 std::vector<int> pad_list; in SetPad() local
446 int *pad_list, int *stride_pad_list) { // pad_mode_ = same in UpdatePaddingAndDilation()
467 void UsePadProcess(const std::vector<int> &pad_list, const int *strideA, const int *dilaA) { in UsePadProcess()
Dconv3d_grad_input_gpu_kernel.h112 std::vector<int> pad_list; in Init() local
294 std::vector<int> pad_list; in SetPad() local
Dconv2d_grad_input_gpu_kernel.h150 std::vector<int> pad_list; in Init() local
296 std::vector<int> pad_list; in SetPad() local
Dim2col_gpu_kernel.h207 std::vector<int> pad_list; in SetPad() local
/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dconv_pad_fusion.cc48 auto pad_list = std::dynamic_pointer_cast<tensor::Tensor>(paddings->default_param()); in ReplaceParamsAndNodes() local
118 auto pad_list = pad_cnode->input(kInputIndexTwo)->cast<ParameterPtr>(); in IsPrimitiveProper() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dmulti_conv_info.cc283 auto pad_list = GetSplitPadList(conv_prim, input_h, input_w); in AdJustConvPrim() local

12