Searched refs:shape_base (Results 1 – 5 of 5) sorted by relevance
115 ShapePtr shape_base = tensor_base->shape(); in CheckShapeSame() local116 MS_EXCEPTION_IF_NULL(shape_base); in CheckShapeSame()120 if (*shape != *shape_base) { in CheckShapeSame()122 << " are not consistent with second arg shape " << shape_base->ToString(); in CheckShapeSame()124 return shape_base; in CheckShapeSame()
949 ShapeVector shape_base = tensor_base->shape()->shape(); in InferImplConcat() local950 int64_t rank_base = SizeToLong(shape_base.size()); in InferImplConcat()953 CheckMinMaxShape(shape_base, &min_shape_base, &max_shape_base); in InferImplConcat()964 int64_t all_shp = shape_base[axis_value]; in InferImplConcat()984 if (j != axis_value && shape_tensor[j] != shape_base[j]) { in InferImplConcat()988 if (all_shp == -1 || shape_base[axis_value] == -1) { in InferImplConcat()
112 auto shape_base = shape_abstract->BuildShape(); in InferShape() local113 MS_EXCEPTION_IF_NULL(shape_base); in InferShape()114 auto shape = shape_base->cast<abstract::ShapePtr>(); in InferShape()
94 self.shape_base = P.Shape()139 dtype_tensor = self.fill_base(self.dtype, self.shape_base(value), 0.0)
112 self.shape_base = P.Shape()183 broadcast_shape = self.shape_base(arg)188 broadcast_shape = self.shape_base(arg + broadcast_shape_tensor)