Home
last modified time | relevance | path

Searched refs:current_cpu_scalars_on_non_cpu (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_subclasses/
H A Dfake_impls.py876 current_cpu_scalars_on_non_cpu = 0
882 if current_cpu_scalars_on_non_cpu >= max_cpu_scalars_on_non_cpu:
884 current_cpu_scalars_on_non_cpu += 1
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp469 int current_cpu_scalars_on_non_cpu = 0; in compute_types() local
496 TORCH_CHECK(current_cpu_scalars_on_non_cpu < max_cpu_scalars_on_non_cpu, in compute_types()
498 ++current_cpu_scalars_on_non_cpu; in compute_types()