1 #pragma once 2 #ifdef __cplusplus 3 extern "C" { 4 #endif 5 6 struct _PytorchRecordFunctionState; 7 typedef struct _PytorchRecordFunctionState _PytorchRecordFunctionState; 8 9 _PytorchRecordFunctionState* _pytorch_record_function_enter(const char* name); 10 _PytorchRecordFunctionState* _pytorch_record_function_enter_with_context( 11 const char* name, 12 const char* context); 13 void _pytorch_record_function_exit(_PytorchRecordFunctionState* state); 14 15 #ifdef __cplusplus 16 } // extern "C" 17 #endif 18