Home
last modified time | relevance | path

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

12

/third_party/mindspore/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/nn/layer/
Dpooling.py29 def __init__(self, kernel_size, stride, pad_mode, data_format="NCHW"): argument
133 def __init__(self, kernel_size=1, stride=1, pad_mode="valid", data_format="NCHW"): argument
295 data_format="NCHW"): argument
Dconv.py48 data_format='NCHW', argument
235 data_format='NCHW'): argument
603 data_format='NCDHW'): argument
809 data_format='NCDHW'): argument
Dnormalization.py60 data_format='NCHW'): argument
468 data_format='NCHW'): argument
570 data_format='NCDHW'): argument
/third_party/mindspore/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
Dbatchnorm_fold.py102 momentum=0.9, epsilon=1e-5, is_training=True, freeze_bn=0, data_format="NCHW", argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/proto/
Dtensor_shape.proto34 int32 data_format = 4; field
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dslice_grad_gpu_kernel.h59 auto data_format = AnfAlgo::GetInputFormat(kernel_node, 0); in Init() local
110 void CalcBeginAndSize(const std::string data_format) { in CalcBeginAndSize()
Dslice_gpu_kernel.h116 auto data_format = AnfAlgo::GetInputFormat(kernel_node, 0); in Init() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/expanders/
Dbias_add.cc48 auto data_format = GetValue<std::string>(attrs_["format"]); in Expand() local
/third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/
Dconfig.py30 data_format = 1 variable in DataConfig
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dmax_pool3d_grad_grad_fission.cc36 auto data_format = AnfAlgo::GetNodeAttr<std::string>(cnode, "format"); in CreateTensor() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_nn.cc33 int64_t data_format; in GetAndCheckFormat() local
153 int64_t data_format = GetAndCheckFormat(data_format_ptr); in InferImplBatchNorm() local
267 int64_t data_format = GetAndCheckFormat(primitive->GetAttr("format")); in InferImplConv2D() local
360 int64_t data_format = Format::NCHW; in InferImplBiasAdd() local
/third_party/mindspore/mindspore/core/ops/
Dbias_add.cc53 int64_t data_format = Format::NCHW; in InferShape() local
/third_party/mesa3d/src/gallium/drivers/r600/sfn/
Dsfn_instruction_fetch.h85 EVTXDataFormat data_format() const { return m_data_format;} in data_format() function
Dsfn_instruction_fetch.cpp81 EVTXDataFormat data_format, in FetchInstruction()
/third_party/mindspore/mindspore/ops/operations/
Dnn_ops.py905 def __init__(self, mode=0, epsilon=1e-5, momentum=0.1, data_format="NCHW"): argument
1258 def __init__(self, is_training=False, epsilon=1e-5, momentum=0.1, data_format="NCHW"): argument
1403 data_format="NCHW"): argument
1629 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1753 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1821 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
1908 …self, kernel_size=1, strides=1, pad_mode="VALID", pad_list=0, ceil_mode=None, data_format="NCDHW"): argument
2069 def __init__(self, kernel_size=1, strides=1, pad_mode="valid", data_format="NCHW"): argument
2153 data_format="NCHW"): argument
2253 pad_list=None, mode=1, stride=1, dilation=1, group=1, data_format="NCHW"): argument
[all …]
D_grad_ops.py182 def __init__(self, is_training=False, epsilon=1e-5, data_format='NCHW'): argument
217 def __init__(self, data_format="NCHW"): argument
346 data_format="NCDHW"): argument
464 data_format="NCHW"): argument
763 def __init__(self, kernel_size, strides, pad_mode="VALID", data_format="NCHW"): argument
827 def __init__(self, kernel_size=1, strides=1, pad_mode="VALID", data_format="NCHW"): argument
856 count_include_pad=True, divisor_override=0, data_format="NCDHW"): argument
879 def __init__(self, kernel_size=1, strides=1, pad_mode="VALID", data_format="NCHW"): argument
964 pad_mode='VALID', pad_list=0, data_format="NCDHW"): argument
1007 …self, kernel_size=(1, 1, 1, 1, 1), strides=(1, 1, 1, 1, 1), pad_mode='VALID', data_format="NCDHW"): argument
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dmodel.py309 def __init__(self, name, shape, dtype, data_format=DataFormat.DEFAULT, para_type=0): argument
343 def __init__(self, name, dtype, value, data_format=DataFormat.DEFAULT): argument
Dmodel_builder.py72 …def tensor(self, shape, dtype, data_format="DefaultFormat", name=None, para_type=Tensor.PARA_NONE): argument
/third_party/libexif/libexif/
Dexif-loader.c69 ExifLoaderDataFormat data_format; member
/third_party/mesa3d/src/amd/vulkan/
Dradv_formats.c557 unsigned data_format, num_format; in radv_is_storage_image_format_supported() local
612 unsigned data_format, num_format; in radv_is_buffer_format_supported() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dpooling_grad_gpu_kernel.h91 auto data_format = AnfAlgo::GetInputFormat(kernel_node, 0); in InitShape() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dconv2d_unify_mindir.cc56 int64_t data_format; in NeedUpdate() local
/third_party/mesa3d/src/gallium/drivers/radeonsi/
Dsi_state.c2210 unsigned data_format; in si_is_vertex_format_supported() local
3758 unsigned num_format, data_format; in si_make_buffer_descriptor() local
3983 unsigned num_format, data_format, type, num_samples; in si_make_texture_descriptor() local
4171 uint32_t data_format, num_format; in si_make_texture_descriptor() local
4871 unsigned data_format, num_format; in si_create_vertex_elements() local

12