(cRank: number, shape: number[], rank: number)
| 30 | size = true; |
| 31 | |
| 32 | constructor(cRank: number, shape: number[], rank: number) { |
| 33 | this.outputShape = shape; |
| 34 | this.dispatchLayout = flatDispatchLayout(this.outputShape); |
| 35 | this.dispatch = computeDispatch( |
| 36 | this.dispatchLayout, this.outputShape, this.workgroupSize); |
| 37 | |
| 38 | this.cRank = cRank; |
| 39 | this.rank = rank; |
| 40 | this.shaderKey = 'select'; |
| 41 | } |
| 42 | |
| 43 | getUserCode(): string { |
| 44 | // TODO(WGSL): below code can be merged with getUserCode. |
nothing calls this directly
no test coverage detected