Home
last modified time | relevance | path

Searched defs:segment_ids_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1145 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local
1166 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local
1228 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h1499 const RuntimeShape& segment_ids_shape, in SegmentSum()