HomeSort by relevance Sort by last modified time
    Searched refs:restored_tensor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
save_restore_tensor.cc 253 Tensor* restored_tensor = nullptr; local
264 context->allocate_output(i, restored_full_shape, &restored_tensor));
265 TF_RETURN_IF_ERROR(reader.Lookup(tensor_name, restored_tensor));
284 context->allocate_output(i, parsed_slice_shape, &restored_tensor));
286 reader.LookupSlice(tensor_name, parsed_slice, restored_tensor));
288 if (dtypes[i] != restored_tensor->dtype()) {
292 DataTypeString(restored_tensor->dtype()));
  /external/tensorflow/tensorflow/python/training/
saver.py 154 restored_tensor = restored_tensors[0]
156 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
159 restored_tensor,
193 restored_tensor = restored_tensors[0]
195 restored_tensor = array_ops.reshape(restored_tensor, restored_shapes[0])
198 restored_tensor = array_ops.identity(restored_tensor)
200 self.handle_op, self._var_shape, restored_tensor)
    [all...]

Completed in 133 milliseconds