Searched refs:ReduceFnUse (Results 1 – 1 of 1) sorted by relevance
129 return ReduceFnUse(BinaryFn.add, None, *self._compute_reduce_dims(rhs))(rhs)352 class ReduceFnUse: class386 def __getitem__(self, reduce_dims: Tuple[DimDef]) -> ReduceFnUse:387 return ReduceFnUse(self.binary_fn, None, *reduce_dims)618 def __getitem__(self, reduce_dims: Tuple[DimDef]) -> ReduceFnUse:619 return ReduceFnUse(None, self, *reduce_dims)