Home
last modified time | relevance | path

Searched defs:AssignLayouts (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc51 void AssignLayouts(HloModule* module, in AssignLayouts() function in xla::__anonfdf205500111::CpuLayoutAssignmentTest
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_assignment_test.cc54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() function in xla::__anoned7fa0720111::LayoutAssignmentTest
H A Dlayout_assignment.cc1955 Status LayoutAssignment::AssignLayouts(LayoutConstraints& constraints) { in AssignLayouts() function in xla::LayoutAssignment