Fix the inference operation of the deformable convolutional layer.
This commit removes the malloc operation in the inference method and adds the sigmoid kernel. Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
@@ -219,7 +219,12 @@ public:
|
||||
int kernelH, kernelW, strideH, strideW, paddingH, paddingW;
|
||||
protected:
|
||||
|
||||
dnnType *ones_d1;
|
||||
dnnType *ones_d2;
|
||||
cudnnTensorDescriptor_t biasTensorDesc;
|
||||
int chunk_dim;
|
||||
dnnType *offset, *mask;
|
||||
dnnType *output_conv;
|
||||
|
||||
void initCUDNN();
|
||||
|
||||
|
||||
Reference in New Issue
Block a user