| 55 | } |
| 56 | |
| 57 | TensorDescriptor PermuteLayerNode::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 output_desc = src->desc(); |
| 66 | permute(output_desc.shape, _perm); |
| 67 | if (_layout != DataLayout::UNKNOWN) |
| 68 | { |
| 69 | output_desc.layout = _layout; |
| 70 | } |
| 71 | |
| 72 | return output_desc; |
| 73 | } |
| 74 | |
| 75 | NodeType PermuteLayerNode::type() const |
| 76 | { |