Searched refs:inplaceable_foreach_ops (Results 1 – 2 of 2) sorted by relevance
17 inplaceable_foreach_ops as inplaceable_foreach_ops_lowerings,354 inplaceable_foreach_ops: Dict[torch._ops.OpOverload, InplaceableOp] = {} variable356 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)
88 inplaceable_foreach_ops: Dict[torch._ops.OpOverload, torch._ops.OpOverload] = {} variable5992 inplaceable_foreach_ops[outplace_aten_op] = aten_op