Home
last modified time | relevance | path

Searched refs:DEVICE_GPU_XLA_JIT (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Dxla_gpu_device.cc104 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
119 RegisterXlaDeviceKernels(DEVICE_XLA_GPU, DEVICE_GPU_XLA_JIT); in CreateDevices()
154 options.compilation_device_name = DEVICE_GPU_XLA_JIT; in CreateDevices()
161 LOG(INFO) << "Ignoring visible " << DEVICE_GPU_XLA_JIT in CreateDevices()
Dxla_cpu_device.cc99 errors::AppendToMessage(&status, "while setting up ", DEVICE_GPU_XLA_JIT); in CreateDevices()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_gpu_backend.cc32 REGISTER_XLA_BACKEND(DEVICE_GPU_XLA_JIT, kGpuAllTypes, GpuOpFilter);
Dxla_op_registry.cc38 const char* const DEVICE_GPU_XLA_JIT = "XLA_GPU_JIT"; variable
168 registration.compilation_device_name = DEVICE_GPU_XLA_JIT; in GetCompilationDevice()
Dxla_op_registry.h43 extern const char* const DEVICE_GPU_XLA_JIT; // "GPU_XLA_JIT"
Dxla_compiler_test.cc177 REGISTER_XLA_OP(Name("DummyDuplicateOp").Device(DEVICE_GPU_XLA_JIT),
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dindex_ops.cc36 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in XlaArgMinMaxOp()
Dbatch_norm_op.cc67 is_on_gpu_ = ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT; in FusedBatchNormOp()
248 is_on_gpu_ = ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT; in FusedBatchNormGradOp()
Ddiag_op.cc120 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in DiagPartOp()
Dcategorical_op.cc41 is_gpu_(ctx->device_type().type_string() == DEVICE_GPU_XLA_JIT) {} in CategoricalOp()
Dstateless_random_ops_v2.cc86 if (device_type_string == DEVICE_GPU_XLA_JIT || in GetKeyCounter()
101 if (device_type_string == DEVICE_GPU_XLA_JIT || in GetAlg()
Dstateless_random_ops.cc43 if (device_type_string == DEVICE_GPU_XLA_JIT || in GetBitGeneratorForDevice()
Dmatrix_diag_ops.cc348 is_gpu_(context->device_type().type_string() == DEVICE_GPU_XLA_JIT) { in MatrixDiagPartOp()