Searched defs:shape_layout (Results 1 – 4 of 4) sorted by relevance
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout()
100 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function123 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
83 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint()1253 const ShapeLayout& shape_layout, const HloInstruction* instruction, in PropagateUseConstraintToDefs()