forked from mindspore-Ecosystem/mindspore
commit
5c7bbd0074
|
@ -57,8 +57,8 @@ abstract::ShapePtr InplaceOpV2InferShape(const PrimitivePtr &primitive,
|
|||
auto indices = CheckAndConvertUtils::ConvertShapePtrToShapeMap(indices_shape_ptr)[kShape];
|
||||
|
||||
// check indices
|
||||
(void)CheckAndConvertUtils::CheckValue<size_t>("size of indices", indices.at(0), kEqual, "v.shape[0]",
|
||||
v_in_shape.at(0), primitive->name());
|
||||
(void)CheckAndConvertUtils::CheckValue<size_t>("size of indices", LongToSize(indices.at(0)), kEqual, "v.shape[0]",
|
||||
LongToSize(v_in_shape.at(0)), primitive->name());
|
||||
|
||||
return x_shape_ptr->cast<abstract::ShapePtr>();
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue