Home
last modified time | relevance | path

Searched refs:nested_int_memo (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_tensor.py617 nested_int_memo = SymIntMemoDescriptor(is_nested_int=True) variable in FakeTensor
715 self.nested_int_memo = None
891 if self.nested_int_memo is None:
892 self.nested_int_memo = self.fake_mode.create_symbolic_nested_int(
895 return self.nested_int_memo * coeff
H A Dmeta_utils.py1580 r.nested_int_memo = r.fake_mode.create_symbolic_nested_int(
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dnested_tensor.py273 ragged_source.nested_int_memo = ragged_size
H A Dops.py529 tgt.nested_int_memo = src.nested_int_memo