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<Context>& X, Tensor<Context>* Y,
                    Tensor<Context>& mean, Tensor<Context>& std,
                    Context* context);

}  // namespace caffe2

#endif