Searched refs:patch_depth (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 360 const int patch_depth = static_cast<int>(patch_depth_raw); in ComputeConv2DDimension() local
361 TF_REQUIRES(in_depth % patch_depth == 0, in ComputeConv2DDimension()
364 in_depth, " vs ", patch_depth)); in ComputeConv2DDimension()
423 dimensions->patch_depth = patch_depth; in ComputeConv2DDimension()
475 << ", patch_depth = " << dimensions.patch_depth in Compute()
|
| D | conv_ops.h | 89 int patch_depth;
|
| D | eigen_spatial_convolutions-inl.h | 104 Index patch_depth; in TensorContractionInputMapper() local
106 patch_depth = tensor.impl().dimensions()[0]; in TensorContractionInputMapper()
112 patch_depth = tensor.impl().dimensions()[NumDims - 1]; in TensorContractionInputMapper()
119 m_patch_row_stride = patch_depth; in TensorContractionInputMapper()
143 m_rowInputStride = patch_depth; in TensorContractionInputMapper()
144 m_colInputStride = patch_depth * m_inputRows; in TensorContractionInputMapper()
145 m_patchInputStride = patch_depth * m_inputRows * m_inputCols; in TensorContractionInputMapper()
161 m_fastDimZero = internal::TensorIntDivisor<Index>(patch_depth); in TensorContractionInputMapper()
|
| D | eigen_spatial_convolutions_test.cc | 1505 const Index patch_depth = evaluators[0].impl().dimensions()[0]; in PackRhsHelper() local
1515 const Index patch_size = patch_depth * patch_rows * patch_cols; in PackRhsHelper()
1547 absl::StrCat("patch: ", patch_rows, "x", patch_cols, " D", patch_depth, in PackRhsHelper()
|
| D | conv_ops_fused_impl.h | 914 << ", patch_depth = " << dimensions.patch_depth
|