| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 85 ExecuteAsyncOnStream(run_options, arguments, hlo_execution_profile); in ExecuteOnStream()
102 StatusOr<ScopedShapedBuffer> Executable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::Executable
112 ExecuteAsyncOnStream(run_options, std::move(args), in ExecuteAsyncOnStream()
121 StatusOr<ExecutionOutput> result = ExecuteAsyncOnStream( in ExecuteOnStream()
150 auto rv, ExecuteAsyncOnStream(&run_options[i], arguments[i], in ExecuteOnStreams()
283 ExecuteAsyncOnStream(run_options, arguments, state.profile_ptr.get()); in ExecuteAsyncOnStreamWrapper()
293 StatusOr<ExecutionOutput> return_value = ExecuteAsyncOnStream( in ExecuteAsyncOnStreamWrapper()
|
| D | executable.h | 279 virtual StatusOr<ScopedShapedBuffer> ExecuteAsyncOnStream(
291 virtual StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | service.cc | 444 executables[i]->ExecuteAsyncOnStream( in ExecuteParallelAndRegisterResult()
|
| /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/compiler/xla/stream_executor/tpu/ |
| D | tpu_executable_interface.h | 47 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | tpu_executable.h | 33 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | tpu_executable_interface.cc | 183 StatusOr<ExecutionOutput> TpuExecutableInterface::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::TpuExecutableInterface
|
| D | tpu_executable.cc | 75 StatusOr<ExecutionOutput> TpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::TpuExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.h | 184 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
189 StatusOr<ScopedShapedBuffer> ExecuteAsyncOnStream(
|
| D | gpu_executable.cc | 555 StatusOr<ExecutionOutput> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
562 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream() function in xla::gpu::GpuExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.h | 56 StatusOr<ExecutionOutput> ExecuteAsyncOnStream(
|
| D | cpu_executable.cc | 318 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 | 95 executable->ExecuteAsyncOnStream( in RunMlirModule()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | buffer_donation_test.cc | 129 executable->ExecuteAsyncOnStream(&service_run_options, std::move(args), in RunAndCheck()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 503 tpu_executable->ExecuteAsyncOnStream(&service_run_options, in TPUExecute()
|