Searched refs:StreamPool (Results 1 – 21 of 21) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | stream_pool_test.cc | 36 TEST_F(StreamPoolTest, EmptyPool) { StreamPool pool; } in TEST_F()
40 StreamPool pool; in TEST_F()
43 StreamPool::Ptr stream1 = pool.BorrowStream(executor.get()); in TEST_F()
49 StreamPool::Ptr stream2 = pool.BorrowStream(executor.get()); in TEST_F()
61 StreamPool pool; 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()
98 StreamPool pool; in TEST_F()
[all …]
|
| D | stream_pool.h | 30 class StreamPool {
34 StreamPool* pool;
41 StreamPool() {} in StreamPool() function
|
| D | stream_pool.cc | 23 StreamPool::Ptr StreamPool::BorrowStream(se::StreamExecutor* executor) { in BorrowStream()
55 void StreamPool::ReturnStream(se::Stream* stream) { in ReturnStream()
|
| D | backend.h | 124 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal);
125 StatusOr<StreamPool::Ptr> BorrowStream(se::StreamExecutor* executor);
130 std::function<StatusOr<StreamPool::Ptr>(int)> StreamBorrower() { in StreamBorrower()
178 absl::flat_hash_map<se::StreamExecutor*, std::unique_ptr<StreamPool>>
|
| D | execution_tracker.h | 43 AsyncExecution(Backend* backend, std::vector<StreamPool::Ptr> streams,
57 std::vector<StreamPool::Ptr> streams_;
75 std::vector<StreamPool::Ptr> stream,
|
| D | service_executable_run_options.h | 30 using StreamBorrower = std::function<StatusOr<StreamPool::Ptr>(int)>;
51 StatusOr<StreamPool::Ptr> BorrowStream(int device_ordinal) const { in BorrowStream()
|
| D | backend.cc | 130 StatusOr<StreamPool::Ptr> Backend::BorrowStream(int device_ordinal) { in BorrowStream()
135 StatusOr<StreamPool::Ptr> Backend::BorrowStream(se::StreamExecutor* executor) { in BorrowStream()
138 stream_pools_.emplace(executor, absl::make_unique<StreamPool>()); in BorrowStream()
|
| D | execution_tracker.cc | 28 std::vector<StreamPool::Ptr> streams, in AsyncExecution()
50 std::vector<StreamPool::Ptr> streams, in Register()
|
| D | service.cc | 401 std::vector<StreamPool::Ptr> streams; in ExecuteParallelAndRegisterResult()
428 TF_ASSIGN_OR_RETURN(StreamPool::Ptr stream, in ExecuteParallelAndRegisterResult()
523 std::vector<StreamPool::Ptr> streams; in ExecuteAndRegisterResult()
528 TF_ASSIGN_OR_RETURN(StreamPool::Ptr stream, in ExecuteAndRegisterResult()
541 for (const StreamPool::Ptr& stream : streams) { in ExecuteAndRegisterResult()
|
| /external/webrtc/webrtc/base/ |
| D | socketpool.h | 32 class StreamPool {
34 virtual ~StreamPool() { } in ~StreamPool()
46 class StreamCache : public StreamPool, public sigslot::has_slots<> {
48 StreamCache(StreamPool* pool);
63 StreamPool* pool_;
75 class NewSocketPool : public StreamPool {
95 class ReuseSocketPool : public StreamPool, public sigslot::has_slots<> {
119 class LoggingPoolAdapter : public StreamPool {
121 LoggingPoolAdapter(StreamPool* pool, LoggingSeverity level,
131 StreamPool* pool_;
|
| D | httpclient.h | 63 HttpClient(const std::string& agent, StreamPool* pool,
67 void set_pool(StreamPool* pool) { pool_ = pool; } in set_pool()
166 StreamPool* pool_;
|
| D | socketpool.cc | 27 StreamCache::StreamCache(StreamPool* pool) : pool_(pool) { in StreamCache()
237 StreamPool* pool, LoggingSeverity level, const std::string& label, in LoggingPoolAdapter()
|
| D | httpclient.cc | 267 HttpClient::HttpClient(const std::string& agent, StreamPool* pool, in HttpClient()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | hlo_execution_profiler.h | 43 const std::vector<StreamPool::Ptr>& sub_streams,
75 const std::vector<StreamPool::Ptr>& sub_streams_;
|
| D | hlo_execution_profiler.cc | 42 const std::vector<StreamPool::Ptr>& sub_streams, in GetCyclesTaken()
56 const std::vector<StreamPool::Ptr>& sub_streams, in HloExecutionProfiler()
|
| D | gpu_executable.cc | 89 std::vector<StreamPool::Ptr> sub_streams; in ExecuteThunks()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 35 StatusOr<StreamPool::Ptr> BorrowStreamForDevice(int device_ordinal, in BorrowStreamForDevice()
149 StreamPool::Ptr stream; in Run()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 220 stream_executor::port::StatusOr<xla::StreamPool::Ptr> ptr_or_status = in CopyDeviceTensorToCPU()
|
| D | xla_device.cc | 268 xla::StreamPool::Ptr ptr; in EnsureStreamOkLocked()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 64 StreamPool::Ptr stream_ptr_;
|
| D | xla_hlo_profile_test.cc | 141 StreamPool::Ptr stream_ptr, in ExecuteAndFetchProfile()
|