Searched defs: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
150 Eigen::DSizes<int32, 4> broad_cast_dims(batch, 1, height, width); in Compute() local
158 Eigen::DSizes<int32, 5> broad_cast_dims(batch, 1, height, width, in Compute() local
|