Home
last modified time | relevance | path

Searched refs:TensorLayoutPtr (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/
Dparallel_strategy_checkpoint.h34 using TensorLayoutPtr = std::shared_ptr<TensorLayout>; variable
35 using TensorInfoMap = std::unordered_map<std::string, TensorLayoutPtr>;
Dparallel_strategy_checkpoint.cc163 TensorLayoutPtr tensor_layout = node_tensor_info.second; in Save()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.h32 using TensorLayoutPtr = std::shared_ptr<TensorLayout>; variable