Searched refs:broad_cast_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 141 Eigen::DSizes<int32, 3> broad_cast_dims(batch, 1, height); in Compute() local
146 .broadcast(broad_cast_dims); in Compute()
150 Eigen::DSizes<int32, 4> broad_cast_dims(batch, 1, height, width); in Compute() local
154 bias.tensor<T, 1>().reshape(four_dims).broadcast(broad_cast_dims); in Compute()
158 Eigen::DSizes<int32, 5> broad_cast_dims(batch, 1, height, width, in Compute() local
163 bias.tensor<T, 1>().reshape(five_dims).broadcast(broad_cast_dims); in Compute()
|