Searched refs:indicesShape (Results 1 – 1 of 1) sorted by relevance
723 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents() local724 if (indicesShape.hasRank()) { in inferReturnTypeComponents()726 outputShape[0] = indicesShape.getDimSize(0); in inferReturnTypeComponents()728 outputShape[1] = indicesShape.getDimSize(1); in inferReturnTypeComponents()820 ShapeAdaptor indicesShape = operands.getShape(1); in inferReturnTypeComponents() local821 if (indicesShape.hasRank()) { in inferReturnTypeComponents()823 outputShape[0] = indicesShape.getDimSize(0); in inferReturnTypeComponents()