Searched defs:KernelTypes (Results 1 – 3 of 3) sorted by relevance
122 struct KernelTypes { struct123 using Element = Element_;124 using Fragment =126 using Fragment4 = cutlass::Array<Element, 4>;127 using ValuesPacked = cutlass::Array<Element, 8>; // 4 first col, 4 second col129 struct Params {166 struct Tile4x4Accessor {184 struct Tile4x4Packed {208 Tile4x4Packed packed;210 for (int row = 0; row < 4; ++row) {[all …]
73 def comp_types(val: KernelTypes, input_val: ValueType) -> Optional[bool]: # noqa
139 KernelTypes = Union[ variable