Searched refs:weightHeight (Results 1 – 2 of 2) sorted by relevance
188 int64_t weightHeight = weightTy.getDimSize(1); in matchAndRewrite() local195 weightHeight % stride[0] ? stride[0] - weightHeight % stride[0] : 0; in matchAndRewrite()217 weightHeight = weightTy.getDimSize(1); in matchAndRewrite()222 outputChannels, weightHeight / stride[0], in matchAndRewrite()240 outputChannels * stride[0] * stride[1], weightHeight / stride[0], in matchAndRewrite()
1023 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1038 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()1059 !ShapedType::isDynamic(weightHeight)) { in inferReturnTypeComponents()1092 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1108 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()1129 !ShapedType::isDynamic(weightHeight)) { in inferReturnTypeComponents()1184 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1199 weightHeight = weightShape.getDimSize(0); in inferReturnTypeComponents()1231 !ShapedType::isDynamic(weightHeight)) { in inferReturnTypeComponents()1281 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()[all …]