Searched refs:SetComputationLayoutIfExists (Results 1 – 3 of 3) sorted by relevance
44 void HloModuleConfig::SetComputationLayoutIfExists( in SetComputationLayoutIfExists() function in xla::HloModuleConfig
95 void SetComputationLayoutIfExists(const ProgramShape& program_shape);
83 config_.SetComputationLayoutIfExists( in AddComputationInternal()