Searched refs:weightWidth (Results 1 – 2 of 2) sorted by relevance
189 int64_t weightWidth = weightTy.getDimSize(2); in matchAndRewrite() local197 weightWidth % stride[1] ? stride[1] - weightWidth % stride[1] : 0; in matchAndRewrite()218 weightWidth = weightTy.getDimSize(2); in matchAndRewrite()223 stride[0], weightWidth / stride[1], in matchAndRewrite()241 weightWidth / stride[1], inputChannels}; in matchAndRewrite()
1022 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1039 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()1067 !ShapedType::isDynamic(weightWidth)) { in inferReturnTypeComponents()1091 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1109 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()1137 !ShapedType::isDynamic(weightWidth)) { in inferReturnTypeComponents()1183 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents() local1200 weightWidth = weightShape.getDimSize(1); in inferReturnTypeComponents()1239 !ShapedType::isDynamic(weightWidth)) { in inferReturnTypeComponents()1282 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()[all …]