aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/c_api_experimental.cc
diff options
context:
space:
mode:
authorGravatar Richard Wei <rxwei@google.com>2018-04-06 11:47:44 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-06 11:50:31 -0700
commit76719deb3c12d84c4945a444e92493956530e165 (patch)
treead39a71292dd820dc3dbe42e0ed740b0b19cb2f5 /tensorflow/c/c_api_experimental.cc
parent98b8b786036172d33c85b6b5f81347440d0594df (diff)
Remove `TF_InitializeTPU` and `TF_ShutdownTPU` from experimental C API as they are no longer needed. Also remove a duplicate function declaration.
PiperOrigin-RevId: 191918408
Diffstat (limited to 'tensorflow/c/c_api_experimental.cc')
-rw-r--r--tensorflow/c/c_api_experimental.cc51
1 files changed, 0 insertions, 51 deletions
diff --git a/tensorflow/c/c_api_experimental.cc b/tensorflow/c/c_api_experimental.cc
index bea9378571..e82a546092 100644
--- a/tensorflow/c/c_api_experimental.cc
+++ b/tensorflow/c/c_api_experimental.cc
@@ -56,57 +56,6 @@ void TF_EnableXLACompilation(TF_SessionOptions* options, unsigned char enable) {
}
}
-void TF_InitializeTPU(TF_Session* session, TF_Status* status) {
- VLOG(1) << "Initializing TPU";
- TF_Operation* config_op =
- TF_GraphOperationByName(session->graph, "ConfigureDistributedTPU");
- if (config_op == nullptr) {
- status->status = tensorflow::errors::Internal(
- "Unable to find node ConfigureDistributedTPU in the TF graph.");
- return;
- }
-
- TF_Output config_node{config_op, 0};
-
- TF_Tensor* dummy_output;
- TF_SessionRun(session, /*run_options*/ nullptr,
- // input related parameters
- /*inputs*/ nullptr, /*input_values*/ nullptr, /*ninputs*/ 0,
- // output related parameters
- /*outputs*/ &config_node, /*output_values*/ &dummy_output,
- /*noutputs*/ 1,
- /*targets*/ nullptr, /*ntargets*/ 0,
- /*run_metadata*/ nullptr, status);
- if (status->status.ok()) {
- TF_DeleteTensor(dummy_output);
- }
-}
-
-void TF_ShutdownTPU(TF_Session* session, TF_Status* status) {
- {
- tensorflow::mutex_lock c(session->graph->mu);
- VLOG(1) << "Shutting down TPU, with input graph: "
- << session->graph->graph.ToGraphDefDebug().DebugString();
- }
-
- TF_Operation* shutdown_op =
- TF_GraphOperationByName(session->graph, "ShutdownDistributedTPU");
- if (shutdown_op == nullptr) {
- status->status = tensorflow::errors::Internal(
- "Unable to find node ShutdownDistributedTPU in the TF graph.");
- return;
- }
-
- TF_SessionRun(session, /*run_options*/ nullptr,
- // input related parameters
- /*inputs*/ nullptr, /*input_values*/ nullptr, /*ninputs*/ 0,
- // output related parameters
- /*outputs*/ nullptr, /*output_values*/ nullptr,
- /*noutputs*/ 0,
- /*targets*/ &shutdown_op, /*ntargets*/ 1,
- /*run_metadata*/ nullptr, status);
-}
-
const char* TF_GraphDebugString(TF_Graph* graph, size_t* len) {
tensorflow::mutex_lock c(graph->mu);
const auto& debug_str = graph->graph.ToGraphDefDebug().DebugString();