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