Home
last modified time | relevance | path

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

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/
Dunsorted_segment.cc52 const tflite::RuntimeShape& segment_ids_shape, in UnsortedSegmentRef()
176 const tflite::RuntimeShape& segment_ids_shape, in EvalType()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops.cc1157 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local
1178 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local
1240 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/tensorflow/tensorflow/lite/kernels/
Dunsorted_segment.cc144 const RuntimeShape& segment_ids_shape, in EvalType()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h995 const RuntimeShape& segment_ids_shape, in SegmentSum()
1015 const RuntimeShape& segment_ids_shape, in UnsortedSegmentRef()