Searched refs:_compute_reduce_dims (Results 1 – 1 of 1) sorted by relevance
117 def _compute_reduce_dims(self, rhs: TensorExpression) -> Set[DimDef]: member in TensorUse129 return ReduceFnUse(BinaryFn.add, None, *self._compute_reduce_dims(rhs))(rhs)