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_def.
Gets the src node names and output indices for the incoming data edges of
node_def.
Gets the src node names and output indices for the incoming data edges of node_def .
std::vector<NodeDefBuilder::NodeOut> GetDataInputs(const NodeDef& node_def,
                                                   DataType dt) {
  std::vector<NodeDefBuilder::NodeOut> data_inputs;
  for (const std::string& input : node_def.input()) {
    const TensorId tensor_id = ParseTensorName(input);
    if (tensor_id.index() != Graph::kControlSlot) {
      data_inputs.emplace_back(tensor_id.node(), tensor_id.index(), dt);
    }
  }
  return data_inputs;
}
std :: vector < NodeDefBuilder :: NodeOut > GetDataInputs ( const NodeDef & node_def , DataType dt ) { std :: vector < NodeDefBuilder :: NodeOut > data_inputs ; for ( const std :: string & input : node_def . input ( ) ) { const TensorId tensor_id = ParseTensorName ( input ) ; if ( tensor_id . index ( ) != Graph :: kControlSlot ) { data_inputs . emplace_back ( tensor_id . node ( ) , tensor_id . index ( ) , dt ) ; } } return data_inputs ; }
Gets the src node names and output indices for the incoming data edges of
node_def.
Gets the src node names and output indices for the incoming data edges of node_def .