Searched refs:xla_device (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | transfer_ops.cc | 130 XlaDevice* xla_device = in RunTransfer() local
132 if (((xla_device == nullptr) || (xla_device->device_type() == DEVICE_CPU)) && in RunTransfer()
|
| /external/tensorflow/tensorflow/compiler/plugin/ |
| D | BUILD | 45 # This target is added purely for the purpose of ensuring that `:xla_device` is
50 "//tensorflow/compiler/jit:xla_device",
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 112 string xla_device = absl::StrCat("/device:XLA_", test_device, ":0"); in TEST() local
118 root.WithOpName("fused_batch_norm_xla").WithDevice(xla_device), input, in TEST()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | BUILD | 46 "//tensorflow/compiler/jit:xla_device",
77 "//tensorflow/compiler/jit:xla_device",
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 173 XlaDevice* xla_device = dynamic_cast<XlaDevice*>(device->UnderlyingDevice()); in GetMetadataFromDevice() local
174 if (xla_device == nullptr) { in GetMetadataFromDevice()
181 *metadata = &(xla_device->xla_metadata_); in GetMetadataFromDevice()
|
| D | BUILD | 124 ":xla_device",
149 ":xla_device",
176 ":xla_device",
281 "xla_device.cc",
289 "xla_device.h",
300 name = "xla_device",
303 "xla_device.h",
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | stateful_random_ops_test.py | 44 def xla_device(): function
59 return str(xla_device().name)
292 device = xla_device()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | BUILD | 985 "//tensorflow/compiler/jit:xla_device",
|