| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_cuboid_convolutions_test.cc | 32 const int input_planes = 5; in TEST() local
40 const int output_planes = input_planes - patch_planes + 1; in TEST()
43 Tensor<float, 4> input_backward(input_depth, input_planes, input_rows, in TEST()
55 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
59 EXPECT_EQ(input_backward.dimension(1), input_planes); in TEST()
63 for (int i = 0; i < input_planes; ++i) { in TEST()
95 const int input_planes = 5; in TEST() local
103 const int output_planes = input_planes - patch_planes + 1; in TEST()
107 input_planes, input_depth); in TEST()
118 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
[all …]
|
| D | eigen_pooling_test.cc | 126 const int input_planes = 5; in TEST() local
137 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols, in TEST()
186 const int input_planes = 5; in TEST() local
198 input_planes, channels); in TEST()
246 const int input_planes = 5; in TEST() local
257 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols, in TEST()
308 const int input_planes = 5; in TEST() local
320 input_planes, channels); in TEST()
370 const int input_planes = 5; in TEST() local
379 const int output_planes = input_planes; in TEST()
[all …]
|
| D | eigen_benchmark_cpu_test.cc | 250 int input_planes, int input_depth, in CuboidConvolution() argument
261 input_batches, input_height, input_width, input_planes, input_depth); in CuboidConvolution()
278 int input_width, int input_planes, in CuboidConvolutionBackwardInput() argument
290 input_batches, input_height, input_width, input_planes, input_depth); in CuboidConvolutionBackwardInput()
306 int input_width, int input_planes, in CuboidConvolutionBackwardKernel() argument
318 input_batches, input_height, input_width, input_planes, input_depth); in CuboidConvolutionBackwardKernel()
326 (input_batches * input_height * input_width * input_planes); in CuboidConvolutionBackwardKernel()
|
| D | eigen_benchmark.h | 222 Eigen::Index input_planes = input_dims[3]; in CuboidConvolutionBackwardInput() local
240 filter, output_backward, input_planes, input_rows, input_cols); in CuboidConvolutionBackwardInput()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 157 int input_planes = static_cast<int>(input_planes_raw); in GetInputSizeInMklOrder() local
173 mkldnn_sizes[MklDnnDims3D::Dim3d_D] = input_planes; in GetInputSizeInMklOrder()
330 int input_planes, input_rows, input_cols; variable
335 input_planes = GetTensorDim(input_shape, data_format_, '0');
421 input_planes, filter_planes, dilation_planes,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_3d_test.py | 467 input_planes, input_rows, input_cols = input_shape
470 input_shape = [batch, input_planes, input_rows, input_cols, in_depth]
482 math.ceil((input_planes - filter_planes + 1.0) / strides[1]))
488 output_planes = int(math.ceil(float(input_planes) / strides[1]))
|