Searched refs:output_planes (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_cuboid_convolutions_test.cc | 40 const int output_planes = input_planes - patch_planes + 1; in TEST() local
47 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST()
74 if (output_i >= 0 && output_i < output_planes && in TEST()
103 const int output_planes = input_planes - patch_planes + 1; in TEST() local
111 output_planes, output_depth); in TEST()
137 if (output_i >= 0 && output_i < output_planes && in TEST()
166 const int output_planes = input_planes; in TEST() local
173 Tensor<float, 4> output_backward(output_depth, output_planes, output_rows, in TEST()
208 if (output_i >= 0 && output_i < output_planes && in TEST()
237 const int output_planes = input_planes; in TEST() local
[all …]
|
| 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 …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | conv_ops_3d_test.py | 488 output_planes = int(
495 output_planes = int(math.ceil(float(input_planes) / strides[1]))
498 output_shape = [batch, output_planes, output_rows, output_cols, out_depth]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 861 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
865 padding, -1, -1, &output_planes)); in Conv3DShape()
876 output_planes, output_rows, output_cols}); in Conv3DShape()
878 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Conv3DShape()
1856 DimensionHandle output_planes, output_rows, output_cols; in Pool3DShape() local
1858 c, in_planes_dim, kernel_planes, stride_planes, padding, &output_planes)); in Pool3DShape()
1867 output_planes, output_rows, output_cols}); in Pool3DShape()
1869 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows, in Pool3DShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2786 int64_t output_planes, output_rows, output_cols; in __anon091a71b73a02() local
2789 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anon091a71b73a02()
2798 c->MakeShape({batch_size_dim, output_planes, output_rows, output_cols, in __anon091a71b73a02()
|