Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_custom_class_registrations.cpp414 struct ContainsTensor : public torch::CustomClassHolder { struct
415 explicit ContainsTensor(at::Tensor t) : t_(t) {} in ContainsTensor() argument
417 at::Tensor get() { in get()
421 std::tuple<std::tuple<std::string, at::Tensor>> __obj_flatten__() { in __obj_flatten__()
425 at::Tensor t_;