diff options
Diffstat (limited to 'src/FbgemmConv.cc')
| -rw-r--r-- | src/FbgemmConv.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/FbgemmConv.cc b/src/FbgemmConv.cc index 164411d..6a1e55b 100644 --- a/src/FbgemmConv.cc +++ b/src/FbgemmConv.cc @@ -115,6 +115,7 @@ int fbgemmConv( outProcess.getColOffsets(), outProcess.getBias(), outProcess.RELU_FUSED, // fuse_relu + 1.0f, // act_scale * weight_scale thread_id, num_threads); } else if ( @@ -140,6 +141,7 @@ int fbgemmConv( outProcess.getColOffsets(), outProcess.getBias(), outProcess.RELU_FUSED, // fuse_relu + nullptr, // act_scale * weight_scale thread_id, num_threads); } else { @@ -167,6 +169,7 @@ int fbgemmConv( outProcess.getColOffsets(), outProcess.getBias(), outProcess.RELU_FUSED, // fuse_relu + 1.0f, // act_scale * weight_scale thread_id, num_threads); } else if ( @@ -191,6 +194,7 @@ int fbgemmConv( outProcess.getColOffsets(), outProcess.getBias(), outProcess.RELU_FUSED, // fuse_relu + nullptr, // act_scale * weight_scale thread_id, num_threads); } else { |