Searched defs:RecordFunction (Results 1 – 6 of 6) sorted by relevance
60 [](const at::RecordFunction&) -> std::unique_ptr<at::ObserverContext> { in TEST()64 auto end_callback = [](const at::RecordFunction&, at::ObserverContext*) { in TEST()164 -> std::unique_ptr<at::ObserverContext> { return nullptr; }, in TEST()165 [](const at::RecordFunction&, at::ObserverContext*) { ++call_count; }) in TEST()
24 [](const at::RecordFunction&) -> std::unique_ptr<at::ObserverContext> { return nullptr; }) { in addTestCallback()27 [](const at::RecordFunction&, at::ObserverContext*) {}) in __anon0260e61d0302()111 [](const at::RecordFunction&) -> std::unique_ptr<at::ObserverContext> { in main()
529 RecordFunction::RecordFunction(RecordScope scope) in RecordFunction() function in at::RecordFunction532 RecordFunction::RecordFunction(StepCallbacks&& step_callbacks) in RecordFunction() function in at::RecordFunction
61 [](const at::RecordFunction&, at::ObserverContext*) { in pushITTCallbacks()
1156 /* unused */) -> std::unique_ptr<at::ObserverContext> { in TEST()1163 [](const RecordFunction& /* unused */, ObserverContext* ctx_ptr) { in TEST()1181 /* unused */) -> std::unique_ptr<at::ObserverContext> { in TEST()1188 [](const RecordFunction& /* unused */, ObserverContext* ctx_ptr) { in TEST()1373 [](const RecordFunction&) -> std::unique_ptr<at::ObserverContext> { in TEST()
72 static void onFunctionExit(const at::RecordFunction&, at::ObserverContext*) { in onFunctionExit()