Home
last modified time | relevance | path

Searched refs:statically_known_equals (Results 1 – 9 of 9) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dtriton_utils.py162 and V.graph.sizevars.statically_known_equals(arg.expr, 1) # type: ignore[arg-type]
Dsimd.py168 if V.graph.sizevars.statically_known_equals(divisor * length, self.numel):
218 if not V.graph.sizevars.statically_known_equals(node.divisor, divisor):
223 if not V.graph.sizevars.statically_known_equals(self.numel, divisor):
553 if sv.statically_known_equals(size, 1): # type: ignore[arg-type]
557 while current_group < len(remaining) and sv.statically_known_equals(
Dhalide.py639 if V.graph.sizevars.statically_known_equals(left, right):
1003 elif not V.graph.sizevars.statically_known_equals(dims[0].stride, 1):
Dtriton.py305 not sizevars.statically_known_equals(
2905 if V.graph.sizevars.statically_known_equals(tree.numel, 1): # type: ignore[arg-type]
Dwrapper.py1305 ) and V.graph.sizevars.statically_known_equals(
/external/pytorch/torch/_inductor/kernel/
Dconv.py588 … and V.graph.sizevars.statically_known_equals(in_chan, x.get_size()[1]) # type: ignore[arg-type]
/external/pytorch/torch/_inductor/
Dsizevars.py340 def statically_known_equals( member in SizeVarAllocator
354 self.statically_known_equals(l, r) for l, r in zip(left, right)
Dgraph.py1139 self.sizevars.statically_known_equals(s1, s2)
1149 if not self.sizevars.statically_known_equals(s1, s2):
Dlowering.py6177 if V.graph.sizevars.statically_known_equals(old_numel, 0): # type: ignore[arg-type]