Searched refs:GatherTree (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/seq2seq/kernels/ |
| D | beam_search_ops.cc | 90 functor::GatherTree<Device, T>()(ctx, device, step_ids_t, parent_ids_t, in Compute()
106 struct GatherTree<CPUDevice, int32> { struct
172 void GatherTree<GPUDevice, T>::operator()( \
178 extern template struct GatherTree<GPUDevice, T>;
|
| D | beam_search_ops_gpu.cu.cc | 81 struct GatherTree<GPUDevice, T> { struct
102 #define DEFINE_GPU_SPECS(T) template struct GatherTree<GPUDevice, T>;
|
| D | beam_search_ops.h | 29 struct GatherTree { struct
|