Home
last modified time | relevance | path

Searched refs:runner_table (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/runtime_fallback/kernel/
Dkernel_fallback_compat_request_state.cc66 core::RefCountPtr<Rendezvous> rendezvous, OpKernelRunnerTable* runner_table, in KernelFallbackCompatRequestState() argument
80 runner_table_(runner_table), in KernelFallbackCompatRequestState()
105 OpKernelRunnerTable* runner_table, FallbackResourceArray* resource_array, in KernelFallbackCompatRequestState() argument
130 runner_table, resource_array, user_intra_op_threadpool,
Dkernel_fallback_execute_compat.cc120 auto* runner_table = in SetUpKernelFallbackCompatRequestContext() local
130 runner_table, resource_array, user_intra_op_threadpool, model_metadata, in SetUpKernelFallbackCompatRequestContext()
137 tfrt::RequestContextBuilder* builder, OpKernelRunnerTable* runner_table, in SetUpKernelFallbackCompatRequestContext() argument
145 if (runner_table == nullptr) in SetUpKernelFallbackCompatRequestContext()
146 runner_table = in SetUpKernelFallbackCompatRequestContext()
160 runner_table, resource_array, user_intra_op_threadpool, in SetUpKernelFallbackCompatRequestContext()
582 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackExecuteOp() local
583 DCHECK(runner_table); in KernelFallbackExecuteOp()
585 auto* kernel_runner = runner_table->Get(frame.op_key().GetValue()); in KernelFallbackExecuteOp()
614 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackCreateOp() local
[all …]
Dkernel_fallback_compat_request_state.h80 tfrt_stub::OpKernelRunnerTable* runner_table,
90 tfrt_stub::OpKernelRunnerTable* runner_table,
119 tfrt_stub::OpKernelRunnerTable* runner_table() const { return runner_table_; } in runner_table() function
Dkernel_fallback_execute_compat.h60 tfrt_stub::OpKernelRunnerTable* runner_table,
/external/tensorflow/tensorflow/core/tfrt/eager/
Dfunction_cache.h50 runner_table) { in CreateFunctionState()
53 std::move(fn), std::move(runner_table))); in CreateFunctionState()
70 std::unique_ptr<tensorflow::tfrt_stub::OpKernelRunnerTable> runner_table) in FunctionState() argument
76 runner_table_(std::move(runner_table)) {} in FunctionState()
Dfunction_cache.cc132 auto runner_table = in GetOrAddFunction() local
135 TF_RETURN_IF_ERROR(request_ctx_fn(runner_table.get(), &request_ctx)); in GetOrAddFunction()
143 std::move(expected_fn.get()), std::move(runner_table)); in GetOrAddFunction()
Dfunction_cache_test.cc236 tensorflow::tfrt_stub::OpKernelRunnerTable* runner_table, in TEST_P()
Dc_api_tfrt.cc1010 tensorflow::tfrt_stub::OpKernelRunnerTable* runner_table, in BuildFunctionRequestContext() argument
1017 &request_context_builder, runner_table, GetEagerContext())); in BuildFunctionRequestContext()
1543 [this](tensorflow::tfrt_stub::OpKernelRunnerTable* runner_table, in Initialize()
1545 return context_->BuildFunctionRequestContext(runner_table, request_ctx); in Initialize()
Dc_api_tfrt.h225 tensorflow::tfrt_stub::OpKernelRunnerTable* runner_table,