Lines Matching refs:kDynamicSize
383 outputShape.resize(operandShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
389 if (outputShape[i] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
411 concatDimSize = ShapedType::kDynamicSize; in inferReturnTypeComponents()
434 outShape.resize(2, ShapedType::kDynamicSize); in inferReturnTypeComponents()
445 outShape[1] = outShape[1] == ShapedType::kDynamicSize in inferReturnTypeComponents()
465 outShape.resize(3, ShapedType::kDynamicSize); in inferReturnTypeComponents()
473 outShape[0] = outShape[0] == ShapedType::kDynamicSize in inferReturnTypeComponents()
506 outputShape.resize(paddingShape.getDimSize(0), ShapedType::kDynamicSize); in inferReturnTypeComponents()
514 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
527 outputShape.push_back(ShapedType::kDynamicSize); in inferReturnTypeComponents()
579 outputShape.resize(multiples.size(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
595 if (dim != ShapedType::kDynamicSize) in inferReturnTypeComponents()
628 if (val != ShapedType::kDynamicSize) { in inferReturnTypeComponents()
635 if (val == ShapedType::kDynamicSize) in inferReturnTypeComponents()
668 outputShape.resize(permsShape.getDimSize(0), ShapedType::kDynamicSize); in inferReturnTypeComponents()
696 outputShape.resize(inputShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
715 outputShape.resize(3, ShapedType::kDynamicSize); in inferReturnTypeComponents()
725 if (outputShape[0] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
727 if (outputShape[1] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
741 outputShape.resize(4, ShapedType::kDynamicSize); in inferReturnTypeComponents()
743 int32_t inHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
744 int32_t inWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
778 if (outputShape[1] == ShapedType::kDynamicSize && inHeight >= 0 && fpMode) { in inferReturnTypeComponents()
784 if (outputShape[2] == ShapedType::kDynamicSize && inWidth >= 0 && fpMode) { in inferReturnTypeComponents()
790 if (outputShape[1] == ShapedType::kDynamicSize && inHeight >= 0 && !fpMode) { in inferReturnTypeComponents()
796 if (outputShape[2] == ShapedType::kDynamicSize && inWidth >= 0 && !fpMode) { in inferReturnTypeComponents()
811 outputShape.resize(3, ShapedType::kDynamicSize); in inferReturnTypeComponents()
822 if (outputShape[0] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
828 if (outputShape[0] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
830 if (outputShape[2] == ShapedType::kDynamicSize) in inferReturnTypeComponents()
1017 llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamicSize); in inferReturnTypeComponents()
1020 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1021 int32_t inputHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1022 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1023 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1084 llvm::SmallVector<int64_t> outputShape(5, ShapedType::kDynamicSize); in inferReturnTypeComponents()
1087 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1088 int32_t inputHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1089 int32_t inputDepth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1091 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1092 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1093 int32_t weightDepth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1176 llvm::SmallVector<int64_t> outputShape(4, ShapedType::kDynamicSize); in inferReturnTypeComponents()
1179 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1180 int32_t inputHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1181 int32_t inputChannels = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1183 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1184 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1185 int32_t depthChannels = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1260 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1261 int32_t inputHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1262 int32_t weightWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1263 int32_t weightHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()