Searched refs:num_true_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | candidate_sampler_ops.cc | 37 OP_REQUIRES_OK(context, context->GetAttr("num_true", &num_true_)); in BaseCandidateSamplerOp()
48 context, true_classes.dim_size(1) == num_true_, in Compute()
51 true_classes.dim_size(1), " was: ", num_true_)); in Compute()
67 1, TensorShape({batch_size, num_true_}), in Compute()
75 batch_size * num_true_); in Compute()
80 batch_size * num_true_); in Compute()
105 int32 num_true_; member in tensorflow::BaseCandidateSamplerOp
194 OP_REQUIRES_OK(context, context->GetAttr("num_true", &num_true_)); in ComputeAccidentalHitsOp()
202 in_true_candidates_shape.dim_size(1) == num_true_, in Compute()
226 for (int64 j = 0; j < num_true_; ++j) { in Compute()
[all …]
|