Searched defs:UnflattenImpl (Results 1 – 3 of 3) sorted by relevance
138 UnflattenImpl(int64_t dim, std::vector<int64_t> sizes) in UnflattenImpl() function140 UnflattenImpl(std::string dimname, UnflattenOptions::namedshape_t namedshape) in UnflattenImpl() function
84 UnflattenImpl::UnflattenImpl(UnflattenOptions options_) in UnflattenImpl() function in torch::nn::UnflattenImpl
239 py::object PyTreeDef::UnflattenImpl(T leaves) const { in UnflattenImpl() function in xla::PyTreeDef