Searched refs:num_sampled_ (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | candidate_sampler_ops.cc | 36 OP_REQUIRES_OK(context, context->GetAttr("num_sampled", &num_sampled_)); in BaseCandidateSamplerOp()
55 OP_REQUIRES(context, num_sampled_ <= sampler_->range(), in Compute()
62 context->allocate_output(0, TensorShape({num_sampled_}), in Compute()
71 context->allocate_output(2, TensorShape({num_sampled_}), in Compute()
77 out_sampled_candidates->vec<int64>().data(), num_sampled_); in Compute()
82 out_sampled_expected_count->vec<float>().data(), num_sampled_); in Compute()
87 const int64 samples32 = 2048 * num_sampled_; in Compute()
106 int32 num_sampled_; member in tensorflow::BaseCandidateSamplerOp
|