Home
last modified time | relevance | path

Searched defs:multinomial (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/types/operations/src/
DMultinomial.cpp24 namespace multinomial { namespace
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomialTest.cpp116 MultinomialOpModel multinomial(kBatchSize, kNumClasses, kNumSamples); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmultinomial.cc29 namespace multinomial { namespace
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Drandom_ops.py475 def multinomial(logits, num_samples, seed=None, name=None, output_dtype=None): function
/aosp_15_r20/external/rust/android-crates-io/crates/num-integer/src/
Dlib.rs1170 pub fn multinomial<T: Integer + Clone>(k: &[T]) -> T in multinomial() function
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dconstraints.py668 multinomial = _Multinomial variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistributions.cpp627 Tensor multinomial( in multinomial() function
/aosp_15_r20/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1064 Multinomial multinomial(operation, operands); in executeOperation() local
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py5602 def multinomial( function