Searched defs:kernel_runner (Results 1 – 2 of 2) sorted by relevance
228 const OpKernelRunner& kernel_runner, in KernelFallbackExecuteCompatAsyncInternal()301 OpKernelRunState* run_state, const OpKernelRunner& kernel_runner, in KernelFallbackExecuteCompatSyncInternal()399 const OpKernelRunner& kernel_runner) { in GetTracingMetadata()489 const OpKernelRunner& kernel_runner, bool is_async, in KernelFallbackExecuteOpInternal()585 auto* kernel_runner = runner_table->Get(frame.op_key().GetValue()); in KernelFallbackExecuteOp() local848 auto* kernel_runner = runner_table->Get(attr_frame.op_key().GetValue()); in KernelFallbackExecuteOpCustomAllocatorInternal() local
53 const tfrt_stub::OpKernelRunner& kernel_runner) { in GetDeviceFromFallbackState()