Searched refs:kernel_size_x (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 928 const int kernel_size_x = m_kernelImpl.dimensions()[idxX];
935 …actor = sqrtf(static_cast<float>(maxSharedMem) / (sizeof(Scalar) * kernel_size_y * kernel_size_x));
938 …int inner_dim = ((static_cast<int>(scaling_factor * kernel_size_x) - kernel_size_x + 1 + 32) / 32)…
940 …const int maxY = numext::mini<int>(maxSharedMem / (sizeof(Scalar) * (maxX + kernel_size_x - 1)) - …
941 …const int maxP = numext::mini<int>(maxSharedMem / ((kernel_size_x - 1 + maxX) * (kernel_size_y - 1…
948 …const int shared_mem = block_size.z * (maxX + kernel_size_x - 1) * (maxY + kernel_size_y - 1) * si…
966 switch (kernel_size_x) {
994 …ce, m_inputImpl, indexMapper, m_kernel, numP, numX, maxX, numY, maxY, kernel_size_x, kernel_size_y…
1009 const int kernel_size_x = m_kernelImpl.dimensions()[idxX];
1018 …i<int>(maxSharedMem / (sizeof(Scalar) * kernel_size_y * kernel_size_z) - kernel_size_x + 1, numX));
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.cc | 132 std::string kernel_size_x = in GenerateDepthwiseConvolutionCode() local
187 c += " for (int kx = 0; kx < " + kernel_size_x + "; ++kx) {\n"; in GenerateDepthwiseConvolutionCode()
|