Searched refs:mat_b_swizzled (Results 1 – 1 of 1) sorted by relevance
| /dports/lang/halide/Halide-release_2019_08_27-2654-g664dc4993/apps/nn_ops/ |
| H A D | MatrixMultiply_generator.cpp | 77 Func mat_b_swizzled("mat_b_swizzled"); in generate() local
79 mat_b_swizzled(x, y, k) = mat_b_(x, 4 * y + k); in generate()
112 u32(u16(mat_a_(4 * rk + 0, y)) * u16(mat_b_swizzled(x, rk, 0))) + in generate()
113 u32(u16(mat_a_(4 * rk + 1, y)) * u16(mat_b_swizzled(x, rk, 1))) + in generate()
114 u32(u16(mat_a_(4 * rk + 2, y)) * u16(mat_b_swizzled(x, rk, 2))) + in generate()
115 u32(u16(mat_a_(4 * rk + 3, y)) * u16(mat_b_swizzled(x, rk, 3))); in generate()
179 mat_b_swizzled.compute_at(output_, xo) in generate()
|