MCPcopy Create free account

hub / github.com/AliaksandrSiarohin/cuda-gridsample-grad2 / functions

Functions53 in github.com/AliaksandrSiarohin/cuda-gridsample-grad2

↓ 8 callersMethodcreate_random_input
(self, oob=False, max_dim=20, with_grad_output=False)
test3d.py:154
↓ 8 callersMethodcreate_random_input
(self, oob=False, max_dim=20, with_grad_output=False)
test2d.py:154
↓ 8 callersMethodgradcheck
(self, image, optical, padding_mode='border', align_corners=True)
test3d.py:210
↓ 8 callersMethodgradcheck
(self, image, optical, padding_mode='border', align_corners=True)
test2d.py:209
↓ 4 callersMethodcmp_with_naive
(self, image, optical, double=True)
test3d.py:176
↓ 4 callersMethodcmp_with_naive
(self, image, optical, double=True)
test2d.py:176
↓ 1 callersFunction_should_use_custom_op
()
stylegan_gridsample.py:35
FunctionPYBIND11_MODULE
gridsample_cuda.cpp:53
Methodbackward
(ctx, grad_output)
stylegan_gridsample.py:50
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
stylegan_gridsample.py:72
Methodbackward
(ctx, grad_output)
cuda_gridsample.py:36
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
cuda_gridsample.py:58
Methodbackward
(ctx, grad_output)
cuda_gridsample.py:87
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
cuda_gridsample.py:111
Methodfn
(image, optical)
test3d.py:140
Methodfn
(image, optical)
test2d.py:140
Methodforward
(ctx, input, grid)
stylegan_gridsample.py:42
Methodforward
(ctx, grad_output, input, grid)
stylegan_gridsample.py:59
Methodforward
(ctx, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:22
Methodforward
(ctx, grad_output, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:43
Methodforward
(ctx, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:73
Methodforward
(ctx, grad_output, input, grid, padding_mode=0, align_corners=True)
cuda_gridsample.py:96
Functiongrid_sample
(input, grid)
stylegan_gridsample.py:28
Functiongrid_sample2d_grad2
gridsample_cuda.cpp:26
Functiongrid_sample3d_grad2
gridsample_cuda.cpp:39
Functiongrid_sample_2d
(input, grid, padding_mode='zeros', align_corners=True)
cuda_gridsample.py:7
Functiongrid_sample_2d
(image, optical)
naive_gridsample.py:115
Functiongrid_sample_3d
(input, grid, padding_mode='zeros', align_corners=True)
cuda_gridsample.py:12
Functiongrid_sample_3d
(image, optical)
naive_gridsample.py:3
Methodtest_float
(self)
test3d.py:74
Methodtest_float
(self)
test2d.py:77
Methodtest_grad_output
(self)
test3d.py:52
Methodtest_grad_output
(self)
test2d.py:43
Methodtest_gradcheck_constant
(self)
test3d.py:32
Methodtest_gradcheck_constant
(self)
test2d.py:33
Methodtest_gradcheck_random
(self)
test3d.py:37
Methodtest_gradcheck_random
(self)
test2d.py:38
Methodtest_gradcheck_random_nocorners
(self)
test3d.py:69
Methodtest_gradcheck_random_nocorners
(self)
test2d.py:72
Methodtest_gradcheck_random_oob_border
(self)
test3d.py:47
Methodtest_gradcheck_random_oob_border
(self)
test2d.py:66
Methodtest_gradcheck_random_oob_zeros
(self)
test3d.py:42
Methodtest_gradcheck_random_oob_zeros
(self)
test2d.py:61
Methodtest_naive_constant
(self)
test3d.py:17
Methodtest_naive_constant
(self)
test2d.py:17
Methodtest_naive_oob
(self)
test3d.py:22
Methodtest_naive_oob
(self)
test2d.py:22
Methodtest_naive_random
(self)
test3d.py:27
Methodtest_naive_random
(self)
test2d.py:28
Methodtest_strided
(self)
test3d.py:80
Methodtest_strided
(self)
test2d.py:83
Methodtest_use_case
(self)
test3d.py:121
Methodtest_use_case
(self)
test2d.py:121