Searched refs:_compute_reduce_dims (Results 1 – 4 of 4) sorted by relevance
119 return ReduceFn.add(*self._compute_reduce_dims(rhs))(rhs)121 def _compute_reduce_dims(self, rhs: TensorExpression) -> Set[DimDef]: member in TensorUse
118 return ReduceFn.add(*self._compute_reduce_dims(rhs))(rhs)120 def _compute_reduce_dims(self, rhs: TensorExpression) -> Set[DimDef]: member in TensorUse