aboutsummaryrefslogtreecommitdiffhomepage
path: root/configure.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-19 12:55:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-19 13:03:19 -0700
commitf72126c164ea67b226368bf51811c8528d81093b (patch)
tree116987724ac043ba9e8b00692f4e5eed821b85f7 /configure.py
parentc95353498e180e50f701dcb8331b994d9e5fad0b (diff)
Two improvements in resolve_tensorflow_matmul:
1. Before inserting a new Transpose node, check if there already is one that may be reused. In practice, there are two cases: either the array being transposed is a constant (by far the most common case) or it's not. * If it is constant, then this doesn't really make a difference: ResolveConstantTranspose runs anyway, eliminating these Transpose nodes and also mootifying this change as it leaves no Transpose node to be reused. So in that case, constant-array-deduping is really the only thing that prevents duplication of data. * If it is not constant, that's where this new logic really helps, as the resulting Transpose nodes are here to stay in the final graph, and this avoids inserting more than are needed. 2. transpose_a is not supported. However, rather than CHECK-fail, it's more useful to have this graph transformation bail with a log message. The resulting 'unresolved' MatMul node could still be handled in some way at the TFLite level, or we could end up having support for MatMul per se. PiperOrigin-RevId: 213678294
Diffstat (limited to 'configure.py')
0 files changed, 0 insertions, 0 deletions