Searched refs:kRandomSeedsTensor (Results 1 – 3 of 3) sorted by relevance
44 static constexpr int kRandomSeedsTensor = 2; variable
58 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
79 ASSERT_EQ(execution.setInput(Multinomial::kRandomSeedsTensor, seeds.data(), in Invoke()