Searched defs: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
176 void operator()(OpKernelContext* ctx, const Index num_segments, in operator ()()
|
| D | segment_reduction_ops.h | 75 struct UnsortedSegmentFunctor { struct
|
| D | segment_reduction_ops.cc | 371 struct UnsortedSegmentFunctor<CPUDevice, T, Index, InitialValueF, ReductionF> { struct
372 void operator()(OpKernelContext* ctx, const Index num_segments, in operator ()()
|