Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:GatherOpKernel (Results 1 - 2 of 2) sorted by null

/external/tensorflow/tensorflow/core/kernels/
gather_functor_gpu.cu.h	33 __global__ void GatherOpKernel(const T* params, const Index* indices, T* out, 96 GatherOpKernel<T, Index, true>, config.block_count, 101 GatherOpKernel<T, Index, false>, config.block_count,
dynamic_partition_op_gpu.cu.cc	108 GatherOpKernel<T, int32, true>, config.block_count,

Completed in 117 milliseconds