| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_3d.h | 46 typename TTypes<T, 5>::ConstTensor filter, int stride_planes,
50 input, filter, stride_planes, stride_rows, stride_cols, padding);
60 int stride_planes, int stride_rows, int stride_cols) {
67 stride_cols, stride_rows, stride_planes);
77 int stride_planes, int stride_rows, int stride_cols) {
84 stride_cols, stride_rows, stride_planes);
|
| D | extract_volume_patches_op.h | 31 int stride_planes, int stride_rows, int stride_cols, in operator()
41 stride_cols, stride_rows, stride_planes, in operator()
48 stride_cols, stride_rows, stride_planes, in operator()
|
| D | extract_volume_patches_op.cc | 85 const int stride_planes = strides_[1]; in Compute() local
112 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
136 ksize_rows, ksize_cols, stride_planes, stride_rows, stride_cols, in Compute()
172 int stride_planes, int stride_rows, int stride_cols, \
|
| D | eigen_backward_spatial_convolutions_test.cc | 1809 const int stride_planes = 2; in TEST() local
1816 ceil_div(input_planes - patch_planes + 1, stride_planes); in TEST()
1831 stride_planes, stride_rows, stride_cols); in TEST()
1853 output_i / stride_planes < output_planes && in TEST()
1856 output_i % stride_planes == 0 && in TEST()
1860 output_backward(od, output_i / stride_planes, in TEST()
1888 const int stride_planes = 2; in TEST() local
1895 ceil_div(input_planes - patch_planes + 1, stride_planes); in TEST()
1910 stride_planes, stride_rows, stride_cols); in TEST()
1932 output_i / stride_planes < output_planes && in TEST()
[all …]
|
| D | mkl_conv_ops.h | 80 int stride_planes = GetTensorDim(strides_, data_format_, '0'); in GetStridesInMklOrder() local
83 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder()
353 int stride_planes, stride_rows, stride_cols; variable
363 stride_planes = strides[0];
415 input_planes, filter_planes, stride_planes,
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 29 `input`. Must be: `[1, stride_planes, stride_rows, stride_cols, 1]`.
41 strides = [1, stride_planes, strides_rows, strides_cols, 1]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 582 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
591 stride_planes = strides[2]; in Conv3DShape()
598 stride_planes = strides[1]; in Conv3DShape()
625 c, in_planes_dim, filter_planes_dim, dilation_planes, stride_planes, in Conv3DShape()
1066 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
1076 stride_planes = strides[2]; in Pool3DShape()
1083 stride_planes = strides[1]; in Pool3DShape()
1105 c, in_planes_dim, kernel_planes, stride_planes, padding, &output_planes)); in Pool3DShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2555 int32 stride_planes = strides[1]; in __anon7c94107b3a02() local
2597 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anon7c94107b3a02()
|