Searched refs:instruction_can_change_layout_func_ (Results 1 – 4 of 4) sorted by relevance
232 instruction_can_change_layout_func_( in target_metadata_()234 CHECK(instruction_can_change_layout_func_ == nullptr || layout_sensitive); in target_metadata_()252 instruction_can_change_layout_func_; variable
1330 : instruction_can_change_layout_func_( in InstructionVerifier()1438 if (instruction_can_change_layout_func_ && in Postprocess()1440 !instruction_can_change_layout_func_(instruction)) { in Postprocess()1463 instruction_can_change_layout_func_; member in xla::__anond588ae4e0811::InstructionVerifier1485 instruction_can_change_layout_func_); in Run()
976 instruction_can_change_layout_func_( in LayoutAssignment()995 !instruction_can_change_layout_func_(instruction)) { in ChooseOperandLayoutFromOutputLayout()1062 !instruction_can_change_layout_func_(user)) { in ChooseOutputLayoutFromOperandLayout()1259 if (instruction_can_change_layout_func_(user) && !user->shape().IsArray()) { in PropagateOperandConstraint()1277 if (!instruction_can_change_layout_func_(user)) { in PropagateOperandConstraint()1396 if (!instruction_can_change_layout_func_(instruction)) { in PropagateBufferConstraintToOperands()
529 instruction_can_change_layout_func_; variable