PandA-2024.02
02_vecmul_b.cc
Go to the documentation of this file.
1 #include "tvm/runtime/c_runtime_api.h"
2 #include "tvm/runtime/c_backend_api.h"
3 extern void* __tvm_module_ctx = NULL;
4 #ifdef __cplusplus
5 extern "C"
6 #endif
7 TVM_DLL int32_t fused_multiply( void* args, void* arg_type_ids, int32_t num_args) {
8  void* arg0 = (((TVMValue*)args)[0].v_handle);
9  float* placeholder = (float*)(((TVMArray*)arg0)[0].data);
10 
11  void* arg1 = (((TVMValue*)args)[1].v_handle);
12  float* placeholder1 = (float*)(((TVMArray*)arg1)[0].data);
13 
14  void* arg2 = (((TVMValue*)args)[2].v_handle);
15  float* T_multiply = (float*)(((TVMArray*)arg2)[0].data);
16 
17  for (int32_t ax1 = 0; ax1 < 64; ++ax1) {
18  T_multiply[ax1] = (placeholder[ax1] * placeholder1[ax1]);
19  }
20  return 0;
21 }
22 
#define NULL
Union type of values being passed through API and function calls.
void * __tvm_module_ctx
#define TVM_DLL
Definition: c_runtime_api.h:59
Plain C Tensor object, does not manage memory.
Definition: dlpack.h:111
TVM_DLL int32_t fused_multiply(void *args, void *arg_type_ids, int32_t num_args)
Definition: 02_vecmul_b.cc:7

Generated on Mon Feb 12 2024 13:02:50 for PandA-2024.02 by doxygen 1.8.13