Searched refs:inputDepth (Results 1 – 5 of 5) sorted by relevance
41 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \85 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvFloat32()86 wInput * inputDepth + dInput; in groupedConvFloat32()101 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvFloat32()157 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8()158 wInput * inputDepth + dInput; in groupedConvQuant8()179 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8()262 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8PerChannel()263 wInput * inputDepth + dInput; in groupedConvQuant8PerChannel()284 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8PerChannel()
109 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \145 for (uint32_t j = 0; j < filterWidth; j++, filterBase += inputDepth) { in transposeConvNhwc()150 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()160 inputBase += inputDepth; in transposeConvNhwc()225 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()237 k * filterHeight * filterWidth * inputDepth + in transposeConvNhwc()238 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvNhwc()367 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvQuant8PerChannelNhwc()379 k * filterHeight * filterWidth * inputDepth + in transposeConvQuant8PerChannelNhwc()380 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvQuant8PerChannelNhwc()
288 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local324 for (uint32_t ic = 0; ic < inputDepth; ic++) { in depthwiseConvQuant8PerChannelNhwc()342 uint32_t inputIndex = hInput * inputWidth * inputDepth + in depthwiseConvQuant8PerChannelNhwc()343 wInput * inputDepth + ic; in depthwiseConvQuant8PerChannelNhwc()362 inputBase += inputHeight * inputWidth * inputDepth; in depthwiseConvQuant8PerChannelNhwc()
362 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local416 uint32_t inputIndex = hInput * inputWidth * inputDepth + in convQuant8PerChannelNhwc()417 wInput * inputDepth + dInput; in convQuant8PerChannelNhwc()436 inputBase += inputHeight * inputWidth * inputDepth; in convQuant8PerChannelNhwc()455 [[maybe_unused]] uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local
127 const int inputDepth = extInputShape.Dims(3); in padGeneric() local159 memcpy(out, in, inputDepth * sizeof(T)); in padGeneric()