Home
last modified time | relevance | path

Searched refs:shape_0 (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Daddn.cc39 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/
Dop_infer.py375 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/
Dconv2d.py97 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/
Drcnn.py90 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/
Daudio_utils.h200 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/
Darray_ops.py5475 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))