Home
last modified time | relevance | path

Searched refs:prepare_indexing (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dhalide.py522 index = V.kernel.prepare_indexing(expr)
728 indices = dict.fromkeys(map(super().prepare_indexing, indices))
783 all_used_symbols.update(super().prepare_indexing(index).free_symbols)
914 def prepare_indexing( member in HalideKernel
918 index = super().prepare_indexing(index)
1111 index = self.prepare_indexing(index)
1154 index = self.prepare_indexing(index)
Dsimd.py670 def prepare_indexing( member in SIMDKernel
Dtriton.py1308 index = self.prepare_indexing(index)