Searched refs:weightsWidth (Results 1 – 2 of 2) sorted by relevance
40 const unsigned int weightsWidth = weightsShape[widthIndex]; in TransposeConvolution2dImpl() local71 for (unsigned int xWeights = 0u; xWeights < weightsWidth; ++xWeights) in TransposeConvolution2dImpl()91 … weightsIndex = dOutput * weightsHeight * weightsWidth * weightsDepth + in TransposeConvolution2dImpl()92 yWeights * weightsWidth * weightsDepth + in TransposeConvolution2dImpl()108 … weightsIndex = dOutput * weightsDepth * weightsHeight * weightsWidth + in TransposeConvolution2dImpl()109 dInput * weightsHeight * weightsWidth + in TransposeConvolution2dImpl()110 yWeights * weightsWidth + in TransposeConvolution2dImpl()
71 const unsigned int weightsWidth = depthwise ? weightsShape[2] : in Run() local97 for (unsigned int w = 0; w < weightsWidth; ++w) in Run()105 h * weightsWidth * outputChannels; in Run()109 weightsIdx = cOut * weightsHeight * weightsWidth * inputChannels + in Run()110 h * weightsWidth * inputChannels + in Run()116 weightsIdx = cOut * weightsWidth * weightsHeight * inputChannels + in Run()117 cInput * weightsWidth * weightsHeight + in Run()118 h * weightsWidth + in Run()