| 147 | return Conv2DProblemSize_(self) |
| 148 | |
| 149 | def implicit_gemm_size(self, kind: ConvKind): |
| 150 | if kind == ConvKind.Fprop: |
| 151 | return GemmCoord( |
| 152 | self.N * self.P * self.Q, |
| 153 | self.K, |
| 154 | self.R * self.S * self.C // self.groups |
| 155 | ) |
| 156 | elif kind == ConvKind.Dgrad: |
| 157 | return GemmCoord( |
| 158 | self.N * self.H * self.W, |
| 159 | self.C, |
| 160 | self.R * self.S * self.K |
| 161 | ) |
| 162 | elif kind == ConvKind.Wgrad: |
| 163 | return GemmCoord( |
| 164 | self.K, |
| 165 | self.R * self.S * self.C, |
| 166 | self.N * self.P * self.Q |
| 167 | ) |
| 168 | |
| 169 | @staticmethod |
| 170 | def from_sizes(input_size, weight_size): |