| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stream_pool_test.cc | 43 StreamPool::Ptr stream1 = pool.BorrowStream(executor.get()); in TEST_F()
49 StreamPool::Ptr stream2 = pool.BorrowStream(executor.get()); in TEST_F()
64 StreamPool::Ptr stream1 = pool.BorrowStream(executor.get()); in TEST_F()
67 StreamPool::Ptr stream2 = pool.BorrowStream(executor.get()); in TEST_F()
77 StreamPool::Ptr stream3 = pool.BorrowStream(executor.get()); in TEST_F()
87 StreamPool::Ptr stream4 = pool.BorrowStream(executor.get()); in TEST_F()
101 StreamPool::Ptr stream1 = pool.BorrowStream(executor.get()); in TEST_F()
111 StreamPool::Ptr stream2 = pool.BorrowStream(executor.get()); in TEST_F()
129 StreamPool::Ptr stream3 = pool.BorrowStream(executor.get()); in TEST_F()
140 StreamPool::Ptr stream1 = pool.BorrowStream(executor.get()); in TEST_F()
[all …]
|
| D | backend.cc | 108 StatusOr<StreamPool::Ptr> Backend::BorrowStream(int device_ordinal) { in BorrowStream() function in xla::Backend
110 return BorrowStream(executor); in BorrowStream()
113 StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) { in BorrowStream() function in xla::Backend
118 return stream_pools_.at(executor)->BorrowStream(executor); in BorrowStream()
|
| D | backend.h | 124 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal);
125 StatusOr<StreamPool::Ptr> BorrowStream(se::StreamExecutor* executor);
131 return [this](int device_ordinal) { return BorrowStream(device_ordinal); }; in StreamBorrower()
|
| D | stream_pool.h | 48 Ptr BorrowStream(se::StreamExecutor* executor);
|
| D | service_executable_run_options.h | 53 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream() function
|
| D | stream_pool.cc | 23 StreamPool::Ptr StreamPool::BorrowStream(se::StreamExecutor* executor) { in BorrowStream() function in xla::StreamPool
|
| D | service.cc | 379 backend->BorrowStream(replicas[replica])); in ExecuteParallelAndRegisterResult()
481 backend->BorrowStream(executor)); in ExecuteAndRegisterResult()
653 execute_backend_->BorrowStream( in ExecuteGraphParallel()
720 execute_backend_->BorrowStream(all_executors[i][0])); in ExecuteGraphParallel()
868 execute_backend_->BorrowStream( in Execute()
931 TF_ASSIGN_OR_RETURN(auto stream, execute_backend_->BorrowStream( in TransferToClient()
971 TF_ASSIGN_OR_RETURN(auto stream, execute_backend_->BorrowStream(executor)); in TransferToServer()
|
| D | hlo_runner.cc | 55 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralToDevice()
86 auto stream, backend().BorrowStream(backend().default_stream_executor())); in TransferLiteralFromDevice()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 76 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal)); in AllocateScopedShapedBuffer()
178 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal)); in CreateAndTransfer()
252 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal())); in StoreToLiteral()
274 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal())); in WriteLiteral()
310 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal())); in SwapIn()
460 TF_ASSIGN_OR_RETURN(auto stream, backend->BorrowStream(device_ordinal)); in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 40 return backend->BorrowStream(device_ordinal); in BorrowStreamForDevice()
409 mutable_backend()->BorrowStream(device_ordinal)); in LiteralToShapedBuffer()
417 TF_ASSIGN_OR_RETURN(auto stream, mutable_backend()->BorrowStream( in ShapedBufferToLiteral()
456 mutable_backend()->BorrowStream(device_ordinal)); in TransferToLocalServer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 121 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in RunMlirModuleWithHostBuffers()
168 auto stream = backend_->BorrowStream(backend_->default_device_ordinal()) in CompileMlirText()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 229 client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
279 auto status_or_new_stream = client_->mutable_backend()->BorrowStream( in CopyDeviceTensorToCPU()
|
| D | xla_device.cc | 283 TF_ASSIGN_OR_RETURN(ptr, backend->BorrowStream(device_ordinal_)); in EnsureStreamOkLocked()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_test_base.cc | 207 ->BorrowStream(device_ordinal) in ExecuteLocally()
|
| D | xla_hlo_profile_test.cc | 142 backend->BorrowStream(backend->default_device_ordinal())); in ExecuteAndFetchProfile()
|
| D | transfer_manager_test.cc | 48 ->BorrowStream(stream_executor_)
|
| D | local_client_execute_test.cc | 776 ->BorrowStream(0) in XLA_TEST_F()
972 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_LocalClientOverhead()
|
| D | dynamic_ops_test.cc | 772 client->mutable_backend()->BorrowStream(device_ordinal).ValueOrDie(); in BM_DynamicSlice()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_execute_op.cc | 211 backend->BorrowStream(device_ordinal)); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 152 run_options->BorrowStream(executor->device_ordinal())); in ExecuteThunks()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_execute_op.cc | 687 backend->BorrowStream(device_ordinal)); in DoWork()
|