Searched defs:ComputationLayout (Results 1 – 3 of 3) sorted by relevance
36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout() function
27 ComputationLayout::ComputationLayout(const ProgramShape& program_shape, in ComputationLayout() function in xla::ComputationLayout
91 TEST_F(LayoutAssignmentTest, ComputationLayout) { in TEST_F() argument