mxnet
mxnet::ext::CustomOp Member List

This is the complete list of members for mxnet::ext::CustomOp, including all inherited members.

backward_ctx_cstrmxnet::ext::CustomOp
backward_fpmxnet::ext::CustomOp
create_op_ctx_cstrmxnet::ext::CustomOp
create_op_fpmxnet::ext::CustomOp
CustomOp(const char *op_name)mxnet::ext::CustomOpexplicit
forward_ctx_cstrmxnet::ext::CustomOp
forward_fpmxnet::ext::CustomOp
infer_shapemxnet::ext::CustomOp
infer_storage_typemxnet::ext::CustomOp
infer_typemxnet::ext::CustomOp
isSGopmxnet::ext::CustomOp
mapToVector()mxnet::ext::CustomOp
mutate_inputsmxnet::ext::CustomOp
namemxnet::ext::CustomOp
parse_attrsmxnet::ext::CustomOp
setBackward(fcomp_t fgrad, const char *ctx)mxnet::ext::CustomOp
setCreateOpState(createOpState_t func, const char *ctx)mxnet::ext::CustomOp
setForward(fcomp_t fcomp, const char *ctx)mxnet::ext::CustomOp
setInferShape(inferShape_t func)mxnet::ext::CustomOp
setInferSType(inferSType_t func)mxnet::ext::CustomOp
setInferType(inferType_t func)mxnet::ext::CustomOp
setIsSubgraphOp()mxnet::ext::CustomOp
setMutateInputs(mutateInputs_t func)mxnet::ext::CustomOp
setParseAttrs(parseAttrs_t func)mxnet::ext::CustomOp