Searched refs:EnableColumnMajorColumnMajorLayouts (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseSemiStructuredLinear.cu | 323 bool EnableColumnMajorColumnMajorLayouts,
404 if constexpr (EnableColumnMajorColumnMajorLayouts) {
446 bool EnableColumnMajorColumnMajorLayouts,
463 EnableColumnMajorColumnMajorLayouts,
482 EnableColumnMajorColumnMajorLayouts,
504 bool EnableColumnMajorColumnMajorLayouts,
525 EnableColumnMajorColumnMajorLayouts,
546 EnableColumnMajorColumnMajorLayouts,
567 EnableColumnMajorColumnMajorLayouts,
723 const auto EnableColumnMajorColumnMajorLayouts = false; in _sparse_semi_structured_linear() local
[all …]
|
| D | SparseSemiStructuredOps.cu | 311 bool EnableColumnMajorColumnMajorLayouts,
401 if constexpr (EnableColumnMajorColumnMajorLayouts) {
445 bool EnableColumnMajorColumnMajorLayouts>
462 EnableColumnMajorColumnMajorLayouts,
483 EnableColumnMajorColumnMajorLayouts,
635 const auto EnableColumnMajorColumnMajorLayouts = false; in sparse_semi_structured_mad_op()
649 EnableColumnMajorColumnMajorLayouts>( in sparse_semi_structured_mad_op()
670 EnableColumnMajorColumnMajorLayouts>( in sparse_semi_structured_mad_op()
693 const auto EnableColumnMajorColumnMajorLayouts = true; in sparse_semi_structured_mad_op()
705 EnableColumnMajorColumnMajorLayouts>( in sparse_semi_structured_mad_op()
[all …]
|