Searched refs:XlaDeviceContext (Results 1 – 4 of 4) sorted by relevance
71 XlaDeviceContext::XlaDeviceContext( in XlaDeviceContext() function in tensorflow::XlaDeviceContext99 void XlaDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()106 void XlaDeviceContext::CopyCPUTensorToDevice(const Tensor* cpu_tensor, in CopyCPUTensorToDevice()199 void XlaDeviceContext::CopyDeviceTensorToCPU(const Tensor* device_tensor, in CopyDeviceTensorToCPU()259 se::Stream* XlaDeviceContext::GetDeviceToDeviceStream() { in GetDeviceToDeviceStream()
52 class XlaDeviceContext : public DeviceContext {54 explicit XlaDeviceContext(
278 xla::StatusOr<XlaDeviceContext*> XlaDevice::GetDeviceContextLocked() { in GetDeviceContextLocked()324 device_context_ = new XlaDeviceContext( in GetDeviceContextLocked()361 TF_ASSIGN_OR_RETURN(XlaDeviceContext * device_context, in FillContextMap()456 TF_ASSIGN_OR_RETURN(XlaDeviceContext * device_context, in MakeTensorFromProto()
187 xla::StatusOr<XlaDeviceContext*> GetDeviceContextLocked()231 XlaDeviceContext* device_context_ GUARDED_BY(mu_) = nullptr;