| /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 | eigen_backward_cuboid_convolutions_test.cc | 709 const int stride_planes = 2; in TEST() local
716 ceil_div(input_planes - patch_planes + 1, stride_planes); in TEST()
731 stride_planes, stride_rows, stride_cols); in TEST()
753 output_i / stride_planes < output_planes && in TEST()
756 output_i % stride_planes == 0 && in TEST()
760 output_backward(od, output_i / stride_planes, in TEST()
788 const int stride_planes = 2; in TEST() local
795 ceil_div(input_planes - patch_planes + 1, stride_planes); in TEST()
810 stride_planes, stride_rows, stride_cols); in TEST()
832 output_i / stride_planes < output_planes && in TEST()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| 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 | 88 const int stride_planes = strides_[1]; in Compute() local
115 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
139 ksize_rows, ksize_cols, stride_planes, stride_rows, stride_cols, in Compute()
175 int stride_planes, int stride_rows, int stride_cols, \
|
| /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/kernels/mkl/ |
| D | mkl_conv_ops.h | 86 int stride_planes = GetTensorDim(strides_, data_format_, '0'); in GetStridesInMklOrder() local
89 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder()
359 int stride_planes, stride_rows, stride_cols; variable
369 stride_planes = strides[0];
422 stride_planes, padding_, &out_planes,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 765 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
774 stride_planes = strides[2]; in Conv3DShape()
781 stride_planes = strides[1]; in Conv3DShape()
828 c, in_planes_dim, filter_planes_dim, dilation_planes, stride_planes, in Conv3DShape()
1728 int32 stride_planes, stride_rows, stride_cols; in Pool3DShape() local
1738 stride_planes = strides[2]; in Pool3DShape()
1745 stride_planes = strides[1]; in Pool3DShape()
1767 c, in_planes_dim, kernel_planes, stride_planes, padding, &output_planes)); in Pool3DShape()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2714 int32 stride_planes = strides[1]; in __anon42d741193a02() local
2756 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anon42d741193a02()
|