Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
44c66270
Commit
44c66270
authored
Jun 26, 2019
by
Shahriar
Committed by
Soumith Chintala
Jun 25, 2019
Browse files
Renamed vision.h files to vision_cpu.h and vision_cuda.h (#1051)
parent
8616227c
Changes
7
Hide whitespace changes
Inline
Side-by-side
Showing
7 changed files
with
8 additions
and
8 deletions
+8
-8
torchvision/csrc/ROIAlign.h
torchvision/csrc/ROIAlign.h
+2
-2
torchvision/csrc/ROIPool.h
torchvision/csrc/ROIPool.h
+2
-2
torchvision/csrc/cpu/ROIAlign_cpu.cpp
torchvision/csrc/cpu/ROIAlign_cpu.cpp
+1
-1
torchvision/csrc/cpu/nms_cpu.cpp
torchvision/csrc/cpu/nms_cpu.cpp
+1
-1
torchvision/csrc/cpu/vision_cpu.h
torchvision/csrc/cpu/vision_cpu.h
+0
-0
torchvision/csrc/cuda/vision_cuda.h
torchvision/csrc/cuda/vision_cuda.h
+0
-0
torchvision/csrc/nms.h
torchvision/csrc/nms.h
+2
-2
No files found.
torchvision/csrc/ROIAlign.h
View file @
44c66270
#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
...
...
torchvision/csrc/ROIPool.h
View file @
44c66270
#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
(
...
...
torchvision/csrc/cpu/ROIAlign_cpu.cpp
View file @
44c66270
#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
>
...
...
torchvision/csrc/cpu/nms_cpu.cpp
View file @
44c66270
#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
(
...
...
torchvision/csrc/cpu/vision.h
→
torchvision/csrc/cpu/vision
_cpu
.h
View file @
44c66270
File moved
torchvision/csrc/cuda/vision.h
→
torchvision/csrc/cuda/vision
_cuda
.h
View file @
44c66270
File moved
torchvision/csrc/nms.h
View file @
44c66270
#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
(
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment