Home
last modified time | relevance | path

Searched defs:runner_table (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_compat_request_state.cc70 core::RefCountPtr<Rendezvous> rendezvous, OpKernelRunnerTable* runner_table, in KernelFallbackCompatRequestState()
108 OpKernelRunnerTable* runner_table, FallbackResourceArray* resource_array, in KernelFallbackCompatRequestState()
Dkernel_fallback_compat_request_state.h107 OpKernelRunnerTable* runner_table() const { return runner_table_; } in runner_table() function
Dkernel_fallback_execute_compat.cc101 auto* runner_table = in SetUpKernelFallbackCompatRequestContext() local
117 tfrt::RequestContextBuilder* builder, OpKernelRunnerTable* runner_table, in SetUpKernelFallbackCompatRequestContext()
678 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackExecuteOp() local
749 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackCreateOp() local
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache.h49 std::unique_ptr<tensorflow::tfd::OpKernelRunnerTable> runner_table) { in CreateFunctionState()
67 std::unique_ptr<tensorflow::tfd::OpKernelRunnerTable> runner_table) in FunctionState()
Dfunction_cache.cc128 auto runner_table = absl::make_unique<tensorflow::tfd::OpKernelRunnerTable>(); in GetOrAddFunction() local
Dfunction_cache_test.cc236 RCReference<RequestContext>* request_ctx) { in TEST_P()
Dc_api_tfrt.cc969 tensorflow::tfd::OpKernelRunnerTable* runner_table, in BuildFunctionRequestContext()
1478 RCReference<RequestContext>* request_ctx) { in Initialize()