PandA-2024.02
12_maxp_a.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_nn_max_pool2d( void* args, void* arg_type_ids, int32_t num_args) {
8  void* arg0 = (((TVMValue*)args)[0].v_handle);
9  void* arg1 = (((TVMValue*)args)[1].v_handle);
10  float* placeholder = (float*)(((TVMArray*)arg0)[0].data);
11  float* tensor = (float*)(((TVMArray*)arg1)[0].data);
12 
13  for (int32_t ax2 = 0; ax2 < 2; ++ax2) {
14  for (int32_t ax3 = 0; ax3 < 2; ++ax3) {
15  tensor[((ax2 * 2) + ax3)] = -3.402823e+38f;
16  for (int32_t rv = 0; rv < 3; ++rv) {
17  for (int32_t rv1 = 0; rv1 < 3; ++rv1) {
18  float _1 = tensor[((ax2 * 2) + ax3)];
19  float _2 = placeholder[((((ax2 * 24) + (rv * 8)) + (ax3 * 3)) + rv1)];
20  tensor[((ax2 * 2) + ax3)] = ((_1) > (_2) ? (_1) : (_2));
21  }
22  }
23  }
24  }
25  return 0;
26 }
27 
#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
TVM_DLL int32_t fused_nn_max_pool2d(void *args, void *arg_type_ids, int32_t num_args)
Definition: 12_maxp_a.cc:7
Plain C Tensor object, does not manage memory.
Definition: dlpack.h:111

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