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
967 int64_t kz = in_minor_channel_index >= 0 in ConvolutionDimensionsFromInputs() local2167 int64_t kz = iz; in OpDimensionsFromInputs() local
117 for (int64_t kz = 0; kz < input.depth(); ++kz) { in SeparableConvArray4D() local