Home
last modified time | relevance | path

Searched refs:_is_tensor_constructor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dtwo_tensor.py84 if torch._subclasses.fake_tensor._is_tensor_constructor(func):
/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py115 def _is_tensor_constructor(func: OpOverload): function
150 @register_op_impl(_is_tensor_constructor)
H A Dfake_tensor.py2411 _is_tensor_constructor,
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dpost_grad.py1218 if not torch._subclasses.fake_tensor._is_tensor_constructor(node.target):
/aosp_15_r20/external/pytorch/test/
H A Dtest_fake_tensor.py1218 if not torch._subclasses.fake_tensor._is_tensor_constructor(op):