Searched refs:broad_cast_dims (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 140 Eigen::DSizes<int32, 3> broad_cast_dims(batch, 1, height); in Compute() local
145 .broadcast(broad_cast_dims); in Compute()
149 Eigen::DSizes<int32, 4> broad_cast_dims(batch, 1, height, width); in Compute() local
153 bias.tensor<T, 1>().reshape(four_dims).broadcast(broad_cast_dims); in Compute()
157 Eigen::DSizes<int32, 5> broad_cast_dims(batch, 1, height, width, in Compute() local
162 bias.tensor<T, 1>().reshape(five_dims).broadcast(broad_cast_dims); in Compute()
|