Searched refs:inputDepth (Results 1 – 5 of 5) sorted by relevance
33 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \77 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvFloat32()78 wInput * inputDepth + dInput; in groupedConvFloat32()93 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvFloat32()148 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8()149 wInput * inputDepth + dInput; in groupedConvQuant8()170 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8()234 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8PerChannel()235 wInput * inputDepth + dInput; in groupedConvQuant8PerChannel()256 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8PerChannel()
104 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \138 for (uint32_t j = 0; j < filterWidth; j++, filterBase += inputDepth) { in transposeConvNhwc()143 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()153 inputBase += inputDepth; in transposeConvNhwc()218 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()230 k * filterHeight * filterWidth * inputDepth + in transposeConvNhwc()231 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvNhwc()359 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvQuant8PerChannelNhwc()371 k * filterHeight * filterWidth * inputDepth + in transposeConvQuant8PerChannelNhwc()372 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvQuant8PerChannelNhwc()
164 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in depthwiseConvQuant8PerChannel() local200 for (uint32_t ic = 0; ic < inputDepth; ic++) { in depthwiseConvQuant8PerChannel()218 uint32_t inputIndex = hInput * inputWidth * inputDepth + in depthwiseConvQuant8PerChannel()219 wInput * inputDepth + ic; in depthwiseConvQuant8PerChannel()238 inputBase += inputHeight * inputWidth * inputDepth; in depthwiseConvQuant8PerChannel()
113 const int inputDepth = extInputShape.Dims(3); in padGeneric() local145 memcpy(out, in, inputDepth * sizeof(T)); in padGeneric()
295 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local349 uint32_t inputIndex = hInput * inputWidth * inputDepth + in convQuant8PerChannelNhwc()350 wInput * inputDepth + dInput; in convQuant8PerChannelNhwc()369 inputBase += inputHeight * inputWidth * inputDepth; in convQuant8PerChannelNhwc()