HomeSort by relevance Sort by last modified time
    Searched refs:UnsortedSegmentFunctor (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
segment_reduction_ops_gpu.cu.cc 170 struct UnsortedSegmentFunctor<GPUDevice, T, Index, InitialValueF, ReductionF> {
213 template struct UnsortedSegmentFunctor< \
215 template struct UnsortedSegmentFunctor< \
217 template struct UnsortedSegmentFunctor<GPUDevice, T, Index, functor::One<T>, \
222 template struct UnsortedSegmentFunctor< \
segment_reduction_ops.h 54 struct UnsortedSegmentFunctor {
segment_reduction_ops.cc 367 struct UnsortedSegmentFunctor<CPUDevice, T, Index, InitialValueF, ReductionF> {
508 functor::UnsortedSegmentFunctor<CPUDevice, type, index_type, \
563 functor::UnsortedSegmentFunctor<GPUDevice, type, index_type, \
    [all...]

Completed in 59 milliseconds