Searched refs:Multinomial (Results 1 – 5 of 5) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Multinomial.cpp | 60 Multinomial::Multinomial(const Operation& operation, RunTimeOperandInfo* operands) { in Multinomial() function in android::nn::Multinomial
69 bool Multinomial::Prepare(const Operation& operation, RunTimeOperandInfo* operands, in Prepare()
75 const RunTimeOperandInfo* input = GetInput(operation, operands, Multinomial::kInputTensor); in Prepare()
90 bool Multinomial::Eval() { in Eval()
111 void Multinomial::EvalFloat32(const float* inputData) { in EvalFloat32()
|
| D | MultinomialTest.cpp | 76 ASSERT_EQ(execution.setInput(Multinomial::kInputTensor, input_.data(), in Invoke()
79 ASSERT_EQ(execution.setInput(Multinomial::kSampleCountParam, &sample_size_, in Invoke()
84 ASSERT_EQ(execution.setInput(Multinomial::kRandomSeedsTensor, seeds.data(), in Invoke()
89 ASSERT_EQ(execution.setOutput(Multinomial::kOutputTensor, output_.data(), in Invoke()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | Multinomial.h | 33 class Multinomial {
35 Multinomial(const Operation& operation, RunTimeOperandInfo* operands);
|
| /packages/modules/NeuralNetworks/common/ |
| D | Android.bp | 159 "cpu_operations/Multinomial.cpp",
346 "cpu_operations/Multinomial.cpp",
|
| D | CpuExecutor.cpp | 1061 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1064 Multinomial multinomial(operation, operands); in executeOperation()
1066 success = Multinomial::Prepare(operation, operands, &outputShape) && in executeOperation()
|