Home
last modified time | relevance | path

Searched refs:TFE_Profiler (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental.h45 typedef struct TFE_Profiler TFE_Profiler; typedef
47 TF_CAPI_EXPORT extern TFE_Profiler* TFE_NewProfiler();
48 TF_CAPI_EXPORT extern bool TFE_ProfilerIsOk(TFE_Profiler* profiler);
49 TF_CAPI_EXPORT extern void TFE_DeleteProfiler(TFE_Profiler* profiler);
53 TF_CAPI_EXPORT extern void TFE_ProfilerSerializeToString(TFE_Profiler* profiler,
Dc_api_experimental.cc50 TFE_Profiler* TFE_NewProfiler() { return new TFE_Profiler(); } in TFE_NewProfiler()
52 bool TFE_ProfilerIsOk(TFE_Profiler* profiler) { in TFE_ProfilerIsOk()
56 void TFE_DeleteProfiler(TFE_Profiler* profiler) { delete profiler; } in TFE_DeleteProfiler()
58 void TFE_ProfilerSerializeToString(TFE_Profiler* profiler, TF_Buffer* buf, in TFE_ProfilerSerializeToString()
Dc_api_internal.h96 struct TFE_Profiler { struct
97 explicit TFE_Profiler() { profiler = tensorflow::ProfilerSession::Create(); } in TFE_Profiler() argument
Dc_api_experimental_test.cc47 TFE_Profiler* profiler = TFE_NewProfiler(); in ExecuteWithProfiling()
114 TFE_Profiler* profiler1 = TFE_NewProfiler(); in TEST()
117 TFE_Profiler* profiler2 = TFE_NewProfiler(); in TEST()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc45 PYBIND11_MAKE_OPAQUE(TFE_Profiler);
321 py::class_<TFE_Profiler> TFE_Profiler_class(m, "TFE_Profiler"); in PYBIND11_MODULE()
505 [](TFE_Profiler& profiler, TF_Buffer& buf) { in PYBIND11_MODULE()