/external/pytorch/torch/_inductor/ |
D | jagged_lowerings.py | 18 begin_idx = ops.indirect_indexing( 96 batch_idx = ops.indirect_indexing( 103 seq_idx = ops.indirect_indexing(seq, max_seq_len, check=False)
|
D | ops_handler.py | 206 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]
|
D | virtualized.py | 284 def indirect_indexing(index, size, check=True, wrap_neg=True): member in OpsWrapper 287 return _ops.indirect_indexing(index, size, check, wrap_neg)
|
D | loop_body.py | 380 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
|
D | index_propagation.py | 330 def indirect_indexing( member in IndexPropagation
|
D | lowering.py | 2944 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 …]
|
D | dependencies.py | 701 def indirect_indexing( member in FreeUnbackedSymbolsOpsHandler
|
D | select_algorithm.py | 387 def indirect_indexing(self, index_var, size, check, wrap_neg=True): member in TritonTemplateKernel.modification.PlaceholderSubstitution
|
D | ir.py | 8152 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/ |
D | test_custom_lowering.py | 71 begin_idx = ops.indirect_indexing(
|
/external/pytorch/torch/_inductor/codegen/ |
D | common.py | 1851 def indirect_indexing( member in Kernel.__enter__.CSEProxy 1891 sympy_var = parent_handler.indirect_indexing(var, size, check)
|
D | triton.py | 1723 indirect_indexing = self.is_indirect_indexing(index) 1754 evict_last = not last_use and (has_rindex or indirect_indexing)
|
D | halide.py | 533 def indirect_indexing(cls, index_var, size, check=True, wrap_neg=True): member in HalideOverrides
|