Home
last modified time | relevance | path

Searched refs:BaseSaverBuilder (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dsaver_test_utils.py75 class CustomSaveable(saver_module.BaseSaverBuilder.SaveableObject):
81 saver_module.BaseSaverBuilder.SaveSpec(tensors[0], "",
83 saver_module.BaseSaverBuilder.SaveSpec(tensors[1], "",
Dsaver.py70 class BaseSaverBuilder(object): class
559 class BulkSaverBuilder(BaseSaverBuilder):
Dsaver_test.py2988 saver_module.BaseSaverBuilder.ResourceVariableSaveable):
3085 class _CountingSaveable(saver_module.BaseSaverBuilder.SaveableObject):
3095 [saver_module.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py70 class QuantileAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
83 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
165 class _TreeEnsembleSavable(saver.BaseSaverBuilder.SaveableObject):
183 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
185 saver.BaseSaverBuilder.SaveSpec(serialized, slice_spec,
Dtensor_forest_ops.py26 class TreeVariableSaveable(saver.BaseSaverBuilder.SaveableObject):
44 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name)]
Dlookup_ops.py42 from tensorflow.python.training.saver import BaseSaverBuilder
1804 class _Saveable(BaseSaverBuilder.SaveableObject):
1810 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
1811 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
2100 class _Saveable(BaseSaverBuilder.SaveableObject):
2106 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
2107 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py36 from tensorflow.python.training.saver import BaseSaverBuilder
789 class _IteratorSaveable(BaseSaverBuilder.SaveableObject):
800 BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/core/protobuf/
Dsaver.proto35 // Usually, each subclass of BaseSaverBuilder works with a particular
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py714 class _MirroredSaveable(saver.BaseSaverBuilder.ResourceVariableSaveable):
987 class _SyncOnReadSaveable(saver.BaseSaverBuilder.SaveableObject):
999 spec = saver.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_test.py224 class _MirroringSaveable(saver_lib.BaseSaverBuilder.SaveableObject):
230 spec = saver_lib.BaseSaverBuilder.SaveSpec(