Searched refs:manual_shape_map (Results 1 – 4 of 4) sorted by relevance
143 ManualShapeMap *manual_shape_map) { in Save() argument181 MS_EXCEPTION_IF_NULL(manual_shape_map); in Save()182 auto manual_shape = (*manual_shape_map)[node_tensor_info.first]; in Save()
54 …StrategyMap &strategy_map, const TensorInfoMap &tensor_info_map, ManualShapeMap *manual_shape_map);
33 ManualShapeMap *manual_shape_map) { return SUCCESS; } in Save() argument
2658 ManualShapeMap manual_shape_map; in CheckpointStrategy() local2693 manual_shape_map[param_name] = manual_shape; in CheckpointStrategy()2712 …if (StrategyCheckpoint::GetInstance().Save(stra_map, tensor_info_map, &manual_shape_map) != SUCCES… in CheckpointStrategy()