Searched defs:entry_computation_layout (Results 1 – 7 of 7) sorted by relevance
32 explicit GpuLayoutAssignment(ComputationLayout* entry_computation_layout, in GpuLayoutAssignment()
32 ComputationLayout* entry_computation_layout, in CpuLayoutAssignment()
52 ComputationLayout* entry_computation_layout) { in AssignLayouts()
59 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
131 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts()517 ComputationLayout* entry_computation_layout) in OperandsMustBeTheSameLayoutAssignment()
968 ComputationLayout* entry_computation_layout, in LayoutAssignment()