Home
last modified time | relevance | path

Searched refs:enableRecordFunction (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/torch/csrc/api/include/torch/
Dutils.h106 using at::enableRecordFunction;
/external/pytorch/aten/src/ATen/
Drecord_function.h720 TORCH_API void enableRecordFunction(bool enable = true);
732 enableRecordFunction(is_enabled);
736 enableRecordFunction(prev_value_);
Drecord_function.cpp693 void enableRecordFunction(bool enable) { in enableRecordFunction() function
/external/pytorch/binaries/
Drecord_function_benchmark.cc88 at::enableRecordFunction(); in main()
/external/pytorch/torch/csrc/jit/mobile/
Dinterpreter.cpp119 enableRecordFunction(true); in run()
369 enableRecordFunction(false); in run()
/external/pytorch/docs/source/notes/
Dlarge_scale_deployments.rst66 // call enableRecordFunction() in the thread before running a model
/external/pytorch/torch/csrc/autograd/
Dinit.cpp401 at::enableRecordFunction(enable); in THPAutograd_initExtension()