Searched refs:EnableColumnMajorRowMajorLayouts (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseSemiStructuredLinear.cu | 322 bool EnableColumnMajorRowMajorLayouts,
382 if constexpr (EnableColumnMajorRowMajorLayouts) {
445 bool EnableColumnMajorRowMajorLayouts,
462 EnableColumnMajorRowMajorLayouts,
481 EnableColumnMajorRowMajorLayouts,
503 bool EnableColumnMajorRowMajorLayouts,
524 EnableColumnMajorRowMajorLayouts,
545 EnableColumnMajorRowMajorLayouts,
566 EnableColumnMajorRowMajorLayouts,
722 const auto EnableColumnMajorRowMajorLayouts = false; in _sparse_semi_structured_linear() local
[all …]
|
| D | SparseSemiStructuredOps.cu | 310 bool EnableColumnMajorRowMajorLayouts,
376 if constexpr (EnableColumnMajorRowMajorLayouts) {
444 bool EnableColumnMajorRowMajorLayouts,
461 EnableColumnMajorRowMajorLayouts,
482 EnableColumnMajorRowMajorLayouts,
634 const auto EnableColumnMajorRowMajorLayouts = false; in sparse_semi_structured_mad_op()
648 EnableColumnMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
669 EnableColumnMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
692 const auto EnableColumnMajorRowMajorLayouts = true; in sparse_semi_structured_mad_op()
704 EnableColumnMajorRowMajorLayouts, in sparse_semi_structured_mad_op()
[all …]
|