diff options
| author |  Eugene Zhulenev <ezhulenev@google.com> | 2018-09-14 15:25:27 -0700 |
|---|---|---|
| committer | Eugene Zhulenev <ezhulenev@google.com> | 2018-09-14 15:25:27 -0700 |
| commit | 1b8d70a22b83d63667bbefe3899d9a2e0c2c8b78 (patch) | |
| tree | e50af92d4d253a94d1e9cc87aa748e5c9a579014 /unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h | |
| parent | 9b864cdb3789dbddaa26e53dd85393713b24ce94 (diff) | |
Support reshaping with static shapes and dimensions conversion in tensor broadcasting
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h b/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h index e5cf93ab0..c102a43fb 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorBroadcasting.h @@ -641,7 +641,7 @@ struct TensorEvaluator<const TensorBroadcastingOp<Broadcast, ArgType>, Device> return; } - const Dimensions& input_dims = m_impl.dimensions(); + const Dimensions& input_dims = Dimensions(m_impl.dimensions()); // Pre-fill input_block_sizes, broadcast_block_sizes, // broadcast_block_strides, and broadcast_tensor_strides. Later on we will |