Searched refs:scalar_indexed (Results 1 – 2 of 2) sorted by relevance
635 const Shape& shape, ScalarIndexedConstantArray* scalar_indexed) { in FoldReshapeOfGatherNoDegenerateDims() argument636 VLOG(3) << "FoldReshapeOfGatherNoDegenerateDims(" << ToString(scalar_indexed) in FoldReshapeOfGatherNoDegenerateDims()639 CHECK(!ShapeUtil::HasDegenerateDimensions(scalar_indexed->shape())); in FoldReshapeOfGatherNoDegenerateDims()691 /*operand_shape=*/AsInt64Slice(scalar_indexed->shape().dimensions()), in FoldReshapeOfGatherNoDegenerateDims()699 if (!absl::c_all_of(scalar_indexed->output_dims(), in FoldReshapeOfGatherNoDegenerateDims()702 << ToString(scalar_indexed); in FoldReshapeOfGatherNoDegenerateDims()710 for (int64 i = scalar_indexed->output_dims().size() - 1; i >= 0; i--) { in FoldReshapeOfGatherNoDegenerateDims()711 int64 output_dim = scalar_indexed->output_dims()[i]; in FoldReshapeOfGatherNoDegenerateDims()724 const Shape& scalar_indexed_source_shape = scalar_indexed->source()->shape(); in FoldReshapeOfGatherNoDegenerateDims()731 scalar_indexed->source_dim()); in FoldReshapeOfGatherNoDegenerateDims()[all …]
321 const Shape& shape, ScalarIndexedConstantArray* scalar_indexed);