#13361: allow keepdim for generic reductions

tenstorrent · Nov 4, 2024 · e7d0dbf · e7d0dbf
1 parent a837dde
commit e7d0dbf
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/ttnn/cpp/ttnn/operations/reduction/generic/generic_reductions.cpp b/ttnn/cpp/ttnn/operations/reduction/generic/generic_reductions.cpp
@@ -21,9 +21,6 @@ static Tensor reduce_impl(
     float scalar,
     bool reshape) {
     using ttnn::operations::experimental::auto_format::AutoFormat;
-    if (not keepdim) {
-        TT_THROW("keepdim=False is not supported");
-    }
 
     auto input_shape = input_tensor_arg.get_shape();
     auto rank = input_shape.size();