mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-09 18:40:18 +08:00
[Other] Optimize code style (#1032)
* Optimize code * optimize code * optimize code * fix compile error
This commit is contained in:
@@ -15,21 +15,18 @@
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <cstdint>
|
||||
#include <cuda.h>
|
||||
#include <cuda_runtime.h>
|
||||
#include <cstdint>
|
||||
#include <iostream>
|
||||
#include <vector>
|
||||
#include <math.h>
|
||||
#include <vector>
|
||||
|
||||
namespace fastdeploy {
|
||||
|
||||
void CudaAdaptivePool(const std::vector<int64_t>& input_dims,
|
||||
const std::vector<int64_t>& output_dims,
|
||||
float* output,
|
||||
const float* input,
|
||||
void* compute_stream,
|
||||
const std::vector<int64_t>& output_dims, float* output,
|
||||
const float* input, void* compute_stream,
|
||||
const std::string& pooling_type);
|
||||
|
||||
|
||||
} // namespace fastdeploy
|
||||
|
Reference in New Issue
Block a user