Home
last modified time | relevance | path

Searched refs:async_task_runner (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dcwise_op_unary_benchmark.h135 HostContextAsyncTaskRunner async_task_runner(b.exec_ctx.host()); in TestUnaryMlirBenchmark()
136 opts.async_task_runner = &async_task_runner; in TestUnaryMlirBenchmark()
168 HostContextAsyncTaskRunner async_task_runner(b.exec_ctx.host()); in RunUnaryMlirBenchmark()
169 opts.async_task_runner = &async_task_runner; in RunUnaryMlirBenchmark()
H A Dbenchmark_mlir_function.cc146 HostContextAsyncTaskRunner async_task_runner(host.get()); in RunJitRtBenchmark() local
147 opts.async_task_runner = &async_task_runner; in RunJitRtBenchmark()
H A Dmatmul_op_benchmark.h94 HostContextAsyncTaskRunner async_task_runner(host.get()); in RunMatMulMlirBenchmark()
95 opts.async_task_runner = &async_task_runner; in RunMatMulMlirBenchmark()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/
H A Dtf_jitrt_executor.cc235 HostContextAsyncTaskRunner async_task_runner(&host_context_); in Execute() local
236 opts.async_task_runner = &async_task_runner; in Execute()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_jitrt_kernels.cc686 EigenThreadPoolAsyncTaskRunner async_task_runner(*worker_threads); in ExecuteImpl() local
689 opts.async_task_runner = &async_task_runner; in ExecuteImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/
H A Dexecutable.h181 AsyncTaskRunner* async_task_runner = nullptr; member
H A Dexecutable.cc308 AsyncRuntime::Set(AsyncRuntime(opts.async_task_runner)); in Execute()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc633 opts.async_task_runner = in ExecuteJitRt()