commit d28d569c0e1cd92e407cb1b882db034277d47928
author Michael K. Sanders <mks@google.com> Thu Jan 10 10:59:00 2019 -0800
committer android-build-merger <android-build-merger@google.com> Thu Jan 10 10:59:00 2019 -0800
tree 135f4238cdb6e9c389bfe5546961da664a475bac
parent e52aa140b83a9f121c81b31d6ae22039e6b832ba
parent 41e6732a66856cb9a6f36ca71bec3ebcf1a2835c

Adds float16 support for RANDOM_MULTINOMIAL.
am: 41e6732a66

Change-Id: I57c83709d1ea2ea9a2fd31223e62cbc043ec7126

neuralnetworks/1.2/vts/functional/ValidateModel.cpp

diff --git a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
index 6c3b7b8..7fccac3 100644
--- a/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
+++ b/neuralnetworks/1.2/vts/functional/ValidateModel.cpp
@@ -325,6 +325,7 @@
         // - ARGMIN and ARGMAX's first argument can be any of
         // TENSOR_(FLOAT16|FLOAT32|INT32|QUANT8_ASYMM).
         // - CAST's argument can be any of TENSOR_(FLOAT16|FLOAT32|INT32|QUANT8_ASYMM).
+        // - RANDOM_MULTINOMIAL's argument can be either TENSOR_FLOAT16 or TENSOR_FLOAT32.
         switch (operation.type) {
             case OperationType::LSH_PROJECTION: {
                 if (operand == operation.inputs[1]) {
@@ -339,6 +340,11 @@
                     return true;
                 }
             } break;
+            case OperationType::RANDOM_MULTINOMIAL: {
+                if (type == OperandType::TENSOR_FLOAT16 || type == OperandType::TENSOR_FLOAT32) {
+                    return true;
+                }
+            } break;
             default:
                 break;
         }