transform_gpu.h

#ifndef CAFFE2_IMAGE_TRANSFORM_GPU_H_
#define CAFFE2_IMAGE_TRANSFORM_GPU_H_

#include "caffe2/core/context.h"

namespace caffe2 {

template <typename T_IN, typename T_OUT, class Context>
bool TransformOnGPU(
    Tensor& X,
    Tensor* Y,
    Tensor& mean,
    Tensor& std,
    Context* context);

}  // namespace caffe2

#endif