Searched refs:ForEachAliasWithStatus (Results 1 – 6 of 6) sorted by relevance
167 Status HloInputOutputAliasConfig::ForEachAliasWithStatus( in ForEachAliasWithStatus() function in xla::HloInputOutputAliasConfig187 return ForEachAliasWithStatus([&](const ShapeIndex& output_index, in Verify()
126 Status ForEachAliasWithStatus(AliasFnWithStatus fn) const;
411 TF_RETURN_IF_ERROR(module->input_output_alias_config().ForEachAliasWithStatus( in AddCopiesForAliasedInputOutputs()
65 TF_RETURN_IF_ERROR(alias_config.ForEachAliasWithStatus( in AllocateOutputMemoryWithInputReuse()
227 TF_RETURN_IF_ERROR(config.ForEachAliasWithStatus( in GetParametersThatMustBeDonated()
309 return input_output_alias.ForEachAliasWithStatus(alias_function); in RebuildOutputAliases()