Searched refs:manual_shape (Results 1 – 2 of 2) sorted by relevance
182 auto manual_shape = (*manual_shape_map)[node_tensor_info.first]; in Save() local183 for (auto dim_pair : manual_shape) { in Save()
2689 std::vector<std::pair<int64_t, int64_t>> manual_shape; in CheckpointStrategy() local2691 …manual_shape.emplace_back(std::make_pair(param_split_shapes[LongToSize(i)], index_offsets[LongToSi… in CheckpointStrategy()2693 manual_shape_map[param_name] = manual_shape; in CheckpointStrategy()