Home
last modified time | relevance | path

Searched refs:SaveSpec (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py47 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.py60 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
62 saver.BaseSaverBuilder.SaveSpec(ensemble_config, slice_spec,
Dquantile_ops.py54 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
/external/tensorflow/tensorflow/python/training/
Dsaver_test_utils.py81 saver_module.BaseSaverBuilder.SaveSpec(tensors[0], "",
83 saver_module.BaseSaverBuilder.SaveSpec(tensors[1], "",
Dsaver.py77 SaveSpec = saveable_object.SaveSpec variable in BaseSaverBuilder
Dsaver_test.py2902 [saver_module.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py77 return saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name + suffix)
177 saver.BaseSaverBuilder.SaveSpec(stamp_token, slice_spec,
179 saver.BaseSaverBuilder.SaveSpec(serialized, slice_spec,
Dtensor_forest_ops.py44 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name)]
Dlookup_ops.py1738 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
1739 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
2029 BaseSaverBuilder.SaveSpec(tensors[0], "", name + "-keys"),
2030 BaseSaverBuilder.SaveSpec(tensors[1], "", name + "-values")
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object.py21 class SaveSpec(object): class
Dsaveable_object_util.py62 spec = saveable_object.SaveSpec(var, slice_spec, name, dtype=var.dtype)
103 spec = saveable_object.SaveSpec(tensor, slice_spec, name,
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dsplit_dependency.py32 spec = saver_lib.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py75 saver_lib.BaseSaverBuilder.SaveSpec(serialized_iterator, "",
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/
Dstats_ops.py68 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
Dmodel_ops.py71 specs = [saver.BaseSaverBuilder.SaveSpec(tensor, slice_spec, name),]
/external/tensorflow/tensorflow/python/training/tracking/
Dbase.py100 spec = saveable_object.SaveSpec(tensor, "", name, dtype=dtype)
156 spec = saveable_object.SaveSpec(
Dutil_test.py218 spec = saver_lib.BaseSaverBuilder.SaveSpec(
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py662 BaseSaverBuilder.SaveSpec(serialized_iterator, "", name + "_STATE")
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py720 saver.BaseSaverBuilder.SaveSpec(param, slice_spec, param_name)
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py1205 spec = saver.BaseSaverBuilder.SaveSpec(