mxnet::ext::Node Class Reference

#include <lib_api.h>

Collaboration diagram for mxnet::ext::Node:

Public Member Functions
	Node ()
void	_setPassResource (PassResource *res_)
void	alloc_arg (const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype)
void	alloc_aux (const std::vector< int64_t > &shapes, const MXContext &ctx, MXDType dtype)

Public Attributes
std::string	op
std::string	name
MXTensor *	tensor
std::vector< NodeEntry >	inputs
std::vector< NodeEntry >	outputs
std::vector< Graph * >	subgraphs
std::unordered_map< std::string, std::string >	attrs

Constructor & Destructor Documentation

Node()

mxnet::ext::Node::Node

(

)

Member Function Documentation

_setPassResource()

void mxnet::ext::Node::_setPassResource

(

PassResource *

res_

)

alloc_arg()

void mxnet::ext::Node::alloc_arg	(	const std::vector< int64_t > &	shapes,
		const MXContext &	ctx,
		MXDType	dtype
	)

alloc_aux()

void mxnet::ext::Node::alloc_aux	(	const std::vector< int64_t > &	shapes,
		const MXContext &	ctx,
		MXDType	dtype
	)

Member Data Documentation

attrs

std::unordered_map<std::string, std::string> mxnet::ext::Node::attrs

inputs

std::vector<NodeEntry> mxnet::ext::Node::inputs

name

std::string mxnet::ext::Node::name

op

std::string mxnet::ext::Node::op

outputs

std::vector<NodeEntry> mxnet::ext::Node::outputs

subgraphs

std::vector<Graph*> mxnet::ext::Node::subgraphs

tensor

MXTensor* mxnet::ext::Node::tensor

---

The documentation for this class was generated from the following file:

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