Home
last modified time | relevance | path

Searched refs:ExecuteAsyncOnStreamWrapper (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.cc165 ExecuteAsyncOnStreamWrapper(run_options, arguments); in ExecuteOnStreamWrapper()
176 ExecuteAsyncOnStreamWrapper(run_options, std::move(arguments)); in ExecuteOnStreamWrapper()
278 StatusOr<ScopedShapedBuffer> Executable::ExecuteAsyncOnStreamWrapper( in ExecuteAsyncOnStreamWrapper() function in xla::Executable
289 StatusOr<ExecutionOutput> Executable::ExecuteAsyncOnStreamWrapper( in ExecuteAsyncOnStreamWrapper() function in xla::Executable
H A Dexecutable.h325 StatusOr<ScopedShapedBuffer> ExecuteAsyncOnStreamWrapper(
329 StatusOr<ExecutionOutput> ExecuteAsyncOnStreamWrapper(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dlocal_client.cc261 executable_->ExecuteAsyncOnStreamWrapper( in RunAsync()
313 executable_->ExecuteAsyncOnStreamWrapper( in RunAsync()