Searched refs:enableRecordFunction (Results 1 – 7 of 7) sorted by relevance
106 using at::enableRecordFunction;
720 TORCH_API void enableRecordFunction(bool enable = true);732 enableRecordFunction(is_enabled);736 enableRecordFunction(prev_value_);
693 void enableRecordFunction(bool enable) { in enableRecordFunction() function
88 at::enableRecordFunction(); in main()
119 enableRecordFunction(true); in run()369 enableRecordFunction(false); in run()
66 // call enableRecordFunction() in the thread before running a model
401 at::enableRecordFunction(enable); in THPAutograd_initExtension()