Searched refs:BaseSaverBuilder (Results 1 – 19 of 19) sorted by relevance
/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/ |
D | stats_accumulator_ops.py | 35 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,
|
D | model_ops.py | 43 class TreeEnsembleVariableSavable(saver.BaseSaverBuilder.SaveableObject): 60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec, 62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
|
D | quantile_ops.py | 42 class QuantileAccumulatorSaveable(saver.BaseSaverBuilder.SaveableObject): 54 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
|
/external/tensorflow/tensorflow/python/training/ |
D | saver_test_utils.py | 75 class CustomSaveable(saver_module.BaseSaverBuilder.SaveableObject): 81 saver_module.BaseSaverBuilder.SaveSpec(tensors[0], "", 83 saver_module.BaseSaverBuilder.SaveSpec(tensors[1], "",
|
D | saver.py | 71 class BaseSaverBuilder(object): class 563 class BulkSaverBuilder(BaseSaverBuilder):
|
D | saver_test.py | 2795 saver_module.BaseSaverBuilder.ResourceVariableSaveable): 2892 class _CountingSaveable(saver_module.BaseSaverBuilder.SaveableObject): 2902 [saver_module.BaseSaverBuilder.SaveSpec(
|
/external/tensorflow/tensorflow/python/ops/ |
D | boosted_trees_ops.py | 64 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,
|
D | tensor_forest_ops.py | 26 class TreeVariableSaveable(saver.BaseSaverBuilder.SaveableObject): 44 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name)]
|
D | lookup_ops.py | 41 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/ |
D | split_dependency.py | 27 class _CallbackSaveable(saver_lib.BaseSaverBuilder.SaveableObject): 32 spec = saver_lib.BaseSaverBuilder.SaveSpec(
|
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/ |
D | stats_ops.py | 50 class FertileStatsVariableSavable(saver.BaseSaverBuilder.SaveableObject): 68 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
|
D | model_ops.py | 53 class TreeVariableSavable(saver.BaseSaverBuilder.SaveableObject): 71 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
|
/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | iterator_ops.py | 69 class _Saveable(saver_lib.BaseSaverBuilder.SaveableObject): 75 saver_lib.BaseSaverBuilder.SaveSpec(serialized_iterator, "",
|
/external/tensorflow/tensorflow/python/data/ops/ |
D | iterator_ops.py | 34 from tensorflow.python.training.saver import BaseSaverBuilder 656 class _IteratorSaveable(BaseSaverBuilder.SaveableObject): 662 BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE")
|
/external/tensorflow/tensorflow/core/protobuf/ |
D | saver.proto | 35 // Usually, each subclass of BaseSaverBuilder works with a particular
|
/external/tensorflow/tensorflow/python/distribute/ |
D | values.py | 632 class _MirroredSaveable(saver.BaseSaverBuilder.ResourceVariableSaveable): 1194 class _SyncOnReadSaveable(saver.BaseSaverBuilder.SaveableObject): 1205 spec = saver.BaseSaverBuilder.SaveSpec(
|
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/ |
D | cudnn_rnn_ops.py | 629 class CudnnOpaqueParamsSaveable(saver.BaseSaverBuilder.SaveableObject): 720 saver.BaseSaverBuilder.SaveSpec(param, slice_spec, param_name)
|
/external/tensorflow/tensorflow/python/training/tracking/ |
D | util_test.py | 212 class _MirroringSaveable(saver_lib.BaseSaverBuilder.SaveableObject): 218 spec = saver_lib.BaseSaverBuilder.SaveSpec(
|
/external/tensorflow/tensorflow/contrib/optimizer_v2/ |
D | checkpointable_utils_test.py | 77 core_saver.BaseSaverBuilder.ResourceVariableSaveable):
|