| /external/tensorflow/tensorflow/python/tpu/profiler/ |
| D | capture_tpu_profile.py | 114 def monitoring_helper(service_addr, duration_ms, monitoring_level, argument
132 res = profiler_client.monitor(service_addr, duration_ms, monitoring_level,
150 if not FLAGS.service_addr and not FLAGS.tpu:
154 if FLAGS.service_addr:
158 service_addr = FLAGS.service_addr
165 service_addr = tpu_cluster_resolver.get_master()
170 service_addr = service_addr.replace('grpc://', '').replace(':8470', ':8466')
183 print('Since monitoring level is provided, profile', service_addr, ' for ',
186 monitoring_helper(service_addr, duration_ms, FLAGS.monitoring_level,
196 profiler_client.start_tracing(service_addr,
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | profiler_client.py | 27 def start_tracing(service_addr, argument
51 service_addr, logdir, worker_list, include_dataset_ops, duration_ms,
56 def monitor(service_addr, argument
75 pywrap_tfe.TFE_ProfilerClientMonitor(service_addr, duration_ms,
|
| /external/tensorflow/tensorflow/core/profiler/rpc/client/ |
| D | capture_profile.cc | 79 Status Profile(const string& service_addr, const string& logdir, in Profile() argument
93 "dns:///" + service_addr, ::grpc::InsecureChannelCredentials(), in Profile()
120 Status NewSession(const string& service_addr, in NewSession() argument
145 "dns:///" + service_addr, ::grpc::InsecureChannelCredentials(), in NewSession()
203 Status Trace(const string& service_addr, const string& logdir, in Trace() argument
225 status = Profile(service_addr, logdir, duration_ms, repository_root, in Trace()
228 string tpu_master = service_addr; in Trace()
248 Status Monitor(const string& service_addr, int duration_ms, in Monitor() argument
259 "dns:///" + service_addr, ::grpc::InsecureChannelCredentials(), in Monitor()
|
| D | capture_profile.h | 30 Status Monitor(const string& service_addr, int duration_ms,
36 Status Trace(const string& service_addr, const string& logdir,
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 85 bool TFE_ProfilerClientStartTracing(const char* service_addr, in TFE_ProfilerClientStartTracing() argument
91 tensorflow::profiler::ValidateHostPortPair(service_addr); in TFE_ProfilerClientStartTracing()
96 s = tensorflow::profiler::Trace(service_addr, logdir, worker_list, in TFE_ProfilerClientStartTracing()
103 void TFE_ProfilerClientMonitor(const char* service_addr, int duration_ms, in TFE_ProfilerClientMonitor() argument
107 tensorflow::profiler::ValidateHostPortPair(service_addr); in TFE_ProfilerClientMonitor()
113 s = tensorflow::profiler::Monitor(service_addr, duration_ms, monitoring_level, in TFE_ProfilerClientMonitor()
|
| D | c_api_experimental.h | 83 const char* service_addr, const char* logdir, const char* worker_list,
94 const char* service_addr, int duration_ms, int monitoring_level,
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 514 [](const char* service_addr, const char* logdir, const char* worker_list, in PYBIND11_MODULE()
519 service_addr, logdir, worker_list, include_dataset_ops, duration_ms, in PYBIND11_MODULE()
525 [](const char* service_addr, int duration_ms, int monitoring_level, in PYBIND11_MODULE() argument
529 TFE_ProfilerClientMonitor(service_addr, duration_ms, monitoring_level, in PYBIND11_MODULE()
|