Searched defs:MultinomialFunctor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | multinomial_op_gpu.cu.cc | 58 struct MultinomialFunctor<GPUDevice, T, OutputType> { struct
59 void operator()(OpKernelContext* ctx, const GPUDevice& d, in operator ()()
120 template struct MultinomialFunctor<GPUDevice, Eigen::half, int32>; variable
121 template struct MultinomialFunctor<GPUDevice, float, int32>; variable
122 template struct MultinomialFunctor<GPUDevice, double, int32>; variable
123 template struct MultinomialFunctor<GPUDevice, int32, int32>; variable
124 template struct MultinomialFunctor<GPUDevice, int64, int32>; variable
126 template struct MultinomialFunctor<GPUDevice, Eigen::half, int64>; variable
127 template struct MultinomialFunctor<GPUDevice, float, int64>; variable
128 template struct MultinomialFunctor<GPUDevice, double, int64>; variable
[all …]
|
| H A 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 ()()
|