Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dprofiler_kineto.cpp515 void pushProfilingCallbacks(const std::unordered_set<at::RecordScope>& scopes) { in pushProfilingCallbacks() function
628 config.global() ? pushProfilingCallbacks</*global=*/true>(scopes) in toggleTorchOpCollectionDynamic()
629 : pushProfilingCallbacks</*global=*/false>(scopes); in toggleTorchOpCollectionDynamic()
743 config.global() ? pushProfilingCallbacks</*global=*/true>(scopes) in enableProfiler()
744 : pushProfilingCallbacks</*global=*/false>(scopes); in enableProfiler()
771 pushProfilingCallbacks</*global=*/false>(state_info_ptr->scopes); in enableProfilerInChildThread()