Home
last modified time | relevance | path

Searched refs:async_value (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir/utils/runtime/
H A Dasync_runtime_api.cc44 auto *async_value = AsyncRuntime::GetAsyncValue(token); in ConvertAsyncTokenToChain() local
45 auto out_chain = AsyncValueRef<Chain>(FormRef(async_value)); in ConvertAsyncTokenToChain()
53 auto *async_value = AsyncRuntime::GetAsyncValue(value); in ExtractAsyncValue() local
56 if (async_value->IsAvailable()) { in ExtractAsyncValue()
64 async_value->AndThen([value, emplace_fn, dst = FormRef(dst)]() { in ExtractAsyncValue()
75 auto *async_value = AsyncRuntime::GetAsyncValue(value); in ExtractAsyncValue() local
78 if (async_value->IsAvailable()) { in ExtractAsyncValue()
86 async_value->AndThen([value, emplace_fn, context, dst = FormRef(dst)]() { in ExtractAsyncValue()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_future.h162 tfrt::AsyncValueRef<T> async_value,
167 : promise_ref_(std::move(async_value)), in promise_ref_() argument