Home
last modified time | relevance | path

Searched refs:restored_tensor (Results 1 – 6 of 6) sorted by relevance

/external/federated-compute/fcp/artifact_building/
Dtensor_utils_test.py79 restored_tensor = tensor_utils.restore(filename, tensor_name, tensor.dtype)
80 self.assertAllEqual(tensor, restored_tensor)
90 restored_tensor = tensor_utils.restore(
93 self.assertAllEqual(tensor, restored_tensor)
121 expected_tensor, restored_tensor = sess.run([tensor, restored])
122 self.assertAllEqual(expected_tensor, restored_tensor)
146 expected_tensor, restored_tensor = sess.run([tensor, restored])
147 self.assertAllEqual(expected_tensor, restored_tensor)
/external/tensorflow/tensorflow/core/kernels/
Dsave_restore_tensor.cc303 Tensor* restored_tensor; in run() local
307 context->allocate_output(idx, restored_full_shape, &restored_tensor)); in run()
308 TF_RETURN_IF_ERROR(reader->Lookup(tensor_name, restored_tensor)); in run()
327 context->allocate_output(idx, parsed_slice_shape, &restored_tensor)); in run()
329 reader->LookupSlice(tensor_name, parsed_slice, restored_tensor)); in run()
332 if (restored_tensor->dtype() == DT_FLOAT) { in run()
333 const float* t_data = restored_tensor->flat<float>().data(); in run()
337 for (int i = 0; i < restored_tensor->NumElements(); ++i) { in run()
343 << avg / restored_tensor->NumElements() << " total elts " in run()
344 << restored_tensor->NumElements(); in run()
/external/tensorflow/tensorflow/python/training/saving/
Dsaveable_object_util.py77 restored_tensor = restored_tensors[0]
79 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
82 restored_tensor,
126 restored_tensor = restored_tensors[0]
128 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
131 restored_tensor = array_ops.identity(restored_tensor)
134 self.handle_op, self._var_shape, restored_tensor)
/external/tensorflow/tensorflow/python/saved_model/registration/
DREADME.md160 for trackable, restored_tensor in zip(restored_trackables, restored_tensors):
162 restored_tensor = array_ops.reshape(restored_tensor, expected_shape)
163 parts = array_ops.unstack(restored_tensor)
Dregistration_saving_test.py103 for trackable, restored_tensor in zip(restored_trackables, restored_tensors):
105 restored_tensor = array_ops.reshape(restored_tensor, expected_shape)
106 parts = array_ops.unstack(restored_tensor)
/external/tensorflow/tensorflow/python/checkpoint/
Dfunctional_saver.py116 restored_tensor = restored_tensors.pop(0)
118 restored_tensor)