Home
last modified time | relevance | path

Searched defs:DimConv2DKernel (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/impl/
H A DCommon.h34 struct DimConv2DKernel { struct
35 static constexpr uint32_t Width = 1u;
36 static constexpr uint32_t Height = 2u;
37 static constexpr uint32_t InChannels = 3u;
38 static constexpr uint32_t OutChannels = 4u;
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/
H A DDimUtils.h49 struct DimConv2DKernel { struct
50 static constexpr uint32_t Width = 1u;
51 static constexpr uint32_t Height = 2u;
52 static constexpr uint32_t InChannels = 3u;
53 static constexpr uint32_t OutChannels = 4u;