e7ec22d8b8
* add swin-transformer * change dir * reformat * fix gen wts and polish readme Co-authored-by: wdhao <873665182@qq.com>
21 lines
333 B
C++
21 lines
333 B
C++
#ifndef UPSAMPLE_KERNEL_H
|
|
#define UPSAMPLE_KERNEL_H
|
|
|
|
#include <iostream>
|
|
#include "NvInfer.h"
|
|
|
|
int UpsampleInference(
|
|
cudaStream_t stream,
|
|
int n,
|
|
int input_b,
|
|
int input_c,
|
|
int input_h,
|
|
int input_w,
|
|
float scale_h,
|
|
float scale_w,
|
|
const void* inputs,
|
|
void* outputs);
|
|
|
|
|
|
#endif
|