Searched refs:_set_save_slice_info (Results 1 – 6 of 6) sorted by relevance
654 v0._set_save_slice_info(656 v1._set_save_slice_info(702 v0._set_save_slice_info(704 v1._set_save_slice_info(717 v0._set_save_slice_info(734 v0._set_save_slice_info(736 v1._set_save_slice_info(738 v2._set_save_slice_info(740 v3._set_save_slice_info(
88 slot._set_save_slice_info(variables.Variable.SaveSliceInfo(
426 v2._set_save_slice_info(1872 v2._set_save_slice_info(
1138 def _set_save_slice_info(self, save_slice_info): member in TPUMirroredVariable1139 return self.primary._set_save_slice_info(save_slice_info)
1056 def _set_save_slice_info(self, save_slice_info): member in Variable
786 var._set_save_slice_info(variables.Variable.SaveSliceInfo(