Searched refs:scalar_indexed_source_shape (Results 1 – 1 of 1) sorted by relevance
724 const Shape& scalar_indexed_source_shape = scalar_indexed->source()->shape(); in FoldReshapeOfGatherNoDegenerateDims() local729 scalar_indexed_source_shape.dimensions()), in FoldReshapeOfGatherNoDegenerateDims()758 << StrJoin(scalar_indexed_source_shape.dimensions(), ",") in FoldReshapeOfGatherNoDegenerateDims()766 scalar_indexed_source_shape.dimensions(scalar_indexed->source_dim())); in FoldReshapeOfGatherNoDegenerateDims()770 ShapeUtil::ElementsIn(scalar_indexed_source_shape)); in FoldReshapeOfGatherNoDegenerateDims()775 scalar_indexed_source_shape.dimensions()), in FoldReshapeOfGatherNoDegenerateDims()