mxnet
|
The resources that can be requested by Operator. More...
#include <resource.h>
Public Types | |
enum | Type { kRandom, kTempSpace, kParallelRandom } |
Resource type, indicating what the pointer type is. More... | |
Public Member Functions | |
ResourceRequest () | |
default constructor More... | |
ResourceRequest (Type type) | |
constructor, allow implicit conversion More... | |
Public Attributes | |
Type | type |
type of resources More... | |
The resources that can be requested by Operator.
Resource type, indicating what the pointer type is.
Enumerator | |
---|---|
kRandom |
mshadow::Random<xpu> object |
kTempSpace |
A dynamic temp space that can be arbitrary size. |
kParallelRandom |
common::RandGenerator<xpu> object, which can be used in GPU kernel functions |
|
inline |
default constructor
|
inline |
constructor, allow implicit conversion
type | type of resources |
Type mxnet::ResourceRequest::type |
type of resources