Home
last modified time | relevance | path

Searched defs:GatherTree (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/seq2seq/kernels/
Dbeam_search_ops.h29 struct GatherTree { struct
Dbeam_search_ops_gpu.cu.cc81 struct GatherTree<GPUDevice, T> { struct
82 void operator()(OpKernelContext* ctx, const GPUDevice& d, in operator ()()
Dbeam_search_ops.cc106 struct GatherTree<CPUDevice, int32> { struct
107 void operator()(OpKernelContext* ctx, const CPUDevice& d, in operator ()()