mxnet
mxnet
ext
CustomOp
mxnet::ext::CustomOp Member List
This is the complete list of members for
mxnet::ext::CustomOp
, including all inherited members.
backward_ctx_cstr
mxnet::ext::CustomOp
backward_fp
mxnet::ext::CustomOp
create_op_ctx_cstr
mxnet::ext::CustomOp
create_op_fp
mxnet::ext::CustomOp
CustomOp
(const char *op_name)
mxnet::ext::CustomOp
explicit
forward_ctx_cstr
mxnet::ext::CustomOp
forward_fp
mxnet::ext::CustomOp
infer_shape
mxnet::ext::CustomOp
infer_storage_type
mxnet::ext::CustomOp
infer_type
mxnet::ext::CustomOp
isSGop
mxnet::ext::CustomOp
mapToVector
()
mxnet::ext::CustomOp
mutate_inputs
mxnet::ext::CustomOp
name
mxnet::ext::CustomOp
parse_attrs
mxnet::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
Generated on Thu Jan 5 2023 03:47:41 for mxnet by
1.8.17