Home
last modified time | relevance | path

Searched refs:AssignLayouts (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() function in xla::__anona66179c30111::LayoutAssignmentTest
115 AssignLayouts(m.get(), &computation_layout); in TEST_F()
151 AssignLayouts(m.get(), &computation_layout); in TEST_F()
196 AssignLayouts(m.get(), &computation_layout); in TEST_F()
237 AssignLayouts(m.get(), &computation_layout); in TEST_F()
279 AssignLayouts(m.get(), &computation_layout); in TEST_F()
318 AssignLayouts(m.get(), &computation_layout); in TEST_F()
375 AssignLayouts(m.get(), &computation_layout); in TEST_F()
416 AssignLayouts(m.get(), &computation_layout); in TEST_F()
455 AssignLayouts(m.get(), &computation_layout); in TEST_F()
[all …]
Dlayout_assignment.h415 Status AssignLayouts(const LayoutConstraints& constraints,
Dlayout_assignment.cc1724 Status LayoutAssignment::AssignLayouts(const LayoutConstraints& constraints, in AssignLayouts() function in xla::LayoutAssignment
1917 TF_RETURN_IF_ERROR(AssignLayouts(constraints, computation)); in RunOnComputation()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment_test.cc51 void AssignLayouts(HloModule* module, in AssignLayouts() function in xla::__anon5efc9fa10111::CpuLayoutAssignmentTest
84 AssignLayouts(module.get(), &computation_layout); in TEST_F()
125 AssignLayouts(module.get(), &computation_layout); in TEST_F()
171 AssignLayouts(module.get(), &computation_layout); in TEST_F()
203 AssignLayouts(module.get(), &computation_layout); in TEST_F()
243 AssignLayouts(module.get(), &computation_layout); in TEST_F()
465 AssignLayouts(module.get(), &computation_layout); in TEST_F()