Searched refs:XlaDevice (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 140 XlaDevice::Metadata::Metadata( in Metadata()
151 int XlaDevice::Metadata::device_ordinal() const { return device_ordinal_; } in device_ordinal()
153 se::Platform* XlaDevice::Metadata::platform() const { return platform_; } in platform()
155 xla::LocalClient* XlaDevice::Metadata::client() const { in client()
160 const DeviceType& XlaDevice::Metadata::jit_device_type() const { in jit_device_type()
164 /*static*/ Status XlaDevice::GetMetadataFromDevice( in GetMetadataFromDevice()
165 DeviceBase* device, const XlaDevice::Metadata** metadata) { in GetMetadataFromDevice()
167 XlaDevice* xla_device = dynamic_cast<XlaDevice*>(device->UnderlyingDevice()); in GetMetadataFromDevice()
179 /* static */ Status XlaDevice::GetMetadata(OpKernelContext* ctx, in GetMetadata()
184 /* static */ Status XlaDevice::GetMetadata(OpKernelConstruction* ctx, in GetMetadata()
[all …]
|
| D | xla_device.h | 50 class XlaDevice : public LocalDevice {
136 XlaDevice(const SessionOptions& session_options, const Options& options);
137 ~XlaDevice() override;
191 const XlaDevice::Metadata** metadata);
|
| D | xla_compile_on_demand_op.cc | 51 const XlaDevice::Metadata& metadata, in Run()
114 OpKernelContext* ctx, const XlaDevice::Metadata& metadata, in Compile()
214 const XlaDevice::Metadata* metadata; in Compute()
215 OP_REQUIRES_OK(ctx, XlaDevice::GetMetadata(ctx, &metadata)); in Compute()
|
| D | xla_compile_on_demand_op.h | 45 Status Compile(OpKernelContext* ctx, const XlaDevice::Metadata& metadata,
48 Status Run(OpKernelContext* ctx, const XlaDevice::Metadata& metadata,
|
| D | xla_interpreter_device.cc | 57 XlaDevice::Options options; in CreateDevices()
64 devices->push_back(absl::make_unique<XlaDevice>(session_options, options)); in CreateDevices()
|
| D | xla_cpu_device.cc | 59 XlaDevice::Options options; in CreateDevices()
66 auto device = absl::make_unique<XlaDevice>(session_options, options); in CreateDevices()
|
| D | xla_gpu_device.cc | 94 XlaDevice::Options options; in CreateDevices()
102 auto device = absl::make_unique<XlaDevice>(session_options, options); in CreateDevices()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_device.cc | 35 const XlaDevice::Metadata* metadata; in InitScopedRef()
36 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in InitScopedRef()
48 const XlaDevice::Metadata* metadata; in InitScopedRef()
49 TF_RETURN_IF_ERROR(XlaDevice::GetMetadata(ctx, &metadata)); in InitScopedRef()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.h | 41 const XlaDevice::Metadata* xla_device_metadata, in XlaPlatformInfo()
69 const XlaDevice::Metadata* xla_device_metadata() const { in xla_device_metadata()
81 const XlaDevice::Metadata* xla_device_metadata_;
|
| D | xla_ops.cc | 63 const XlaDevice::Metadata* xla_device_metadata = nullptr; in PlatformInfoFromContext()
75 } else if (XlaDevice::GetMetadata(ctx, &xla_device_metadata).ok()) { in PlatformInfoFromContext()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 64 tensorflow::XlaDevice* xla_device = in TFE_TensorHandleTensorDebugInfo()
65 dynamic_cast<tensorflow::XlaDevice*>(device); in TFE_TensorHandleTensorDebugInfo()
67 tensorflow::XlaDevice::PaddedShapeFn shape_fn = in TFE_TensorHandleTensorDebugInfo()
|