Home
last modified time | relevance | path

Searched defs:padding_mode (Results 1 – 19 of 19) sorted by relevance

/third_party/mindspore/mindspore/core/ops/fusion/
Dpad_fusion.cc26 void PadFusion::Init(const PaddingMode &padding_mode, const float constant_value) { in Init()
31 void PadFusion::set_padding_mode(const PaddingMode &padding_mode) { in set_padding_mode()
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dpad_ir.cc32 BorderType padding_mode) in PadOperation()
114 BorderType padding_mode = static_cast<BorderType>(op_params["padding_mode"]); in from_json() local
Drandom_crop_with_bbox_ir.cc34 BorderType padding_mode) in RandomCropWithBBoxOperation()
139 BorderType padding_mode = static_cast<BorderType>(op_params["padding_mode"]); in from_json() local
Drandom_crop_ir.cc32 BorderType padding_mode) in RandomCropOperation()
138 BorderType padding_mode = static_cast<BorderType>(op_params["padding_mode"]); in from_json() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_pad_parser.cc28 mindspore::PaddingMode padding_mode; in Parse() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Drandom_crop_with_bbox_op.h35 …pOp(crop_height, crop_width, pad_top, pad_bottom, pad_left, pad_right, pad_if_needed, padding_mode, in RandomCropOp() argument
Dpad_op.cc29 …(int32_t pad_top, int32_t pad_bottom, int32_t pad_left, int32_t pad_right, BorderType padding_mode, in PadOp()
Drandom_crop_op.cc36 … int32_t pad_left, int32_t pad_right, bool pad_if_needed, BorderType padding_mode, in RandomCropOp()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Dpad_tensorrt.cc47 schema::PaddingMode padding_mode = pad_primitive->padding_mode(); in IsSupport() local
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pynative/
Dops_test.py55 dilation=1, groups=1, padding_mode='zeros'): argument
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dprimitive_mapper.cc119 std::string padding_mode = "CALCULATED"; in AdjustOnnxPoolAttr() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/image/
Dbindings.cc268 BorderType padding_mode) { in __anon574f76b42402()
331 const std::vector<uint8_t> &fill_value, BorderType padding_mode) { in __anon574f76b42e02()
345 … bool pad_if_needed, const std::vector<uint8_t> &fill_value, BorderType padding_mode) { in __anon574f76b43002()
/third_party/mindspore/mindspore/dataset/vision/
Dpy_transforms_util.py400 def random_crop(img, size, padding, pad_if_needed, fill_value, padding_mode): argument
832 def pad(img, padding, fill_value, padding_mode): argument
Dc_transforms.py661 def __init__(self, padding, fill_value=0, padding_mode=Border.CONSTANT): argument
907 …def __init__(self, size, padding=None, pad_if_needed=False, fill_value=0, padding_mode=Border.CONS… argument
1032 …def __init__(self, size, padding=None, pad_if_needed=False, fill_value=0, padding_mode=Border.CONS… argument
Dpy_transforms.py407 …def __init__(self, size, padding=None, pad_if_needed=False, fill_value=0, padding_mode=Border.CONS… argument
959 def __init__(self, padding, fill_value=0, padding_mode=Border.CONSTANT): argument
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/api/
Dvision.cc444 Pad::Pad(std::vector<int32_t> padding, std::vector<uint8_t> fill_value, BorderType padding_mode) in Pad()
532 std::vector<uint8_t> fill_value, BorderType padding_mode) in RandomCrop()
579 std::vector<uint8_t> fill_value, BorderType padding_mode) in RandomCropWithBBox()
/third_party/mindspore/tests/vm_impl/
Dvm_me.py232 dilation=1, groups=1, padding_mode='zeros'): argument
/third_party/mindspore/mindspore/ops/operations/
D_thor_ops.py555 padding_mode="SAME", argument
/third_party/mindspore/mindspore/lite/mindir/src/
Dmindir.cc2580 PaddingMode padding_mode, float constant_value) { in MindIR_PadFusion_CreatePrimitive()
2648 void MindIR_PadFusion_SetPaddingMode(PrimitivePtr *primitive, PaddingMode padding_mode) { in MindIR_PadFusion_SetPaddingMode()