Searched refs:instruction_can_change_layout_func_ (Results 1 – 4 of 4) sorted by relevance
273 instruction_can_change_layout_func_( in target_metadata_()275 CHECK(instruction_can_change_layout_func_ == nullptr || layout_sensitive); in target_metadata_()293 instruction_can_change_layout_func_; variable
1033 instruction_can_change_layout_func_( in LayoutAssignment()1052 !instruction_can_change_layout_func_(instruction)) { in ChooseOperandLayoutFromOutputLayout()1120 !instruction_can_change_layout_func_(user)) { in ChooseOutputLayoutFromOperandLayout()1342 if (instruction_can_change_layout_func_(user) && !user->shape().IsArray()) { in PropagateOperandConstraint()1360 if (!instruction_can_change_layout_func_(user)) { in PropagateOperandConstraint()1508 if (!instruction_can_change_layout_func_(instruction)) { in PropagateBufferConstraintToOperands()
1792 : instruction_can_change_layout_func_( in InstructionVerifier()1900 if (instruction_can_change_layout_func_ && in Postprocess()1902 !instruction_can_change_layout_func_(instruction)) { in Postprocess()1925 instruction_can_change_layout_func_; member in xla::__anon45533f2c0811::InstructionVerifier1944 InstructionVerifier instruction_verifier(instruction_can_change_layout_func_); in Run()
543 instruction_can_change_layout_func_; variable