Home
last modified time | relevance | path

Searched refs:indirect_indexing (Results 1 – 13 of 13) sorted by relevance

/external/pytorch/torch/_inductor/
Djagged_lowerings.py18 begin_idx = ops.indirect_indexing(
96 batch_idx = ops.indirect_indexing(
103 seq_idx = ops.indirect_indexing(seq, max_seq_len, check=False)
Dops_handler.py206 def indirect_indexing( member in OpsHandler
774 def indirect_indexing(index_var, size, check=True, wrap_neg=True) -> sympy.Symbol: member in NoopHandler
818 def indirect_indexing(index_var, size, check=True, wrap_neg=True) -> sympy.Symbol: member in MockHandler
997 def indirect_indexing(self, *args, **kwargs): member in OpCounterCSE
999 return self.parent_handler.indirect_indexing(*args, **kwargs)
1069 def indirect_indexing(self, *args, **kwargs) -> sympy.Expr: member in SimpleCSEHandler
1070 return super().indirect_indexing(*args, **kwargs) # type: ignore[misc]
Dvirtualized.py284 def indirect_indexing(index, size, check=True, wrap_neg=True): member in OpsWrapper
287 return _ops.indirect_indexing(index, size, check, wrap_neg)
Dloop_body.py380 var, V.ops.indirect_indexing(new_var, size, check, wrap_neg)
526 def indirect_indexing(index_proxy, size, check=True, wrap_neg=True): member in LoopBodyBlock.__init__.CaptureIndexing
Dindex_propagation.py330 def indirect_indexing( member in IndexPropagation
Dlowering.py2944 gather_idx = ops.indirect_indexing(index_loader(idx), size[dim])
2975 weight_idx = [ops.indirect_indexing(var_index, weight_size[0])] + [
3068 ops.indirect_indexing(
3468 indirect_idx[dim] = ops.indirect_indexing(
3560 return ops.indirect_indexing(x, size, check=False)
4033 ops.indirect_indexing(
4038 ops.indirect_indexing(
4437 h_start_index = ops.indirect_indexing(h_index_fn(prefix, bh), inp_h)
4438 w_start_index = ops.indirect_indexing(w_index_fn(prefix, bw), inp_w)
4823 ops.indirect_indexing(
[all …]
Ddependencies.py701 def indirect_indexing( member in FreeUnbackedSymbolsOpsHandler
Dselect_algorithm.py387 def indirect_indexing(self, index_var, size, check, wrap_neg=True): member in TritonTemplateKernel.modification.PlaceholderSubstitution
Dir.py8152 def indirect_indexing(index_proxy, size, check=True): member in LoopBodyBlock.__init__.CaptureIndexing
8162 var, V.ops.indirect_indexing(new_var, size, check)
/external/pytorch/test/inductor/
Dtest_custom_lowering.py71 begin_idx = ops.indirect_indexing(
/external/pytorch/torch/_inductor/codegen/
Dcommon.py1851 def indirect_indexing( member in Kernel.__enter__.CSEProxy
1891 sympy_var = parent_handler.indirect_indexing(var, size, check)
Dtriton.py1723 indirect_indexing = self.is_indirect_indexing(index)
1754 evict_last = not last_use and (has_rindex or indirect_indexing)
Dhalide.py533 def indirect_indexing(cls, index_var, size, check=True, wrap_neg=True): member in HalideOverrides