Searched refs:restored_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/training/saving/ |
| D | saveable_object_util.py | 66 restored_tensor = restored_tensors[0]
68 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
71 restored_tensor,
108 restored_tensor = restored_tensors[0]
110 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
113 restored_tensor = array_ops.identity(restored_tensor)
115 self.handle_op, self._var_shape, restored_tensor)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | save_restore_tensor.cc | 275 Tensor* restored_tensor; in run() local
279 context->allocate_output(idx, restored_full_shape, &restored_tensor)); in run()
280 TF_RETURN_IF_ERROR(reader->Lookup(tensor_name, restored_tensor)); in run()
299 context->allocate_output(idx, parsed_slice_shape, &restored_tensor)); in run()
301 reader->LookupSlice(tensor_name, parsed_slice, restored_tensor)); in run()
|