Searched refs:TraceMeWrapper (Results 1 – 3 of 3) sorted by relevance
29 class TraceMeWrapper : public tensorflow::profiler::TraceMeWrapper { class31 using tensorflow::profiler::TraceMeWrapper::TraceMeWrapper;50 py::class_<TraceMeWrapper> traceme_class(profiler, "TraceMe", in BuildProfilerSubmodule()58 py::cast<TraceMeWrapper*>(self)->Stop(); in BuildProfilerSubmodule()61 .def("set_metadata", &TraceMeWrapper::SetMetadata) in BuildProfilerSubmodule()62 .def_static("is_enabled", &TraceMeWrapper::IsEnabled); in BuildProfilerSubmodule()
23 using ::tensorflow::profiler::TraceMeWrapper;26 py::class_<TraceMeWrapper>(m, "TraceMe", py::module_local()) in PYBIND11_MODULE()28 .def("SetMetadata", &TraceMeWrapper::SetMetadata) in PYBIND11_MODULE()29 .def("Stop", &TraceMeWrapper::Stop); in PYBIND11_MODULE()
32 class TraceMeWrapper {36 TraceMeWrapper(const pybind11::str& name, const pybind11::kwargs& kwargs) in TraceMeWrapper() function