| /third_party/mindspore/mindspore/ops/_op_impl/_custom_op/ |
| D | img2col_impl.py | 822 n_shape, c1_shape, h_shape, w_shape, c0_shape = input_x_shape
853 …output_tmp = [n_shape * int(h_shape // stride_h) * int(w_shape // stride_w), filter_h * filter_w *…
863 if tuple([h_shape, w_shape]) == (224, 224):
866 if tuple([h_shape, w_shape]) == (56, 56):
869 if tuple([h_shape, w_shape]) == (28, 28):
872 if tuple([h_shape, w_shape]) == (14, 14):
875 if tuple([h_shape, w_shape]) == (7, 7):
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3971 def infer_shape(self, x_shape, h_shape, c_shape, w_shape): argument
3976 validator.check_equal_int(len(h_shape), 3, "h rank", self.name)
3977 validator.check("h_shape", h_shape, "c_shape", c_shape, Rel.EQ, self.name)
3979 … validator.check_int(h_shape[0], self.num_layers * self.num_directions, Rel.EQ, "h[0]", self.name)
3980 validator.check_equal_int(h_shape[1], x_shape[1], "h[1]", self.name)
3981 validator.check_int(h_shape[2], self.hidden_size, Rel.EQ, "h[2]", self.name)
3988 return y_shape, h_shape, c_shape, reserved_shape, state_shape
7588 def infer_shape(self, x_shape, h_shape, c_shape, w_shape, b_shape): argument
7590 validator.check_int(len(h_shape), 2, Rel.EQ, "h rank", self.name)
7594 validator.check("x_shape[0]", x_shape[0], "h_shape[0]", h_shape[0], Rel.EQ, self.name)
[all …]
|
| D | _grad_ops.py | 1382 def infer_shape(self, x_shape, w_shape, b_shape, y_shape, init_h_shape, init_c_shape, h_shape, argument
1394 validator.check("h_shape", h_shape, "excepted shape", valid_shape, Rel.EQ, self.name)
1492 def infer_shape(self, x_shape, winput_shape, whidden_shape, y_shape, init_h_shape, h_shape, argument
1513 validator.check("h_shape", h_shape, "excepted shape", valid_shape, Rel.EQ, self.name)
2118 def infer_shape(self, x_shape, h_shape, dgate_shape): argument
2120 validator.check("h rank", len(h_shape), " x rank", len(x_shape), Rel.EQ, self.name)
2122 validator.check("h_shape[0]", h_shape[0], "x_shape[0]", x_shape[0], Rel.EQ, self.name)
2123 … validator.check("dgate_shape[0]", dgate_shape[0], "h_shape[0]", h_shape[0], Rel.EQ, self.name)
2124 …validator.check("dgate_shape[1]", dgate_shape[1], "4*h_shape[1]", 4 * h_shape[1], Rel.EQ, self.nam…
2126 hidden_size = h_shape[1]
|
| D | array_ops.py | 5714 def __infer__(self, h_indices, h_values, h_shape, truth_indices, truth_values, truth_shape): argument
5715 validator.check_valid_input('hypothesis_shape', h_shape['value'], self.name)
5717 args_int = {"hypothesis_indices": h_indices['dtype'], "hypothesis_shape": h_shape['dtype'],
5727 … validator.check("hypothesis_shape rank", len(h_shape['shape']), "expected", 1, Rel.EQ, self.name)
5732 validator.check("hypothesis_shape", h_shape['shape'][0],
5736 validator.check("hypothesis_shape", h_shape['shape'][0],
5738 hypothesis_shape_v = h_shape['value'].asnumpy()
|
| /third_party/mindspore/mindspore/ccsrc/common/ |
| D | trans.cc | 671 int64_t h_shape = shape[shape.size() - kDim2]; in FracNZDeviceDynamicShape() local
673 …int64_t h1 = (h_shape == Shape::SHP_ANY) ? Shape::SHP_ANY : (h_shape - 1) / SizeToLong(kCubeSize) … in FracNZDeviceDynamicShape()
699 const int64_t h_shape = shape.at(kN); in FracNZLSTMDeviceDynamicShape() local
701 const int64_t h = (h_shape == Shape::SHP_ANY) ? Shape::SHP_ANY : h_shape / c0; in FracNZLSTMDeviceDynamicShape()
704 if (h_shape != Shape::SHP_ANY && i_shape != Shape::SHP_ANY) { in FracNZLSTMDeviceDynamicShape()
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1778 int64_t h_shape = 1; in ExportMergeLayerNorm() local
1785 new_input_shape.push_back(h_shape); in ExportMergeLayerNorm()
|