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

  /external/tensorflow/tensorflow/core/kernels/
gather_functor.cc 30 int64 GatherFunctor<GPUDevice, T, Index>::operator()( \
34 extern template struct GatherFunctor<GPUDevice, T, Index>;
gather_functor_gpu.cu.cc 28 template struct functor::GatherFunctor<GPUDevice, T, Index>
gather_functor.h 150 struct GatherFunctor {
158 struct GatherFunctor<CPUDevice, T, Index> {
168 struct GatherFunctor<GPUDevice, Variant, Index> {
gather_functor_gpu.cu.h 74 struct GatherFunctor<GPUDevice, T, Index> {
gather_op.cc 110 functor::GatherFunctor<Device, T, Index> functor;
resource_variable_ops.cc 693 functor::GatherFunctor<Device, T, Index> functor;
    [all...]

Completed in 321 milliseconds