Searched refs:GetAliasedOutput (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias_test.cc | 89 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0}); in TEST_F()
90 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{1}); in TEST_F()
91 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{2}); in TEST_F()
92 EXPECT_EQ(config_.GetAliasedOutput(0, {3}), ShapeIndex{3}); in TEST_F()
105 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0}); in TEST_F()
106 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1}); in TEST_F()
119 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{3}); in TEST_F()
120 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{2}); in TEST_F()
121 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1}); in TEST_F()
122 EXPECT_EQ(config_.GetAliasedOutput(0, {3}), ShapeIndex{0}); in TEST_F()
[all …]
|
| D | hlo_input_output_alias_config.h | 81 return GetAliasedOutput(param_number, param_index).has_value(); in ParameterHasAlias()
97 absl::optional<ShapeIndex> GetAliasedOutput(
|
| D | hlo_input_output_alias_config_test.cc | 41 config.GetAliasedOutput(param_number, param_index); in expect_aliased()
58 config.GetAliasedOutput(param_number, param_index); in expect_not_aliased()
|
| D | hlo_input_output_alias_config.cc | 137 absl::optional<ShapeIndex> HloInputOutputAliasConfig::GetAliasedOutput( in GetAliasedOutput() function in xla::HloInputOutputAliasConfig
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 1159 auto alias_p0 = config.GetAliasedOutput(0, {}); in TEST_F()
1163 auto alias_p1 = config.GetAliasedOutput(1, {}); in TEST_F()
|