Function IsKvOps

serving/processor/framework/graph_optimizer.cc:755–759

}

bool IsKvOps(const Node* node) {
  return node->op_def().name() == "KvResourceGather" ||
         node->op_def().name() == "KvVarHandleOp" ||
         node->op_def().name() == "KvResourceImportV2";
}

Status GetInputNodesInfo(std::vector<SrcInfo>* input_info,
                         const Node* node) {