Searched refs:must_alias (Results 1 – 5 of 5) sorted by relevance
31 HloInputOutputAliasConfig::AliasKind must_alias) { in SetUpAlias() argument47 Alias(param_number, param_index, must_alias); in SetUpAlias()67 if (data->must_alias()) { in ToProto()130 alias->parameter_index == param_index && alias->must_alias()) { in ParameterMustAlias()
55 bool must_alias() const { return kind == kMustAlias; } in must_alias() function75 AliasKind must_alias = kMayAlias);
68 if (alias && alias->must_alias()) { in AllocateOutputMemoryWithInputReuse()
266 if (alias->must_alias() && !maybe_owning_memory->HasOwnership()) { in CreateResultShapedBuffer()
486 if (output_info.alias_config->must_alias() && maybe_owning_memory && in ExecuteAsyncOnStreamImpl()