Home
last modified time | relevance | path

Searched refs:nested_int (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DNestedIntSymNodeImpl.cpp10 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()
H A DNestedIntSymNodeImpl.h137 std::optional<int64_t> nested_int() override { in nested_int() function
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dmeta_utils.py338 nested_int=(
339 _tensor_symint_registry[t].node.nested_int()
478 nested_int: Optional[int] = None variable in MetaTensorDesc
1579 if t.nested_int is not None:
1581 nt_tensor_id=t.nested_int
/aosp_15_r20/external/pytorch/test/cpp/api/
H A DCMakeLists.txt45 ${TORCH_API_TEST_DIR}/nested_int.cpp
/aosp_15_r20/external/pytorch/c10/core/
H A DSymNodeImpl.h214 virtual std::optional<int64_t> nested_int() { in nested_int() function
/aosp_15_r20/external/pytorch/torch/fx/experimental/
H A Dsym_node.py516 def nested_int(self): member in SymNode
H A Dsymbolic_shapes.py3501 …self.var_to_val[sympy_expr] = SingletonInt(val.node.nested_int(), coeff=val.node.nested_int_coeff(…
/aosp_15_r20/external/pytorch/torch/
H A D__init__.py527 return hash(self.node.nested_int())
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp1308 return node->nested_int(); in initJITBindings()
/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py3645 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))