Home
last modified time | relevance | path

Searched refs:_save_slice_info (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py152 if not variable._save_slice_info:
155 slice_name = variable._save_slice_info.full_name
156 elif slice_name != variable._save_slice_info.full_name:
159 (slice_name, variable._save_slice_info.full_name))
163 variable, variable._save_slice_info.spec, name)
166 variable, variable._save_slice_info.spec, name)
244 elif isinstance(var, variables.Variable) and var._save_slice_info:
245 name = var._save_slice_info.full_name
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dcheckpoint_utils.py131 slice_name = v._save_slice_info.full_name
132 elif slice_name != v._save_slice_info.full_name:
134 (slice_name, v._save_slice_info.full_name))
136 v._save_slice_info.spec)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables.py606 if var._save_slice_info:
607 return var._save_slice_info.full_name
681 if not var._save_slice_info:
690 slice_dims = zip(var._save_slice_info.var_offset,
691 var._save_slice_info.var_shape)
695 slice_value = slice_value.reshape(var._save_slice_info.var_shape)
/external/tensorflow/tensorflow/python/kernel_tests/
Dpartitioned_variables_test.py324 self.assertEquals(expected_specs[i], slices[i]._save_slice_info.spec)
358 var1_name = vs1[0]._save_slice_info.full_name
359 var2_name = vs2[0]._save_slice_info.full_name
378 var1_name = vs1[0]._save_slice_info.full_name
379 var2_name = vs2[0]._save_slice_info.full_name
395 var1_name = vs1[0]._save_slice_info.full_name
396 var2_name = vs2[0]._save_slice_info.full_name
/external/tensorflow/tensorflow/python/training/
Dslot_creator.py78 if isinstance(primary, variables.Variable) and primary._save_slice_info:
87 slice_info = primary._save_slice_info
Dcheckpoint_utils.py351 slice_info = v._save_slice_info # pylint:disable=protected-access
/external/tensorflow/tensorflow/python/ops/
Dvariables.py1062 self._save_slice_info = save_slice_info
1065 return self._save_slice_info
1610 self._save_slice_info = None
1643 self._save_slice_info = Variable.SaveSliceInfo(
1647 self._save_slice_info = None
2250 if self._save_slice_info:
2251 var_def.save_slice_info_def.MergeFrom(self._save_slice_info.to_proto(
Dresource_variable_ops.py516 self._save_slice_info = None
679 self._save_slice_info = variables.Variable.SaveSliceInfo(
683 self._save_slice_info = None
941 if self._save_slice_info:
943 self._save_slice_info.to_proto(export_scope=export_scope))
1439 self._save_slice_info = None
/external/tensorflow/tensorflow/contrib/eager/python/
Dparameter_server.py138 self._save_slice_info = None
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py124 self._save_slice_info = None
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py1132 def _save_slice_info(self): member in TPUMirroredVariable
1133 return self.primary._save_slice_info