Searched refs:outputs_strategy (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 183 Strategys outputs_strategy; in GetOutputsStrategy() local
188 outputs_strategy.push_back(strategy); in GetOutputsStrategy()
189 return outputs_strategy; in GetOutputsStrategy()
257 Strategys outputs_strategy = GetOutputsStrategy(); in InferTensorInfo() local
258 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
|
| D | reduce_method_info.cc | 357 Strategys outputs_strategy = {output_strategy}; in InferTensorInfo() local
358 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
486 Strategys outputs_strategy = {output_strategy, output_strategy}; in InferTensorInfo() local
487 …if (InferSliceShape(inputs_strategy, outputs_strategy, &inputs_slice_shape, &outputs_slice_shape) … in InferTensorInfo()
|
| D | operator_info.h | 218 Status InferSliceShape(const Strategys &inputs_strategy, const Strategys &outputs_strategy,
|
| D | operator_info.cc | 692 … OperatorInfo::InferSliceShape(const Strategys &inputs_strategy, const Strategys &outputs_strategy, in InferSliceShape() argument
704 if (InferSliceShapeByStrategy(outputs_strategy, outputs_shape_, outputs_slice_shape) != SUCCESS) { in InferSliceShape()
|