Searched refs:kernel_size_y (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 929 const int kernel_size_y = m_kernelImpl.dimensions()[idxY];
935 …g_factor = sqrtf(static_cast<float>(maxSharedMem) / (sizeof(Scalar) * kernel_size_y * kernel_size_…
940 …:mini<int>(maxSharedMem / (sizeof(Scalar) * (maxX + kernel_size_x - 1)) - kernel_size_y + 1, numY);
941 …axP = numext::mini<int>(maxSharedMem / ((kernel_size_x - 1 + maxX) * (kernel_size_y - 1 + maxY) * …
948 …const int shared_mem = block_size.z * (maxX + kernel_size_x - 1) * (maxY + kernel_size_y - 1) * si…
968 switch (kernel_size_y) {
974 …_device, m_inputImpl, indexMapper, m_kernel, numP, numX, maxX, numY, maxY, 4, kernel_size_y, data);
981 switch (kernel_size_y) {
987 …_device, m_inputImpl, indexMapper, m_kernel, numP, numX, maxX, numY, maxY, 7, kernel_size_y, data);
994 …nputImpl, indexMapper, m_kernel, numP, numX, maxX, numY, maxY, kernel_size_x, kernel_size_y, data);
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.cc | 134 std::string kernel_size_y = in GenerateDepthwiseConvolutionCode() local
180 c += " for (int ky = 0; ky < " + kernel_size_y + "; ++ky) {\n"; in GenerateDepthwiseConvolutionCode()
|