Searched refs:alias_config (Results 1 – 2 of 2) sorted by relevance
41 HloInputOutputAliasConfig* alias_config) { in Build() argument74 if (!alias_config->ParameterHasAlias(0, input_index) && in Build()75 !alias_config->OutputHasAlias(output_index)) { in Build()76 TF_RETURN_IF_ERROR(alias_config->SetUpAlias( in Build()98 HloInputOutputAliasConfig* alias_config = in Run() local103 alias_config); in Run()
65 HloInputOutputAliasConfig* alias_config);