Searched refs:weightsHeight (Results 1 – 2 of 2) sorted by relevance
39 const unsigned int weightsHeight = weightsShape[heightIndex]; in TransposeConvolution2dImpl() local69 for (unsigned int yWeights = 0u; yWeights < weightsHeight; ++yWeights) in TransposeConvolution2dImpl()91 … weightsIndex = dOutput * weightsHeight * weightsWidth * weightsDepth + in TransposeConvolution2dImpl()108 … weightsIndex = dOutput * weightsDepth * weightsHeight * weightsWidth + in TransposeConvolution2dImpl()109 dInput * weightsHeight * weightsWidth + in TransposeConvolution2dImpl()
69 const unsigned int weightsHeight = depthwise ? weightsShape[1] : in Run() local95 for (unsigned int h = 0; h < weightsHeight; ++h) in Run()109 weightsIdx = cOut * weightsHeight * weightsWidth * inputChannels + in Run()116 weightsIdx = cOut * weightsWidth * weightsHeight * inputChannels + in Run()117 cInput * weightsWidth * weightsHeight + in Run()