Home
last modified time | relevance | path

Searched refs:restored_tensors (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/contrib/boosted_trees/python/ops/
Dstats_accumulator_ops.py84 def restore(self, restored_tensors, unused_restored_shapes): argument
97 stamp_token=restored_tensors[0],
98 num_updates=restored_tensors[1],
99 partition_ids=restored_tensors[2],
100 feature_ids=restored_tensors[3],
101 gradients=restored_tensors[4],
102 hessians=restored_tensors[5])
Dquantile_ops.py63 def restore(self, restored_tensors, unused_restored_shapes): argument
76 stamp_token = restored_tensors[:1]
77 state = restored_tensors[1:2]
78 are_buckets_ready = restored_tensors[2:3]
79 buckets = restored_tensors[3]
Dmodel_ops.py70 def restore(self, restored_tensors, unused_restored_shapes): argument
84 stamp_token=restored_tensors[0],
85 tree_ensemble_config=restored_tensors[1])
/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver.py129 restored_tensors = io_ops.restore_v2(
132 tensor_structure, restored_tensors)
134 for saveable, restored_tensors in zip(saveable_objects,
136 restore_ops.append(saveable.restore(restored_tensors,
Dsaveable_object_util.py65 def restore(self, restored_tensors, restored_shapes): argument
66 restored_tensor = restored_tensors[0]
107 def restore(self, restored_tensors, restored_shapes): argument
108 restored_tensor = restored_tensors[0]
Dsaveable_object.py88 def restore(self, restored_tensors, restored_shapes): argument
/external/tensorflow/tensorflow/python/ops/
Dboosted_trees_ops.py86 def restore(self, restored_tensors, unused_tensor_shapes): argument
87 bucket_boundaries = restored_tensors
186 def restore(self, restored_tensors, unused_restored_shapes): argument
200 stamp_token=restored_tensors[0],
201 tree_ensemble_serialized=restored_tensors[1])
Dtensor_forest_ops.py54 def restore(self, restored_tensors, unused_restored_shapes): argument
68 restored_tensors[0],
Dlookup_ops.py1744 def restore(self, restored_tensors, restored_shapes, name=None): argument
1750 self.op.resource_handle, restored_tensors[0], restored_tensors[1])
2035 def restore(self, restored_tensors, restored_shapes, name=None): argument
2041 self.op.resource_handle, restored_tensors[0], restored_tensors[1])
/external/tensorflow/tensorflow/contrib/checkpoint/python/
Dsplit_dependency.py40 def restore(self, restored_tensors, restored_shapes): argument
42 tensor, = restored_tensors
/external/tensorflow/tensorflow/contrib/tensor_forest/python/ops/
Dstats_ops.py74 def restore(self, restored_tensors, unused_restored_shapes): argument
87 self._stats_handle, restored_tensors[0],
Dmodel_ops.py77 def restore(self, restored_tensors, unused_restored_shapes): argument
91 restored_tensors[0],
/external/tensorflow/tensorflow/python/data/experimental/ops/
Diterator_ops.py81 def restore(self, restored_tensors, unused_restored_shapes): argument
83 return gen_dataset_ops.deserialize_iterator(self.op, restored_tensors[0])
/external/tensorflow/tensorflow/python/data/ops/
Diterator_ops.py666 def restore(self, restored_tensors, restored_shapes): argument
668 return gen_dataset_ops.deserialize_iterator(self.op, restored_tensors[0])
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py229 restored_tensors = []
240 restored_tensors.append(array_ops.identity(restored))
251 saveable.restore(restored_tensors=restored_tensors,
Dbase.py103 def restore(self, restored_tensors, restored_shapes): argument
185 def restore(self, restored_tensors, restored_shapes): argument
Dutil_test.py225 def restore(self, restored_tensors, restored_shapes): argument
227 tensor, = restored_tensors
/external/tensorflow/tensorflow/python/distribute/
Dvalues.py639 def restore(self, restored_tensors, restored_shapes): argument
641 tensor, = restored_tensors
1212 def restore(self, restored_tensors, restored_shapes): argument
1214 tensor, = restored_tensors
/external/tensorflow/tensorflow/contrib/cudnn_rnn/python/ops/
Dcudnn_rnn_ops.py734 def restore(self, restored_tensors, restored_shapes): argument
736 restored_tensors)
751 restored_tensors=tensors,
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py85 def restore(self, restored_tensors, restored_shapes): argument
87 tensor, = restored_tensors
/external/tensorflow/tensorflow/python/training/
Dsaver_test.py2803 def restore(self, restored_tensors, restored_shapes): argument
2805 tensor, = restored_tensors
2906 def restore(self, restored_tensors, restored_shapes): argument