Joseph Redmon
2015-03-08 f047cfff99e00e28c02eb59b6d32386c122f9af6
src/crop_layer.h
@@ -1,7 +1,6 @@
#ifndef CROP_LAYER_H
#define CROP_LAYER_H
#include "opencl.h"
#include "image.h"
typedef struct {
@@ -12,16 +11,16 @@
    int flip;
    float *output;
#ifdef GPU
    cl_mem output_cl;
    float *output_gpu;
#endif
} crop_layer;
image get_crop_image(crop_layer layer);
crop_layer *make_crop_layer(int batch, int h, int w, int c, int crop_height, int crop_width, int flip);
void forward_crop_layer(const crop_layer layer, float *input);
void forward_crop_layer(const crop_layer layer, int train, float *input);
#ifdef GPU
void forward_crop_layer_gpu(crop_layer layer, cl_mem input);
void forward_crop_layer_gpu(crop_layer layer, int train, float *input);
#endif
#endif