experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Revert the To32Bit optimization from cl/218760066 since it doesn't compile with the new version of Eigen.

Browse Source 
PiperOrigin-RevId: 218785999
This commit is contained in:
A. Unique TensorFlower 2018-10-25 18:25:18 -07:00 committed by TensorFlower Gardener
parent 24963954a8
commit ba60fcf712
1 changed files with 0 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/broadcast_to_op.h
View File

@ -57,16 +57,8 @@ struct BroadcastTo {
#define HANDLE_BROADCAST_CASE(dim_i) \
case dim_i: { \
if (std::is_same<Eigen::GpuDevice, Device>::value && \
output_tensor.NumElements() < kint32max && \
input_tensor.NumElements() < kint32max) { \
auto input = input_tensor.tensor<T, dim_i>(); \
To32Bit(output).device(d) = \
To32Bit(input).reshape(reshape).broadcast(broadcast); \
} else { \
auto input = input_tensor.tensor<T, dim_i>(); \
output.device(d) = input.reshape(reshape).broadcast(broadcast); \
} \
} break;
if (output_shape.num_elements() == 0) {