Searched refs:shape_0 (Results 1 – 6 of 6) sorted by relevance
/third_party/mindspore/mindspore/core/ops/ |
D | addn.cc | 39 auto shape_0 = elements[0]->BuildShape(); in AddNInferShape() local 40 auto element0_shape_map = CheckAndConvertUtils::ConvertShapePtrToShapeMap(shape_0); in AddNInferShape() 43 if (shape->isa<abstract::Shape>() && shape_0->isa<abstract::Shape>()) { in AddNInferShape() 45 const auto &shape_0_vec = shape_0->cast<abstract::ShapePtr>()->shape(); in AddNInferShape() 49 << " are consistent with the shape of input[0]" << shape_0->ToString(); in AddNInferShape() 53 if (*shape != *shape_0) { in AddNInferShape() 56 << ", shape of input[0]:" << shape_0->ToString(); in AddNInferShape()
|
/third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
D | op_infer.py | 375 shape_0 = list(self.inputs[0].shape) 377 check_nd(shape_0, 4) 383 n, h, w, out_channel = shape_0[0], shape_0[1], shape_0[2], shape_1[0] 414 shape_0 = list(self.inputs[0].shape) 416 if len(shape_0) != 2 or len(shape_1) != 2: 417 …raise GKException("MatMul's inputs shape must be 2D, but got {}, {}".format(len(shape_0), len(shap… 420 m, k1 = (shape_0[-1], shape_0[-2]) if transpose_a else (shape_0[-2], shape_0[-1])
|
/third_party/mindspore/mindspore/_extends/graph_kernel/expanders/ |
D | conv2d.py | 97 shape_0 = self.inputs[0]['shape'] 100 check_nd(shape_0, 4) 103 n0, h0, w0, c0 = shape_0
|
/third_party/mindspore/tests/st/fl/cross_silo_faster_rcnn/src/FasterRcnn/ |
D | rcnn.py | 90 shape_0 = (self.rcnn_fc_out_channels, representation_size) 91 … weights_0 = initializer("XavierUniform", shape=shape_0[::-1], dtype=self.ms_type).to_tensor()
|
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
D | audio_utils.h | 200 auto shape_0 = input->shape()[0]; in LFilter() local 204 std::vector<T> out_vect(shape_0 * shape_1); in LFilter()
|
/third_party/mindspore/mindspore/ops/operations/ |
D | array_ops.py | 5475 shape_0 = [] 5478 shape_0.append(s[0]) 5480 shape_0[0], shape_0[1] = shape_0[1], shape_0[0] 5481 out_shape = tuple(tuple(shape_0) for _ in range(n))
|