Searched refs:restored_tensor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 276 Tensor* restored_tensor; in run() local
280 context->allocate_output(idx, restored_full_shape, &restored_tensor)); in run()
281 TF_RETURN_IF_ERROR(reader->Lookup(tensor_name, restored_tensor)); in run()
300 context->allocate_output(idx, parsed_slice_shape, &restored_tensor)); in run()
302 reader->LookupSlice(tensor_name, parsed_slice, restored_tensor)); in run()
305 if (restored_tensor->dtype() == DT_FLOAT) { in run()
306 const float* t_data = restored_tensor->flat<float>().data(); in run()
310 for (int i = 0; i < restored_tensor->NumElements(); ++i) { in run()
316 << avg / restored_tensor->NumElements() << " total elts " in run()
317 << restored_tensor->NumElements(); in run()
|
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 78 restored_tensor = restored_tensors[0]
80 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
83 restored_tensor,
125 restored_tensor = restored_tensors[0]
127 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
130 restored_tensor = array_ops.identity(restored_tensor)
132 self.handle_op, self._var_shape, restored_tensor)
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | saving.py | 47 restored_tensor, = tensors
49 resource=self._handle, value=restored_tensor)
|