Searched defs:entry_computation_layout (Results 1 – 9 of 9) sorted by relevance
36 : LayoutAssignment(entry_computation_layout, in LayoutAssignment() argument
32 ComputationLayout* entry_computation_layout, in CpuLayoutAssignment()
52 ComputationLayout* entry_computation_layout) { in AssignLayouts()
36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig()
149 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
96 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()
1113 ComputationLayout* entry_computation_layout, in LayoutAssignment()
289 XLA_ComputationLayout entry_computation_layout; member