Searched defs:AsyncState (Results 1 – 4 of 4) sorted by relevance
517 struct ExecutorState<PropagatorStateType>::AsyncState { struct in tensorflow::__anon79fb0f210111::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()
127 pub struct AsyncState { struct138 impl BuilderState for AsyncState {} implementation
273 struct AsyncState { struct274 recently_imported: HashSet<(u32, String)>,275 legacy_loader: LegacyBlobLoader,276 db_path: PathBuf,