Searched defs:entry_computation_layout (Results 1 – 10 of 10) sorted by relevance
35 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument
52 ComputationLayout* entry_computation_layout) { in AssignLayouts()
36 : LayoutAssignment(entry_computation_layout, channel_constraints), in LayoutAssignment() argument
36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig()
152 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
99 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
54 void AssignLayouts(HloModule* m, ComputationLayout* entry_computation_layout, in AssignLayouts()480 ComputationLayout* entry_computation_layout) in OperandsMustBeTheSameLayoutAssignment()
1241 ComputationLayout* entry_computation_layout, in LayoutAssignment()
926 std::optional<ComputationLayout> entry_computation_layout; in ParseHloModule() local
316 XLA_ComputationLayout entry_computation_layout; member