Searched defs:operand_layout (Results 1 – 3 of 3) sorted by relevance
42 const Layout &operand_layout = operand_shape.layout(); in HandleReduce() local
860 const ShapeLayout& operand_layout, HloInstruction* instruction, in CopyOperandIfLayoutsDiffer()1086 auto operand_layout = aligned_operand_shape.value().layout(); in ChooseOperandLayoutFromOutputLayout() local1102 Layout operand_layout = LayoutUtil::MakeLayout(new_minor_to_major); in ChooseOperandLayoutFromOutputLayout() local1112 const Layout& operand_layout, const HloInstruction* user, in ChooseOutputLayoutFromOperandLayout()1528 std::unique_ptr<Layout> operand_layout = in PropagateBufferConstraintToOperands() local1779 const ShapeLayout* operand_layout = in AssignLayouts() local
1909 const Layout& operand_layout = operand_shape.layout(); in Postprocess() local