Searched defs:AssignLayouts (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ | ||
| D | cpu_layout_assignment_test.cc | 51 void AssignLayouts(HloModule* module, in AssignLayouts() function in xla::__anon9e2aafd00111::CpuLayoutAssignmentTest |
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | layout_assignment_test.cc | 54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts() function in xla::__anoncef5caf20111::LayoutAssignmentTest |
| D | layout_assignment.cc | 1955 Status LayoutAssignment::AssignLayouts(LayoutConstraints& constraints) { in AssignLayouts() function in xla::LayoutAssignment |