diff --git a/packages/deep-learning/src/ops/reduce.cairo b/packages/deep-learning/src/ops/reduce.cairo index 63e9a7fe3..9e4ebc0e8 100644 --- a/packages/deep-learning/src/ops/reduce.cairo +++ b/packages/deep-learning/src/ops/reduce.cairo @@ -12,7 +12,7 @@ pub(crate) struct ReduceOpMetadata { output_size: usize, } -pub(crate) fn tensor_sum_reduce_1d, +Zero, +Copy, +Drop, +Debug>( +pub(crate) fn tensor_sum_reduce_1d, +Zero, +Copy, +Drop>( mut input: Tensor ) -> Tensor { let mut result = Zero::::zero(); @@ -30,7 +30,7 @@ pub(crate) fn tensor_sum_reduce_1d, +Zero, +Copy, +Drop, +De Tensor { data: result_data.span() } } -pub(crate) fn tensor_sum_reduce_nd, +Copy, +Drop, +Debug, +Zero>( +pub(crate) fn tensor_sum_reduce_nd, +Copy, +Drop, +Zero>( mut input: Tensor, ref metadata: ReduceOpMetadata ) -> MutTensor { let mut result_data: NullableVec = VecTrait::new(metadata.output_size); @@ -56,7 +56,7 @@ pub(crate) fn tensor_sum_reduce_nd, +Copy, +Drop, +Debug, +Z } -pub(crate) fn tensor_max_reduce_1d, +Drop, +Debug, +PartialOrd>( +pub(crate) fn tensor_max_reduce_1d, +Drop, +PartialOrd>( mut input: Tensor ) -> Tensor { let mut result: Option = Option::None(()); @@ -79,7 +79,7 @@ pub(crate) fn tensor_max_reduce_1d, +Drop, +Debug, +PartialOrd Tensor { data: result_data.span() } } -pub(crate) fn tensor_max_reduce_nd, +Drop, +Debug, +PartialOrd, +Zero>( +pub(crate) fn tensor_max_reduce_nd, +Drop, +PartialOrd, +Zero>( mut input: Tensor, ref metadata: ReduceOpMetadata ) -> MutTensor { let mut result_data: NullableVec = VecTrait::new(metadata.output_size);