Searched refs:inputDepth (Results 1 – 5 of 5) sorted by relevance
37 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \81 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvFloat32()82 wInput * inputDepth + dInput; in groupedConvFloat32()97 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvFloat32()153 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8()154 wInput * inputDepth + dInput; in groupedConvQuant8()175 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8()257 uint32_t inputIndex = hInput * inputWidth * inputDepth + in groupedConvQuant8PerChannel()258 wInput * inputDepth + dInput; in groupedConvQuant8PerChannel()279 inputBase += inputHeight * inputWidth * inputDepth; in groupedConvQuant8PerChannel()
115 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); \149 for (uint32_t j = 0; j < filterWidth; j++, filterBase += inputDepth) { in transposeConvNhwc()154 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()164 inputBase += inputDepth; in transposeConvNhwc()229 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvNhwc()241 k * filterHeight * filterWidth * inputDepth + in transposeConvNhwc()242 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvNhwc()371 for (uint32_t d = 0; d < inputDepth; d++) { in transposeConvQuant8PerChannelNhwc()383 k * filterHeight * filterWidth * inputDepth + in transposeConvQuant8PerChannelNhwc()384 i * filterWidth * inputDepth + j * inputDepth + d; in transposeConvQuant8PerChannelNhwc()
292 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local328 for (uint32_t ic = 0; ic < inputDepth; ic++) { in depthwiseConvQuant8PerChannelNhwc()346 uint32_t inputIndex = hInput * inputWidth * inputDepth + in depthwiseConvQuant8PerChannelNhwc()347 wInput * inputDepth + ic; in depthwiseConvQuant8PerChannelNhwc()366 inputBase += inputHeight * inputWidth * inputDepth; in depthwiseConvQuant8PerChannelNhwc()
121 const int inputDepth = extInputShape.Dims(3); in padGeneric() local153 memcpy(out, in, inputDepth * sizeof(T)); in padGeneric()
365 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local419 uint32_t inputIndex = hInput * inputWidth * inputDepth + in convQuant8PerChannelNhwc()420 wInput * inputDepth + dInput; in convQuant8PerChannelNhwc()439 inputBase += inputHeight * inputWidth * inputDepth; in convQuant8PerChannelNhwc()458 uint32_t inputDepth = getSizeOfDimension(inputShape, 3); in convQuant8PerChannelNhwc() local