#include <lib_api.h>
|
| 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) |
|
MXTensor * | alloc_arg (const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const |
|
MXTensor * | alloc_aux (const std::string &name, const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype) const |
|
◆ PassResource()
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 |
|
) |
| |
◆ alloc_arg()
MXTensor* mxnet::ext::PassResource::alloc_arg |
( |
const std::string & |
name, |
|
|
const std::vector< int64_t > & |
shapes, |
|
|
const MXContext & |
ctx, |
|
|
MXDType |
dtype |
|
) |
| const |
◆ alloc_aux()
MXTensor* mxnet::ext::PassResource::alloc_aux |
( |
const std::string & |
name, |
|
|
const std::vector< int64_t > & |
shapes, |
|
|
const MXContext & |
ctx, |
|
|
MXDType |
dtype |
|
) |
| const |
The documentation for this class was generated from the following file: