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