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.cc987 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local
1008 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local
1070 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_gpu.cu.cc133 const TensorShape& segment_ids_shape, in operator ()()