Searched defs:segment_ids_shape (Results 1 – 3 of 3) sorted by relevance
1124 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local1145 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local1207 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
137 const TensorShape& segment_ids_shape, in operator ()()
1800 const RuntimeShape& segment_ids_shape, in SegmentSum()