Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMaxPooling.h65 struct PoolingParams1D { struct
66 int64_t NB; // Number of batches
67 int64_t NC; // Number of channels
68 int64_t IW; // Input width
69 int64_t OW; // Output width
93 using pooling_fn = void (*)(Tensor&, const Tensor&, const PoolingParams1D&); argument