pytorch-inference
|
This is the complete list of members for pytorch::AvgPool2d, including all inherited members.
AvgPool2d(const pooling_params_t ¶ms) | pytorch::AvgPool2d | inline |
forward(const std::vector< tensor > &input) | pytorch::AvgPool2d | inlinevirtual |
operator()(const std::vector< tensor > &input) | pytorch::AvgPool2d | inlinevirtual |
params | pytorch::AvgPool2d | private |