Searched defs:NVTXThreadLocalState (Results 1 – 1 of 1) sorted by relevance
8 struct NVTXThreadLocalState : ProfilerStateBase { struct9 explicit NVTXThreadLocalState(const ProfilerConfig& config) in NVTXThreadLocalState() function19 ActiveProfilerType profilerType() override { in profilerType()23 void reportMemoryUsage(void*, int64_t, size_t, size_t, c10::Device) override { in reportMemoryUsage()26 static NVTXThreadLocalState* getTLS() { in getTLS()35 void setProducerTensorMap( in setProducerTensorMap()50 producer_tensor_map_;