Home
last modified time | relevance | path

Searched refs:SetUpAlias (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config_test.cc88 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
120 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
125 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
159 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
164 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
190 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
216 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()
221 ASSERT_IS_NOT_OK(config.SetUpAlias( in TEST_F()
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()
Dhlo_input_output_alias_config.cc26 Status HloInputOutputAliasConfig::SetUpAlias(const ShapeIndex& output_index, in SetUpAlias() function in xla::HloInputOutputAliasConfig
100 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
Dhlo_input_output_alias_config.h66 Status SetUpAlias(const ShapeIndex& output_index, int64 param_number,
Doptimize_input_output_buffer_alias.cc76 TF_RETURN_IF_ERROR(alias_config->SetUpAlias( in Build()
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_test.cc489 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc963 b.SetUpAlias({1}, 0, {}); in TEST_F()
964 b.SetUpAlias({0}, 1, {}); in TEST_F()
Dxla_builder.h302 void SetUpAlias(const ShapeIndex& output_index, int64 param_number, in SetUpAlias() function
Dxla_builder.cc410 TF_RETURN_IF_ERROR(config.SetUpAlias( in PopulateInputOutputAlias()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc1099 builder.SetUpAlias({1}, 0, {0}); in TEST()
1100 builder.SetUpAlias({0}, 0, {1}); in TEST()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc1098 builder.SetUpAlias(alias.output_index, alias.param_number, in CompileGraph()