xref: /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/etnaviv_ml_tp.h (revision 6104692788411f58d303aa86923a9ff6ecaded22)
1 /*
2  * Copyright (c) 2023-2024 Tomeu Vizoso <[email protected]>
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_ml_operation *first_operation,
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_compile_operation_tp(struct etna_ml_subgraph *subgraph,
27                              const struct etna_operation *operation,
28                              struct etna_vip_instruction *instruction);
29 
30 void
31 etna_ml_emit_operation_tp(struct etna_ml_subgraph *subgraph,
32                           struct etna_vip_instruction *operation,
33                           unsigned idx);