Searched refs:SparseSegmentReductionFunctor (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu_1.cu.cc | 78 template struct SparseSegmentReductionFunctor<T, int64, int32>; \
79 template struct SparseSegmentReductionFunctor<T, int64, int64>;
|
| D | segment_reduction_ops_gpu_0.cu.cc | 90 template struct SparseSegmentReductionFunctor<T, int32, int32>; \
91 template struct SparseSegmentReductionFunctor<T, int32, int64>;
|
| D | segment_reduction_ops.h | 177 struct SparseSegmentReductionFunctor { struct
|
| D | segment_reduction_ops_gpu.cu.h | 732 Status SparseSegmentReductionFunctor<T, Index, SegmentId>::operator()(
|
| D | segment_reduction_ops_impl.h | 835 functor::SparseSegmentReductionFunctor<T, Index, SegmentId> functor;
|