Searched refs:patch_depth (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 162 const int64_t patch_depth = filter.dim_size(2); in operator ()() local
163 const int64_t num_groups = in_depth / patch_depth; in operator ()()
264 const int64_t patch_depth = filter.dim_size(2); in operator ()() local
266 if (patch_depth <= 0) { in operator ()()
268 "filter depth must be stricly positive, got ", patch_depth)); in operator ()()
271 if (in_depth % patch_depth != 0) { in operator ()()
274 " vs ", patch_depth)); in operator ()()
284 const int64_t num_groups = in_depth / patch_depth; in operator ()()
297 if (in_depth != patch_depth) { in operator ()()
561 const int patch_depth = static_cast<int>(patch_depth_raw); in ComputeConv2DDimension() local
[all …]
|
| D | conv_ops.h | 89 int patch_depth;
|
| D | eigen_spatial_convolutions-inl.h | 109 Index patch_depth; in TensorContractionInputMapper() local
111 patch_depth = tensor.impl().dimensions()[0]; in TensorContractionInputMapper()
117 patch_depth = tensor.impl().dimensions()[NumDims - 1]; in TensorContractionInputMapper()
124 m_patch_row_stride = patch_depth; in TensorContractionInputMapper()
149 m_rowInputStride = patch_depth; in TensorContractionInputMapper()
150 m_colInputStride = patch_depth * m_inputRows; in TensorContractionInputMapper()
151 m_patchInputStride = patch_depth * m_inputRows * m_inputCols; in TensorContractionInputMapper()
167 m_fastDimZero = internal::TensorIntDivisor<Index>(patch_depth); in TensorContractionInputMapper()
|
| D | eigen_spatial_convolutions_test.cc | 1526 const Index patch_depth = evaluators[0].impl().dimensions()[0]; in PackRhsHelper() local
1536 const Index patch_size = patch_depth * patch_rows * patch_cols; in PackRhsHelper()
1572 absl::StrCat("patch: ", patch_rows, "x", patch_cols, " D", patch_depth, in PackRhsHelper()
|
| D | conv_ops_fused_impl.h | 806 << ", patch_depth = " << dimensions.patch_depth
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2021-015.md | 25 const int64 patch_depth = filter.dim_size(2);
26 if (in_depth % patch_depth != 0) { ... }
|