mxnet
|
This is the complete list of members for CustomOp, including all inherited members.
backward_ctx_cstr | CustomOp | |
backward_fp | CustomOp | |
create_op_ctx_cstr | CustomOp | |
create_op_fp | CustomOp | |
CustomOp(const char *op_name) | CustomOp | inlineexplicit |
forward_ctx_cstr | CustomOp | |
forward_fp | CustomOp | |
infer_shape | CustomOp | |
infer_storage_type | CustomOp | |
infer_type | CustomOp | |
isSGop | CustomOp | |
mapToVector() | CustomOp | inline |
mutate_inputs | CustomOp | |
name | CustomOp | |
parse_attrs | CustomOp | |
setBackward(fcomp_t fgrad, const char *ctx) | CustomOp | inline |
setCreateOpState(createOpState_t func, const char *ctx) | CustomOp | inline |
setForward(fcomp_t fcomp, const char *ctx) | CustomOp | inline |
setInferShape(inferShape_t func) | CustomOp | inline |
setInferSType(inferSType_t func) | CustomOp | inline |
setInferType(inferType_t func) | CustomOp | inline |
setIsSubgraphOp() | CustomOp | inline |
setMutateInputs(mutateInputs_t func) | CustomOp | inline |
setParseAttrs(parseAttrs_t func) | CustomOp | inline |
~CustomOp() | CustomOp | inline |