diff --git a/mindspore/core/abstract/prim_nn.cc b/mindspore/core/abstract/prim_nn.cc index 31cceaa2f66..6e54d004367 100644 --- a/mindspore/core/abstract/prim_nn.cc +++ b/mindspore/core/abstract/prim_nn.cc @@ -296,7 +296,7 @@ AbstractBasePtr InferImplConv2D(const AnalysisEnginePtr &, const PrimitivePtr &p CheckShapeAnyAndPositive(op_name + " w_shape", w_shape); CheckShapeAllPositive(op_name + " w_min_shape", w_min_shape); CheckShapeAllPositive(op_name + " w_max_shape", w_max_shape); - int64_t n_axis = 0; + const int64_t n_axis = 0; int64_t c_axis = 1; int64_t h_axis = 2; int64_t w_axis = 3; diff --git a/mindspore/ops/operations/math_ops.py b/mindspore/ops/operations/math_ops.py index d51d8d61cdd..c6e522dbcbf 100644 --- a/mindspore/ops/operations/math_ops.py +++ b/mindspore/ops/operations/math_ops.py @@ -4548,11 +4548,11 @@ class IndexAdd(PrimitiveWithInfer): Inputs: - **input_x** (Parameter) - The input tensor to add to, with data type float64, float32, float16, int32, int16, int8, uint8. - - **indices** (Tensor) - The index of `input_x` on the `axis`th dimension to add to, with data type int32. - The `indices` must be 1D with the same size as the size of the `axis`th dimension of `input_y`. The values - of `indices` should be in the range of 0 to the size of the `axis`th dimension of `input_x`. + - **indices** (Tensor) - The index of `input_x` on the `axis` th dimension to add to, with data type int32. + The `indices` must be 1D with the same size as the size of the `axis` th dimension of `input_y`. The values + of `indices` should be in the range of 0 to the size of the `axis` th dimension of `input_x`. - **input_y** (Tensor) - The input tensor with the value to add. Must have same data type as `input_x`. - The shape must be the same as `input_x` except the `axis`th dimension. + The shape must be the same as `input_x` except the `axis` th dimension. Outputs: Tensor, has the same shape and dtype as input_x. @@ -4564,7 +4564,7 @@ class IndexAdd(PrimitiveWithInfer): ValueError: If axis is out of `input_x` rank's range. ValueError: If `input_x` rank is not the same as `input_y` rank. ValueError: If size of `indices` is not equal to dimension of y[axis]. - ValueError: If `input_y`'s shape is not the same as `input_x` except the `axis`th dimension. + ValueError: If `input_y`'s shape is not the same as `input_x` except the `axis` th dimension. Supported Platforms: ``GPU``