Functions — github.com/deepsense-ai/roi-pooling

CodeHub MCPcopy Create free account

hub / github.com/deepsense-ai/roi-pooling / functions

Functions18 in github.com/deepsense-ai/roi-pooling

⨍ Functions 18 ◇ Types & classes 4

↓ 6 callersFunctionroi_pooling

returns a tensorflow operation for computing the Region of Interest Pooling @arg input: feature maps on which to perform the pooling

roi_pooling/roi_pooling_ops.py:10

↓ 3 callersMethodrun

↓ 1 callersFunctionRoiPoolingGradKernelLauncher

roi_pooling/roi_pooling.cu.cc:201

↓ 1 callersFunctionRoiPoolingKernelLauncher

roi_pooling/roi_pooling.cu.cc:115

FunctionCAFFE_GET_BLOCKS

CUDA: number of blocks for threads.

roi_pooling/roi_pooling.cu.cc:34

FunctionCUDA_KERNEL_LOOP

roi_pooling/roi_pooling.cu.cc:50

roi_pooling/roi_pooling.cc:39

roi_pooling/roi_pooling.cc:123

FunctionRoiPoolingGradKernel

////////// Grad

roi_pooling/roi_pooling.cu.cc:125

MethodRoiPoolingGradOp

roi_pooling/roi_pooling.cc:114

FunctionRoiPoolingKernel

roi_pooling/roi_pooling.cu.cc:44

MethodRoiPoolingOp

roi_pooling/roi_pooling.cc:30

Function_RoiPoolingGrad

roi_pooling/roi_pooling_ops.py:25

Function_RoiPoolingShape

roi_pooling/roi_pooling_ops.py:40

Methodtest_roi_pooling_grad

roi_pooling/roi_pooling_test.py:10

Methodtest_shape_inference_1

roi_pooling/roi_pooling_test.py:45

Methodtest_shape_inference_2

roi_pooling/roi_pooling_test.py:64

Methodtest_very_big_output

This test checks whether the layer can handle a corner case where the number of output pixels is very large, possibly larger

roi_pooling/roi_pooling_test.py:83