Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::autograd::profiler::CUDAStubs Struct Reference

Public Member Functions

virtual void record (int *device, CUDAEventStub *event, int64_t *cpu_ns)
virtual float elapsed (CUDAEventStub event, CUDAEventStub event2)
virtual void nvtxMarkA (const char *name)
virtual void nvtxRangePushA (const char *name)
virtual void nvtxRangePop ()
virtual bool enabled ()
virtual void onEachDevice (std::function< void(int)> op)
virtual void synchronize ()

Detailed Description

Definition at line 30 of file profiler.h.

The documentation for this struct was generated from the following files: