Home
last modified time | relevance | path

Searched defs:UnflattenOptions (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dlinear.cpp17 UnflattenOptions::UnflattenOptions(int64_t dim, std::vector<int64_t> sizes) in UnflattenOptions() function in torch::nn::UnflattenOptions
20 UnflattenOptions::UnflattenOptions(const char* dimname, namedshape_t namedshape) in UnflattenOptions() function in torch::nn::UnflattenOptions
25 UnflattenOptions::UnflattenOptions(std::string dimname, namedshape_t namedshape) in UnflattenOptions() function in torch::nn::UnflattenOptions
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dlinear.h54 struct TORCH_API UnflattenOptions { struct
57 UnflattenOptions(int64_t dim, std::vector<int64_t> sizes); argument