D | random_ops_test.cc | 255 const int kNumClasses = 2; in TEST_P() local 257 MultinomialOpModel m1(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P() 267 MultinomialOpModel m2(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P() 285 const int kNumClasses = 2; in TEST_P() local 287 MultinomialOpModel m1(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P() 298 MultinomialOpModel m2(GetParam(), kLogits, kNumBatches, kNumClasses, in TEST_P() 327 const int kNumClasses = 2; in TEST() local 331 kNumClasses, kNumSamples, /*seed=*/1234, /*seed2=*/5678, in TEST() 353 const int kNumClasses = 3; in TEST() local 357 kNumClasses, kNumSamples, /*seed=*/5678, /*seed2=*/1234); in TEST() [all …]
|