Home
last modified time | relevance | path

Searched refs:ThreadLocalSubqueue (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.cpp296 ThreadLocalSubqueue::TorchOpStorage::EventBlock<T, ChunkSize>::EventBlock() { in EventBlock()
302 std::pair<KinetoObserverContext::Event*, uint64_t> ThreadLocalSubqueue::
309 uint64_t ThreadLocalSubqueue::TorchOpStorage::OpList::correlationID( in correlationID()
315 uint64_t ThreadLocalSubqueue::TorchOpStorage::EventBlock<T, ChunkSize>::
325 std::unique_ptr<KinetoObserverContext> ThreadLocalSubqueue::begin_op( in begin_op()
424 void ThreadLocalSubqueue::TorchOpStorage::materialize( in materialize()
472 ThreadLocalSubqueue::TorchOpStorage::OpList::correlationID(event), in materialize()
528 ThreadLocalSubqueue* ref_;
655 ThreadLocalSubqueue::ThreadLocalSubqueue( in ThreadLocalSubqueue() function in torch::profiler::impl::ThreadLocalSubqueue
684 ThreadLocalSubqueue* RecordQueue::getSubqueue() { in getSubqueue()
[all …]
H A Dcollection.h515 class TORCH_API ThreadLocalSubqueue {
517 ThreadLocalSubqueue(const uint64_t tid, ProfilerConfig config);
643 ThreadLocalSubqueue* getSubqueue();
660 ska::flat_hash_map<uint64_t, std::unique_ptr<ThreadLocalSubqueue>>