| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 86 ExecuteAsyncOnStream(run_options, arguments, hlo_execution_profile); in ExecuteOnStream()
103 StatusOr<ScopedShapedBuffer> Executable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::Executable
113 ExecuteAsyncOnStream(run_options, std::move(args), in ExecuteAsyncOnStream()
122 StatusOr<ExecutionOutput> result = ExecuteAsyncOnStream( in ExecuteOnStream()
151 auto rv, ExecuteAsyncOnStream(&run_options[i], arguments[i], in ExecuteOnStreams()
284 ExecuteAsyncOnStream(run_options, arguments, state.profile_ptr.get()); in ExecuteAsyncOnStreamWrapper()
294 StatusOr<ExecutionOutput> return_value = ExecuteAsyncOnStream( in ExecuteAsyncOnStreamWrapper()
|
| D | executable.h | 281 virtual StatusOr<ScopedShapedBuffer> ExecuteAsyncOnStream(
293 virtual StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | service.cc | 414 executables[i]->ExecuteAsyncOnStream( in ExecuteParallelAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 114 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
119 StatusOr<ScopedShapedBuffer> ExecuteAsyncOnStream(
|
| D | gpu_executable.cc | 398 StatusOr<ExecutionOutput> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
406 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable_base.h | 40 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | executable_base.cc | 41 StatusOr<ExecutionOutput> InterpreterExecutableBase::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::interpreter::InterpreterExecutableBase
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 47 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | tpu_executable_interface.cc | 168 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::TpuExecutableInterface
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 58 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | cpu_executable.cc | 325 StatusOr<ExecutionOutput> CpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::cpu::CpuExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 100 executable->ExecuteAsyncOnStream( in RunMlirModule()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 126 executable->ExecuteAsyncOnStream(&service_run_options, std::move(args), in RunAndCheck()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 499 tpu_executable->ExecuteAsyncOnStream(&service_run_options, in TPUExecute()
|
| D | tpu_on_demand_compiler.cc | 99 StatusOr<ExecutionOutput> ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::__anoncef202870111::TpuExecutable
|