skipgram op

skipgram_definition.cc

REGISTER_OP(“Skipgram”)
 .Output(“vocab_word: string”)
 .Output(“vocab_freq: int32”)
 .Output(“words_per_epoch: int64”)
 .Output(“current_epoch: int32”)
 .Output(“total_words_processed: int64”)
 .Output(“examples: int32”)
 .Output(“labels: int32”)
 .Attr(“filename: string”)
 .Attr(“batch_size: int”)
 .Attr(“window_size: int = 5”)
 .Attr(“min_count: int = 5”)
 .Attr(“subsample: float = 1e-3”)
 .Doc(R”doc(
Parses a text file and creates a batch of examples.
... more documentation here ...
…
)doc”);