Searched refs:SetUpAlias (Results 1 – 20 of 20) sorted by relevance
87 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()118 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()122 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()155 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()159 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()184 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()209 TF_ASSERT_OK(config.SetUpAlias( in TEST_F()213 ASSERT_IS_NOT_OK(config.SetUpAlias( in TEST_F()
79 alias_config->SetUpAlias(output_index, 0, input_index)); in Build()
243 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()245 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()249 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()286 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()288 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()292 ASSERT_IS_NOT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()374 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()376 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias( in TEST_F()
28 Status HloInputOutputAliasConfig::SetUpAlias( in SetUpAlias() function in xla::HloInputOutputAliasConfig90 result.SetUpAlias(output_index, param_number, param_index, kind)); in CreateFromProto()
73 Status SetUpAlias(const ShapeIndex& output_index, int64 param_number,
1588 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1590 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1621 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1623 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1687 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1728 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()1769 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()2600 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()2604 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()2627 ASSERT_IS_OK(module->input_output_alias_config().SetUpAlias( in TEST_F()
240 TF_ASSERT_OK(module_->input_output_alias_config().SetUpAlias({}, 0, {})); in TEST_F()
3536 TF_CHECK_OK(module->input_output_alias_config().SetUpAlias({0}, 0, {0})); in TEST_P()3537 TF_CHECK_OK(module->input_output_alias_config().SetUpAlias({1}, 0, {1})); in TEST_P()
496 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({}, 0, {})); in TEST_F()
821 alias_config.SetUpAlias(p.first, p.second.parameter_number, in ParseHloModule()
254 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({0}, 0, {0})); in TEST_F()255 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({1}, 0, {1})); in TEST_F()271 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({0}, 0, {0})); in TEST_F()272 TF_ASSERT_OK(module->input_output_alias_config().SetUpAlias({1}, 0, {1})); in TEST_F()329 TF_ASSERT_OK(module->get()->input_output_alias_config().SetUpAlias( in TEST_F()
40 TF_RETURN_IF_ERROR(module->input_output_alias_config().SetUpAlias( in Run()
43 [`XlaBuilder::SetUpAlias`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
410 builder.SetUpAlias( in BuildXlaCompilerSubmodule()
1150 b.SetUpAlias({1}, 0, {}); in TEST_F()1151 b.SetUpAlias({0}, 1, {}); in TEST_F()
401 void SetUpAlias(const ShapeIndex& output_index, int64 param_number,
543 TF_RETURN_IF_ERROR(config.SetUpAlias(alias.output_index, alias.param_number, in PopulateInputOutputAlias()
1804 builder.SetUpAlias({1}, 0, {0}); in TEST()1805 builder.SetUpAlias({0}, 0, {1}); in TEST()1903 builder.SetUpAlias({1}, 0, {0}); in TEST()1904 builder.SetUpAlias({0}, 0, {1}); in TEST()
1419 builder.SetUpAlias(/*output_index=*/{aliasing_output.getInt()}, in RunOnFunction()1422 builder.SetUpAlias(/*output_index=*/{aliasing_output.getInt()}, in RunOnFunction()
415 builder->SetUpAlias(alias.output_index, alias.param_number, in BuildComputation()