All the possible information needed by Operator.Forward and Backward This is the superset of RunContext. We use this data structure to bookkeep everything needed by Forward and Backward.
More...
#include <op_attr_types.h>
All the possible information needed by Operator.Forward and Backward This is the superset of RunContext. We use this data structure to bookkeep everything needed by Forward and Backward.
- See also
- Resource
◆ get_gpu_aux_stream()
get auxilary gpu stream auto-syncing object from Context
- Returns
- the aux stream auto-syncing object
◆ get_stream()
get mshadow stream from Context
- Returns
- the mshadow stream
- Template Parameters
-
xpu | the device type of the stream |
◆ async_on_complete
the callback when operation completes, used by asynchronize ops
◆ is_train
bool mxnet::OpContext::is_train |
whether it is training phase
◆ need_grad
bool mxnet::OpContext::need_grad |
whether there is a backward phase to compute gradients.
◆ requested
std::vector<Resource> mxnet::OpContext::requested |
Resources requested by the operator.
◆ run_ctx
The documentation for this struct was generated from the following file: