Searched refs:SetDefaultComputationLayout (Results 1 – 3 of 3) sorted by relevance
36 void HloModuleConfig::SetDefaultComputationLayout( in SetDefaultComputationLayout() function in xla::HloModuleConfig
55 void SetDefaultComputationLayout(const ProgramShape& program_shape);
61 config_.SetDefaultComputationLayout( in AddComputationInternal()