Searched refs:DEVICE_GPU_XLA_JIT (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 84 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
99 RegisterXlaDeviceKernels(DEVICE_XLA_GPU, DEVICE_GPU_XLA_JIT); in CreateDevices()
134 options.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
144 LOG(INFO) << "Ignoring visible " << DEVICE_GPU_XLA_JIT in CreateDevices()
|
| D | xla_cpu_device.cc | 106 errors::AppendToMessage(&status, "while setting up ", DEVICE_GPU_XLA_JIT); in CreateDevices()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | light_outside_compilation_kernels_for_test.cc | 61 REGISTER_XLA_OP(Name("TestStaticTf").Device(DEVICE_GPU_XLA_JIT),
107 REGISTER_XLA_OP(Name("TestStaticMultipleOutputTf").Device(DEVICE_GPU_XLA_JIT),
174 REGISTER_XLA_OP(Name("TestDynamicTf").Device(DEVICE_GPU_XLA_JIT),
236 .Device(DEVICE_GPU_XLA_JIT)
247 REGISTER_XLA_OP(Name("DynamicUnranked").Device(DEVICE_GPU_XLA_JIT),
302 .Device(DEVICE_GPU_XLA_JIT)
365 REGISTER_XLA_OP(Name("TestDynamicTfWithBound").Device(DEVICE_GPU_XLA_JIT),
|
| D | xla_gpu_backend.cc | 32 REGISTER_XLA_BACKEND(DEVICE_GPU_XLA_JIT, kGpuAllTypes, GpuOpFilter);
|
| D | xla_op_registry.cc | 38 const char* const DEVICE_GPU_XLA_JIT = "XLA_GPU_JIT"; variable
168 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in GetCompilationDevice()
|
| D | xla_op_registry.h | 43 extern const char* const DEVICE_GPU_XLA_JIT; // "GPU_XLA_JIT"
|
| D | xla_compiler_test.cc | 178 REGISTER_XLA_OP(Name("DummyDuplicateOp").Device(DEVICE_GPU_XLA_JIT),
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 71 is_on_gpu_ = ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT; in FusedBatchNormOp()
252 is_on_gpu_ = ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT; in FusedBatchNormGradOp()
|
| D | stateless_random_ops_v2.cc | 88 if (device_type_string == DEVICE_GPU_XLA_JIT || in GetKeyCounter()
103 if (device_type_string == DEVICE_GPU_XLA_JIT || in DefaultRngAlgForDeviceType()
|
| D | where_op.cc | 148 return t == DEVICE_CPU_XLA_JIT || t == DEVICE_GPU_XLA_JIT || in ShouldUsePrefixSumImpl()
|
| D | matrix_diag_ops.cc | 351 is_gpu_(context->device_type().type_string() == DEVICE_GPU_XLA_JIT) { in MatrixDiagPartOp()
|
| D | stateless_random_ops.cc | 44 if (device_type_string == DEVICE_GPU_XLA_JIT || in GetBitGeneratorForDevice()
|
| D | image_resize_ops.cc | 713 if (ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) { in ResizeBilinearGradOp()
|