Commit 356f360e authored by Yong Tang's avatar Yong Tang
Browse files

Add additional shape validation to `compute_accidental_hits`



In `compute_accidental_hits`, the `sampled_candidates` must
be a vector, as is shown in the kernel implementation in
`tensorflow/core/kernels/candidate_sampler_ops.cc`.

This fix adds shape validation of `sampled_candidates`
in the shape function whenever possible.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent f27033fb
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment