Searched refs:_PytorchRecordFunctionState (Results 1 – 3 of 3) sorted by relevance
4 struct _PytorchRecordFunctionState { struct7 _PytorchRecordFunctionState() : guard(at::RecordScope::FUNCTION) {} in _PytorchRecordFunctionState() argument10 _PytorchRecordFunctionState* _pytorch_record_function_enter(const char* name) { in _pytorch_record_function_enter()11 _PytorchRecordFunctionState* state = new _PytorchRecordFunctionState(); in _pytorch_record_function_enter()16 static inline _PytorchRecordFunctionState*20 _PytorchRecordFunctionState* state = new _PytorchRecordFunctionState(); in _pytorch_record_function_enter_with_kwinputs()26 _PytorchRecordFunctionState* _pytorch_record_function_enter_with_context( in _pytorch_record_function_enter_with_context()34 void _pytorch_record_function_exit(_PytorchRecordFunctionState* state) { in _pytorch_record_function_exit()
6 struct _PytorchRecordFunctionState;7 typedef struct _PytorchRecordFunctionState _PytorchRecordFunctionState; typedef9 _PytorchRecordFunctionState* _pytorch_record_function_enter(const char* name);10 _PytorchRecordFunctionState* _pytorch_record_function_enter_with_context(13 void _pytorch_record_function_exit(_PytorchRecordFunctionState* state);
489 …_PytorchRecordFunctionState* rf = _pytorch_record_function_enter_with_context("Torch-Compiled Regi… in eval_custom_code()617 _PytorchRecordFunctionState* rf = _pytorch_record_function_enter(cache_lookup_profiler_str); in _custom_eval_frame()646 _PytorchRecordFunctionState* rf = _pytorch_record_function_enter(cache_lookup_profiler_str); in _custom_eval_frame()