Searched refs:SetDefaultComputationLayout (Results 1 – 3 of 3) sorted by relevance
39 void HloModuleConfig::SetDefaultComputationLayout( in SetDefaultComputationLayout() function in xla::HloModuleConfig
91 void SetDefaultComputationLayout(const ProgramShape& program_shape);
69 config_.SetDefaultComputationLayout( in ReplaceEntryComputation()88 config_.SetDefaultComputationLayout( in AddComputationInternal()