Searched refs:GatherFunctor (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_functor.cc | 30 int64 GatherFunctor<GPUDevice, T, Index>::operator()( \
34 extern template struct GatherFunctor<GPUDevice, T, Index>;
|
| D | gather_functor.h | 150 struct GatherFunctor { struct
158 struct GatherFunctor<CPUDevice, T, Index> {
168 struct GatherFunctor<GPUDevice, Variant, Index> {
|
| D | gather_functor_gpu.cu.cc | 28 template struct functor::GatherFunctor<GPUDevice, T, Index>
|
| D | gather_functor_gpu.cu.h | 74 struct GatherFunctor<GPUDevice, T, Index> {
|
| D | gather_op.cc | 110 functor::GatherFunctor<Device, T, Index> functor; in Compute()
|
| D | resource_variable_ops.cc | 693 functor::GatherFunctor<Device, T, Index> functor; in Compute()
|