Function FindDicePattern

tensorflow/core/grappler/optimizers/dice_fusion.cc:225–258

}

bool FindDicePattern(const utils::MutableGraphView& graph_view, int node_index,
                     DicePattern* matched) {
  DicePattern pattern;
  std::vector<int> outs;
  auto* node_view = graph_view.GetNode(node_index);
  if (node_view->NumControllingFanins() > 0 ||
      node_view->NumControlledFanouts() > 0)
    return false;

  pattern.sub_1_id = node_index;
  // Match sub_1, get input index and mul_1 index
  if (!IsNodeViewMatchSub_1(graph_view, node_index, &pattern)) return false;
  // Match mul, get constant input index and sigmoid index
  if (!IsNodeViewMatchMul_1(graph_view, pattern.mul_1_id, &pattern))
    return false;
  // Match sigmoid, get sub_2 & mul_4 index
  if (!IsNodeViewMatchSigmoid(graph_view, pattern.sigmoid_id, &pattern))
    return false;
  // Match sub_2, get constant input index
  if (!IsNodeViewMatchSub_2(graph_view, pattern.sub_2_id, &pattern))
    return false;
  // Match mul_2, get mul_3 index and check input index
  if (!IsNodeViewMatchMul_2(graph_view, pattern.mul_2_id, &pattern))
    return false;
  // Match mul_3, get add index and input index
  if (!IsNodeViewMatchMul_3(graph_view, pattern.mul_3_id, &pattern))
    return false;
  // Match mul_4, check add and input index
  if (!IsNodeViewMatchMul_4(graph_view, pattern.mul_4_id, &pattern))
    return false;

  *matched = pattern;
  return true;
}

string get_node_by_tensor(string tensor_name) {
  auto position = tensor_name.find(":");