HomeSort by relevance Sort by last modified time
    Searched refs:REGISTER_XLA_BACKEND (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_cpu_backend.cc 32 REGISTER_XLA_BACKEND(DEVICE_CPU_XLA_JIT, kCpuAllTypes, CpuOpFilter);
xla_gpu_backend.cc 32 REGISTER_XLA_BACKEND(DEVICE_GPU_XLA_JIT, kGpuAllTypes, GpuOpFilter);
xla_op_registry.h 320 // REGISTER_XLA_BACKEND() registers an XLA backend. Example usage:
321 // REGISTER_XLA_BACKEND(DEVICE_GPU_XLA_JIT, kGpuAllTypes, GpuOpFilter);
322 #define REGISTER_XLA_BACKEND(NAME, ...) \
  /external/tensorflow/tensorflow/compiler/jit/
xla_interpreter_device.cc 87 REGISTER_XLA_BACKEND(DEVICE_INTERPRETER_XLA_JIT, kExecAllTypes, OpFilter);

Completed in 50 milliseconds