Searched refs:nested_int (Results 1 – 10 of 10) sorted by relevance
10 std::optional<int64_t> c = rhs->nested_int(); in _eq()12 c.has_value() && lhs->nested_int() == *c && in _eq()16 if (auto mb_si = lhs->nested_int()) { in _ge()17 if (auto mb_si2 = rhs->nested_int()) { in _ge()28 } else if (rhs->nested_int()) { in _ge()70 TORCH_CHECK(!other->nested_int(), "nested int cannot be multiplied by nested int"); in mul()
137 std::optional<int64_t> nested_int() override { in nested_int() function
338 nested_int=(339 _tensor_symint_registry[t].node.nested_int()478 nested_int: Optional[int] = None variable in MetaTensorDesc1579 if t.nested_int is not None:1581 nt_tensor_id=t.nested_int
45 ${TORCH_API_TEST_DIR}/nested_int.cpp
214 virtual std::optional<int64_t> nested_int() { in nested_int() function
516 def nested_int(self): member in SymNode
3501 …self.var_to_val[sympy_expr] = SingletonInt(val.node.nested_int(), coeff=val.node.nested_int_coeff(…
527 return hash(self.node.nested_int())
1308 return node->nested_int(); in initJITBindings()
3645 nested_int = torch.nested._internal.nested_tensor.get_tensor_symint(3648 self.assertEqual(nt.size(), (3, nested_int, 3))3649 self.assertEqual(nt.shape, (3, nested_int, 3))