Home
last modified time | relevance | path

Searched defs:same_dtype (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/
H A Dtensor_ops.py77 def same_dtype(*args, **kwargs): function
/aosp_15_r20/external/pytorch/torch/_inductor/fx_passes/
H A Dfreezing_patterns.py209 def same_dtype(match): function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DCopy.cu267 bool same_dtype = iter.dtype(0) == iter.dtype(1); in copy_requires_temporaries() local
/aosp_15_r20/external/pytorch/test/
H A Dtest_tensor_creation_ops.py3834 …def _check(self, original, cvt=lambda t: t, is_alias=True, same_dtype=True, same_device=True, **kw… argument