Searched refs:layout_assignment (Results 1 – 7 of 7) sorted by relevance
77 GpuLayoutAssignment layout_assignment( in TEST_F() local80 EXPECT_TRUE(layout_assignment.Run(module.get()).ValueOrDie()); in TEST_F()166 GpuLayoutAssignment layout_assignment( in TEST_F() local169 EXPECT_TRUE(layout_assignment.Run(module.get()).ValueOrDie()); in TEST_F()237 GpuLayoutAssignment layout_assignment( in TEST_F() local240 EXPECT_TRUE(layout_assignment.Run(module.get()).ValueOrDie()); in TEST_F()319 GpuLayoutAssignment layout_assignment( in TEST_F() local322 EXPECT_TRUE(layout_assignment.Run(module.get()).ValueOrDie()); in TEST_F()356 GpuLayoutAssignment layout_assignment( in TEST_F() local359 EXPECT_TRUE(layout_assignment.Run(module.get()).ValueOrDie()); in TEST_F()[all …]
951 "//tensorflow/compiler/xla/service:layout_assignment",
57 cpu::CpuLayoutAssignment layout_assignment( in AssignLayouts() local60 EXPECT_IS_OK(layout_assignment.Run(module).status()); in AssignLayouts()325 cpu::CpuLayoutAssignment layout_assignment( in RunDotOutputFusion() local329 layout_assignment.Run(module)); in RunDotOutputFusion()
798 "//tensorflow/compiler/xla/service:layout_assignment",
56 LayoutAssignment layout_assignment( in AssignLayouts() local59 EXPECT_IS_OK(layout_assignment.Run(m).status()); in AssignLayouts()317 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local570 OperandsMustBeTheSameLayoutAssignment layout_assignment(&computation_layout); in TEST_F() local571 EXPECT_IS_OK(layout_assignment.Run(m.get()).status()); in TEST_F()832 LayoutAssignment layout_assignment(&computation_layout); in TEST_F() local833 Status error_status = layout_assignment.Run(m.get()).status(); in TEST_F()1310 LayoutAssignment layout_assignment( in AssignLayoutsToComputation() local1313 return layout_assignment.Run(m).status(); in AssignLayoutsToComputation()
2592 name = "layout_assignment",2594 "layout_assignment.cc",2597 "layout_assignment.h",2733 ":layout_assignment",2843 ":layout_assignment",
50 "//tensorflow/compiler/xla/service:layout_assignment",