mxnet
|
This is the complete list of members for mxnet::ext::PassResource, including all inherited members.
alloc_arg(const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const | mxnet::ext::PassResource | |
alloc_aux(const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const | mxnet::ext::PassResource | |
PassResource(std::unordered_map< std::string, MXTensor > *new_args, std::unordered_map< std::string, MXTensor > *new_aux, nd_malloc_t nd_malloc, const void *nd_alloc) | mxnet::ext::PassResource |