Searched refs:seeds (Results 1 – 6 of 6) sorted by relevance
| /frameworks/ml/nn/runtime/test/generated/models/ |
| D | random_multinomial_float16.model.cpp | 11 auto seeds = model->addOperand(&type2); in CreateModel() local
17 model->setOperandValue(seeds, seeds_init, sizeof(int32_t) * 2); in CreateModel()
18 model->addOperation(ANEURALNETWORKS_RANDOM_MULTINOMIAL, {input0, sample_count, seeds}, {output}); in CreateModel()
39 auto seeds = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
45 model->setOperandValue(seeds, seeds_init, sizeof(int32_t) * 2); in CreateModel_dynamic_output_shape()
46 model->addOperation(ANEURALNETWORKS_RANDOM_MULTINOMIAL, {input0, sample_count, seeds}, {output}); in CreateModel_dynamic_output_shape()
|
| D | random_multinomial.model.cpp | 11 auto seeds = model->addOperand(&type2); in CreateModel() local
17 model->setOperandValue(seeds, seeds_init, sizeof(int32_t) * 2); in CreateModel()
18 model->addOperation(ANEURALNETWORKS_RANDOM_MULTINOMIAL, {input0, sample_count, seeds}, {output}); in CreateModel()
39 auto seeds = model->addOperand(&type2); in CreateModel_dynamic_output_shape() local
45 model->setOperandValue(seeds, seeds_init, sizeof(int32_t) * 2); in CreateModel_dynamic_output_shape()
46 model->addOperation(ANEURALNETWORKS_RANDOM_MULTINOMIAL, {input0, sample_count, seeds}, {output}); in CreateModel_dynamic_output_shape()
|
| /frameworks/ml/nn/runtime/test/specs/V1_2/ |
| D | random_multinomial_float16.mod.py | 19 seeds = Parameter("seeds", "TENSOR_INT32", "{2}", [37, 42]) variable
22 model = Model().Operation("RANDOM_MULTINOMIAL", input0, sample_count, seeds).To(output0)
|
| D | random_multinomial.mod.py | 19 seeds = Parameter("seeds", "TENSOR_INT32", "{2}", [37, 42]) variable
22 model = Model().Operation("RANDOM_MULTINOMIAL", input0, sample_count, seeds).To(output0)
|
| /frameworks/ml/nn/common/operations/ |
| D | Multinomial.cpp | 103 int32_t* seeds = GetBuffer<int32_t>(random_seeds_); in EvalFloat32() local
104 random_generator.Init(seeds[0], seeds[1]); in EvalFloat32()
|
| D | MultinomialTest.cpp | 78 std::vector<uint32_t> seeds{kFixedRandomSeed1, kFixedRandomSeed2}; in Invoke() local
79 ASSERT_EQ(execution.setInput(Multinomial::kRandomSeedsTensor, seeds.data(), in Invoke()
80 sizeof(uint32_t) * seeds.size()), in Invoke()
|