| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_info.h | 35 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo()
45 Shape slice_shape() const { return slice_shape_; } in slice_shape() function
|
| D | tensor_redistribution.cc | 211 Shape slice_shape = op_cost.second; in ComputeCost() local
|
| D | tensor_layout.cc | 322 Arrangement TensorLayout::slice_shape() const { in slice_shape() function in mindspore::parallel::TensorLayout
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.h | 36 Shape slice_shape; member
|
| D | parameter_manager.cc | 362 Shape slice_shape = parameter_shape[0]; in HandleNoUsedParameter() local
514 auto slice_shape = cloned_from_parameter->user_data<TensorLayout>()->slice_shape().array(); in SetClonedTensorShapeForOptimizer() local
571 auto slice_shape = tensor_layout->slice_shape().array(); in HandleAdaFactorOpt() local
|
| D | step_parallel.cc | 1681 Shape slice_shape = tensor_layout.slice_shape().array(); in SetParallelShape() local
2381 auto slice_shape = loss_grad_layout.slice_shape().array(); in SplitSens() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | get_parallel_info.cc | 44 auto slice_shape = tensor_layout->slice_shape().array(); in GetParameterLayout() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 1085 bool IsDataParallel(const Shape &shape, const Shape &slice_shape, int64_t stage_id) { in IsDataParallel()
1350 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
1391 Shape slice_shape = tensor_info.slice_shape(); in GetBackwardCommCost() local
1416 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
1784 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | reshape_info.cc | 246 Shape slice_shape; in InferTensorInfo() local
|
| D | matmul_info.cc | 580 auto slice_shape = one_input_tensor.slice_shape(); in CheckForTensorSliceValid() local
|
| D | gather_v2_p_info.cc | 323 …auto slice_shape = param_shape.at(param_shape.size() - 1) / param_strategy.at(param_strategy.size(… in CheckStrategy() local
|
| D | operator_info.cc | 642 Shape slice_shape; in GetSliceShape() local
685 Shape slice_shape = GetSliceShape(shapes.at(i), strategys.at(i)); in InferSliceShapeByStrategy() local
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 490 def set_data(self, data, slice_shape=False): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 592 auto slice_shape = tensor_info.slice_shape(); in InsertSend() local
648 Shape slice_shape = tensor_info.slice_shape(); in InsertReceive() local
|
| /third_party/mindspore/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 813 def compute_slice_shape(slice_shape, broadcast_shape_len, slice_cnt, fancy_position): argument
|
| /third_party/python/Lib/test/ |
| D | test_buffer.py | 357 def slice_shape(lst, slices): function
|