Searched refs:GetAliasedParameter (Results 1 – 10 of 10) sorted by relevance
40 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()42 EXPECT_EQ(1, alias_config.GetAliasedParameter({2})->parameter_number); in TEST_F()56 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
617 hlo_module.input_output_alias_config().GetAliasedParameter(index); in GetOutputInfo()
47 config.GetAliasedParameter(output_index); in expect_aliased()63 config.GetAliasedParameter(output_index); in expect_not_aliased()
103 absl::optional<Alias> GetAliasedParameter(
151 HloInputOutputAliasConfig::GetAliasedParameter( in GetAliasedParameter() function in xla::HloInputOutputAliasConfig
194 module_->input_output_alias_config().GetAliasedParameter( in ComputeInputOutputAliasedBuffers()
1110 module->input_output_alias_config().GetAliasedParameter(index); in AddSpecialCaseCopies()
113 alias_config.GetAliasedParameter(result_index); in AllocateOutputMemoryWithInputReuse()
260 input_output_alias.GetAliasedParameter(index); in CreateResultShapedBuffer()
314 input_output_alias.GetAliasedParameter(output_index)) { in GetOrCreateTensorForOutput()