Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AliaksandrSiarohin/cuda-gridsample-grad2
/ functions
Functions
53 in github.com/AliaksandrSiarohin/cuda-gridsample-grad2
⨍
Functions
53
◇
Types & classes
8
↓ 8 callers
Method
create_random_input
(self, oob=False, max_dim=20, with_grad_output=False)
test3d.py:154
↓ 8 callers
Method
create_random_input
(self, oob=False, max_dim=20, with_grad_output=False)
test2d.py:154
↓ 8 callers
Method
gradcheck
(self, image, optical, padding_mode='border', align_corners=True)
test3d.py:210
↓ 8 callers
Method
gradcheck
(self, image, optical, padding_mode='border', align_corners=True)
test2d.py:209
↓ 4 callers
Method
cmp_with_naive
(self, image, optical, double=True)
test3d.py:176
↓ 4 callers
Method
cmp_with_naive
(self, image, optical, double=True)
test2d.py:176
↓ 1 callers
Function
_should_use_custom_op
()
stylegan_gridsample.py:35
Function
PYBIND11_MODULE
gridsample_cuda.cpp:53
Method
backward
(ctx, grad_output)
stylegan_gridsample.py:50
Method
backward
(ctx, grad2_grad_input, grad2_grad_grid)
stylegan_gridsample.py:72
Method
backward
(ctx, grad_output)
cuda_gridsample.py:36
Method
backward
(ctx, grad2_grad_input, grad2_grad_grid)
cuda_gridsample.py:58
Method
backward
(ctx, grad_output)
cuda_gridsample.py:87
Method
backward
(ctx, grad2_grad_input, grad2_grad_grid)
cuda_gridsample.py:111
Method
fn
(image, optical)
test3d.py:140
Method
fn
(image, optical)
test2d.py:140
Method
forward
(ctx, input, grid)
stylegan_gridsample.py:42
Method
forward
(ctx, grad_output, input, grid)
stylegan_gridsample.py:59
Method
forward
(ctx, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:22
Method
forward
(ctx, grad_output, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:43
Method
forward
(ctx, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:73
Method
forward
(ctx, grad_output, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:96
Function
grid_sample
(input, grid)
stylegan_gridsample.py:28
Function
grid_sample2d_grad2
gridsample_cuda.cpp:26
Function
grid_sample3d_grad2
gridsample_cuda.cpp:39
Function
grid_sample_2d
(input, grid, padding_mode='zeros', align_corners=True)
cuda_gridsample.py:7
Function
grid_sample_2d
(image, optical)
naive_gridsample.py:115
Function
grid_sample_3d
(input, grid, padding_mode='zeros', align_corners=True)
cuda_gridsample.py:12
Function
grid_sample_3d
(image, optical)
naive_gridsample.py:3
Method
test_float
(self)
test3d.py:74
Method
test_float
(self)
test2d.py:77
Method
test_grad_output
(self)
test3d.py:52
Method
test_grad_output
(self)
test2d.py:43
Method
test_gradcheck_constant
(self)
test3d.py:32
Method
test_gradcheck_constant
(self)
test2d.py:33
Method
test_gradcheck_random
(self)
test3d.py:37
Method
test_gradcheck_random
(self)
test2d.py:38
Method
test_gradcheck_random_nocorners
(self)
test3d.py:69
Method
test_gradcheck_random_nocorners
(self)
test2d.py:72
Method
test_gradcheck_random_oob_border
(self)
test3d.py:47
Method
test_gradcheck_random_oob_border
(self)
test2d.py:66
Method
test_gradcheck_random_oob_zeros
(self)
test3d.py:42
Method
test_gradcheck_random_oob_zeros
(self)
test2d.py:61
Method
test_naive_constant
(self)
test3d.py:17
Method
test_naive_constant
(self)
test2d.py:17
Method
test_naive_oob
(self)
test3d.py:22
Method
test_naive_oob
(self)
test2d.py:22
Method
test_naive_random
(self)
test3d.py:27
Method
test_naive_random
(self)
test2d.py:28
Method
test_strided
(self)
test3d.py:80
Method
test_strided
(self)
test2d.py:83
Method
test_use_case
(self)
test3d.py:121
Method
test_use_case
(self)
test2d.py:121