Home
last modified time | relevance | path

Searched refs:ParameterHasAlias (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc76 if (!alias_config->ParameterHasAlias(0, input_index) && in Build()
Dhlo_input_output_alias_config.h79 bool ParameterHasAlias(int64 param_number, in ParameterHasAlias() function
Dhlo_live_range.cc195 !module->input_output_alias_config().ParameterHasAlias( in CalculateBufferStartEndMap()
Dhlo_runner.cc136 if (alias_config.ParameterHasAlias(param_num, index)) { in ExecutionInputsFromScopedShapedBuffers()
Dbuffer_assignment.cc113 module->input_output_alias_config().ParameterHasAlias( in HloBufferIsReadOnly()
1127 assignment->module().input_output_alias_config().ParameterHasAlias( in AssignSingleHloBuffer()
Dcopy_insertion.cc48 !computation->parent()->input_output_alias_config().ParameterHasAlias( in IsReadonlyEntryParameterValue()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc332 if (input_output_alias.ParameterHasAlias(i, index)) { in GetArgumentsBuffers()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc253 input_output_alias.ParameterHasAlias(i, xla::ShapeIndex{}); in PopulateInputs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc1156 EXPECT_TRUE(config.ParameterHasAlias(0, {})); in TEST_F()
1157 EXPECT_TRUE(config.ParameterHasAlias(1, {})); in TEST_F()