Searched refs:AssignLayouts (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment_test.cc | 54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() function in xla::__anoned7fa0720111::LayoutAssignmentTest
116 AssignLayouts(m.get(), &computation_layout); in TEST_F()
152 AssignLayouts(m.get(), &computation_layout); in TEST_F()
197 AssignLayouts(m.get(), &computation_layout); in TEST_F()
238 AssignLayouts(m.get(), &computation_layout); in TEST_F()
284 AssignLayouts(m.get(), &computation_layout); in TEST_F()
341 AssignLayouts(m.get(), &computation_layout); in TEST_F()
380 AssignLayouts(m.get(), &computation_layout); in TEST_F()
419 AssignLayouts(m.get(), &computation_layout); in TEST_F()
471 AssignLayouts(m.get(), &computation_layout); in TEST_F()
[all …]
|
| H A D | layout_assignment.h | 513 Status AssignLayouts(LayoutConstraints& constraints);
|
| H A D | layout_assignment.cc | 1955 Status LayoutAssignment::AssignLayouts(LayoutConstraints& constraints) { in AssignLayouts() function in xla::LayoutAssignment
2522 TF_RETURN_IF_ERROR(AssignLayouts(*constraints)); in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_layout_assignment_test.cc | 51 void AssignLayouts(HloModule* module, in AssignLayouts() function in xla::__anonfdf205500111::CpuLayoutAssignmentTest
83 AssignLayouts(module.get(), &computation_layout); in TEST_F()
124 AssignLayouts(module.get(), &computation_layout); in TEST_F()
170 AssignLayouts(module.get(), &computation_layout); in TEST_F()
202 AssignLayouts(module.get(), &computation_layout); in TEST_F()
242 AssignLayouts(module.get(), &computation_layout); in TEST_F()
463 AssignLayouts(module.get(), &computation_layout); in TEST_F()
|