Searched refs:_get_save_slice_info (Results 1 – 3 of 3) sorted by relevance
1064 def _get_save_slice_info(self): member in Variable2503 if not all(v._get_save_slice_info() is not None for v in variable_list):2510 if v._get_save_slice_info().full_shape != shape:2514 % (shape, str([v._get_save_slice_info().full_shape])))2516 variable_list, key=lambda v: v._get_save_slice_info().var_offset)
254 slice_info = v._get_save_slice_info()
1135 def _get_save_slice_info(self): member in TPUMirroredVariable1136 return self.primary._get_save_slice_info()