diff options
author | Brian Patton <bjp@google.com> | 2018-06-22 07:18:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-22 07:20:52 -0700 |
commit | 1fb0c68b57aa3c08c55f8538bc27c48887ef1a15 (patch) | |
tree | be759e56d02d16e501b9217e76a1b753d558606b /tensorflow/compiler/xla/service/layout_assignment.h | |
parent | 440feef1344968e840571bd78530bff8389f8cd6 (diff) |
Modifies LayoutAssignment to keep a record of the HLO instructions having unconstrained layout at the last layout pass.
PiperOrigin-RevId: 201681232
Diffstat (limited to 'tensorflow/compiler/xla/service/layout_assignment.h')
-rw-r--r-- | tensorflow/compiler/xla/service/layout_assignment.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/service/layout_assignment.h b/tensorflow/compiler/xla/service/layout_assignment.h index 0d7dde9c55..b75ecb311a 100644 --- a/tensorflow/compiler/xla/service/layout_assignment.h +++ b/tensorflow/compiler/xla/service/layout_assignment.h @@ -506,6 +506,11 @@ class LayoutAssignment : public HloPassInterface { // case we have to undo operations due to the multiple passes over the // computations/instructions. ChannelLayoutConstraints channel_constraints_; + + // The set of HLO instructions which lacked any layout constraint, thus + // receiving propagated default layouts. + tensorflow::gtl::FlatSet<const HloInstruction*> + unconstrained_layout_instructions_; }; } // namespace xla |