Searched refs:max_cpu_scalars_on_non_cpu (Results 1 – 2 of 2) sorted by relevance
877 max_cpu_scalars_on_non_cpu = 1 # hard coded atm882 if current_cpu_scalars_on_non_cpu >= max_cpu_scalars_on_non_cpu:
468 int max_cpu_scalars_on_non_cpu = config.allow_cpu_scalars_ ? 1 : 0; in compute_types() local496 TORCH_CHECK(current_cpu_scalars_on_non_cpu < max_cpu_scalars_on_non_cpu, in compute_types()