Searched defs:KronImpl (Results 1 – 1 of 1) sorted by relevance
3366 struct KronImpl final { struct3368 explicit KronImpl(const Tensor& self, const Tensor& other) { in KronImpl() argument3386 Tensor& kron_out(Tensor& result) const { in kron_out()3401 Tensor kron() const { in kron()3405 int64_t maxdim;3406 Tensor self_view;3407 Tensor other_view;3408 c10::SmallVector<int64_t, 10> result_reshape;3409 c10::SmallVector<int64_t, 10> a_reshape;3410 c10::SmallVector<int64_t, 10> b_reshape;