Lines Matching refs:getDimSize

360     outShape.push_back(inputShape.getDimSize(i));  in inferReturnTypeComponents()
390 outputShape[i] = operandShape.getDimSize(i); in inferReturnTypeComponents()
391 if (outputShape[i] != operandShape.getDimSize(i)) in inferReturnTypeComponents()
415 concatDimSize += operandShape.getDimSize(axis); in inferReturnTypeComponents()
437 outShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
441 outShape[1] = weightShape.getDimSize(0); in inferReturnTypeComponents()
446 ? biasShape.getDimSize(0) in inferReturnTypeComponents()
468 outShape[0] = lhsShape.getDimSize(0); in inferReturnTypeComponents()
469 outShape[1] = lhsShape.getDimSize(1); in inferReturnTypeComponents()
474 ? rhsShape.getDimSize(0) in inferReturnTypeComponents()
476 outShape[2] = rhsShape.getDimSize(2); in inferReturnTypeComponents()
506 outputShape.resize(paddingShape.getDimSize(0), ShapedType::kDynamicSize); in inferReturnTypeComponents()
531 outputShape.push_back(inputShape.getDimSize(i) + paddingValues[i * 2] + in inferReturnTypeComponents()
594 int dim = inputShape.getDimSize(i); in inferReturnTypeComponents()
660 if (permsShape.getDimSize(0) != inputShape.getRank()) { in inferReturnTypeComponents()
668 outputShape.resize(permsShape.getDimSize(0), ShapedType::kDynamicSize); in inferReturnTypeComponents()
682 if (inputShape.getDimSize(0) != inputShape.getDimSize(i)) { in inferReturnTypeComponents()
691 outputShape.resize(inputShape.getRank(), inputShape.getDimSize(0)); in inferReturnTypeComponents()
702 outputShape[i] = inputShape.getDimSize(permShape.getDimSize(i)); in inferReturnTypeComponents()
719 outputShape[0] = valuesShape.getDimSize(0); in inferReturnTypeComponents()
720 outputShape[2] = valuesShape.getDimSize(2); in inferReturnTypeComponents()
726 outputShape[0] = indicesShape.getDimSize(0); in inferReturnTypeComponents()
728 outputShape[1] = indicesShape.getDimSize(1); in inferReturnTypeComponents()
748 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
749 outputShape[3] = inputShape.getDimSize(3); in inferReturnTypeComponents()
751 inHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
752 inWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
815 outputShape[0] = valuesInShape.getDimSize(0); in inferReturnTypeComponents()
816 outputShape[1] = valuesInShape.getDimSize(1); in inferReturnTypeComponents()
817 outputShape[2] = valuesInShape.getDimSize(2); in inferReturnTypeComponents()
823 outputShape[0] = indicesShape.getDimSize(0); in inferReturnTypeComponents()
829 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
831 outputShape[2] = inputShape.getDimSize(2); in inferReturnTypeComponents()
892 auto dim2 = shape.getDimSize(i); in REDUCE_SHAPE_INFER()
985 outputShape[0] = inputShape.getDimSize(0); in NARY_SHAPE_INFER()
986 outputShape[3] = inputShape.getDimSize(3); in NARY_SHAPE_INFER()
988 int32_t height = inputShape.getDimSize(1); in NARY_SHAPE_INFER()
989 int32_t width = inputShape.getDimSize(2); in NARY_SHAPE_INFER()
1029 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1030 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1031 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1037 outputShape[3] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1038 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1039 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1046 ? biasShape.getDimSize(0) in inferReturnTypeComponents()
1098 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1099 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1100 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1101 inputDepth = inputShape.getDimSize(3); in inferReturnTypeComponents()
1107 outputShape[4] = weightShape.getDimSize(0); in inferReturnTypeComponents()
1108 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1109 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1110 weightDepth = weightShape.getDimSize(3); in inferReturnTypeComponents()
1117 (outputShape[4] == -1) ? biasShape.getDimSize(0) : outputShape[4]; in inferReturnTypeComponents()
1190 outputShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
1191 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1192 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1193 inputChannels = inputShape.getDimSize(3); in inferReturnTypeComponents()
1199 weightHeight = weightShape.getDimSize(0); in inferReturnTypeComponents()
1200 weightWidth = weightShape.getDimSize(1); in inferReturnTypeComponents()
1202 ? weightShape.getDimSize(2) in inferReturnTypeComponents()
1204 depthChannels = weightShape.getDimSize(3); in inferReturnTypeComponents()
1218 ? biasShape.getDimSize(0) in inferReturnTypeComponents()
1269 ? inputShape.getDimSize(0) in inferReturnTypeComponents()
1271 inputHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
1272 inputWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
1279 ? weightShape.getDimSize(0) in inferReturnTypeComponents()
1281 weightHeight = weightShape.getDimSize(1); in inferReturnTypeComponents()
1282 weightWidth = weightShape.getDimSize(2); in inferReturnTypeComponents()
1289 ? biasShape.getDimSize(0) in inferReturnTypeComponents()