1 #include "tvm/runtime/c_runtime_api.h" 2 #include "tvm/runtime/c_backend_api.h" 8 void* arg0 = (((
TVMValue*)args)[0].v_handle);
9 float* placeholder = (
float*)(((
TVMArray*)arg0)[0].data);
11 void* arg1 = (((
TVMValue*)args)[1].v_handle);
12 float* placeholder1 = (
float*)(((
TVMArray*)arg1)[0].data);
14 void* arg2 = (((
TVMValue*)args)[2].v_handle);
15 float* T_multiply = (
float*)(((
TVMArray*)arg2)[0].data);
17 for (int32_t ax1 = 0; ax1 < 64; ++ax1) {
18 T_multiply[ax1] = (placeholder[ax1] * placeholder1[ax1]);
Union type of values being passed through API and function calls.
Plain C Tensor object, does not manage memory.
TVM_DLL int32_t fused_multiply(void *args, void *arg_type_ids, int32_t num_args)