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

  /external/tensorflow/tensorflow/core/kernels/
gather_nd_op.h 33 struct GatherNdSlice {
gather_nd_op_cpu_impl.h 19 // Specialization of GatherNdSlice to CPU
97 struct GatherNdSlice<CPUDevice, T, Index, IXDIM> {
128 template Index GatherNdSlice<CPUDevice, T, Index, CPU_PROVIDED_IXDIM>:: \
gather_nd_op_gpu.cu.cc 68 struct GatherNdSlice<GPUDevice, T, Index, IXDIM> {
106 template struct functor::GatherNdSlice<GPUDevice, T, Index, NDIM>;
gather_nd_op.cc 168 functor::GatherNdSlice<Device, T, Index, IXDIM> func; \
209 Index GatherNdSlice<GPUDevice, T, Index, NDIM>::operator()( \
215 extern template struct GatherNdSlice<GPUDevice, T, Index, NDIM>;

Completed in 116 milliseconds