Searched refs:inputs_slice_shape (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reduce_method_info.cc | 353 Shapes inputs_slice_shape, outputs_slice_shape; in InferTensorInfo() local
358 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
361 Shape input_slice_shape = inputs_slice_shape.at(0); in InferTensorInfo()
482 Shapes inputs_slice_shape, outputs_slice_shape; in InferTensorInfo() local
487 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
490 Shape input_slice_shape = inputs_slice_shape.at(0); in InferTensorInfo()
|
| D | reshape_info.cc | 255 Shapes inputs_slice_shape, outputs_slice_shape; in InferTensorInfo() local
258 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
269 Shape slice_shape_in = inputs_slice_shape.at(0); in InferTensorInfo()
|
| D | operator_info.cc | 693 Shapes *inputs_slice_shape, Shapes *outputs_slice_shape) { in InferSliceShape() argument
694 if (inputs_slice_shape == nullptr || outputs_slice_shape == nullptr) { in InferSliceShape()
699 if (InferSliceShapeByStrategy(inputs_strategy, inputs_shape_, inputs_slice_shape) != SUCCESS) { in InferSliceShape()
706 inputs_slice_shape->clear(); in InferSliceShape()
|
| D | operator_info.h | 219 Shapes *inputs_slice_shape, Shapes *outputs_slice_shape);
|