diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-19 19:40:00 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-19 19:42:57 -0700 |
| commit | 9ab04addfb80cbf9334bb330acee5fca09353d23 (patch) | |
| tree | 4abcfc65a760370c7e061ab9fd415e6ddca8b7d9 /tensorflow/compiler/xla/service/hlo_parser_test.cc | |
| parent | c04396e3fd7a449429212d37899703bc3cf507e9 (diff) | |
Remove the ambiguity of device/host computation layouts within the HloModuleConfig.
PiperOrigin-RevId: 201284741
Diffstat (limited to 'tensorflow/compiler/xla/service/hlo_parser_test.cc')
| -rw-r--r-- | tensorflow/compiler/xla/service/hlo_parser_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/compiler/xla/service/hlo_parser_test.cc b/tensorflow/compiler/xla/service/hlo_parser_test.cc index d551400d1e..d481e07f60 100644 --- a/tensorflow/compiler/xla/service/hlo_parser_test.cc +++ b/tensorflow/compiler/xla/service/hlo_parser_test.cc @@ -1302,7 +1302,7 @@ ENTRY %Reduce (input: f32[8,16,256]) -> f32[8,16] { auto module = ParseHloString(original); TF_ASSERT_OK(module.status()); - auto program_layout = module.ValueOrDie()->host_entry_computation_layout(); + auto program_layout = module.ValueOrDie()->entry_computation_layout(); ASSERT_EQ(program_layout.parameter_count(), 1); auto param_layout = program_layout.parameter_layout(0).layout(); auto result_layout = program_layout.result_layout().layout(); |