diff options
Diffstat (limited to 'src/graph/nodes/SoftmaxLayerNode.cpp')
| -rw-r--r-- | src/graph/nodes/SoftmaxLayerNode.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/graph/nodes/SoftmaxLayerNode.cpp b/src/graph/nodes/SoftmaxLayerNode.cpp index 4c21ac6ad0..b6241e6654 100644 --- a/src/graph/nodes/SoftmaxLayerNode.cpp +++ b/src/graph/nodes/SoftmaxLayerNode.cpp @@ -63,7 +63,10 @@ TensorDescriptor SoftmaxLayerNode::configure_output(size_t idx) const const Tensor *src = input(0); ARM_COMPUTE_ERROR_ON(src == nullptr); - return src->desc(); + TensorDescriptor out_desc = src->desc(); + out_desc.quant_info = QuantizationInfo(1.f / 256.f, 0); + + return out_desc; } Status SoftmaxLayerNode::validate() |