Searched defs:AsyncState (Results 1 – 2 of 2) sorted by relevance
517 struct ExecutorState<PropagatorStateType>::AsyncState { struct in tensorflow::__anon35c05eb00111::ExecutorState518 AsyncState(const OpKernelContext::Params& p, const TaggedNode& _tagged_node, in AsyncState() argument536 TensorValueVec saved_inputs;537 AllocatorAttributeVec saved_input_alloc_attrs;538 OpKernelContext::Params params;539 TaggedNode tagged_node;540 const NodeItem* item;541 Entry* first_input;542 OpKernelContext ctx;543 NodeExecStatsInterface* stats;[all …]
244 struct AsyncState { in KernelFallbackExecuteCompatAsyncInternal() struct245 explicit AsyncState(const OpKernelRunState& rs, int num_outputs) in KernelFallbackExecuteCompatAsyncInternal() function249 OpKernelRunState run_state; in KernelFallbackExecuteCompatAsyncInternal()250 OpKernelContext context; in KernelFallbackExecuteCompatAsyncInternal()252 tfrt::AsyncValueRef<tfrt::Chain> chain; in KernelFallbackExecuteCompatAsyncInternal()253 llvm::SmallVector<tfrt::AsyncValueRef<TensorType>, 4> result_refs; in KernelFallbackExecuteCompatAsyncInternal()