Method MakeFastMemTensorFromProto

tensorflow/compiler/jit/xla_device.cc:533–544

}

Status XlaDevice::MakeFastMemTensorFromProto(
    const TensorProto& tensor_proto, const AllocatorAttributes alloc_attrs,
    Tensor* tensor) {
  VLOG(1) << "XlaDevice::MakeFastMemTensorFromProto";
  std::pair<XlaDeviceContext*, XlaDeviceContext*> device_contexts;
  {
    mutex_lock lock(mu_);
    TF_ASSIGN_OR_RETURN(device_contexts, GetDeviceContextLocked());
  }
  return MakeTensorFromProto(device_contexts.second, tensor_proto, alloc_attrs,
                             tensor);
}

void XlaDevice::SetAllowsSyncOnCompletion(bool sync_on_completion) {
  mutex_lock lock(mu_);