OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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