Searched refs:three_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op.cc | 140 Eigen::DSizes<int32, 3> three_dims(1, channel, 1); in Compute() local
145 .reshape(three_dims) in Compute()
274 Eigen::DSizes<Eigen::Index, 3> three_dims(batch, channel, in Compute() local
286 .reshape(three_dims) in Compute()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | README.md | 1266 array<int, 3> three_dims{{7, 11, 1}};
1267 Tensor<float, 3> result = input.reshape(three_dims);
|