Function RemoveDevice

tensorflow/tools/graph_transforms/remove_device.cc:30–41

Clears the device field of all ops in the graph.

// Clears the device field of all ops in the graph.
Status RemoveDevice(const GraphDef& input_graph_def,
                    const TransformFuncContext& context,
                    GraphDef* output_graph_def) {
  output_graph_def->Clear();
  for (const NodeDef& node : input_graph_def.node()) {
    NodeDef* new_node = output_graph_def->mutable_node()->Add();
    *new_node = node;
    new_node->set_device("");
  }

  return Status::OK();
}

REGISTER_GRAPH_TRANSFORM("remove_device", RemoveDevice);