HomeSort by relevance Sort by last modified time
    Searched defs:MultinomialFunctor (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
multinomial_op_gpu.cu.cc 55 struct MultinomialFunctor<GPUDevice, T, OutputType> {
116 template struct MultinomialFunctor<GPUDevice, Eigen::half, int32>;
117 template struct MultinomialFunctor<GPUDevice, float, int32>;
118 template struct MultinomialFunctor<GPUDevice, double, int32>;
119 template struct MultinomialFunctor<GPUDevice, int32, int32>;
120 template struct MultinomialFunctor<GPUDevice, int64, int32>;
122 template struct MultinomialFunctor<GPUDevice, Eigen::half, int64>;
123 template struct MultinomialFunctor<GPUDevice, float, int64>;
124 template struct MultinomialFunctor<GPUDevice, double, int64>;
125 template struct MultinomialFunctor<GPUDevice, int32, int64>
    [all...]
multinomial_op.cc 44 struct MultinomialFunctor {
56 struct MultinomialFunctor<CPUDevice, T, OutputType> {
198 functor::MultinomialFunctor<Device, T, OutputType>()(

Completed in 59 milliseconds