| 55 | } |
| 56 | |
| 57 | TensorDescriptor SoftmaxLayerNode::configure_output(size_t idx) const |
| 58 | { |
| 59 | ARM_COMPUTE_UNUSED(idx); |
| 60 | ARM_COMPUTE_ERROR_ON(idx >= _outputs.size()); |
| 61 | |
| 62 | const Tensor *src = input(0); |
| 63 | ARM_COMPUTE_ERROR_ON(src == nullptr); |
| 64 | |
| 65 | TensorDescriptor out_desc = src->desc(); |
| 66 | out_desc.quant_info = get_softmax_output_quantization_info(out_desc.data_type, false); |
| 67 | |
| 68 | return out_desc; |
| 69 | } |
| 70 | |
| 71 | NodeType SoftmaxLayerNode::type() const |
| 72 | { |
nothing calls this directly
no test coverage detected