Searched refs:TensorInfoMap (Results 1 – 4 of 4) sorted by relevance
35 using TensorInfoMap = std::unordered_map<std::string, TensorLayoutPtr>; variable54 …Status Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, ManualShapeMap …
142 Status StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_m… in Save()
32 Status StrategyCheckpoint::Save(const StrategyMap &strategy_map, const TensorInfoMap &tensor_info_m… in Save()
2657 TensorInfoMap tensor_info_map; in CheckpointStrategy()