Searched defs:runner_table (Results 1 – 7 of 7) sorted by relevance
66 core::RefCountPtr<Rendezvous> rendezvous, OpKernelRunnerTable* runner_table, in KernelFallbackCompatRequestState()105 OpKernelRunnerTable* runner_table, FallbackResourceArray* resource_array, in KernelFallbackCompatRequestState()
120 auto* runner_table = in SetUpKernelFallbackCompatRequestContext() local137 tfrt::RequestContextBuilder* builder, OpKernelRunnerTable* runner_table, in SetUpKernelFallbackCompatRequestContext()582 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackExecuteOp() local614 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackCreateOp() local845 auto* runner_table = fallback_request_state->runner_table(); in KernelFallbackExecuteOpCustomAllocatorInternal() local
119 tfrt_stub::OpKernelRunnerTable* runner_table() const { return runner_table_; } in runner_table() function
49 std::unique_ptr<tensorflow::tfrt_stub::OpKernelRunnerTable> in CreateFunctionState()70 std::unique_ptr<tensorflow::tfrt_stub::OpKernelRunnerTable> runner_table) in FunctionState()
132 auto runner_table = in GetOrAddFunction() local
237 RCReference<RequestContext>* request_ctx) { in TEST_P()
1010 tensorflow::tfrt_stub::OpKernelRunnerTable* runner_table, in BuildFunctionRequestContext()1544 RCReference<RequestContext>* request_ctx) { in Initialize()