Searched refs:xla_device (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 64 tensorflow::XlaDevice* xla_device = in TFE_TensorHandleTensorDebugInfo() local
66 if (xla_device != nullptr) { in TFE_TensorHandleTensorDebugInfo()
68 xla_device->metadata().padded_shape_fn(); in TFE_TensorHandleTensorDebugInfo()
|
| D | BUILD | 57 "//tensorflow/compiler/jit:xla_device",
198 "//tensorflow/compiler/jit:xla_device",
|
| /external/tensorflow/tensorflow/compiler/plugin/ |
| D | BUILD | 44 # This target is added purely for the purpose of ensuring that `:xla_device` is
49 "//tensorflow/compiler/jit:xla_device",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 98 string xla_device = absl::StrCat("/device:XLA_", test_device, ":0"); in TEST() local
104 root.WithOpName("fused_batch_norm_xla").WithDevice(xla_device), input, in TEST()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | BUILD | 84 ":xla_device",
103 ":xla_device",
122 ":xla_device",
149 name = "xla_device",
152 "xla_device.cc",
158 "xla_device.h",
|
| D | xla_device.cc | 167 XlaDevice* xla_device = dynamic_cast<XlaDevice*>(device->UnderlyingDevice()); in GetMetadataFromDevice() local
168 if (xla_device == nullptr) { in GetMetadataFromDevice()
175 *metadata = &(xla_device->xla_metadata_); in GetMetadataFromDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | BUILD | 17 "//tensorflow/compiler/jit:xla_device",
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | BUILD | 57 "//tensorflow/compiler/jit:xla_device",
|