Searched refs:convolution_batch_rule (Results 1 – 1 of 1) sorted by relevance
20 convolution_batch_rule(const Tensor& lhs, std::optional<int64_t> lhs_bdim, const Tensor& rhs, std::… in convolution_batch_rule() function459 VMAP_SUPPORT(convolution, convolution_batch_rule); in TORCH_LIBRARY_IMPL()