Searched refs:computeFrontBatchDimsFromLevels (Results 1 – 2 of 2) sorted by relevance
87 static std::tuple<int64_t, int64_t> computeFrontBatchDimsFromLevels(std::bitset<kVmapNumLevels> lev… in computeFrontBatchDimsFromLevels() function219 auto bdim_level = computeFrontBatchDimsFromLevels(levels_); in apply()
91 static BatchDims computeFrontBatchDimsFromLevels(std::bitset<kVmapNumLevels> levels_bitset) { in computeFrontBatchDimsFromLevels() function287 return makeBatched(physical_tensor, computeFrontBatchDimsFromLevels(levels_)); in apply()