Searched refs:DEVICE_TPU_XLA_JIT (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 139 .Device(DEVICE_TPU_XLA_JIT),
144 .Device(DEVICE_TPU_XLA_JIT),
149 .Device(DEVICE_TPU_XLA_JIT),
152 REGISTER_XLA_OP(Name("ResizeBilinearGrad").Device(DEVICE_TPU_XLA_JIT),
|
| D | tpu_compile_op_common.cc | 289 compiler_options.device_type = DeviceType(DEVICE_TPU_XLA_JIT); in CompileTFFunctionToHlo()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | index_ops.cc | 29 .Device(DEVICE_TPU_XLA_JIT)
|
| D | segment_reduction_ops.cc | 120 .Device(DEVICE_TPU_XLA_JIT)
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_tpu_backend.cc | 24 REGISTER_XLA_BACKEND(DEVICE_TPU_XLA_JIT, kTpuAllTypes, TpuOpFilter); in RegisterTpuXlaBackend()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_defs.cc | 23 const char* const DEVICE_TPU_XLA_JIT = "XLA_TPU_JIT"; variable
|
| D | tpu_defs.h | 40 extern const char* const DEVICE_TPU_XLA_JIT; // "XLA_TPU_JIT";
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tpu_device.cc | 341 registration.compilation_device_name = DEVICE_TPU_XLA_JIT; in CreateDevices()
359 RegisterXlaDeviceKernels(DEVICE_TPU_NODE, DEVICE_TPU_XLA_JIT); in CreateDevices()
372 options.compilation_device_name = DEVICE_TPU_XLA_JIT; in CreateDevices()
383 errors::AppendToMessage(&status, "while setting up ", DEVICE_TPU_XLA_JIT, in CreateDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | where_op.cc | 88 REGISTER_XLA_OP(Name("Where").Device(DEVICE_TPU_XLA_JIT), WhereOp);
|
| D | unique_op.cc | 255 REGISTER_XLA_OP(Name("Unique").Device(DEVICE_TPU_XLA_JIT), UniqueOp);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | compile_mlir_util.cc | 223 return device_type == DEVICE_TPU_XLA_JIT; in CanInlineFunctionsPostLegalization()
|