diff options
author | 2017-10-20 12:47:57 -0700 | |
---|---|---|
committer | 2017-10-20 12:55:39 -0700 | |
commit | 5c331cfd573984287778aab02794dd86ba1f3006 (patch) | |
tree | fb36c812fbd87a51f5ecf6763461daa920aaa5bf /tensorflow/compiler/xla/layout_util.h | |
parent | aada11e19a1ceb901f490aa3c064f2778cb2acf2 (diff) |
The new array class provides a way to simplify the implementation of
these classes by eliminating a large number of duplicated code.
Removing the old API is non-trivial because of the existing users
outside of tensorflow.
PiperOrigin-RevId: 172920837
Diffstat (limited to 'tensorflow/compiler/xla/layout_util.h')
-rw-r--r-- | tensorflow/compiler/xla/layout_util.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/compiler/xla/layout_util.h b/tensorflow/compiler/xla/layout_util.h index 5de0a653f6..bc42e22229 100644 --- a/tensorflow/compiler/xla/layout_util.h +++ b/tensorflow/compiler/xla/layout_util.h @@ -40,6 +40,7 @@ class LayoutUtil { static Layout GetDefaultLayoutForShape(const Shape& shape); // Helper functions that create default layouts for various ranks. + static Layout GetDefaultLayoutForRank(int64 rank); static Layout GetDefaultLayoutForR2(); static Layout GetDefaultLayoutForR3(); static Layout GetDefaultLayoutForR4(); |