Searched defs:ScalarTypeClass (Results 1 – 2 of 2) sorted by relevance
7 struct ScalarTypeClass : public torch::CustomClassHolder { struct8 ScalarTypeClass(at::ScalarType s) : scalar_type_(s) {} in ScalarTypeClass() argument9 at::ScalarType scalar_type_;
48 TEST(CustomClassTest, ScalarTypeClass) { in TEST() argument