1 #include "tvm/runtime/c_runtime_api.h" 2 #include "tvm/runtime/c_backend_api.h" 9 void* arg0 = (((
TVMValue*)args)[0].v_handle);
10 float* placeholder = (
float*)(((
TVMArray*)arg0)[0].data);
12 void* arg1 = (((
TVMValue*)args)[1].v_handle);
13 float* tensor = (
float*)(((
TVMArray*)arg1)[0].data);
15 for (int32_t ax2 = 0; ax2 < 10; ++ax2) {
16 for (int32_t ax3 = 0; ax3 < 10; ++ax3) {
17 tensor[((ax2 * 10) + ax3)] = -3.402823e+38f;
18 for (int32_t rv = 0; rv < 3; ++rv) {
19 for (int32_t rv1 = 0; rv1 < 3; ++rv1) {
20 float _1 = tensor[((ax2 * 10) + ax3)];
21 float _2 = placeholder[((((ax2 * 96) + (rv * 32)) + (ax3 * 3)) + rv1)];
22 tensor[((ax2 * 10) + ax3)] = ((_1) > (_2) ? (_1) : (_2));
Union type of values being passed through API and function calls.
TVM_DLL int32_t fused_nn_max_pool2d(void *args, void *arg_type_ids, int32_t num_args)
Plain C Tensor object, does not manage memory.