Searched defs:namedshape (Results 1 – 5 of 5) sorted by relevance
20 UnflattenOptions::UnflattenOptions(const char* dimname, namedshape_t namedshape) in UnflattenOptions()25 UnflattenOptions::UnflattenOptions(std::string dimname, namedshape_t namedshape) in UnflattenOptions()
90 auto namedshape = options.namedshape(); in pretty_print() local114 auto namedshape = options.namedshape(); in forward() local
140 UnflattenImpl(std::string dimname, UnflattenOptions::namedshape_t namedshape) in UnflattenImpl()
28 def unzip_namedshape(namedshape): argument
58 def create(namedshape, factory=torch.randn): argument