xref: /aosp_15_r20/external/pytorch/tools/iwyu/aten.imp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1[
2    { include: ["@<ATen/cpu/vec/vec512/.*>", private, "<ATen/cpu/vec/vec.h>", public] },
3    { include: ["@<ATen/cpu/vec/vec256/.*>", private, "<ATen/cpu/vec/vec.h>", public] },
4    { include: ["<ATen/cpu/vec/vec_base.h>", private, "<ATen/cpu/vec/vec.h>", public] },
5    { include: ["<ATen/cpu/vec/functional_base.h>", private, "<ATen/cpu/vec/functional.h>", public] },
6    { include: ["<ATen/core/IListRef_inl.h>", private, "<ATen/core/IListRef.h>", public] },
7    { include: ["<ATen/cpu/vec/functional_bfloat16.h>", private, "<ATen/cpu/vec/functional.h>", public] },
8
9    { include: ["<ATen/core/Dict_inl.h>", private, "<ATen/core/Dict.h>", public] },
10    { include: ["<ATen/core/ivalue_inl.h>", private, "<ATen/core/ivalue.h>", public] },
11    { include: ["<ATen/core/DimVector.h>", public, "<ATen/DimVector.h>", public] },
12    { include: ["<ATen/core/Dimname.h>", public, "<ATen/Dimname.h>", public] },
13
14    { symbol: ["at::Tensor", private, "<ATen/core/Tensor.h>", public] },
15
16    { include: ["<ATen/Parallel-inl.h>", private, "<ATen/Parallel.h>", public] },
17    { include: ["<ATen/ParallelOpenMP.h>", private, "<ATen/Parallel.h>", public] },
18    { include: ["<ATen/ParallelNative.h>", private, "<ATen/Parallel.h>", public] },
19    { include: ["<ATen/ParallelNativeTBB.h>", private, "<ATen/Parallel.h>", public] },
20
21]
22