Home
last modified time | relevance | path

Searched refs:profiler_context (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dprofiler.py74 profiler_context = pywrap_tensorflow.TFE_NewProfilerContext()
77 profiler_context,
79 _profiler = pywrap_tensorflow.TFE_NewProfiler(profiler_context)
80 pywrap_tensorflow.TFE_DeleteProfilerContext(profiler_context)
162 profiler_context = pywrap_tensorflow.TFE_NewProfilerContext()
165 profiler_context,
167 pywrap_tensorflow.TFE_StartProfilerServer(profiler_context, port)
168 pywrap_tensorflow.TFE_DeleteProfilerContext(profiler_context)
/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental_test.cc44 TFE_ProfilerContext* profiler_context = TFE_NewProfilerContext(); in ExecuteWithProfiling() local
45 TFE_ProfilerContextSetEagerContext(profiler_context, ctx); in ExecuteWithProfiling()
46 TFE_Profiler* profiler = TFE_NewProfiler(profiler_context); in ExecuteWithProfiling()
49 TFE_DeleteProfilerContext(profiler_context); in ExecuteWithProfiling()
114 TFE_ProfilerContext* profiler_context = TFE_NewProfilerContext(); in TEST() local
115 TFE_ProfilerContextSetEagerContext(profiler_context, ctx); in TEST()
117 TFE_Profiler* profiler1 = TFE_NewProfiler(profiler_context); in TEST()
120 TFE_Profiler* profiler2 = TFE_NewProfiler(profiler_context); in TEST()
125 TFE_DeleteProfilerContext(profiler_context); in TEST()
Dc_api_experimental.cc58 void TFE_ProfilerContextSetEagerContext(TFE_ProfilerContext* profiler_context, in TFE_ProfilerContextSetEagerContext() argument
60 profiler_context->profiler_context.eager_context = &eager_context->context; in TFE_ProfilerContextSetEagerContext()
63 void TFE_DeleteProfilerContext(TFE_ProfilerContext* profiler_context) { in TFE_DeleteProfilerContext() argument
64 delete profiler_context; in TFE_DeleteProfilerContext()
70 tensorflow::StartProfilerServer(&context->profiler_context, port).release(); in TFE_StartProfilerServer()
Dc_api_experimental.h53 TFE_ProfilerContext* profiler_context, TFE_Context* eager_context);
57 TFE_ProfilerContext* profiler_context);
Dc_api_internal.h123 tensorflow::ProfilerContext profiler_context; member
128 profiler = tensorflow::ProfilerSession::Create(&ctx->profiler_context); in TFE_Profiler()
/external/tensorflow/tensorflow/core/profiler/rpc/
Dprofiler_server.cc27 ProfilerContext* const profiler_context, int32 port) { in StartProfilerServer() argument
28 Env* env = profiler_context->eager_context != nullptr in StartProfilerServer()
29 ? profiler_context->eager_context->TFEnv() in StartProfilerServer()
33 ProfilerContext ctx = *profiler_context; in StartProfilerServer()
Dprofiler_service_impl.cc28 explicit ProfilerServiceImpl(const ProfilerContext& profiler_context) in ProfilerServiceImpl() argument
29 : profiler_context_(profiler_context) {} in ProfilerServiceImpl()
71 const ProfilerContext& profiler_context) { in CreateProfilerService() argument
72 return MakeUnique<ProfilerServiceImpl>(profiler_context); in CreateProfilerService()
Dprofiler_service_impl.h28 const ProfilerContext& profiler_context);
Dprofiler_server.h23 ProfilerContext* const profiler_context, int32 port);