Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
torch / include / torch / csrc / profiler / standalone / nvtx_observer.h
Size: Mime:
#include <torch/csrc/profiler/api.h>

namespace torch {
namespace profiler {
namespace impl {

void pushNVTXCallbacks(
    const ProfilerConfig& config,
    const std::unordered_set<at::RecordScope>& scopes);

} // namespace impl
} // namespace profiler
} // namespace torch