projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b25e6fe) | patch
[TF:XLA] Separate on-host and on-device shape and layout in HloModule.
authorA. Unique TensorFlower <gardener@tensorflow.org>
Tue, 1 May 2018 23:17:21 +0000 (16:17 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 1 May 2018 23:19:55 +0000 (16:19 -0700)
commit210abebd3febdd2c44ab5021bcebf8f1f5d451c4
tree22810ab30037169b4df8fec07016a8d8f69b7a6etree | snapshot
parentb25e6fe32cccd29ec4cb4014bbb45d62b75835b4commit | diff
[TF:XLA] Separate on-host and on-device shape and layout in HloModule.

Previously, only one layout was stored with an HLO module. This CL allows
HLO passes to modify the on-device layouts without affecting the on-host
layout (provided by the client)

PiperOrigin-RevId: 195014875
24 files changed:
tensorflow/compiler/xla/client/local_client.cc diff | blob | history
tensorflow/compiler/xla/client/local_client.h diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_compiler.cc diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_executable.cc diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_layout_assignment.h diff | blob | history
tensorflow/compiler/xla/service/cpu/cpu_layout_assignment_test.cc diff | blob | history
tensorflow/compiler/xla/service/executable.h diff | blob | history
tensorflow/compiler/xla/service/gpu/gpu_compiler.cc diff | blob | history
tensorflow/compiler/xla/service/gpu/gpu_layout_assignment.h diff | blob | history
tensorflow/compiler/xla/service/gpu/gpu_layout_assignment_test.cc diff | blob | history
tensorflow/compiler/xla/service/hlo_module.cc diff | blob | history
tensorflow/compiler/xla/service/hlo_module.h diff | blob | history
tensorflow/compiler/xla/service/hlo_module_config.cc diff | blob | history
tensorflow/compiler/xla/service/hlo_module_config.h diff | blob | history
tensorflow/compiler/xla/service/interpreter/compiler.cc diff | blob | history
tensorflow/compiler/xla/service/layout_assignment.cc diff | blob | history
tensorflow/compiler/xla/service/layout_assignment.h diff | blob | history
tensorflow/compiler/xla/service/layout_assignment_test.cc diff | blob | history
tensorflow/compiler/xla/service/service.cc diff | blob | history
tensorflow/compiler/xla/service/service.h diff | blob | history
tensorflow/compiler/xla/tests/BUILD diff | blob | history
tensorflow/compiler/xla/tests/hlo_test_base.h diff | blob | history
tensorflow/compiler/xla/tools/parser/hlo_parser.cc diff | blob | history
tensorflow/compiler/xla/tools/parser/hlo_parser_test.cc diff | blob | history
Domain: Machine Learning / ML Framework;