Searched defs:MultinomialFunctor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | multinomial_op_gpu.cu.cc | 58 struct MultinomialFunctor<GPUDevice, T, OutputType> { struct
59 void operator()(OpKernelContext* ctx, const GPUDevice& d, in operator ()()
126 template struct MultinomialFunctor<GPUDevice, Eigen::half, int32>; variable
127 template struct MultinomialFunctor<GPUDevice, float, int32>; variable
128 template struct MultinomialFunctor<GPUDevice, double, int32>; variable
129 template struct MultinomialFunctor<GPUDevice, int32, int32>; variable
130 template struct MultinomialFunctor<GPUDevice, int64, int32>; variable
132 template struct MultinomialFunctor<GPUDevice, Eigen::half, int64>; variable
133 template struct MultinomialFunctor<GPUDevice, float, int64>; variable
134 template struct MultinomialFunctor<GPUDevice, double, int64>; variable
[all …]
|
| D | multinomial_op.cc | 45 struct MultinomialFunctor { struct
71 struct MultinomialFunctor<CPUDevice, T, OutputType> { struct
72 void operator()(OpKernelContext* ctx, const CPUDevice& d, in operator ()()
|