Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py35 class StatsAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
47 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
49 saver.BaseSaverBuilder.SaveSpec(num_updates, slice_spec,
51 saver.BaseSaverBuilder.SaveSpec(partition_ids, slice_spec,
53 saver.BaseSaverBuilder.SaveSpec(feature_ids, slice_spec,
55 saver.BaseSaverBuilder.SaveSpec(gradients, slice_spec,
57 saver.BaseSaverBuilder.SaveSpec(hessians, slice_spec,
Dmodel_ops.py43 class TreeEnsembleVariableSavable(saver.BaseSaverBuilder.SaveableObject):
60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
Dquantile_ops.py42 class QuantileAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
54 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
/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.py71 class BaseSaverBuilder(object): class
563 class BulkSaverBuilder(BaseSaverBuilder):
Dsaver_test.py2795 saver_module.BaseSaverBuilder.ResourceVariableSaveable):
2892 class _CountingSaveable(saver_module.BaseSaverBuilder.SaveableObject):
2902 [saver_module.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py64 class QuantileAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject):
77 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
159 class _TreeEnsembleSavable(saver.BaseSaverBuilder.SaveableObject):
177 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
179 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.py41 from tensorflow.python.training.saver import BaseSaverBuilder
1732 class _Saveable(BaseSaverBuilder.SaveableObject):
1738 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
1739 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
2023 class _Saveable(BaseSaverBuilder.SaveableObject):
2029 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
2030 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dsplit_dependency.py27 class _CallbackSaveable(saver_lib.BaseSaverBuilder.SaveableObject):
32 spec = saver_lib.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/
Dstats_ops.py50 class FertileStatsVariableSavable(saver.BaseSaverBuilder.SaveableObject):
68 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
Dmodel_ops.py53 class TreeVariableSavable(saver.BaseSaverBuilder.SaveableObject):
71 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py69 class _Saveable(saver_lib.BaseSaverBuilder.SaveableObject):
75 saver_lib.BaseSaverBuilder.SaveSpec(serialized_iterator, "",
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py34 from tensorflow.python.training.saver import BaseSaverBuilder
656 class _IteratorSaveable(BaseSaverBuilder.SaveableObject):
662 BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE")
/external/tensorflow/tensorflow/core/protobuf/
Dsaver.proto35 // Usually, each subclass of BaseSaverBuilder works with a particular
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py632 class _MirroredSaveable(saver.BaseSaverBuilder.ResourceVariableSaveable):
1194 class _SyncOnReadSaveable(saver.BaseSaverBuilder.SaveableObject):
1205 spec = saver.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py629 class CudnnOpaqueParamsSaveable(saver.BaseSaverBuilder.SaveableObject):
720 saver.BaseSaverBuilder.SaveSpec(param, slice_spec, param_name)
/external/tensorflow/tensorflow/python/training/tracking/
Dutil_test.py212 class _MirroringSaveable(saver_lib.BaseSaverBuilder.SaveableObject):
218 spec = saver_lib.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py77 core_saver.BaseSaverBuilder.ResourceVariableSaveable):