Searched refs:ForEachAlias (Results 1 – 5 of 5) sorted by relevance
109 ForEachAlias([&](const ShapeIndex& output_index, const Alias& alias) { in ToString()153 void HloInputOutputAliasConfig::ForEachAlias(AliasFn fn) const { in ForEachAlias() function in xla::HloInputOutputAliasConfig
107 void ForEachAlias(AliasFn fn) const;
52 config_.ForEachAlias( in AliasCount()
352 module->input_output_alias_config().ForEachAlias( in AddCopiesForAliasedInputOutputs()
1540 module->input_output_alias_config().ForEachAlias( in BuildColocatedBufferSets()