Commit 44c66270 authored by Shahriar's avatar Shahriar Committed by Soumith Chintala
Browse files

Renamed vision.h files to vision_cpu.h and vision_cuda.h (#1051)

parent 8616227c
#pragma once #pragma once
#include "cpu/vision.h" #include "cpu/vision_cpu.h"
#ifdef WITH_CUDA #ifdef WITH_CUDA
#include "cuda/vision.h" #include "cuda/vision_cuda.h"
#endif #endif
// Interface for Python // Interface for Python
......
#pragma once #pragma once
#include "cpu/vision.h" #include "cpu/vision_cpu.h"
#ifdef WITH_CUDA #ifdef WITH_CUDA
#include "cuda/vision.h" #include "cuda/vision_cuda.h"
#endif #endif
std::tuple<at::Tensor, at::Tensor> ROIPool_forward( std::tuple<at::Tensor, at::Tensor> ROIPool_forward(
......
#include <ATen/TensorUtils.h> #include <ATen/TensorUtils.h>
#include "cpu/vision.h" #include "cpu/vision_cpu.h"
// implementation taken from Caffe2 // implementation taken from Caffe2
template <typename T> template <typename T>
......
#include "cpu/vision.h" #include "cpu/vision_cpu.h"
template <typename scalar_t> template <typename scalar_t>
at::Tensor nms_cpu_kernel( at::Tensor nms_cpu_kernel(
......
#pragma once #pragma once
#include "cpu/vision.h" #include "cpu/vision_cpu.h"
#ifdef WITH_CUDA #ifdef WITH_CUDA
#include "cuda/vision.h" #include "cuda/vision_cpu.h"
#endif #endif
at::Tensor nms( at::Tensor nms(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment