Searched refs:GetAliasedOutput (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias_test.cc | 94 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0}); in TEST_F()
95 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{1}); in TEST_F()
96 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{2}); in TEST_F()
97 EXPECT_EQ(config_.GetAliasedOutput(0, {3}), ShapeIndex{3}); in TEST_F()
110 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{0}); in TEST_F()
111 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1}); in TEST_F()
124 EXPECT_EQ(config_.GetAliasedOutput(0, {0}), ShapeIndex{3}); in TEST_F()
125 EXPECT_EQ(config_.GetAliasedOutput(0, {1}), ShapeIndex{2}); in TEST_F()
126 EXPECT_EQ(config_.GetAliasedOutput(0, {2}), ShapeIndex{1}); in TEST_F()
127 EXPECT_EQ(config_.GetAliasedOutput(0, {3}), ShapeIndex{0}); in TEST_F()
[all …]
|
| D | hlo_input_output_alias_config_test.cc | 42 config.GetAliasedOutput(param_number, param_index); in expect_aliased()
59 config.GetAliasedOutput(param_number, param_index); in expect_not_aliased()
|
| D | hlo_input_output_alias_config.h | 94 absl::optional<ShapeIndex> GetAliasedOutput(
|
| D | hlo_input_output_alias_config.cc | 133 absl::optional<ShapeIndex> HloInputOutputAliasConfig::GetAliasedOutput( in GetAliasedOutput() function in xla::HloInputOutputAliasConfig
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 252 auto output_alias = input_output_alias.GetAliasedOutput( in CreateResultShapedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 331 auto output_alias = input_output_alias.GetAliasedOutput( in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 972 auto alias_p0 = config.GetAliasedOutput(0, {}); in TEST_F()
976 auto alias_p1 = config.GetAliasedOutput(1, {}); in TEST_F()
|