int type of numpy in windows default int32, need to set int64

test=develop
revert-15774-anakin_subgraph_engine
xuezhong 6 years ago
parent 9b24ac34dd
commit 9505850e33

@ -305,7 +305,8 @@ class TestSampleLogitsOpV2(OpTest):
out = sample_logits(self.inputs["Logits"], self.inputs["Label"],
self.attrs["num_samples"], self.attrs["seed"],
self.attrs["remove_accidental_hits"], True,
self.fetched_samples, self.probabilities)
self.fetched_samples.astype(np.int64),
self.probabilities)
self.outputs = {
'SampledLogits': out[0],
'Samples': out[1],
@ -365,7 +366,6 @@ class TestSampleLogitsOpV3(OpTest):
batch_size, num_true = label.shape
use_custom_samples = False
#import pdb; pdb.set_trace()
num_sampled_classes = num_samples + num_true
logits = np.random.randn(batch_size, num_classes)
@ -391,7 +391,8 @@ class TestSampleLogitsOpV3(OpTest):
out = sample_logits(self.inputs["Logits"], self.inputs["Label"],
self.attrs["num_samples"], self.attrs["seed"],
self.attrs["remove_accidental_hits"], True,
self.fetched_samples, self.probabilities)
self.fetched_samples.astype(np.int64),
self.probabilities)
self.outputs = {
'SampledLogits': out[0],
'Samples': out[1],

Loading…
Cancel
Save