Home
last modified time | relevance | path

Searched refs:ReorderedLayoutInputE (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseSemiStructuredLinear.cu171 using ReorderedLayoutInputE = typename Gemm::LayoutE;
173 std::is_same<ReorderedLayoutInputE,
243 cutlass::TensorRef<ElementInputE, ReorderedLayoutInputE>(
245 ReorderedLayoutInputE::packed({length_m, meta_ncols}));
H A DSparseSemiStructuredOps.cu163 using ReorderedLayoutInputE = typename Gemm::LayoutE;
165 std::is_same<ReorderedLayoutInputE,
208 cutlass::TensorRef<ElementInputE, ReorderedLayoutInputE>(
210 ReorderedLayoutInputE::packed({length_m, tensor_e_ncols}));
H A DSparseSemiStructuredPack.h17 using ReorderedLayoutInputE = cutlass::layout::ColumnMajorInterleaved<2>; variable