| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | minimum_grad_cpu_kernel.cc | 27 …::vector<size_t> *cargo, const std::vector<size_t> &shape, const std::vector<size_t> &dout_shape) { in GetCargo() argument
28 int i = dout_shape.size() - 1; in GetCargo()
47 …:vector<size_t> *shape, const std::vector<size_t> &shape_, const std::vector<size_t> &dout_shape) { in GetShape() argument
48 int k = dout_shape.size() - 1; in GetShape()
68 dout_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel()
74 CheckShape(&dout_shape); in InitKernel()
103 const std::vector<size_t> &dout_shape) { in MinimumGradRecTask() argument
104 for (size_t i = 0; i < dout_shape[dim]; i++) { in MinimumGradRecTask()
105 size_t x_i = x_shape[dim] == dout_shape[dim] ? i * x_cargo[dim] : 0; in MinimumGradRecTask()
106 size_t y_i = y_shape[dim] == dout_shape[dim] ? i * y_cargo[dim] : 0; in MinimumGradRecTask()
[all …]
|
| D | maximum_grad_cpu_kernel.cc | 40 dout_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 2); in InitKernel()
46 CheckShape(&dout_shape); in InitKernel()
76 const std::vector<size_t> &y_shape, const std::vector<size_t> &dout_shape) { in MaximumGradRecTask() argument
77 for (size_t i = 0; i < dout_shape[dim]; i++) { in MaximumGradRecTask()
78 size_t x_i = x_shape[dim] == dout_shape[dim] ? i * x_cargo[dim] : 0; in MaximumGradRecTask()
79 size_t y_i = y_shape[dim] == dout_shape[dim] ? i * y_cargo[dim] : 0; in MaximumGradRecTask()
82 if (dim == dout_shape.size() - 1) { in MaximumGradRecTask()
90 y_cargo, dout_cargo, x_shape, y_shape, dout_shape); in MaximumGradRecTask()
95 …::vector<size_t> *cargo, const std::vector<size_t> &shape, const std::vector<size_t> &dout_shape) { in GetCargo() argument
96 int i = dout_shape.size() - 1; in GetCargo()
[all …]
|
| D | maximum_grad_cpu_kernel.h | 44 std::vector<size_t> dout_shape; variable
|
| D | minimum_grad_cpu_kernel.h | 44 std::vector<size_t> dout_shape; variable
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _quant_ops.py | 267 def infer_shape(self, dout_shape, x_shape, alpha_shape, quant_max_shape): argument
268 validator.check("dout shape", dout_shape, "x_shape", x_shape, Rel.EQ, self.name)
271 return dout_shape, alpha_shape
298 def infer_shape(self, dout_shape, x_shape, alpha_shape, quant_max_shape): argument
299 validator.check("dout shape", dout_shape, "x_shape", x_shape, Rel.EQ, self.name)
302 return dout_shape, dout_shape
432 def infer_shape(self, dout_shape, x_shape, alpha_shape, quant_max_shape): argument
433 validator.check("dout shape", dout_shape, "x_shape", x_shape, Rel.EQ, self.name)
434 return dout_shape, alpha_shape
463 def infer_shape(self, dout_shape, x_shape, alpha_shape, quant_max_shape): argument
[all …]
|
| D | _grad_ops.py | 127 def infer_shape(self, x_shape, dout_shape): argument
128 validator.check("x shape", x_shape, "dout shape", dout_shape, Rel.EQ, self.name)
144 def infer_shape(self, x_shape, dout_shape): argument
145 validator.check("x shape", x_shape, "dout shape", dout_shape, Rel.EQ, self.name)
167 def infer_shape(self, x_shape, dout_shape): argument
168 validator.check("x shape", x_shape, "dout shape", dout_shape, Rel.EQ, self.name)
393 dout_shape = doutput['shape']
407 …pad_needed_d = max(0, (dout_shape[2] - 1) * stride_d + dilation_d * (kernel_d - 1) + 1 - x_shape[2…
411 …pad_needed_h = max(0, (dout_shape[3] - 1) * stride_h + dilation_h * (kernel_h - 1) + 1 - x_shape[3…
415 …pad_needed_w = max(0, (dout_shape[4] - 1) * stride_w + dilation_w * (kernel_w - 1) + 1 - x_shape[4…
[all …]
|
| D | nn_ops.py | 8558 dout_shape = doutput['shape']
8572 …pad_needed_d = max(0, (dout_shape[2] - 1) * stride_d + dilation_d * (kernel_d - 1) + 1 - x_size_v[…
8576 …pad_needed_h = max(0, (dout_shape[3] - 1) * stride_h + dilation_h * (kernel_h - 1) + 1 - x_size_v[…
8580 …pad_needed_w = max(0, (dout_shape[4] - 1) * stride_w + dilation_w * (kernel_w - 1) + 1 - x_size_v[…
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | conv2d_backprop_input.cc | 136 …auto dout_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kDoutIndex]->BuildSha… in Conv2DBackpropInputInferShape() local
139 ShapeVector tmp_shape = {dout_shape[0], dout_shape[2], dout_shape[3], dout_shape[1]}; in Conv2DBackpropInputInferShape()
140 auto dout_shape_norm = format == Format::NCHW ? dout_shape : tmp_shape; in Conv2DBackpropInputInferShape()
|
| D | sigmoid_cross_entropy_with_logits_grad.cc | 41 …auto dout_shape = CheckAndConvertUtils::ConvertShapePtrToShapeMap(input_args[kInputIndex2]->BuildS… in SigmoidCrossEntropyWithLogitsGradInfer() local
43 …CheckAndConvertUtils::Check("x_shape", x_shape, kEqual, "dout_shape", dout_shape, prim_name, TypeE… in SigmoidCrossEntropyWithLogitsGradInfer()
|
| D | conv2d_backprop_filter.cc | 36 auto dout_shape = CheckAndConvertUtils::GetTensorInputShape(prim_name, input_args, kDoutIndex); in TransStrideTo4D() local
37 if (!x_shape->IsDynamic() && !dout_shape->IsDynamic()) { in TransStrideTo4D()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | pooling_grad_gpu_kernel.h | 89 auto dout_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2); in InitShape() local
97 … = CHECK_NULL_INPUT(input_shape) || CHECK_NULL_INPUT(input_mask) || CHECK_NULL_INPUT(dout_shape) || in InitShape()
104 CheckTensorSize({input_shape, input_mask, dout_shape, output_shape}); in InitShape()
114 SetDimA(dout_shape, dimAdy, nbDims, data_format); in InitShape()
115 SetStrideA(dout_shape, strideAdy, nbDims, data_format); in InitShape()
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_math_ops.py | 43 dout_shape = F.shape(dout)
44 dout_dim = len(dout_shape)
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | thor_layer.py | 158 dout_shape = self.shape(dout)
159 normalizer = dout_shape[0]
449 dout_shape = self.shape(dout)
451 dout = self.reshape(dout, (-1, dout_shape[1]))
452 dout_shape = self.shape(dout)
453 normalizer = dout_shape[0]
461 dout_shape = self.shape(dout)
462 dout = self.reshape(dout, (dout_shape[0], -1))
463 dout_shape = self.shape(dout)
464 normalizer = dout_shape[1]
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_array_ops.py | 1033 dout_shape = shape_op(dout)
1036 if x_shape == dout_shape:
|
| /third_party/mindspore/tests/ut/python/ops/ |
| D | test_math_ops.py | 153 def infer_shape(self, x_shape, out_shape, dout_shape): argument
|
| D | test_nn_ops.py | 125 def infer_shape(self, x_shape, out_shape, dout_shape): argument
|