Searched defs:segment_ids_shape (Results 1 – 2 of 2) sorted by relevance
987 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local1008 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local1070 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
133 const TensorShape& segment_ids_shape, in operator ()()