6 #ifndef MXNET_RESOURCE_H_
7 #define MXNET_RESOURCE_H_
9 #include <dmlc/logging.h>
64 template<
typename xpu,
typename DType>
66 mshadow::Stream<xpu> *stream)
const {
68 mshadow::Random<xpu, DType> *ret =
69 static_cast<mshadow::Random<xpu, DType>*
>(
ptr_);
70 ret->set_stream(stream);
83 template<
typename xpu,
int ndim>
84 inline mshadow::Tensor<xpu, ndim, real_t>
get_space(
85 mshadow::Shape<ndim> shape, mshadow::Stream<xpu> *stream)
const {
86 return get_space_typed<xpu, ndim, real_t>(shape, stream);
98 mshadow::Shape<ndim> shape)
const {
99 return get_host_space_typed<cpu, ndim, real_t>(shape);
111 template<
typename xpu,
int ndim,
typename DType>
113 mshadow::Shape<ndim> shape, mshadow::Stream<xpu> *stream)
const {
115 return mshadow::Tensor<xpu, ndim, DType>(
117 shape, shape[ndim - 1], stream);
128 template<
int ndim,
typename DType>
130 mshadow::Shape<ndim> shape)
const {
131 return mshadow::Tensor<cpu, ndim, DType>(
133 shape, shape[ndim - 1], NULL);
174 #endif // MXNET_RESOURCE_H_
Engine that schedules all the operations according to dependency.
virtual Resource Request(Context ctx, const ResourceRequest &req)=0
Get resource of requested type.
engine::VarHandle var
engine variable
Definition: resource.h:48
int32_t id
identifier of id information, used for debug purpose
Definition: resource.h:50
The resources that can be requested by Operator.
Definition: resource.h:18
ResourceRequest(Type type)
constructor, allow implicit conversion
Definition: resource.h:34
ResourceRequest req
The original request.
Definition: resource.h:46
Type type
type of resources
Definition: resource.h:27
A dynamic temp space that can be arbitrary size.
Definition: resource.h:24
mshadow::Tensor< cpu, ndim, DType > get_host_space_typed(mshadow::Shape< ndim > shape) const
Get CPU space as mshadow Tensor in specified type. The caller can request arbitrary size...
Definition: resource.h:129
Global resource manager.
Definition: resource.h:150
Resources used by mxnet operations. A resource is something special other than NDArray, but will still participate.
Definition: resource.h:44
mshadow::Random< xpu, DType > * get_random(mshadow::Stream< xpu > *stream) const
Get random number generator.
Definition: resource.h:65
mshadow::Tensor< xpu, ndim, DType > get_space_typed(mshadow::Shape< ndim > shape, mshadow::Stream< xpu > *stream) const
Get space requested as mshadow Tensor in specified type. The caller can request arbitrary size...
Definition: resource.h:112
mshadow::Tensor< xpu, ndim, real_t > get_space(mshadow::Shape< ndim > shape, mshadow::Stream< xpu > *stream) const
Get space requested as mshadow Tensor. The caller can request arbitrary size.
Definition: resource.h:84
ResourceRequest()
default constructor
Definition: resource.h:29
virtual ~ResourceManager() DMLC_THROW_EXCEPTION
virtual destructor
Definition: resource.h:167
Type
Resource type, indicating what the pointer type is.
Definition: resource.h:20
Resource()
default constructor
Definition: resource.h:57
Var * VarHandle
Variable pointer type, usually hold by user used to specify dependencies.
Definition: engine.h:27
configuation of mxnet as well as basic data structure.
mshadow::Tensor< cpu, ndim, real_t > get_host_space(mshadow::Shape< ndim > shape) const
Get cpu space requested as mshadow Tensor. The caller can request arbitrary size. ...
Definition: resource.h:97
mshadow::Random<xpu> object
Definition: resource.h:22
void * ptr_
pointer to the resource, do not use directly, access using member functions
Definition: resource.h:55
void * get_space_internal(size_t size) const
internal function to get space from resources.
virtual void SeedRandom(uint32_t seed)=0
Seed all the allocated random numbers.
void * get_host_space_internal(size_t size) const
internal function to get cpu space from resources.
static ResourceManager * Get()
Context information about the execution enviroment.
Definition: base.h:90