Home
last modified time | relevance | path

Searched refs:ExecuteOnStreamWrapper (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.h317 StatusOr<ScopedShapedBuffer> ExecuteOnStreamWrapper(
321 StatusOr<ExecutionOutput> ExecuteOnStreamWrapper(
H A Dexecutable.cc161 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStreamWrapper( in ExecuteOnStreamWrapper() function in xla::Executable
172 StatusOr<ExecutionOutput> Executable::ExecuteOnStreamWrapper( in ExecuteOnStreamWrapper() function in xla::Executable
H A Dhlo_runner.cc188 executable->ExecuteOnStreamWrapper(&service_run_options, in ExecuteWithDeviceBuffers()
H A Dservice.cc537 TF_ASSIGN_OR_RETURN(auto result, executable->ExecuteOnStreamWrapper( in ExecuteAndRegisterResult()