diff options
Diffstat (limited to 'src/FbgemmConv.cc')
-rw-r--r-- | src/FbgemmConv.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/FbgemmConv.cc b/src/FbgemmConv.cc index 5a486c0..de833d2 100644 --- a/src/FbgemmConv.cc +++ b/src/FbgemmConv.cc @@ -109,7 +109,7 @@ int fbgemmConv( outProcess.getAZeroPoint(), activations, B_zero_point[0], - *(packed_weights.getPackedWFor3DDW()), + *(packed_weights.getPackedWForDepthwise()), C_multiplier[0], outProcess.getCZeroPoint(), out, @@ -135,7 +135,7 @@ int fbgemmConv( outProcess.getAZeroPoint(), activations, B_zero_point, - *(packed_weights.getPackedWFor3DDW()), + *(packed_weights.getPackedWForDepthwise()), C_multiplier, outProcess.getCZeroPoint(), out, @@ -163,7 +163,7 @@ int fbgemmConv( outProcess.getAZeroPoint(), activations, B_zero_point[0], - *(packed_weights.getPackedWFor2DDW()), + *(packed_weights.getPackedWForDepthwise()), C_multiplier[0], outProcess.getCZeroPoint(), out, @@ -188,7 +188,7 @@ int fbgemmConv( outProcess.getAZeroPoint(), activations, B_zero_point, - *(packed_weights.getPackedWFor2DDW()), + *(packed_weights.getPackedWForDepthwise()), C_multiplier, outProcess.getCZeroPoint(), out, |