Searched refs:vocab_count (Results 1 – 6 of 6) sorted by relevance
248 std::vector<int32> vocab_count; in NegTrainOp() local249 OP_REQUIRES_OK(ctx, ctx->GetAttr("vocab_count", &vocab_count)); in NegTrainOp()252 vocab_weights.reserve(vocab_count.size()); in NegTrainOp()253 for (const auto& f : vocab_count) { in NegTrainOp()
28 name: "vocab_count"
18814 name: "vocab_count"18847 name: "vocab_count"18881 name: "vocab_count"
41838 name: "vocab_count"
41341 name: "vocab_count"
20748 name: "vocab_count"