e1_mlp.cc File Reference

#include "tvm/runtime/c_runtime_api.h"
#include "tvm/runtime/c_backend_api.h"

Include dependency graph for e1_mlp.cc:

Go to the source code of this file.

Functions
TVM_DLL int32_t	fused_nn_dense_add (void *args, void *arg_type_ids, int32_t num_args)
TVM_DLL int32_t	fused_nn_softmax (void *args, void *arg_type_ids, int32_t num_args)

Variables
void *	__tvm_module_ctx = NULL

Function Documentation

fused_nn_dense_add()

TVM_DLL int32_t fused_nn_dense_add	(	void *	args,
		void *	arg_type_ids,
		int32_t	num_args
	)

Definition at line 7 of file e1_mlp.cc.

References compute(), k, NULL, and lenet_tvm::shape.

Here is the call graph for this function:

fused_nn_softmax()

TVM_DLL int32_t fused_nn_softmax	(	void *	args,
		void *	arg_type_ids,
		int32_t	num_args
	)

Definition at line 72 of file e1_mlp.cc.

References NULL, and lenet_tvm::shape.

Referenced by fused_nn_softmax_wrapper().

Here is the caller graph for this function:

Variable Documentation

__tvm_module_ctx

void* __tvm_module_ctx = NULL