Home
last modified time | relevance | path

Searched defs:ExecuteAsyncOnStream (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executable.cc75 StatusOr<ExecutionOutput> TpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::TpuExecutable
H A Dtpu_executable_interface.cc183 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::TpuExecutableInterface
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc555 StatusOr<ExecutionOutput> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
562 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dexecutable_base.cc41 StatusOr<ExecutionOutput> InterpreterExecutableBase::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::interpreter::InterpreterExecutableBase
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_executable.cc318 StatusOr<ExecutionOutput> CpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::cpu::CpuExecutable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dexecutable.cc102 StatusOr<ScopedShapedBuffer> Executable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::Executable