| 102 | TensorLayout getWeightLayout() const { return weightLayout; } |
| 103 | int getTensorBlockC() const { return tensorBlockC; } |
| 104 | int getMinTileAlignment() const { return minTileAlignment; } |
| 105 | virtual bool needWeightAndBiasOnDevice() const { return true; } |
| 106 | |
| 107 | // Memory |