Searched refs:ksize_rows (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.cc | 77 const int ksize_rows = ksizes_[1]; in Compute() local
86 const int ksize_rows_eff = ksize_rows + (ksize_rows - 1) * (rate_rows - 1); in Compute()
99 ksize_rows * ksize_cols * depth}; in Compute()
111 context->eigen_device<Device>(), input.tensor<T, 4>(), ksize_rows, in Compute()
|
| D | extract_volume_patches_op.cc | 85 const int ksize_rows = ksizes_[2]; in Compute() local
118 GetWindowedOutputSize(in_rows, ksize_rows, stride_rows, 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 | 12 4-D Tensor with shape `[batch, out_rows, out_cols, ksize_rows *
14 `ksize_rows x ksize_cols x depth` vectorized in the "depth" dimension. Note
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2615 int32 ksize_rows = ksizes[1]; in __anondb9326b23902() local
2624 int32 ksize_rows_eff = ksize_rows + (ksize_rows - 1) * (rate_rows - 1); in __anondb9326b23902()
2632 c->Dim(input_shape, 3), ksize_rows * ksize_cols, &output_depth_dim)); in __anondb9326b23902()
2711 int32 ksize_rows = ksizes[2]; in __anondb9326b23a02() local
2735 ksize_planes * ksize_rows * ksize_cols, in __anondb9326b23a02()
2759 in_rows, ksize_rows, stride_rows, padding, &output_rows, in __anondb9326b23a02()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 4125 …TF_Uint32, TF_Uint64, TF_Uint8]>, [{4-D Tensor with shape `[batch, out_rows, out_cols, ksize_rows *
4127 `ksize_rows x ksize_cols x depth` vectorized in the "depth" dimension. Note
|