Home
last modified time | relevance | path

Searched refs:SetMaxTuningIterations (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/tunable/
H A DTunable.h182 void SetMaxTuningIterations(int max_iter);
H A DTunable.cpp452 void TuningContext::SetMaxTuningIterations(int max_iter) { in SetMaxTuningIterations() function in at::cuda::tunable::TuningContext
H A DREADME.md143 | void SetMaxTuningIterations(int max_iter); | |
/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A DModule.cpp1566 at::cuda::tunable::getTuningContext()->SetMaxTuningIterations(iterations); in THCPModule_cuda_tunableop_set_max_tuning_iterations()