diff options
author | 2016-02-06 18:17:13 -0800 | |
---|---|---|
committer | 2016-02-06 21:54:10 -0800 | |
commit | 0a00cd3b41fe1c3632a6062e11936e7bc4b3308e (patch) | |
tree | bdef0a4a82f710bc35adb00d3ab3e168c2e17f8f | |
parent | 1115799825078f2037d60bc2dc0283bb40f309f4 (diff) |
Marks a few stateful ops stateful.
Change: 114039491
-rw-r--r-- | tensorflow/core/ops/ops.pbtxt | 2 | ||||
-rw-r--r-- | tensorflow/models/embedding/word2vec_ops.cc | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/ops/ops.pbtxt b/tensorflow/core/ops/ops.pbtxt index 6ef61172c1..bdf69877ae 100644 --- a/tensorflow/core/ops/ops.pbtxt +++ b/tensorflow/core/ops/ops.pbtxt @@ -4934,6 +4934,7 @@ op { description: "Number of negative samples per example." } summary: "Training via negative sampling." + is_stateful: true } op { name: "NextIteration" @@ -7857,6 +7858,7 @@ op { description: "Threshold for word occurrence. Words that appear with higher\nfrequency will be randomly down-sampled. Set to 0 to disable." } summary: "Parses a text file and creates a batch of examples." + is_stateful: true } op { name: "Slice" diff --git a/tensorflow/models/embedding/word2vec_ops.cc b/tensorflow/models/embedding/word2vec_ops.cc index 1c16da6fda..4086a3dde3 100644 --- a/tensorflow/models/embedding/word2vec_ops.cc +++ b/tensorflow/models/embedding/word2vec_ops.cc @@ -25,6 +25,7 @@ REGISTER_OP("Skipgram") .Output("total_words_processed: int64") .Output("examples: int32") .Output("labels: int32") + .SetIsStateful() .Attr("filename: string") .Attr("batch_size: int") .Attr("window_size: int = 5") @@ -55,6 +56,7 @@ REGISTER_OP("NegTrain") .Input("examples: int32") .Input("labels: int32") .Input("lr: float") + .SetIsStateful() .Attr("vocab_count: list(int)") .Attr("num_negative_samples: int") .Doc(R"doc( |