Home
last modified time | relevance | path

Searched defs:request_context_builder (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dbef_thunk.cc270 tfrt::RequestContextBuilder* request_context_builder) { in CreateXcclContext()
337 tfrt::RequestContextBuilder request_context_builder( in ExecuteOnStream() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Dwork_queue_interface.h48 tfrt::RequestContextBuilder* request_context_builder, in InitializeRequest()
H A Dtf_threadpool_concurrent_work_queue.cc38 ::tfrt::RequestContextBuilder* request_context_builder, in InitializeRequest()
H A Dwork_queue_interface.cc82 tfrt::RequestContextBuilder* request_context_builder, in InitializeRequest()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/
H A Dfallback_test_util.cc52 tfrt::RequestContextBuilder request_context_builder(host, resource_context, in CreateFallbackTestExecutionContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/run_handler_thread_pool/
H A Drun_handler_concurrent_work_queue.cc61 tfrt::RequestContextBuilder* request_context_builder, in InitializeRequest()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.cc94 tfrt::RequestContextBuilder request_context_builder( in SetUpRequestContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_batch_tf_opkernels.cc519 tfrt::RequestContextBuilder request_context_builder( in SetUpRequestContext() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.cc1013 RequestContextBuilder request_context_builder( in BuildFunctionRequestContext() local