Searched refs:MultinomialFunctor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | multinomial_op_gpu.cu.cc | 58 struct MultinomialFunctor<GPUDevice, T, OutputType> { struct
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
135 template struct MultinomialFunctor<GPUDevice, int32, int64>; variable
[all …]
|
| D | multinomial_op.cc | 45 struct MultinomialFunctor { struct
57 extern template struct MultinomialFunctor<GPUDevice, Eigen::half, int32>;
58 extern template struct MultinomialFunctor<GPUDevice, float, int32>;
59 extern template struct MultinomialFunctor<GPUDevice, double, int32>;
60 extern template struct MultinomialFunctor<GPUDevice, int32, int32>;
61 extern template struct MultinomialFunctor<GPUDevice, int64, int32>;
63 extern template struct MultinomialFunctor<GPUDevice, Eigen::half, int64>;
64 extern template struct MultinomialFunctor<GPUDevice, float, int64>;
65 extern template struct MultinomialFunctor<GPUDevice, double, int64>;
66 extern template struct MultinomialFunctor<GPUDevice, int32, int64>;
[all …]
|
| D | multinomial_op.h | 25 struct MultinomialFunctor;
|