Skip to content
Snippets Groups Projects

Added Bitshift Operator

Merged Noam Zerah requested to merge noamzerah/aidge_backend_cpu:feat_operator_bitshift into dev
All threads resolved!
1 file
+ 1
0
Compare changes
  • Side-by-side
  • Inline
@@ -46,6 +46,7 @@ void BitShiftImpl_cpu_forward_kernel(
std::size_t idx1 = getFlattenedIndex(input1Dims, indexes);
std::size_t idx2 = getFlattenedIndex(input2Dims, indexes);
if(direction == BitShift_Op::BitShiftDirection::right)
{
output[oIndex]= input_1[idx1] >> input_2[idx2];
}
Loading