Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Protected Attributes
AsyncInputIsOutputTest Class Reference
Inheritance diagram for AsyncInputIsOutputTest:
AsyncTest AsyncAllreduceTest AsyncBroadcastTest

Public Member Functions

 AsyncInputIsOutputTest (const std::string &path, int numTensors)
void wait (std::shared_ptr< ProcessGroup::Work > &work)
std::vector< at::TensorgetTensors ()
- Public Member Functions inherited from AsyncTest
 AsyncTest (const std::string &path)
 AsyncTest (AsyncTest &&other)
::c10d::ProcessGroupGloo & getProcessGroup ()
void start (int rank, int size)

Protected Attributes

const int numTensors_
const int numDevices_
THCState * state_
std::vector< at::Tensorinputs_
std::vector< CUDAStream > streams_
- Protected Attributes inherited from AsyncTest
std::string path_
std::unique_ptr<::c10d::ProcessGroupGloo > pg_

Detailed Description

Definition at line 66 of file ProcessGroupGlooAsyncTest.cpp.

The documentation for this class was generated from the following file: