| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_info.h | 34 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo()
44 Shape slice_shape() const { return slice_shape_; } in slice_shape() function
|
| D | tensor_layout.cc | 468 Arrangement TensorLayout::slice_shape() const { in slice_shape() function in mindspore::parallel::TensorLayout
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | split_layernorm_comm_fp.cc | 72 auto slice_shape = origin_shape; in GetSliceShape() local
85 ShapeVector slice_shape = GetSliceShape(src_cnode_shape, slice_axis); in NewCNodeAndCloneAttrsSetSliceAbstract() local
98 auto slice_shape = common::AnfAlgo::GetOutputInferShape(input_node, LongToSize(index)); in NewTupleGetItemCNodeAndSetAbstract() local
111 auto slice_shape = GetSliceShape(common::AnfAlgo::GetOutputInferShape(src_cnode, kIndex0)); in NewLayerNormCNodeAndCloneAttrsSetSliceAbstract() local
142 ShapeVector slice_shape = GetSliceShape(input_shape); in NewSplitCNodeAndSetAbstract() local
|
| D | slice_activation_in_cell_share_recompute.cc | 129 std::vector<int64_t> slice_shape = out_shape_element; in CreateSliceNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | get_tuple_index_info.cc | 144 static ShapeVector ComputeSliceShape(const ShapeVector &slice_shape, size_t broadcast_shape_len, si… in ComputeSliceShape()
197 [](const ShapeVector &slice_shape) { in ConstGetTupleIndexInfo()
205 [](const ShapeVector &slice_shape) { in ConstGetTupleIndexInfo()
|
| D | strided_slice_v2.cc | 380 … auto slice_shape = CheckAndConvertUtils::GetTensorInputShape("StridedSliceV2", {input_arg}, 0); in CheckAndGetDynamicSliceV2() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/view/ |
| D | split_tensor_strides_calc.cc | 67 std::vector<int64_t> slice_shape = old_shape; in SplitTensorCalc() local
|
| D | split_with_size_strides_calc.cc | 66 std::vector<int64_t> slice_shape = old_shape; in SplitWithSizeCalc() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | normalize_slice_cpu_kernel.cc | 59 …::for_each(input_shapes.begin() + kIndex2, input_shapes.end(), [](const ShapeVector &slice_shape) { in Resize()
|
| D | slice_to_indices_cpu_kernel.cc | 65 …::for_each(input_shapes.begin() + kIndex2, input_shapes.end(), [](const ShapeVector &slice_shape) { in Resize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | get_parallel_info.cc | 279 const auto &slice_shape = tensor_layout->base_slice_shape().array(); in GetParameterLayoutFromGraph() local
307 const auto &slice_shape = layout->get_slice_shape(); in GetParameterLayoutFromResource() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/load_mindir/ |
| D | load_model.h | 38 void set_slice_shape(const std::vector<int64_t> &slice_shape) { slice_shape_ = slice_shape; } in set_slice_shape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | parameter_manager.cc | 415 Shape slice_shape = parameter_shape[0]; in HandleNoUsedParameter() local
496 auto slice_shape = tensor_layout->base_slice_shape().array(); in SliceParameterObj() local
551 auto slice_shape = tensor_layout->slice_shape().array(); in SliceTensorObj() local
727 auto slice_shape = cloned_from_parameter->user_data<TensorLayout>()->slice_shape().array(); in SetClonedTensorShapeForOptimizer() local
1121 auto slice_shape = target_param_layout->slice_shape().array(); in HandleAdaSumReshape() local
1462 auto slice_shape = tensor_layout->slice_shape().array(); in HandleCameAndAdaFactorOpt() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | parameter.py | 817 from_ckpt, slice_shape=False, slice_num=1): argument
842 def set_data(self, data, slice_shape=False): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | fold_pipeline_transformer.cc | 273 auto slice_shape = tensor_info.slice_shape(); in InsertSend() local
351 Shape slice_shape = tensor_info.slice_shape(); in InsertReceive() local
|
| D | pipeline_transformer.cc | 1017 auto slice_shape = tensor_info.slice_shape(); in InsertSend() local
1063 Shape slice_shape = tensor_info.slice_shape(); in InsertReceive() local
1454 auto slice_shape = tensor_info.slice_shape(); in GenNewSendFromOld() local
1575 auto slice_shape = *(cnode->user_data<Shape>(SLICE_SHAPE)); in GenNewRecvFromOld() local
|
| D | pipeline_interleave.cc | 834 Shape slice_shape; in ModifySendRecvAttr() local
1161 auto slice_shape = shape_ptr->shape(); in HandleSendParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/ |
| D | slice_activation_in_recompute.cc | 150 std::vector<int64_t> slice_shape = out_shape_element; in InsertSliceAllGatherNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/auto_parallel/ |
| D | operator_costmodel.cc | 1168 bool IsDataParallel(const Shape &shape, const Shape &slice_shape, int64_t stage_id) { in IsDataParallel()
1434 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
1475 Shape slice_shape = tensor_info.slice_shape(); in GetBackwardCommCost() local
1500 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
1868 Shape slice_shape = tensor_info.slice_shape(); in GetForwardComputationCost() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | neighbor_exchange_v2_fusion.cc | 160 ShapeVector slice_shape(slice_node_info.base_shape); in CreateSliceNode() local
190 ShapeVector slice_shape(slice_node_info.base_shape); in CreateSplitNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/ |
| D | compile_cache_manager.cc | 59 auto slice_shape = tensor_layout->slice_shape().array(); in BuildLayout() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | strided_slice.cc | 157 auto slice_shape = input_arg->GetShape(); in GetSliceInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | ps_embedding_cache_inserter.cc | 600 const std::vector<int64_t> &slice_shape = param_info->parameter_shape(); in BuildDenseEmbeddingStorages() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_index_py.h | 496 …static ShapeVector ComputeSliceShape(const ShapeVector &slice_shape, size_t broadcast_shape_len, s… in ComputeSliceShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/composite/multitype_ops/ |
| D | _constexpr_utils.py | 813 def compute_slice_shape(slice_shape, broadcast_shape_len, slice_cnt, fancy_position): argument
|