mxnet::ext::PassResource Class Reference

#include <lib_api.h>

Collaboration diagram for mxnet::ext::PassResource:

Public Member Functions
	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

Constructor & Destructor Documentation

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
	)

Member Function Documentation

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:

• /work/mxnet/include/mxnet/lib_api.h