Home
last modified time | relevance | path

Searched refs:kRandomSeedsTensor (Results 1 – 3 of 3) sorted by relevance

/packages/modules/NeuralNetworks/common/types/operations/include/
DMultinomial.h43 static constexpr int kRandomSeedsTensor = 2; variable
/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp64 random_seeds_ = GetInput(operation, operands, kRandomSeedsTensor); in Multinomial()
DMultinomialTest.cpp84 ASSERT_EQ(execution.setInput(Multinomial::kRandomSeedsTensor, seeds.data(), in Invoke()