Searched refs:UnsortedSegmentFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | segment_reduction_ops_gpu.cu.cc | 175 struct UnsortedSegmentFunctor<GPUDevice, T, Index, InitialValueF, ReductionF> { struct
220 template struct UnsortedSegmentFunctor< \
222 template struct UnsortedSegmentFunctor< \
224 template struct UnsortedSegmentFunctor<GPUDevice, T, Index, functor::One<T>, \
229 template struct UnsortedSegmentFunctor< \
|
| D | segment_reduction_ops.h | 75 struct UnsortedSegmentFunctor { struct
|
| D | segment_reduction_ops.cc | 371 struct UnsortedSegmentFunctor<CPUDevice, T, Index, InitialValueF, ReductionF> { struct
512 functor::UnsortedSegmentFunctor<CPUDevice, type, index_type, \
567 functor::UnsortedSegmentFunctor<GPUDevice, type, index_type, \
|