• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2023-2024 Tomeu Vizoso <tomeu@tomeuvizoso.net>
3  * SPDX-License-Identifier: MIT
4  */
5 
6 #include "etnaviv_ml.h"
7 
8 void
9 etna_ml_lower_transpose(struct etna_ml_subgraph *subgraph,
10                         const struct pipe_tensor *input_tensor,
11                         struct etna_operation *operation,
12                         unsigned *output_tensor);
13 
14 void
15 etna_ml_lower_detranspose(struct etna_ml_subgraph *subgraph,
16                           struct etna_operation *convolution,
17                           struct etna_operation *operation);
18 
19 void
20 etna_ml_lower_reshuffle(struct etna_ml_subgraph *subgraph,
21                         const struct pipe_ml_operation *first_operation,
22                         struct etna_operation *operation,
23                         unsigned *output_tensor);
24 
25 void
26 etna_ml_lower_pad(struct etna_ml_subgraph *subgraph,
27                   const struct pipe_ml_operation *pad,
28                   struct etna_operation *operation);
29 
30 void
31 etna_ml_compile_operation_tp(struct etna_ml_subgraph *subgraph,
32                              const struct etna_operation *operation,
33                              struct etna_vip_instruction *instruction);
34 
35 void
36 etna_ml_emit_operation_tp(struct etna_ml_subgraph *subgraph,
37                           struct etna_vip_instruction *operation,
38                           unsigned idx);