| /external/tensorflow/tensorflow/core/kernels/ |
| eigen_pooling_test.cc | 136 const int output_planes = 4; local
140 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols,
152 EXPECT_EQ(result.dimension(1), output_planes);
159 for (int i = 0; i < output_planes; ++i) {
196 const int output_planes = 4; local
201 output_planes, channels);
212 EXPECT_EQ(result.dimension(3), output_planes);
219 for (int i = 0; i < output_planes; ++i) {
256 const int output_planes = 4; local
260 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols
318 const int output_planes = 4; local
380 const int output_planes = input_planes; local
458 const int output_planes = input_planes; local
631 const int output_planes = 2; local
691 const int output_planes = 2; local
[all...] |
| eigen_backward_spatial_convolutions_test.cc | 142 const int output_planes = input_planes - patch_planes + 1; local
149 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows,
176 if (output_i >= 0 && output_i < output_planes &&
205 const int output_planes = input_planes - patch_planes + 1; local
213 output_planes, output_depth);
239 if (output_i >= 0 && output_i < output_planes &&
367 const int output_planes = input_planes; local
374 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows,
409 if (output_i >= 0 && output_i < output_planes &&
438 const int output_planes = input_planes local
621 const int output_planes = input_planes - patch_planes + 1; local
688 const int output_planes = input_planes - patch_planes + 1; local
1249 const int output_planes = input_planes - patch_planes + 1; local
1312 const int output_planes = input_planes - patch_planes + 1; local
1551 const int output_planes = input_planes - patch_planes + 1; local
1618 const int output_planes = input_planes - patch_planes + 1; local
1811 const int output_planes = local
1890 const int output_planes = local
1966 const int output_planes = local
2043 const int output_planes = local
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| common_shape_fns.cc | 531 DimensionHandle output_planes, output_rows, output_cols; local
535 stride_planes, padding, &output_planes));
544 output_planes, output_rows, output_cols});
546 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows,
1010 DimensionHandle output_planes, output_rows, output_cols; local
[all...] |
