| 1a736239090dbba23008d518602a429261a1c0e0 |
|
21-Aug-2017 |
A. Unique TensorFlower <gardener@tensorflow.org> |
Automated g4 rollback of changelist 165646100
PiperOrigin-RevId: 165958212
/external/tensorflow/tensorflow/compiler/xla/service/cpu/runtime_matvec.cc
|
| a92bd5d5cb6e0d4685cc672710986877d0afc65d |
|
18-Aug-2017 |
A. Unique TensorFlower <gardener@tensorflow.org> |
Automated g4 rollback of changelist 165630063
PiperOrigin-RevId: 165646100
/external/tensorflow/tensorflow/compiler/xla/service/cpu/runtime_matvec.cc
|
| 02f87fee25552e220c8295b58ab8e58b6fbe598b |
|
17-Aug-2017 |
A. Unique TensorFlower <gardener@tensorflow.org> |
CPU runtime: Improve the performance of matrix-vector and
vector-matrix products.
This change makes the single threaded matrix-vector product explicit
so that Eigen will always delegate to an optimized GEMV kernel. This
is done by using an Eigen Matrix instead of the Eigen Tensor
implementation. This is the same optimization done by TensorFlow's
matmul op for GEMV.
This is used even in the multi-threaded case because it appears to
be faster than the multi-threaded version.
This change also expands the scope of the CPU runtime test to test
vec-mat and mat-vec on both single threaded and multi threaded modes.
PiperOrigin-RevId: 165630063
/external/tensorflow/tensorflow/compiler/xla/service/cpu/runtime_matvec.cc
|