Home
last modified time | relevance | path

Searched refs:XlaDevice (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc146 XlaDevice::Metadata::Metadata( in Metadata()
157 int XlaDevice::Metadata::device_ordinal() const { return device_ordinal_; } in device_ordinal()
159 se::Platform* XlaDevice::Metadata::platform() const { return platform_; } in platform()
161 xla::LocalClient* XlaDevice::Metadata::client() const { in client()
166 const DeviceType& XlaDevice::Metadata::jit_device_type() const { in jit_device_type()
170 /*static*/ Status XlaDevice::GetMetadataFromDevice( in GetMetadataFromDevice()
171 DeviceBase* device, const XlaDevice::Metadata** metadata) { in GetMetadataFromDevice()
173 XlaDevice* xla_device = dynamic_cast<XlaDevice*>(device->UnderlyingDevice()); in GetMetadataFromDevice()
185 /* static */ Status XlaDevice::GetMetadata(OpKernelContext* ctx, in GetMetadata()
190 /* static */ Status XlaDevice::GetMetadata(OpKernelConstruction* ctx, in GetMetadata()
[all …]
Dxla_platform_info.h34 const XlaDevice::Metadata* xla_device_metadata, in XlaPlatformInfo()
60 const XlaDevice::Metadata* xla_device_metadata() const { in xla_device_metadata()
72 const XlaDevice::Metadata* xla_device_metadata_;
Dxla_device.h50 class XlaDevice : public LocalDevice {
100 const XlaDevice::Metadata** metadata);
141 XlaDevice(const SessionOptions& session_options, const Options& options);
142 ~XlaDevice() override;
Dxla_cpu_device.cc83 XlaDevice::Options options; in CreateDevices()
90 auto device = absl::make_unique<XlaDevice>(session_options, options); in CreateDevices()
Dxla_platform_info.cc81 const XlaDevice::Metadata* xla_device_metadata = nullptr; in XlaPlatformInfoFromDevice()
91 } else if (XlaDevice::GetMetadataFromDevice(device, &xla_device_metadata) in XlaPlatformInfoFromDevice()
Dxla_gpu_device.cc149 XlaDevice::Options options; in CreateDevices()
157 auto device = absl::make_unique<XlaDevice>(session_options, options); in CreateDevices()
Dxla_tpu_device.cc367 XlaDevice::Options options; in CreateDevices()
376 auto device = absl::make_unique<XlaDevice>(session_options, options); in CreateDevices()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_device.cc56 const XlaDevice::Metadata* metadata; in GetResourceManager()
57 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in GetResourceManager()
72 const XlaDevice::Metadata* metadata; in InitScopedRef()
73 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in InitScopedRef()
85 const XlaDevice::Metadata* metadata; in InitScopedRef()
86 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in InitScopedRef()
Dxrt_tpu_device.cc45 const XlaDevice::Metadata* metadata; in Acquire()
46 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in Acquire()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc79 const XlaDevice::Metadata* metadata; in RunTransferWithOrdinal()
80 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in RunTransferWithOrdinal()
130 XlaDevice* xla_device = in RunTransfer()
131 dynamic_cast<XlaDevice*>(ctx->device()->UnderlyingDevice()); in RunTransfer()
Dtpu_execute_op.cc622 const XlaDevice::Metadata* metadata; in DoWork()
623 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(context, &metadata)); in DoWork()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_execute_op.cc271 const XlaDevice::Metadata* metadata; in DoWork()
272 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(context, &metadata)); in DoWork()
393 const XlaDevice::Metadata* metadata; in DoWork()
394 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(context, &metadata)); in DoWork()