PandA-2024.02
|
#include "c_backend_api.h"
Go to the source code of this file.
Functions | |
int32_t | fused_nn_max_pool2d (void *args, void *arg_type_ids, int32_t num_args) |
int32_t | fused_nn_max_pool2d_wrapper (float *placeholder, float *tensor) |
Variables | |
TVMValue | param [3] |
TVMArray | a0 [1] |
TVMArray | a1 [1] |
TVMArray | a2 [1] |
int32_t fused_nn_max_pool2d | ( | void * | args, |
void * | arg_type_ids, | ||
int32_t | num_args | ||
) |
Definition at line 7 of file 12_maxp_a.cc.
Referenced by fused_nn_max_pool2d_wrapper().
int32_t fused_nn_max_pool2d_wrapper | ( | float * | placeholder, |
float * | tensor | ||
) |
Definition at line 14 of file 12_maxp_a.wrapper.c.
References __builtin_bambu_time_start(), __builtin_bambu_time_stop(), a0, a1, DLTensor::data, fused_nn_max_pool2d(), and TVMValue::v_handle.
TVMArray a0[1] |
Definition at line 4 of file 12_maxp_a.wrapper.c.
Referenced by fused_nn_max_pool2d_wrapper().
TVMArray a1[1] |
Definition at line 5 of file 12_maxp_a.wrapper.c.
Referenced by fused_nn_max_pool2d_wrapper().
TVMArray a2[1] |
Definition at line 6 of file 12_maxp_a.wrapper.c.
TVMValue param[3] |
Definition at line 3 of file 12_maxp_a.wrapper.c.