Home
last modified time | relevance | path

Searched refs:alias_config (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dalias_passthrough_params_test.cc39 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local
40 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
41 EXPECT_FALSE(alias_config.OutputHasAlias({1})); in TEST_F()
42 EXPECT_EQ(1, alias_config.GetAliasedParameter({2})->parameter_number); in TEST_F()
55 const auto& alias_config = module->input_output_alias_config(); in TEST_F() local
56 EXPECT_EQ(0, alias_config.GetAliasedParameter({0})->parameter_number); in TEST_F()
57 EXPECT_FALSE(alias_config.OutputHasAlias({1})); in TEST_F()
Dgpu_executable.cc450 if (!output_info.alias_config.has_value()) { in ExecuteAsyncOnStreamImpl()
470 if (output_info.alias_config) { in ExecuteAsyncOnStreamImpl()
486 if (output_info.alias_config->must_alias() && maybe_owning_memory && in ExecuteAsyncOnStreamImpl()
616 output[index].alias_config = in GetOutputInfo()
Dgpu_executable.h67 absl::optional<HloInputOutputAliasConfig::Alias> alias_config; member
Dgpu_compiler.cc1072 o.alias_config.emplace(param_attr.cast<mlir::IntegerAttr>().getInt(), in GetMlirAllocationInfo()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc42 HloInputOutputAliasConfig* alias_config) { in Build() argument
76 if (!alias_config->ParameterHasAlias(0, input_index) && in Build()
77 !alias_config->OutputHasAlias(output_index)) { in Build()
79 alias_config->SetUpAlias(output_index, 0, input_index)); in Build()
96 HloInputOutputAliasConfig* alias_config = in Run() local
101 alias_config); in Run()
Doptimize_input_output_buffer_alias.h62 HloInputOutputAliasConfig* alias_config);
Dhlo_runner.cc123 HloInputOutputAliasConfig alias_config, int device_ordinal, in ExecutionInputsFromScopedShapedBuffers() argument
136 if (alias_config.ParameterHasAlias(param_num, index)) { in ExecutionInputsFromScopedShapedBuffers()
Dhlo_parser.cc818 HloInputOutputAliasConfig alias_config(module->result_shape()); in ParseHloModule() local
821 alias_config.SetUpAlias(p.first, p.second.parameter_number, in ParseHloModule()
827 module->input_output_alias_config() = alias_config; in ParseHloModule()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc55 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config, in AllocateOutputMemoryWithInputReuse() argument
65 TF_RETURN_IF_ERROR(alias_config.ForEachAliasWithStatus( in AllocateOutputMemoryWithInputReuse()
107 if (!ShapeUtil::IndexIsValid(alias_config.shape(), result_index)) { in AllocateOutputMemoryWithInputReuse()
113 alias_config.GetAliasedParameter(result_index); in AllocateOutputMemoryWithInputReuse()
182 const HloInputOutputAliasConfig& alias_config = in ExecuteAsyncOnStream() local
188 shape, alias_config, run_options->allocator(), &arguments, stream, in ExecuteAsyncOnStream()
Dtpu_executable_interface.h67 const Shape& host_shape, const HloInputOutputAliasConfig& alias_config,
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc72 HloInputOutputAliasConfig alias_config = in RunAndCheck() local
145 alias_config.ForEachAlias( in RunAndCheck()