Home
last modified time | relevance | path

Searched full:scalaronly (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dufunc.py171 # Note [ScalarOnly and Generic must match names for CUDA]
174 # codegen, both ScalarOnly and Generic are defined, the ufunc name
180 for lk in [UfuncKey.ScalarOnly, UfuncKey.Generic]:
186 # See Note [ScalarOnly and Generic must match names for CUDA]
189 ), "ScalarOnly and Generic must have same ufunc name"
495 # Reindex the ufunc by dtypes; processing generic/scalaronly as well
503 if UfuncKey.ScalarOnly in loops and k is UfuncKey.CPUScalar:
504 lks.append(UfuncKey.ScalarOnly)
/aosp_15_r20/external/pytorch/tools/test/
H A Dtest_codegen_model.py115 ScalarOnly: binop (Bool)
128 ScalarOnly: binop_scalar (Bool)
134 ScalarOnly and Generic must have same ufunc name""",
H A Dtest_executorch_gen.py35 ScalarOnly: add (Bool)
90 ScalarOnly: add (Bool)
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py435 ScalarOnly = auto() # CUDA*, CPUScalar variable in UfuncKey
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml576 ScalarOnly: add (Bool)