tensorflow/core/tpu/graph_rewrite/tpu_embedding_software_deduplication_rewrite_pass.cc
// Gets the src node names and output indices for the incoming data edges of
// node.
Gets the src node names and output indices for the incoming data edges of
node.
Gets the src node names and output indices for the incoming data edges of node .
std::vector<NodeDefBuilder::NodeOut> GetDataInputs(const Node* node,
                                                   DataType dt) {
  std::vector<NodeDefBuilder::NodeOut> data_inputs;
  for (const Edge* edge : node->in_edges()) {
    if (!edge->IsControlEdge()) {
      data_inputs.emplace_back(edge->src()->name(), edge->src_output(), dt);
    }
  }
  return data_inputs;
}
std :: vector < NodeDefBuilder :: NodeOut > GetDataInputs ( const Node * node , DataType dt ) { std :: vector < NodeDefBuilder :: NodeOut > data_inputs ; for ( const Edge * edge : node -> in_edges ( ) ) { if ( ! edge -> IsControlEdge ( ) ) { data_inputs . emplace_back ( edge -> src ( ) -> name ( ) , edge -> src_output ( ) , dt ) ; } } return data_inputs ; }
Gets the src node names and output indices for the incoming data edges of
node.
Gets the src node names and output indices for the incoming data edges of node .