Searched refs:LayoutAssignmentTest (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 52 class LayoutAssignmentTest : public HloTestBase { class
91 TEST_F(LayoutAssignmentTest, ComputationLayout) { in TEST_F() argument
123 TEST_F(LayoutAssignmentTest, ComputationLayoutMixedLayout) { in TEST_F() argument
159 TEST_F(LayoutAssignmentTest, FusionInstruction) { in TEST_F() argument
212 TEST_F(LayoutAssignmentTest, TupleLayout) { in TEST_F() argument
250 TEST_F(LayoutAssignmentTest, ConflictingLayoutTuple) { in TEST_F() argument
315 TEST_F(LayoutAssignmentTest, ElementwiseAndReshape) { in TEST_F() argument
355 TEST_F(LayoutAssignmentTest, ElementwiseAndTranspose) { in TEST_F() argument
392 TEST_F(LayoutAssignmentTest, BroadcastAndTranspose) { in TEST_F() argument
425 TEST_F(LayoutAssignmentTest, ReshapeOperandHasMultipleUsers) { in TEST_F() argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_layout_assignment_test.cc | 44 using LayoutAssignmentTest = HloTestBase; typedef
46 TEST_F(LayoutAssignmentTest, Elementwise) { in TEST_F() argument
95 TEST_F(LayoutAssignmentTest, DotLayoutUnchangedIfValid) { in TEST_F() argument
121 TEST_F(LayoutAssignmentTest, DotLayoutSetToDefaultIfDefaultValid) { in TEST_F() argument
148 TEST_F(LayoutAssignmentTest, DotOperandLayoutSetToBatchRowsColsOtherwise) { in TEST_F() argument
174 TEST_F(LayoutAssignmentTest, DotOperandInconsistentDimLayouts) { in TEST_F() argument
200 TEST_F(LayoutAssignmentTest, TransposedDotLayout) { in TEST_F() argument
230 TEST_F(LayoutAssignmentTest, DotLayoutS8) { in TEST_F() argument
254 TEST_F(LayoutAssignmentTest, SortLayout) { in TEST_F() argument
289 TEST_F(LayoutAssignmentTest, FftLayout) { in TEST_F() argument
|