Searched defs:max_skips (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/text/kernels/
H A Dskip_gram_kernels.cc47 OP_REQUIRES_OK(context, context->input("max_skips", &max_skips_tensor));
48 const int max_skips = *(max_skips_tensor->scalar<int>().data()); variable
51 context, min_skips >= 0 && max_skips >= 0,
52 errors::InvalidArgument("Both min_skips and max_skips must be >= 0."));
53 OP_REQUIRES(context, min_skips <= max_skips,
54 errors::InvalidArgument("min_skips must be <= max_skips."));
84 const int skips = min_skips + rng.Uniform(max_skips - min_skips + 1);

Completed in 163 milliseconds