Searched defs:shape_layout (Results 1 – 4 of 4) sorted by relevance
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout()
98 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function121 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function
109 const ShapeLayout shape_layout(shape); in TEST_F() local191 const ShapeLayout shape_layout(shape); in TEST_F() local
78 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint()1195 const ShapeLayout& shape_layout, const HloInstruction* instruction, in PropagateUseConstraintToDefs()