Searched refs:ForEachAliasWithStatus (Results 1 – 4 of 4) sorted by relevance
162 Status HloInputOutputAliasConfig::ForEachAliasWithStatus( in ForEachAliasWithStatus() function in xla::HloInputOutputAliasConfig182 return ForEachAliasWithStatus([&](const ShapeIndex& output_index, in Verify()
118 Status ForEachAliasWithStatus(AliasFnWithStatus fn) const;
399 TF_RETURN_IF_ERROR(module->input_output_alias_config().ForEachAliasWithStatus( in AddCopiesForAliasedInputOutputs()
248 TF_RETURN_IF_ERROR(input_output_alias.ForEachAliasWithStatus(alias_function)); in DoWork()