Searched refs:SparseToSparseConversionStrategy (Results 1 – 3 of 3) sorted by relevance
116 enum class SparseToSparseConversionStrategy { kAuto, kViaCOO, kDirect }; enum119 SparseToSparseConversionStrategy sparseToSparseConversionStrategy(int32_t flag);123 SparseTensorConversionOptions(SparseToSparseConversionStrategy s2s) in SparseTensorConversionOptions()126 : SparseTensorConversionOptions(SparseToSparseConversionStrategy::kAuto) { in SparseTensorConversionOptions()128 SparseToSparseConversionStrategy sparseToSparseStrategy;
187 SparseToSparseConversionStrategy191 return SparseToSparseConversionStrategy::kAuto; in sparseToSparseConversionStrategy()193 return SparseToSparseConversionStrategy::kViaCOO; in sparseToSparseConversionStrategy()195 return SparseToSparseConversionStrategy::kDirect; in sparseToSparseConversionStrategy()
692 case SparseToSparseConversionStrategy::kViaCOO: in matchAndRewrite()695 case SparseToSparseConversionStrategy::kDirect: in matchAndRewrite()700 case SparseToSparseConversionStrategy::kAuto: in matchAndRewrite()