Searched refs:EnableRowMajorColumnMajorLayouts (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| D | SparseSemiStructuredLinear.cu | 321 bool EnableRowMajorColumnMajorLayouts,
360 if constexpr (EnableRowMajorColumnMajorLayouts) {
444 bool EnableRowMajorColumnMajorLayouts,
461 EnableRowMajorColumnMajorLayouts,
480 EnableRowMajorColumnMajorLayouts,
502 bool EnableRowMajorColumnMajorLayouts,
523 EnableRowMajorColumnMajorLayouts,
544 EnableRowMajorColumnMajorLayouts,
565 EnableRowMajorColumnMajorLayouts,
721 const auto EnableRowMajorColumnMajorLayouts = true; in _sparse_semi_structured_linear() local
[all …]
|
| D | SparseSemiStructuredOps.cu | 309 bool EnableRowMajorColumnMajorLayouts,
351 if constexpr (EnableRowMajorColumnMajorLayouts) {
443 bool EnableRowMajorColumnMajorLayouts,
460 EnableRowMajorColumnMajorLayouts,
481 EnableRowMajorColumnMajorLayouts,
633 const auto EnableRowMajorColumnMajorLayouts = true; in sparse_semi_structured_mad_op()
647 EnableRowMajorColumnMajorLayouts, in sparse_semi_structured_mad_op()
668 EnableRowMajorColumnMajorLayouts, in sparse_semi_structured_mad_op()
691 const auto EnableRowMajorColumnMajorLayouts = true; in sparse_semi_structured_mad_op()
703 EnableRowMajorColumnMajorLayouts, in sparse_semi_structured_mad_op()
[all …]
|