diff --git a/tensorflow/c/eager/c_api_experimental.cc b/tensorflow/c/eager/c_api_experimental.cc index c84b4462325..15df113f902 100644 --- a/tensorflow/c/eager/c_api_experimental.cc +++ b/tensorflow/c/eager/c_api_experimental.cc @@ -85,10 +85,10 @@ void TFE_ContextDisableGraphCollection(TFE_Context* ctx) { } bool TFE_ProfilerClientStartTracing(const char* service_addr, - const char* logdir, TF_Status* status, - const char* worker_list, + const char* logdir, const char* worker_list, bool include_dataset_ops, int duration_ms, - int num_tracing_attempts) { + int num_tracing_attempts, + TF_Status* status) { tensorflow::Status s = tensorflow::profiler::client::ValidateHostPortPair(service_addr); if (!s.ok()) { diff --git a/tensorflow/c/eager/c_api_experimental.h b/tensorflow/c/eager/c_api_experimental.h index 6d2d2d5a142..2cf95d87eb5 100644 --- a/tensorflow/c/eager/c_api_experimental.h +++ b/tensorflow/c/eager/c_api_experimental.h @@ -84,9 +84,9 @@ TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx); // tensorflow/contrib/tpu/profiler/capture_tpu_profile instead following // https://cloud.google.com/tpu/docs/cloud-tpu-tools#capture_trace. TF_CAPI_EXPORT extern bool TFE_ProfilerClientStartTracing( - const char* service_addr, const char* logdir, TF_Status* status, - const char* worker_list, bool include_dataset_ops, int duration_ms, - int num_tracing_attempts); + const char* service_addr, const char* logdir, const char* worker_list, + bool include_dataset_ops, int duration_ms, int num_tracing_attempts, + TF_Status* status); // TODO(fishx): Move these monitoring APIs into a separate file. // -----------------------------------------------------------------------------