Searched defs:op_chain (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/runtime_fallback/kernel/ |
| D | kernel_fallback_execute_compat.cc | 80 tfrt::string_view op_name, tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackEmitError()
229 tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackExecuteCompatAsyncInternal()
302 tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackExecuteCompatSyncInternal()
337 auto op_chain = tfrt::GetReadyChain(); in KernelFallbackExecuteCompatCoreRuntimeDispatch() local
485 tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackExecuteOpInternal()
567 tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackExecuteOp()
722 auto op_chain = tfrt::MakeUnconstructedAsyncValueRef<tfrt::Chain>(); in FallbackAsyncExecuteOp() local
740 tfrt::AsyncValueRef<tfrt::Chain> op_chain(tfrt::FormRef(all_args[0])); in FallbackAsyncExecuteOpSeq() local
830 tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in KernelFallbackExecuteOpCustomAllocatorInternal()
885 tfrt::AsyncValueRef<tfrt::Chain> op_chain(tfrt::FormRef(args.front())); in FallbackAsyncExecuteOpSeqWithAllocator() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_batch_kernels.cc | 51 AsyncValueRef<Chain>* op_chain, in ExecuteFallbackOp()
179 auto op_chain = in_op_chain.ValueRef(); in BatchFunctionFallback() local
|
| D | runtime_fallback_kernels.cc | 1080 StringAttr op_name_attr, tfrt::AsyncValueRef<tfrt::Chain>* op_chain, in RuntimeFallbackExecuteOp()
|