Searched refs:LayoutAssignmentTest (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | layout_assignment_test.cc | 52 class LayoutAssignmentTest : public HloTestBase { class
90 TEST_F(LayoutAssignmentTest, ComputationLayout) { in TEST_F() argument
122 TEST_F(LayoutAssignmentTest, ComputationLayoutMixedLayout) { in TEST_F() argument
158 TEST_F(LayoutAssignmentTest, FusionInstruction) { in TEST_F() argument
211 TEST_F(LayoutAssignmentTest, TupleLayout) { in TEST_F() argument
249 TEST_F(LayoutAssignmentTest, TupleSelect) { in TEST_F() argument
284 TEST_F(LayoutAssignmentTest, ConflictingLayoutTuple) { in TEST_F() argument
349 TEST_F(LayoutAssignmentTest, ElementwiseAndReshape) { in TEST_F() argument
391 TEST_F(LayoutAssignmentTest, ElementwiseAndTranspose) { in TEST_F() argument
428 TEST_F(LayoutAssignmentTest, BroadcastAndTranspose) { in TEST_F() argument
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 39 using LayoutAssignmentTest = HloTestBase; typedef
41 TEST_F(LayoutAssignmentTest, Elementwise) { in TEST_F() argument
109 TEST_F(LayoutAssignmentTest, BatchNormInference) { in TEST_F() argument
180 TEST_F(LayoutAssignmentTest, BatchNormTraining) { in TEST_F() argument
252 TEST_F(LayoutAssignmentTest, BatchNormGrad) { in TEST_F() argument
339 TEST_F(LayoutAssignmentTest, DotLayout) { in TEST_F() argument
368 TEST_F(LayoutAssignmentTest, SortLayout) { in TEST_F() argument
|