Lines Matching refs:getRank

356   outShape.reserve(inputShape.getRank() - 1);  in inferReturnTypeComponents()
357 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
383 outputShape.resize(operandShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
386 for (int i = 0, s = operandShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
514 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
524 outputShape.reserve(inputShape.getRank()); in inferReturnTypeComponents()
525 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
593 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
660 if (permsShape.getDimSize(0) != inputShape.getRank()) { in inferReturnTypeComponents()
674 if (inputShape.getRank() == 0) { in inferReturnTypeComponents()
681 for (int i = 1, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
691 outputShape.resize(inputShape.getRank(), inputShape.getDimSize(0)); in inferReturnTypeComponents()
696 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
700 outputShape.reserve(inputShape.getRank()); in inferReturnTypeComponents()
701 for (int i = 0, s = inputShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
881 outRank = std::max<int64_t>(outRank, shape.getRank()); in REDUCE_SHAPE_INFER()
888 auto rankDiff = outShape.size() - shape.getRank(); in REDUCE_SHAPE_INFER()
890 for (size_t i = 0, e = shape.getRank(); i < e; ++i) { in REDUCE_SHAPE_INFER()