Home
last modified time | relevance | path

Searched defs:output_padding (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dconv3d_transpose.cc54 const std::vector<int64_t> &output_padding, const Format &format) { in Init()
111 std::vector<int64_t> output_padding = get_output_padding(); in set_pad_mode() local
139 void Conv3DTranspose::set_output_padding(const std::vector<int64_t> &output_padding) { in set_output_padding()
291 …std::vector<int64_t> output_padding = CheckTuple(prim_name, kOutputPadding, primitive->GetAttr(kOu… in InferShape() local
333 … std::vector<int64_t> *pad_list, std::vector<int64_t> *output_padding, int64_t *d_out, in CaculateShape()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dconvolution.cc40 … const ArrayValue<int64_t> &dilation, bool transposed, const ArrayValue<int64_t> &output_padding) { in GetOutputHW()
126 const auto &output_padding = output_padding_opt.value(); in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dconvolution.cc31const ValueTuplePtr &output_padding, const Int64ImmPtr &group) { in ConvolutionAscendCustomize()
Dconvolution_grad.cc31 const ValueTuplePtr &output_padding, const Int64ImmPtr &group, const ValueTuplePtr &output_mask) { in ConvolutionGradAscendCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dconv2d_transpose_fusion_mapper.cc116 std::vector<int64_t> output_padding = {0, 0, 0, 0}; in AdjustOutputPadding() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/symbol_ops_impl/
Dconvolution.cc92 auto output_padding = input_as<ListSymbol>(kOutputPaddingIdx); in Eval() local
/third_party/mindspore/mindspore-src/source/tests/ut/python/nn/
Dtest_conv.py132 output_padding=0, argument
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dconv2d_checker.cc157 … [](int64_t output_padding) { return output_padding == 0; }) != output_paddings.end()) { in CheckAttr()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops_generate/
Darg_handler.py95 def to_output_padding(op_name, arg_name, output_padding): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/
Dconv.py1009 output_padding=0, argument
1223 output_padding=0, argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc482 void ConvolutionGrad::set_output_padding(const std::vector<int64_t> &output_padding) { … in set_output_padding()
513 void Convolution::set_output_padding(const std::vector<int64_t> &output_padding) { (voi… in set_output_padding()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/function/
Dnn_func.py5365 output_padding=0): argument
5776 def conv_transpose2d(input, weight, bias=None, stride=1, padding=0, output_padding=0, groups=1, dil… argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
Dnn_ops.py7283 output_padding=0, argument