diff options
Diffstat (limited to 'src/graph')
| -rw-r--r-- | src/graph/nodes/GenerateProposalsLayerNode.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/graph/nodes/GenerateProposalsLayerNode.cpp b/src/graph/nodes/GenerateProposalsLayerNode.cpp index f5a3c02dd5..7367e80539 100644 --- a/src/graph/nodes/GenerateProposalsLayerNode.cpp +++ b/src/graph/nodes/GenerateProposalsLayerNode.cpp @@ -80,7 +80,8 @@ TensorDescriptor GenerateProposalsLayerNode::configure_output(size_t idx) const break; case 2: // Configure num_valid_proposals - output_desc.shape = TensorShape(1); + output_desc.shape = TensorShape(1); + output_desc.data_type = DataType::U32; break; default: ARM_COMPUTE_ERROR("Unsupported output index"); |