Searched defs:shape_layout (Results 1 – 4 of 4) sorted by relevance
114 const ShapeLayout& shape_layout() const { return shape_layout_; } in shape_layout() function164 void ResetResultLayout(const ShapeLayout& shape_layout, int64_t priority) { in ResetResultLayout()
46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout()
110 const ShapeLayout shape_layout(shape); in TEST_F() local192 const ShapeLayout shape_layout(shape); in TEST_F() local
105 const ShapeLayout& shape_layout, const HloInstruction* instruction, in OperandLayoutConstraint()1482 const ShapeLayout& shape_layout, const HloInstruction* instruction, in PropagateUseConstraintToDefs()