Home
last modified time | relevance | path

Searched defs:dimname (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dlinear.cpp20 UnflattenOptions::UnflattenOptions(const char* dimname, namedshape_t namedshape) in UnflattenOptions()
25 UnflattenOptions::UnflattenOptions(std::string dimname, namedshape_t namedshape) in UnflattenOptions()
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A DDimname_test.cpp40 auto dimname = Dimname::fromSymbol(foo); in TEST() local
50 const std::string& dimname, in check_unify_and_match()
/aosp_15_r20/external/pytorch/torch/csrc/
H A Dpython_dimname.cpp49 void InternedStringsTable::addMapping(PyObject* obj, at::Dimname dimname) { in addMapping()
107 auto dimname = at::Dimname::fromSymbol(at::Symbol::dimname(name)); in THPDimname_parse() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DDimname.cpp9 std::ostream& operator<<(std::ostream& out, const Dimname& dimname) { in operator <<()
H A Dsymbol.h135 inline Symbol Symbol::dimname(const std::string & s) { return Symbol::fromQualString("dimname::" + … in dimname() function
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dlinear.h140 UnflattenImpl(std::string dimname, UnflattenOptions::namedshape_t namedshape) in UnflattenImpl()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dlinear.cpp116 auto dimname = in forward() local
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_arg_parser.h852 inline at::Dimname PythonArgs::dimname(int i) { in dimname() function
/aosp_15_r20/external/mesa3d/src/amd/llvm/
H A Dac_llvm_build.c1895 const char *dimname; in ac_build_image_opcode() local