Home
last modified time | relevance | path

Searched refs:XLA_LIGHTWEIGHT_CHECK (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d.cc40 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF32()
64 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenConvF16()
Druntime_lightweight_check.h27 #define XLA_LIGHTWEIGHT_CHECK(cond) \ macro
Druntime_fft.cc35 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in __xla_cpu_runtime_EigenFft()
Druntime_matmul.cc70 XLA_LIGHTWEIGHT_CHECK(run_options->intra_op_thread_pool() != nullptr); in MatMul()