Home
last modified time | relevance | path

Searched defs:tensor_layout (Results 1 – 23 of 23) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_info.h34 TensorInfo(const TensorLayout &tensor_layout, Shape shape, Shape slice_shape) in TensorInfo()
36 explicit TensorInfo(const TensorLayout &tensor_layout) : tensor_layout_(tensor_layout) { in TensorInfo()
43 TensorLayout tensor_layout() const { return tensor_layout_; } in tensor_layout() function
Dtensor_layout.h84 bool IsSameTensorShape(const TensorLayout &tensor_layout) const { in IsSameTensorShape()
88 bool IsSameDeviceArrangement(const TensorLayout &tensor_layout) const { in IsSameDeviceArrangement()
92 …bool IsSameTensorMap(const TensorLayout &tensor_layout) const { return (tensor_map_ == tensor_layo… in IsSameTensorMap()
Dredistribution_operator_infer.cc25 Status RedistributionOperatorInfer::Init(const TensorLayout &tensor_layout, const Map &out_tensor_m… in Init()
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/
Dtensor_layout_test.cc53 TensorLayout tensor_layout; in ReshapeExpandDeviceArrangementTestFunction() local
173 TensorLayout tensor_layout; in ExpandTensorShapeTestFunction() local
226 TensorLayout tensor_layout; in TEST_F() local
244 TensorLayout tensor_layout; in TEST_F() local
267 TensorLayout tensor_layout; in RemoveElementEqualToOneInDeviceArrangementTestFunction() local
324 TensorLayout tensor_layout; in TEST_F() local
342 TensorLayout tensor_layout; in TEST_F() local
363 TensorLayout tensor_layout; in TEST_F() local
384 TensorLayout tensor_layout; in TEST_F() local
Dconstruct_operator_test.cc93 TensorLayout tensor_layout; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dcame_parallel_handler.h59 TensorLayoutPtr tensor_layout; variable
Dparameter_manager.cc432 auto tensor_layout = param_ptr->user_data<parallel::TensorLayout>(); in IsFullySplitParameter() local
474 void SliceParameterObj(const ParameterPtr &parameter, const TensorLayoutPtr &tensor_layout) { in SliceParameterObj()
534 void SliceTensorObj(const ParameterPtr &parameter, const TensorLayoutPtr &tensor_layout, size_t ran… in SliceTensorObj()
720 auto tensor_layout = cloned_from_parameter->user_data<TensorLayout>(); in SetClonedTensorShapeForOptimizer() local
811 auto tensor_layout = param->user_data<TensorLayout>(); in IsOriginWeight() local
1460 auto tensor_layout = param->user_data<TensorLayout>(); in HandleCameAndAdaFactorOpt() local
Dstep_parallel.cc665 TensorLayout tensor_layout; in SplitTensor() local
733 TensorLayout tensor_layout = tensor_info.tensor_layout(); in SplitTensorList() local
761 TensorLayout tensor_layout = tensor_info.tensor_layout(); in SplitTensorList() local
1615 TensorLayout tensor_layout; in SetParallelShape() local
1721 auto tensor_layout = parameter->user_data<TensorLayout>(); in CoverSliceShape() local
Dstep_assigned_parallel.cc614 TensorLayout tensor_layout = tensorinfo_in.tensor_layout(); in SetParallelShape() local
Dstep_parallel_utils.cc2715 auto tensor_layout = parameter_ptr->user_data<TensorLayout>(); in SetSharedParameterFlag() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/parallel_optimizer/
Dopt_param_mgr.cc33 std::string ShardOptGroup(const AnfNodePtr &parameter, TensorLayout *const tensor_layout, in ShardOptGroup()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dweight_quant_batch_matmul_info.cc501 TensorLayout tensor_layout; in InferTensorLayout() local
526 TensorLayout tensor_layout = inputs_layout.at(i); in InferTensorInfo() local
Dactivation_info.cc355 auto tensor_layout = inputs_tensor_info_[kIndex0].tensor_layout(); in CheckInputLayout() local
Doperator_info.cc974 Operator CreateGetTensorSliceOp(const TensorLayout &tensor_layout) { in CreateGetTensorSliceOp()
1079 Status OperatorInfo::CreateGroupForOptShard(TensorLayout *tensor_layout, std::vector<Group> *groups… in CreateGroupForOptShard()
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_parallel_serialization.py75 def _convert_to_layout(param_name, tensor_layout): argument
D_utils.py428 def _remove_repeated_slices(tensor_layout): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dstrategy_checkpoint_info.cc146 TensorLayoutPtr tensor_layout = node_tensor_info.second; in to_protobuf() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dslice_activation_in_cell_share_recompute.cc56 auto tensor_layout = output_info[0].tensor_layout(); in InferRepeatedRankList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dslice_activation_in_recompute.cc61 auto tensor_layout = output_info[0].tensor_layout(); in InferRepeatedRankList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dget_parallel_info.cc273 auto tensor_layout = para->user_data<parallel::TensorLayout>(); in GetParameterLayoutFromGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/
Dcompile_cache_manager.cc50 auto tensor_layout = para->user_data<parallel::TensorLayout>(); in BuildLayout() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dfold_pipeline_transformer.cc350 auto tensor_layout = tensor_info.tensor_layout(); in InsertReceive() local
Dpipeline_transformer.cc1062 auto tensor_layout = tensor_info.tensor_layout(); in InsertReceive() local
1587 auto tensor_layout = node->user_data<TensorLayout>(); in GenNewRecvFromOld() local