Searched defs:BincountFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bincount_op_gpu.cu.cc | 37 struct BincountFunctor<GPUDevice, Tidx, T, false> { struct
38 static Status Compute(OpKernelContext* context, in Compute()
116 struct BincountFunctor<GPUDevice, Tidx, T, true> { struct
117 static Status Compute(OpKernelContext* context, in Compute()
|
| D | bincount_op.h | 30 struct BincountFunctor { struct
|
| D | bincount_op.cc | 39 struct BincountFunctor<CPUDevice, Tidx, T, true> { struct
40 static Status Compute(OpKernelContext* context, in Compute()
84 struct BincountFunctor<CPUDevice, Tidx, T, false> { struct
85 static Status Compute(OpKernelContext* context, in Compute()
|