| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 87 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
124 Status blocking_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStream()
157 TF_RETURN_IF_ERROR(options.stream()->BlockHostUntilDone()); in ExecuteOnStreams()
167 Status block_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStreamWrapper()
178 Status block_status = run_options->stream()->BlockHostUntilDone(); in ExecuteOnStreamWrapper()
222 Status status = stream->BlockHostUntilDone(); in ExecuteWrapperAfterExecution()
236 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ExecuteWrapperAfterExecution()
|
| D | generic_transfer_manager.cc | 89 done(stream->BlockHostUntilDone()); in TransferLiteralFromDevice()
135 return stream->BlockHostUntilDone(); in TransferLiteralToDeviceAsync()
|
| D | transfer_manager.cc | 111 return substream->BlockHostUntilDone(); in TransferLiteralToDevice()
154 return substream->BlockHostUntilDone(); in TransferArrayToDevice()
207 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ReadDynamicShapes()
290 return stream->BlockHostUntilDone(); in WriteTupleIndexTables()
|
| D | execution_tracker.cc | 42 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in BlockUntilDone()
|
| D | hlo_runner.cc | 179 TF_RETURN_IF_ERROR(stream.BlockHostUntilDone()); in ExecuteWithDeviceBuffers()
310 TF_RETURN_IF_ERROR(streams[i]->BlockHostUntilDone()); in ExecuteReplicatedImpl()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 96 [other]() { SE_CHECK_OK(other->BlockHostUntilDone()); }); in CreateStreamDependency()
111 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/compiler/xla/pjrt/ |
| D | local_device_state.cc | 80 status.Update(compute_stream_->BlockHostUntilDone()); in SynchronizeAllActivity()
81 status.Update(callback_stream_->BlockHostUntilDone()); in SynchronizeAllActivity()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | redzone_allocator.cc | 187 .BlockHostUntilDone()); in CheckRedzoneHost()
243 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ReinitializeRedzone()
279 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in CheckRedzonesForBuffer()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | memcpy_test.cc | 38 EXPECT_TRUE(stream.BlockHostUntilDone().ok()); in TEST()
|
| D | redzone_allocator_test.cc | 78 .BlockHostUntilDone()); in TEST()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_stream_test.cc | 45 TF_ASSERT_OK(stream.BlockHostUntilDone()); in TEST()
|
| D | host_gpu_executor.h | 129 port::Status BlockHostUntilDone(Stream *stream) override;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 104 TF_CHECK_OK(stream->BlockHostUntilDone()); in RunMlirModule()
140 TF_CHECK_OK(stream->BlockHostUntilDone()); in RunMlirModuleWithHostBuffers()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | infeed_thunk.cc | 65 Status block_status = stream.BlockHostUntilDone(); in ExecuteOnStream()
|
| D | buffer_comparator.cc | 632 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in DeviceCompare()
647 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in HostCompare()
|
| D | while_thunk.cc | 74 Status block_status = stream.BlockHostUntilDone(); in ExecuteOnStream()
|
| D | outfeed_thunk.cc | 101 Status block_status = stream.BlockHostUntilDone(); in ExecuteOnStream()
|
| D | conditional_thunk.cc | 86 Status block_status = stream.BlockHostUntilDone(); in ExecuteOnStream()
|
| D | hlo_execution_profiler.cc | 46 stream->BlockHostUntilDone().IgnoreError(); in GetCyclesTaken()
|
| D | kernel_thunk.cc | 76 CHECK(stream->BlockHostUntilDone().ok()); in PrintBufferContents()
|
| D | gpu_transfer_manager.cc | 78 Status block_status = stream->BlockHostUntilDone(); in EnqueueBuffersToInfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.h | 111 Status block_status = options.stream()->BlockHostUntilDone(); in AsyncCallAndBlockHostUntilDone()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 211 TF_RETURN_IF_ERROR(stream->BlockHostUntilDone()); in ExecuteLocally()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 475 port::Status StreamExecutor::BlockHostUntilDone(Stream *stream) { in BlockHostUntilDone() function in stream_executor::StreamExecutor
477 SCOPED_TRACE(TraceListener::BlockHostUntilDone, &result, stream); in BlockHostUntilDone()
479 result = implementation_->BlockHostUntilDone(stream); in BlockHostUntilDone()
|