Home
last modified time | relevance | path

Searched refs:inplaceable_foreach_ops (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/_inductor/fx_passes/
Dreinplace.py17 inplaceable_foreach_ops as inplaceable_foreach_ops_lowerings,
354 inplaceable_foreach_ops: Dict[torch._ops.OpOverload, InplaceableOp] = {} variable
356 inplaceable_foreach_ops[outplace_op] = InplaceableOp(inplace_op, 0)
412 or (src.args[0].target in inplaceable_foreach_ops)
663 inplaceable_op := inplaceable_foreach_ops.get(node.target, None)
/external/pytorch/torch/_inductor/
Dlowering.py88 inplaceable_foreach_ops: Dict[torch._ops.OpOverload, torch._ops.OpOverload] = {} variable
5992 inplaceable_foreach_ops[outplace_aten_op] = aten_op