diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-10 11:43:40 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-10 17:42:57 -0700 |
| commit | 32482dea01e2fc672eee9df5305bccc0b432eb7b (patch) | |
| tree | bc289321d33e24a64dfc842ab7eb3254a57087fe /tensorflow/contrib/hvx | |
| parent | b9cdcbe3a019f09eee7e7c9c9039e4533931e1e7 (diff) | |
Change soc op type id by input data types and add new hvx ops
PiperOrigin-RevId: 155655617
Diffstat (limited to 'tensorflow/contrib/hvx')
| -rw-r--r-- | tensorflow/contrib/hvx/hvx_ops_support_checker/hvx_ops_support_checker_main.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/hvx/hvx_ops_support_checker/hvx_ops_support_checker_main.cc b/tensorflow/contrib/hvx/hvx_ops_support_checker/hvx_ops_support_checker_main.cc index 03b4dcbfa8..3a219bb3e6 100644 --- a/tensorflow/contrib/hvx/hvx_ops_support_checker/hvx_ops_support_checker_main.cc +++ b/tensorflow/contrib/hvx/hvx_ops_support_checker/hvx_ops_support_checker_main.cc @@ -54,7 +54,8 @@ static void CheckOpsSupport(const GraphDef& graph_def) { std::unordered_set<string> unsupported_ops; bool all_supported = true; for (const NodeDef& node : graph_def.node()) { - const int op_id = ops_definition.GetOpIdFor(node.op()); + // TODO(satok): Set correct data type if it's given. + const int op_id = ops_definition.GetOpIdFor(node.op(), {}); if (op_id == IGraphTransferOpsDefinitions::INVALID_OP_ID) { all_supported = false; LOG(ERROR) << "OP type: " << node.op() << " is not supported on hvx. " |