Home
last modified time | relevance | path

Searched defs:GatherNdSlice (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dgather_nd_op_gpu.cu.cc68 struct GatherNdSlice<GPUDevice, T, Index, IXDIM> { struct
69 Index operator()(const GPUDevice& d, const Index unused_slice_size, in operator ()()
Dgather_nd_op.h37 struct GatherNdSlice { struct