Searched refs:kernel_size_z (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | depthwise_conv.cc | 136 std::string kernel_size_z = in GenerateDepthwiseConvolutionCode() local
173 c += " for (int kz = 0; kz < " + kernel_size_z + "; ++kz) {\n"; in GenerateDepthwiseConvolutionCode()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 1011 const int kernel_size_z = m_kernelImpl.dimensions()[idxZ];
1018 …8, numext::mini<int>(maxSharedMem / (sizeof(Scalar) * kernel_size_y * kernel_size_z) - kernel_size…
1019 …ni<int>(maxSharedMem / (sizeof(Scalar) * (maxX + kernel_size_x - 1) * kernel_size_z) - kernel_size…
1020 …eof(Scalar) * (maxX + kernel_size_x - 1) * (maxY + kernel_size_y - 1)) - kernel_size_z + 1, numZ));
1028 …m = (maxX + kernel_size_x - 1) * (maxY + kernel_size_y - 1) * (maxZ + kernel_size_z - 1) * sizeof(…
1040 …rnel, numP, numX, maxX, numY, maxY, numZ, maxZ, kernel_size_x, kernel_size_y, kernel_size_z, data);
|