diff options
| author |  Peter Hawkins <phawkins@google.com> | 2017-11-30 06:13:53 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-11-30 06:17:34 -0800 |
| commit | 8a98563eb6d552f0bd0931f83837640481c1f938 (patch) | |
| tree | fa06a07463eda1dfe16465fb9d722565a451c675 /tensorflow/core/kernels/multinomial_op.h | |
| parent | ad3213bc53d9905c788509948412ad9703fa976b (diff) | |
Add support for int32 output types to the Multinomial op.
PiperOrigin-RevId: 177444775
Diffstat (limited to 'tensorflow/core/kernels/multinomial_op.h')
| -rw-r--r-- | tensorflow/core/kernels/multinomial_op.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/multinomial_op.h b/tensorflow/core/kernels/multinomial_op.h index af5e81f219..6e41060aa4 100644 --- a/tensorflow/core/kernels/multinomial_op.h +++ b/tensorflow/core/kernels/multinomial_op.h @@ -21,7 +21,7 @@ namespace tensorflow { namespace functor { // Generic helper functor for the Multinomial Op. -template <typename Device, typename T> +template <typename Device, typename T, typename OutputType> struct MultinomialFunctor; } // namespace functor |