Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dextension.cpp11 struct MatrixMultiplier { struct
12 MatrixMultiplier(int A, int B) { in MatrixMultiplier() function
16 torch::Tensor forward(torch::Tensor weights) { in forward()
19 torch::Tensor get() const { in get()
45 py::class_<MatrixMultiplier>(m, "MatrixMultiplier") in PYBIND11_MODULE() argument