Searched defs:is_layout_sensitive (Results 1 – 7 of 7) sorted by relevance
386 bool Equal(const HloComputation& other, bool is_layout_sensitive) const { in Equal()396 bool is_layout_sensitive) const { in EqualIgnoringChannelIdValues()403 bool is_layout_sensitive, in EqualIgnoringExecutionThread()
105 : is_layout_sensitive_(is_layout_sensitive), in is_layout_sensitive_() argument115 explicit ConditionalCodeMotion(bool is_layout_sensitive, in ConditionalCodeMotion()
70 void set_is_layout_sensitive(bool is_layout_sensitive) { in set_is_layout_sensitive()74 bool is_layout_sensitive() const { return is_layout_sensitive_; } in is_layout_sensitive() function
208 bool is_layout_sensitive) { in InstructionWithinBranchIdentical()351 bool is_layout_sensitive) { in FindSpecialConverts()477 bool is_layout_sensitive) { in ConvertSpecialMove()986 HloInstruction* conditional, bool is_layout_sensitive, in GroupConnectedBoundaries()
34 : is_layout_sensitive_(is_layout_sensitive), in is_layout_sensitive_() argument
892 bool is_layout_sensitive, in EqualInternal()
124 def is_layout_sensitive(self, node: torch.fx.Node) -> bool: member in LayoutTransform