Home | History | Annotate | Download | only in ops

Lines Matching refs:num_classes

116                               num_classes,
175 weights: A `Tensor` or `PartitionedVariable` of shape `[num_classes, dim]`,
177 has shape [num_classes, dim]. The (possibly-sharded) class embeddings.
178 biases: A `Tensor` or `PartitionedVariable` of shape `[num_classes]`.
189 num_classes: An `int`. The number of possible classes.
209 if num_sampled > num_classes:
210 raise ValueError("num_sampled ({}) cannot be greater than num_classes ({})".
211 format(num_sampled, num_classes))
227 range_max=num_classes)
239 num_classes=num_classes,
252 num_classes,
294 weights: A `Tensor` of shape `[num_classes, dim]`, or a list of `Tensor`
296 [num_classes, dim]. The (possibly-sharded) class embeddings.
297 biases: A `Tensor` of shape `[num_classes]`. The class biases.
305 num_classes: An `int`. The number of possible classes.
327 num_classes=num_classes,