Method RunPass

tensorflow/core/graph/mkl_tfconversion_pass.cc:301–415

}

bool MklToTfConversionPass::RunPass(std::unique_ptr<Graph>* g) {
  bool result = false;

  CHECK_NOTNULL(g);

  DumpGraph("Before MklToTfConversionPass", &**g);

  // Since we are looking for an OneDNN-supported op node immediately
  // followed by a non-OneDNN op node, we will just iterate over edge
  // set of the graph.
  // edge set whose source and destination are candidates for
  // inserting conversion node
  std::vector<Edge*> candidate_edges;

  for (const Edge* e : (*g)->edges()) {
    Node* src = e->src();
    Node* dst = e->dst();

    // We skip control edges.
    if (e->IsControlEdge()) {
      continue;
    }

    // We skip adding MklToTf on an edge between X->MklToTf or
    // MklToTf->X, where X is any node.
    if (src->type_string().compare("_MklToTf") == 0 ||
        dst->type_string().compare("_MklToTf") == 0) {
      continue;
    }

    VLOG(1) << "MklToTfConversionPass: InsertConversionNodes: "
            << src->type_string() << " and " << dst->type_string();

    // Let's get source and destination data type.
    // We cannot check datatype on destination node because destination node
    // may not be OneDNN node.
    DataType src_datatype;
    DataType dst_datatype;
    bool src_is_mkl_op =
        (GetNodeAttr(src->def(), "T", &src_datatype) == Status::OK() &&
         IsMklSupportedOp(src->type_string(), src_datatype));
    bool dst_is_mkl_op =
        (GetNodeAttr(dst->def(), "T", &dst_datatype) == Status::OK() &&
         IsMklSupportedOp(dst->type_string(), dst_datatype));

    // Check if src with is OneDNN-compliant, while dst is not OneDNN-compliant.
    if (src_is_mkl_op && !dst_is_mkl_op) {
      VLOG(1) << "MklToTfConversionPass: Scheduled nodes " << src->name()
              << " and " << dst->name() << " for inserting conversion nodes";
      candidate_edges.push_back(const_cast<Edge*>(e));
    }
  }

  // Process all candidate edges and insert conversion nodes on them.
  for (Edge* e : candidate_edges) {
    // Even if we insert conversion node on a single edge, we
    // need to return true.
    string src_name = e->src()->name();