Searched defs:entry_computation_layout (Results 1 – 9 of 9) 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()
36 HloModuleConfig::HloModuleConfig(ComputationLayout entry_computation_layout) in HloModuleConfig()
148 const ComputationLayout& entry_computation_layout() const { in entry_computation_layout() function
95 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()
1025 ComputationLayout* entry_computation_layout, in LayoutAssignment()
286 XLA_ComputationLayout entry_computation_layout; member