(self, *axis_type:AxisType)
| 101 | return replaced_rng, new_rng |
| 102 | |
| 103 | def ranges_of(self, *axis_type:AxisType) -> list[UOp]: return [r for r in self.rngs if r.arg[-1] in axis_type] |
| 104 | def axes_of(self, *axis_type:AxisType) -> list[int]: return [i for i,t in enumerate(self.axis_types) if t in axis_type] |
| 105 | |
| 106 | def upcast_size(self): return prod(self.full_shape[a] for a in self.axes_of(AxisType.UPCAST, AxisType.UNROLL)) |
no outgoing calls
no test coverage detected