Searched refs:patch_depth (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.h | 89 int patch_depth;
|
| D | conv_ops.cc | 427 const int patch_depth = static_cast<int>(patch_depth_raw); in ComputeConv2DDimension() local
428 TF_REQUIRES(in_depth % patch_depth == 0, in ComputeConv2DDimension()
431 in_depth, " vs ", patch_depth)); in ComputeConv2DDimension()
490 dimensions->patch_depth = patch_depth; in ComputeConv2DDimension()
541 << ", patch_depth = " << dimensions.patch_depth in Compute()
|
| D | eigen_spatial_convolutions-inl.h | 108 Index patch_depth; in TensorContractionInputMapper() local
110 patch_depth = tensor.impl().dimensions()[0]; in TensorContractionInputMapper()
116 patch_depth = tensor.impl().dimensions()[NumDims - 1]; in TensorContractionInputMapper()
123 m_patch_row_stride = patch_depth; in TensorContractionInputMapper()
148 m_rowInputStride = patch_depth; in TensorContractionInputMapper()
149 m_colInputStride = patch_depth * m_inputRows; in TensorContractionInputMapper()
150 m_patchInputStride = patch_depth * m_inputRows * m_inputCols; in TensorContractionInputMapper()
166 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 | 780 << ", patch_depth = " << dimensions.patch_depth
|