Searched refs:concatDimSize (Results 1 – 4 of 4) sorted by relevance
367 int concatDimSize = 0; in inferReturnTypeComponents() local374 concatDimSize = -1; in inferReturnTypeComponents()378 concatDimSize += operandTy.getDimSize(axis); in inferReturnTypeComponents()381 outputShape[axis] = concatDimSize; in inferReturnTypeComponents()
570 int concatDimSize = 0; in inferReturnTypeComponents() local577 concatDimSize = ShapedType::kDynamicSize; in inferReturnTypeComponents()581 concatDimSize += operandShape.getDimSize(axis); in inferReturnTypeComponents()584 outputShape[axis] = concatDimSize; in inferReturnTypeComponents()