Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/autograd/engine/
H A Ddist_engine.cpp224 bool will_use_accelerator = false; in computeDependencies() local
233 if (!will_use_accelerator) { in computeDependencies()
234 will_use_accelerator = fn->stream().has_value(); in computeDependencies()
273 if (will_use_accelerator) { in computeDependencies()
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dengine.cpp1156 bool will_use_accelerator = false; in compute_dependencies() local
1167 if (!will_use_accelerator) { in compute_dependencies()
1168 will_use_accelerator = fn->stream().has_value(); in compute_dependencies()
1180 if (will_use_accelerator) { in compute_dependencies()