Searched refs:output_planes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_pooling_test.cc | 135 const int output_planes = 4; in TEST() local
139 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols, in TEST()
151 EXPECT_EQ(result.dimension(1), output_planes); in TEST()
158 for (int i = 0; i < output_planes; ++i) { in TEST()
195 const int output_planes = 4; in TEST() local
200 output_planes, channels); in TEST()
211 EXPECT_EQ(result.dimension(3), output_planes); in TEST()
218 for (int i = 0; i < output_planes; ++i) { in TEST()
255 const int output_planes = 4; in TEST() local
259 Tensor<float, 5> result(channels, output_planes, output_rows, output_cols, in TEST()
[all …]
|
| D | eigen_backward_spatial_convolutions_test.cc | 141 const int output_planes = input_planes - patch_planes + 1; in TEST() local
148 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST()
175 if (output_i >= 0 && output_i < output_planes && in TEST()
204 const int output_planes = input_planes - patch_planes + 1; in TEST() local
212 output_planes, output_depth); in TEST()
238 if (output_i >= 0 && output_i < output_planes && in TEST()
366 const int output_planes = input_planes; in TEST() local
373 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST()
408 if (output_i >= 0 && output_i < output_planes && in TEST()
437 const int output_planes = input_planes; in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 385 output_planes = int(
392 output_planes = int(math.ceil(float(input_planes) / strides[1]))
395 output_shape = [batch, output_planes, output_rows, output_cols, out_depth]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 622 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
626 padding, -1, -1, &output_planes)); in Conv3DShape()
637 output_planes, output_rows, output_cols}); in Conv3DShape()
639 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Conv3DShape()
1103 DimensionHandle output_planes, output_rows, output_cols; in Pool3DShape() local
1105 c, in_planes_dim, kernel_planes, stride_planes, padding, &output_planes)); in Pool3DShape()
1114 output_planes, output_rows, output_cols}); in Pool3DShape()
1116 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Pool3DShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2594 int64 output_planes, output_rows, output_cols; in __anon7c94107b3a02() local
2597 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anon7c94107b3a02()
2606 c->MakeShape({batch_size_dim, output_planes, output_rows, output_cols, in __anon7c94107b3a02()
|