Python Inst2vecEncoder Exemples

Exemple #1

Fichier : inst2vec.py Projet : ChrisCummins/ProGraML

def main(argv):
    init_app(argv)
    encoder = inst2vec_encoder.Inst2vecEncoder()

    proto = ParseStdinOrDie(ProgramGraph())
    if FLAGS.ir:
        with open(FLAGS.ir) as f:
            ir = f.read()
    else:
        ir = None
    encoder.Encode(proto, ir)
    WriteStdout(proto)

Exemple #2

Fichier : inst2vec.py Projet : yibit/ProGraML

def Main():
  encoder = inst2vec_encoder.Inst2vecEncoder()

  if FLAGS.dataset:
    encoder.RunOnDataset(FLAGS.dataset)
    return

  if FLAGS.directory:
    encoder.RunOnDirectory(FLAGS.directory)
    return

  proto = ParseStdinOrDie(program_graph_pb2.ProgramGraph())
  ir = fs.Read(FLAGS.ir) if FLAGS.ir else None
  encoder.Encode(proto, ir)
  WriteStdout(proto)

Exemple #3

def main(argv):
    init_app(argv)
    encoder = inst2vec_encoder.Inst2vecEncoder()

    if FLAGS.dataset:
        encoder.RunOnDataset(Path(FLAGS.dataset))
        return

    if FLAGS.directory:
        encoder.RunOnDirectory(Path(FLAGS.directory))
        return

    proto = ParseStdinOrDie(program_graph_pb2.ProgramGraph())
    if FLAGS.ir:
        with open(FLAGS.ir) as f:
            ir = f.read()
    else:
        ir = None
    encoder.Encode(proto, ir)
    WriteStdout(proto)

Exemple #4

Fichier : inst2vec_encoder_test.py Projet : yibit/ProGraML

def encoder() -> inst2vec_encoder.Inst2vecEncoder:
  """A session-level fixture to re-use a graph encoder instance."""
  return inst2vec_encoder.Inst2vecEncoder()