Searched refs:ksize_planes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | extract_volume_patches_op.cc | 81 const int ksize_planes = ksizes_[1]; in Compute() local
112 GetWindowedOutputSize(in_planes, ksize_planes, stride_planes, in Compute()
123 ksize_planes * ksize_rows * ksize_cols * depth}; in Compute()
135 context->eigen_device<Device>(), input.tensor<T, 5>(), ksize_planes, 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]
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 2551 int32 ksize_planes = ksizes[1]; in __anondb9326b23a02() local
2576 ksize_planes * ksize_rows * ksize_cols, in __anondb9326b23a02()
2597 in_planes, ksize_planes, stride_planes, padding, &output_planes, in __anondb9326b23a02()
|