Searched refs:segment_ids_shape (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.cc | 133 const TensorShape& segment_ids_shape, in operator ()() argument
144 if (data_size == 0 || segment_ids_shape.num_elements() == 0) { in operator ()()
177 const TensorShape& segment_ids_shape, in operator ()()
190 if (data_size == 0 || segment_ids_shape.num_elements() == 0) { in operator ()()
|
| D | segment_reduction_ops.h | 65 const Index output_rows, const TensorShape& segment_ids_shape,
77 const TensorShape& segment_ids_shape,
|
| D | segment_reduction_ops.cc | 373 const TensorShape& segment_ids_shape, in operator ()()
391 "segment_ids", SliceDebugString(segment_ids_shape, i), in operator ()()
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_ops.cc | 987 ShapeHandle segment_ids_shape; in SegmentReductionShapeFn() local
989 TF_RETURN_IF_ERROR(c->WithRank(c->input(1), 1, &segment_ids_shape)); in SegmentReductionShapeFn()
1008 ShapeHandle segment_ids_shape; in SparseSegmentReductionShapeFn() local
1009 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionShapeFn()
1013 TF_RETURN_IF_ERROR(c->Merge(indices_shape, segment_ids_shape, &unused)); in SparseSegmentReductionShapeFn()
1070 ShapeHandle segment_ids_shape; in SparseSegmentReductionWithNumSegmentsShapeFn() local
1071 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 1, &segment_ids_shape)); in SparseSegmentReductionWithNumSegmentsShapeFn()
1078 TF_RETURN_IF_ERROR(c->Merge(indices_shape, segment_ids_shape, &unused)); in SparseSegmentReductionWithNumSegmentsShapeFn()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | math_ops.py | 3213 segment_ids_shape = array_ops.shape_internal(segment_ids)
3214 ones_tensor = array_ops.ones(segment_ids_shape, dtype=data.dtype)
|