Searched refs:ksize_cols (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.cc | 78 const int ksize_cols = ksizes_[2]; in Compute() local
87 const int ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in Compute()
99 ksize_rows * ksize_cols * depth}; in Compute()
112 ksize_cols, stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
|
| D | extract_volume_patches_op.cc | 86 const int ksize_cols = ksizes_[3]; in Compute() local
121 GetWindowedOutputSize(in_cols, ksize_cols, stride_cols, in Compute()
126 ksize_planes * ksize_rows * ksize_cols * depth}; in Compute()
139 ksize_rows, ksize_cols, stride_planes, stride_rows, stride_cols, in Compute()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 13 ksize_planes * ksize_rows * ksize_cols * depth]` containing patches
14 with size `ksize_planes x ksize_rows x ksize_cols x depth` vectorized
40 ksizes = [1, ksize_planes, ksize_rows, ksize_cols, 1]
|
| D | api_def_ExtractImagePatches.pbtxt | 13 ksize_cols * depth]` containing image patches with size
14 `ksize_rows x ksize_cols x depth` vectorized in the "depth" dimension. Note
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2616 int32 ksize_cols = ksizes[2]; in __anondb9326b23902() local
2625 int32 ksize_cols_eff = ksize_cols + (ksize_cols - 1) * (rate_cols - 1); in __anondb9326b23902()
2632 c->Dim(input_shape, 3), ksize_rows * ksize_cols, &output_depth_dim)); in __anondb9326b23902()
2712 int32 ksize_cols = ksizes[3]; in __anondb9326b23a02() local
2735 ksize_planes * ksize_rows * ksize_cols, in __anondb9326b23a02()
2762 in_cols, ksize_cols, stride_cols, padding, &output_cols, in __anondb9326b23a02()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4126 ksize_cols * depth]` containing image patches with size
4127 `ksize_rows x ksize_cols x depth` vectorized in the "depth" dimension. Note
|