Searched defs:kz (Results 1 – 3 of 3) sorted by relevance
221 int64 kz; // Kernel depth (in case of group convolution, this will be member
956 int64 kz = in_minor_channel_index >= 0 in ConvolutionDimensionsFromInputs() local2154 int64 kz = iz; in OpDimensionsFromInputs() local
117 for (int64 kz = 0; kz < input.depth(); ++kz) { in SeparableConvArray4D() local