Home
last modified time | relevance | path

Searched refs:patch_depth_raw (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops.cc566 const int64_t patch_depth_raw = filter.dim_size(2); in ComputeConv2DDimension() local
569 TF_REQUIRES(FastBoundsCheck(patch_depth_raw, std::numeric_limits<int>::max()), in ComputeConv2DDimension()
572 const int patch_depth = static_cast<int>(patch_depth_raw); in ComputeConv2DDimension()