Home
last modified time | relevance | path

Searched refs:CreateMultiThreadedHostContext (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/utils/
H A Dhost_context.h30 std::unique_ptr<tfrt::HostContext> CreateMultiThreadedHostContext(
H A Dhost_context.cc42 std::unique_ptr<HostContext> CreateMultiThreadedHostContext( in CreateMultiThreadedHostContext() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/
H A Dbenchmark_mlir_function.cc89 num_threads > 0 ? CreateMultiThreadedHostContext(num_threads) in RunJitRtBenchmark()
H A Dcwise_op_unary_benchmark.h66 num_threads > 0 ? CreateMultiThreadedHostContext(num_threads) in PrepareUnaryMlirBenchmark()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/
H A Druntime_fallback_executor.cc97 host_context_ = num_threads ? CreateMultiThreadedHostContext(num_threads) in RuntimeFallbackExecutor()