Searched refs:EnableRowMajorRowMajorLayouts (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseSemiStructuredLinear.cu | 320 bool EnableRowMajorRowMajorLayouts,
338 if constexpr (EnableRowMajorRowMajorLayouts) {
443 bool EnableRowMajorRowMajorLayouts,
460 EnableRowMajorRowMajorLayouts,
479 EnableRowMajorRowMajorLayouts,
501 bool EnableRowMajorRowMajorLayouts,
522 EnableRowMajorRowMajorLayouts,
543 EnableRowMajorRowMajorLayouts,
564 EnableRowMajorRowMajorLayouts,
720 const auto EnableRowMajorRowMajorLayouts = false; in _sparse_semi_structured_linear() local
[all …]
|
| D | SparseSemiStructuredOps.cu | 308 bool EnableRowMajorRowMajorLayouts,
326 if constexpr (EnableRowMajorRowMajorLayouts) {
442 bool EnableRowMajorRowMajorLayouts,
459 EnableRowMajorRowMajorLayouts,
480 EnableRowMajorRowMajorLayouts,
632 const auto EnableRowMajorRowMajorLayouts = false; in sparse_semi_structured_mad_op()
646 EnableRowMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
667 EnableRowMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
690 const auto EnableRowMajorRowMajorLayouts = true; in sparse_semi_structured_mad_op()
702 EnableRowMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
[all …]
|