Caffe2 - C++ API
A deep learning, cross platform ML framework
transform_gpu.h
1 
17 #ifndef CAFFE2_IMAGE_TRANSFORM_GPU_H_
18 #define CAFFE2_IMAGE_TRANSFORM_GPU_H_
19 
45 #include "caffe2/core/context.h"
46 
47 namespace caffe2 {
48 
49 template <typename T_IN, typename T_OUT, class Context>
50 bool TransformOnGPU(Tensor<Context>& X, Tensor<Context>* Y,
52  Context* context);
53 
54 } // namespace caffe2
55 
56 #endif
Definition: types.h:88
Copyright (c) 2016-present, Facebook, Inc.
Copyright (c) 2016-present, Facebook, Inc.