Searched defs:AsyncState (Results 1 – 2 of 2) sorted by relevance
496 struct ExecutorState<PropagatorStateType>::AsyncState { struct in tensorflow::__anon194149970111::ExecutorState497 AsyncState(const OpKernelContext::Params& p, const TaggedNode& _tagged_node, in AsyncState() function514 TensorValueVec saved_inputs;515 AllocatorAttributeVec saved_input_alloc_attrs;516 OpKernelContext::Params params;517 TaggedNode tagged_node;518 const NodeItem* item;519 Entry* first_input;520 OpKernelContext ctx;521 NodeExecStatsInterface* stats;[all …]
290 struct AsyncState { in KernelFallbackExecuteCompatAsyncInternal() struct291 explicit AsyncState(const OpKernelRunState& rs, int num_outputs) in KernelFallbackExecuteCompatAsyncInternal() argument295 OpKernelRunState run_state; in KernelFallbackExecuteCompatAsyncInternal()296 OpKernelContext context; in KernelFallbackExecuteCompatAsyncInternal()298 tfrt::AsyncValueRef<tfrt::Chain> chain; in KernelFallbackExecuteCompatAsyncInternal()299 llvm::SmallVector<tfrt::AsyncValueRef<TensorType>, 4> result_refs; in KernelFallbackExecuteCompatAsyncInternal()