Home
last modified time | relevance | path

Searched refs:kUserAlias (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_input_output_alias_config_test.cc91 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
123 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
128 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
162 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
167 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
193 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
219 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
224 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
Dhlo_input_output_alias_config.cc30 TF_RET_CHECK(kind == AliasKind::kUserAlias || kind == AliasKind::kSystemAlias) in SetUpAlias()
61 case AliasKind::kUserAlias: in ToProto()
97 kind = AliasKind::kUserAlias; in CreateFromProto()
110 const char* kind = alias.kind == AliasKind::kUserAlias ? "USER" : "SYSTEM"; in ToString()
Dhlo_alias_analysis_test.cc245 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
248 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
253 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
291 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
294 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
299 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
382 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
385 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
Dhlo_input_output_alias_config.h40 kUserAlias, enumerator
Dcopy_insertion_test.cc1380 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1383 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1415 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1418 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1483 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1525 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
1567 /*kind=*/HloInputOutputAliasConfig::AliasKind::kUserAlias)); in TEST_F()
Dbuffer_assignment_test.cc490 {}, 0, {}, HloInputOutputAliasConfig::kUserAlias)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc242 return alias.kind == xla::HloInputOutputAliasConfig::AliasKind::kUserAlias in DoWork()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc412 HloInputOutputAliasConfig::AliasKind::kUserAlias)); in PopulateInputOutputAlias()