-
Notifications
You must be signed in to change notification settings - Fork 0
/
morphology.h
30 lines (24 loc) · 960 Bytes
/
morphology.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#ifndef IMGDISTORT_SRC_MORPHOLOGY_H_
#define IMGDISTORT_SRC_MORPHOLOGY_H_
#include <cuda_runtime.h>
void dilate_NCHW_f32(
float* dst, const float* src,
const int N, const int C, const int H, const int W,
const bool* M, const int Mn, const int Mh, const int Mw,
cudaStream_t stream = 0);
void dilate_NCHW_f64(
double* dst, const double* src,
const int N, const int C, const int H, const int W,
const bool* M, const int Mn, const int Mh, const int Mw,
cudaStream_t stream = 0);
void erode_NCHW_f32(
float* dst, const float* src,
const int N, const int C, const int H, const int W,
const bool* M, const int Mn, const int Mh, const int Mw,
cudaStream_t stream = 0);
void erode_NCHW_f64(
double* dst, const double* src,
const int N, const int C, const int H, const int W,
const bool* M, const int Mn, const int Mh, const int Mw,
cudaStream_t stream = 0);
#endif // IMGDISTORT_SRC_MORPHOLOGY_H_