| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 58 TF_RETURN_IF_ERROR(options.stream()->BlockHostUntilDone()); in ExecuteOnStreams()
90 Status status = stream->BlockHostUntilDone(); in ExecuteOnStreamWrapper()
101 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ExecuteOnStreamWrapper()
|
| D | generic_transfer_manager.cc | 55 return stream->BlockHostUntilDone(); in WriteSingleTupleIndexTable()
62 Status status = stream->BlockHostUntilDone(); in TransferLiteralFromDevice()
147 return stream->BlockHostUntilDone(); in TransferLiteralToDeviceAsync()
|
| D | transfer_manager.cc | 108 return substream->BlockHostUntilDone(); in TransferLiteralToDevice()
151 return substream->BlockHostUntilDone(); in TransferArrayToDevice()
236 return stream->BlockHostUntilDone(); in WriteTupleIndexTables()
|
| D | execution_tracker.cc | 42 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in BlockUntilDone()
|
| D | hlo_runner.cc | 218 TF_RETURN_IF_ERROR(stream.BlockHostUntilDone()); in ExecuteWithDeviceBuffers()
253 TF_RETURN_IF_ERROR(stream.BlockHostUntilDone()); in ExecuteWithDeviceBuffers()
407 TF_RETURN_IF_ERROR(streams[i]->BlockHostUntilDone()); in ExecuteReplicated()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_thunk.cc | 78 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ExecuteOnStream()
96 Status block_status = stream->BlockHostUntilDone(); in ExecuteOnStream()
|
| D | buffer_comparator.cc | 153 TF_RETURN_IF_ERROR(stream_->BlockHostUntilDone()); in CompareEqualImpl()
170 TF_RETURN_IF_ERROR(stream_->BlockHostUntilDone()); in CompareEqual()
|
| D | while_thunk.cc | 69 Status block_status = stream->BlockHostUntilDone(); in ExecuteOnStream()
|
| D | conditional_thunk.cc | 74 Status block_status = stream->BlockHostUntilDone(); in ExecuteOnStream()
|
| D | gpu_executable.cc | 139 TF_RETURN_IF_ERROR(main_stream->BlockHostUntilDone()); in ExecuteThunks()
160 Status block_status = main_stream->BlockHostUntilDone(); in ExecuteThunks()
|
| D | hlo_execution_profiler.cc | 47 stream->BlockHostUntilDone().IgnoreError(); in GetCyclesTaken()
|
| D | infeed_thunk.cc | 111 Status block_status = stream->BlockHostUntilDone(); in ExecuteOnStream()
|
| D | buffer_comparator_test.cc | 62 TF_CHECK_OK(stream.BlockHostUntilDone()); in CompareEqualFloatBuffers()
|
| D | gpu_transfer_manager.cc | 83 Status block_status = stream->BlockHostUntilDone(); in EnqueueBuffersToInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 95 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
110 port::Status XlaInterpreterExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::interpreter::XlaInterpreterExecutor
|
| D | executor.h | 160 port::Status BlockHostUntilDone(Stream *stream) override;
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.cc | 167 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
182 port::Status HostExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::host::HostExecutor
|
| D | host_gpu_executor.h | 142 port::Status BlockHostUntilDone(Stream *stream) override;
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 489 port::Status StreamExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::StreamExecutor
491 SCOPED_TRACE(TraceListener::BlockHostUntilDone, &result, stream); in BlockHostUntilDone()
493 result = implementation_->BlockHostUntilDone(stream); in BlockHostUntilDone()
|
| D | stream_executor_internal.h | 257 virtual port::Status BlockHostUntilDone(Stream *stream) = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 204 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ExecuteLocally()
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 201 SE_ASSERT_OK(stream->BlockHostUntilDone()); in VerifyResults()
317 SE_ASSERT_OK(stream->BlockHostUntilDone()); in TYPED_TEST()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 182 port::Status BlockHostUntilDone(Stream* stream) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 274 TF_RETURN_IF_ERROR(run_options->stream()->BlockHostUntilDone()); in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 351 return dev_info->stream->BlockHostUntilDone(); in Sync()
|