Home
last modified time | relevance | path

Searched defs:data_format (Results 1 – 25 of 124) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/config/
Dsuper_bar_config.json6 "data_format": "format" string
12 "data_format": "format" string
18 "data_format": "format" string
24 "data_format": "format" string
36 "data_format": "format" string
43 "data_format": "format" string
47 "data_format": "format" string
51 "data_format": "format" string
55 "data_format": "format" string
62 "data_format": "format" string
[all …]
/third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/
Dtest_bias_add.py24 def __init__(self, data_format="NCHW"): argument
32 def get_output(x, b, data_format, enable_graph_kernel): argument
39 def test_bias_add(shape1, shape2, data_format, dtype): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dbias_add.cc62 auto data_format = static_cast<mindspore::Format>(data_format_opt.value()); in CheckShapeValid() local
103 auto data_format = data_format_opt.value(); in InferShape() local
138 auto data_format = static_cast<mindspore::Format>(data_format_opt.value()); in CheckValidation() local
Dbias_add_grad.cc53 auto data_format = data_format_opt.value(); in InferShape() local
77 auto data_format = data_format_opt.value(); in CheckValidation() local
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_dynamic_rank/
Dtest_biasadd_dynamic_rank.py28 def __init__(self, data_format="NCHW"): argument
54 def __init__(self, in_shape, indices, dtype=np.float32, data_format="NCHW"): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dfractional_max_pool_grad_with_fixed_ksize.cc76 auto data_format = GetValue<std::string>(primitive->GetAttr(kFormat)); in FractionalMaxPoolGradWithFixedKsizeInferShape() local
151 void FractionalMaxPoolGradWithFixedKsize::Init(const std::string data_format) { set_data_format(dat… in Init()
153 void FractionalMaxPoolGradWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format()
Dfractional_max_pool3d_grad_with_fixed_ksize.cc72 auto data_format = GetValue<std::string>(primitive->GetAttr(kFormat)); in FractionalMaxPool3DGradWithFixedKsizeInferShape() local
190 void FractionalMaxPool3DGradWithFixedKsize::Init(const std::string data_format) { set_data_format(d… in Init()
192 void FractionalMaxPool3DGradWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/_op_impl/_custom_op/
Dcorrection_mul_grad.py52 def correction_mul_grad_compute(dout, x, batch_std, running_std, channel, data_format, kernel_name=… argument
144 def correction_mul_grad_reduce_compute(mul_dx, channel, data_format, kernel_name="correction_mul"): argument
/third_party/mindspore/mindspore-src/source/tests/st/pi_jit/dynamic_shape_operations/
Dmaxpool.py12 def __init__(self, pad_mode="SAME", kernel_size=3, strides=2, data_format='NCHW'): argument
28 data_format='NCHW', pad_mode='SAME', loss=0.001): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/proto/
Dtensor_shape.proto34 int32 data_format = 4; field
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/
Ddeprecated.py34 def __init__(self, data_format="NCHW"): argument
52 def __init__(self, isRef=True, epsilon=1e-5, factor=0.1, data_format="NCHW"): argument
79 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/
D_grad_ops.py222 data_format="NCDHW"): argument
291 data_format="NCHW"): argument
546 def __init__(self, epsilon=0.0001, data_format='NCHW'): argument
556 def __init__(self, epsilon=0.0001, data_format='NCHW'): argument
566 def __init__(self, send_rank_ids, send_lens, recv_rank_ids, recv_lens, data_format, argument
600 def __init__(self, kernel_size, strides, pad_mode="VALID", data_format="NCHW"): argument
667 def __init__(self, kernel_size=1, strides=1, pad_mode="VALID", data_format="NCHW"): argument
684 def __init__(self, kernel_size=1, strides=1, pad_mode="VALID", data_format="NCHW"): argument
753 count_include_pad=True, divisor_override=0, data_format="NCDHW", pad_mode="pad"): argument
786 def __init__(self, kernel_size=1, strides=1, pad_mode="VALID", data_format="NCHW"): argument
[all …]
Dnn_ops.py588 def __init__(self, mode=0, epsilon=1e-5, momentum=0.1, data_format="NCHW"): argument
965 data_format="NCHW"): argument
1174 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1324 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1394 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1500 …self, kernel_size=1, strides=1, pad_mode="VALID", pad_list=0, ceil_mode=None, data_format="NCDHW"): argument
1625 def __init__(self, ksize, strides=0, pads=0, output_shape=(), data_format="NCHW"): argument
1729 def __init__(self, ksize, strides=0, pads=0, output_shape=(), data_format="NCDHW"): argument
1821 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1872 data_format="NCHW"): argument
[all …]
D_ocr_ops.py90 def __init__(self, data_format="NHWC"): argument
239 def __init__(self, data_format="NHWC"): argument
271 def __init__(self, size, data_format="NHWC"): argument
308 def __init__(self, batch_size, data_format="NHWC", pad_mode="REPLICATE", argument
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_batchnorm.cc37 ValuePtr data_format; member
64 auto data_format = param.data_format->ToAbstract(); in TEST_P() local
Dtest_ops_batchnorm_grad.cc36 ValuePtr data_format; member
65 auto data_format = param.data_format->ToAbstract(); in TEST_P() local
Dtest_ops_batch_norm_grad_grad.cc36 ValuePtr data_format; member
72 auto data_format = param.data_format->ToAbstract(); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dfractional_max_pool3d_with_fixed_ksize.cc144 auto data_format = GetValue<std::string>(primitive->GetAttr(kFormat)); in FractionalMaxPool3DWithFixedKsizeInferShape() local
232 const std::string data_format) { in Init()
246 void FractionalMaxPool3DWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format()
Dfractional_max_pool_with_fixed_ksize.cc71 auto data_format = GetValue<std::string>(primitive->GetAttr(kFormat)); in InputCheck() local
195 const std::string data_format) { in Init()
209 void FractionalMaxPoolWithFixedKsize::set_data_format(const std::string data_format) { in set_data_format()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dtransformation_ops_proto.cc22 … const std::vector<int64_t> &input_dims, const std::string &data_format) { in VerifyDepthToSpaceInputShape()
73 std::string data_format; in IMPLEMT_VERIFIER() local
Dfractional_max_pool_grad_with_fixed_ksize_proto.cc24 std::string data_format; in CUST_IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dslice_grad_gpu_kernel.cc215 auto data_format = inputs[1]->format(); in ProccessAttr() local
236 void SliceGradGpuKernelMod::CalcBeginAndSize(const mindspore::Format &data_format, size_t dim) { in CalcBeginAndSize()
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/
Dtest_deformable_offsets_grad.py26 def __init__(self, data_format): argument
/third_party/mindspore/mindspore-src/source/tests/vm_impl/
Dnn_ops_vm_impl.py99 def vm_impl(x, scale, b, mean, variance, reserve, is_training, epsilon, data_format): argument
185 def vm_impl(x, kernel_size, strides, pad_mode, data_format): argument
277 def vm_impl(wx, bias, data_format): argument
290 def vm_impl(dout, data_format): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/
Derror_util.cc63 const std::string &data_format) { in GetInputFormatNotSupportErrMsg()
84 const std::string &data_format) { in GetInputFormatErrMsg()

12345