Home
last modified time | relevance | path

Searched refs:input_output_alias_config (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.h240 HloInputOutputAliasConfig& input_output_alias_config() { in input_output_alias_config() function
243 const HloInputOutputAliasConfig& input_output_alias_config() const { in input_output_alias_config() function
Dhlo_alias_analysis_test.cc243 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
246 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
251 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
289 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
292 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
297 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
380 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
383 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
Doptimize_input_output_buffer_alias.cc99 &module->input_output_alias_config(); in Run()
Dcopy_insertion_test.cc1378 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1381 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1413 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1416 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1481 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1523 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
1565 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
Dbuffer_assignment.cc1017 assignment->module().input_output_alias_config().ParameterHasAlias( in AssignBuffersForComputation()
1431 !module->input_output_alias_config().ParameterHasAlias( in MergeColocatedBufferSets()
1539 VLOG(4) << module->input_output_alias_config(); in BuildColocatedBufferSets()
1540 module->input_output_alias_config().ForEachAlias( in BuildColocatedBufferSets()
1745 assignment->module().input_output_alias_config().ParameterHasAlias( in AssignColocatedBufferSets()
Dcopy_insertion.cc48 !computation->parent()->input_output_alias_config().ParameterHasAlias( in IsReadonlyEntryParameterValue()
352 module->input_output_alias_config().ForEachAlias( in AddCopiesForAliasedInputOutputs()
399 TF_RETURN_IF_ERROR(module->input_output_alias_config().ForEachAliasWithStatus( in AddCopiesForAliasedInputOutputs()
Dhlo_alias_analysis.cc194 module_->input_output_alias_config().GetAliasedParameter( in ComputeInputOutputAliasedBuffers()
Dhlo_module.cc238 *proto.mutable_input_output_alias() = input_output_alias_config().ToProto(); in ToProto()
Dhlo_verifier.cc1497 TF_RETURN_IF_ERROR(module->input_output_alias_config().Verify( in Run()
Dbuffer_assignment_test.cc489 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc237 executable->executable()->module().input_output_alias_config(); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc217 module().input_output_alias_config(); in CreateResultShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc330 module().input_output_alias_config(); in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc968 const HloInputOutputAliasConfig& config = module->input_output_alias_config(); in TEST_F()