Home
last modified time | relevance | path

Searched refs:GetAliasedParameter (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dalias_passthrough_params_test.cc40 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()
Dgpu_executable.cc617 hlo_module.input_output_alias_config().GetAliasedParameter(index); in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config_test.cc47 config.GetAliasedParameter(output_index); in expect_aliased()
63 config.GetAliasedParameter(output_index); in expect_not_aliased()
Dhlo_input_output_alias_config.h103 absl::optional<Alias> GetAliasedParameter(
Dhlo_input_output_alias_config.cc151 HloInputOutputAliasConfig::GetAliasedParameter( in GetAliasedParameter() function in xla::HloInputOutputAliasConfig
Dhlo_alias_analysis.cc194 module_->input_output_alias_config().GetAliasedParameter( in ComputeInputOutputAliasedBuffers()
Dcopy_insertion.cc1110 module->input_output_alias_config().GetAliasedParameter(index); in AddSpecialCaseCopies()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc113 alias_config.GetAliasedParameter(result_index); in AllocateOutputMemoryWithInputReuse()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc260 input_output_alias.GetAliasedParameter(index); in CreateResultShapedBuffer()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc314 input_output_alias.GetAliasedParameter(output_index)) { in GetOrCreateTensorForOutput()