Searched refs:OptimizationContext (Results 1 – 5 of 5) sorted by relevance
28 OptimizationContext,134 def get_opt_ctx(node: torch.fx.Node) -> OptimizationContext:135 return node.meta.get(OptimizationContext.key, None)138 def get_current_node_opt_ctx() -> OptimizationContext:915 dtypes.add(node.meta[OptimizationContext.key].dtype)
370 OptimizationContext.key in n.meta371 and n.meta[OptimizationContext.key].dtype is not None379 [n.meta[OptimizationContext.key].dtype for n in input_nodes],422 if OptimizationContext.key in node.meta:423 opt_ctx = node.meta[OptimizationContext.key]425 opt_ctx = OptimizationContext()428 node.meta[OptimizationContext.key] = opt_ctx2055 class OptimizationContext: class
61 OptimizationContext,499 self.opt_ctx: Optional[OptimizationContext] = None507 if OptimizationContext.key in self.current_node.meta:508 self.opt_ctx = self.current_node.meta[OptimizationContext.key]510 self.opt_ctx = OptimizationContext()518 self.current_node.meta[OptimizationContext.key] = self.opt_ctx3219 assert OptimizationContext.key in _node.meta3220 opt_ctx: OptimizationContext = _node.meta[OptimizationContext.key]3640 assert OptimizationContext.key in fx_node.meta3641 opt_ctx: OptimizationContext = fx_node.meta[[all …]
22 from torch._inductor.codegen.common import OptimizationContext2240 if OptimizationContext.key in node.meta:2241 opt_ctx = node.meta[OptimizationContext.key]2243 opt_ctx = OptimizationContext()2245 node.meta[OptimizationContext.key] = opt_ctx
42 from torch._inductor.codegen.common import DataTypePropagation, OptimizationContext9312 if OptimizationContext.key in node.meta:9313 return node.meta[OptimizationContext.key].dtype