Searched refs:layout_constraints (Results 1 – 3 of 3) sorted by relevance
80 layout_constraints,1196 elif n.op == "call_function" and n.target in layout_constraints:1198 … args, kwargs = layout_constraints[n.target](n, *args, **kwargs) # type: ignore[index]
611 ChannelLayoutConstraints layout_constraints; in RunHloPassesThroughLayoutAssn() local614 &layout_constraints); in RunHloPassesThroughLayoutAssn()
597 ChannelLayoutConstraints layout_constraints; in OptimizeHloModule() local600 &layout_constraints); in OptimizeHloModule()