Home
last modified time | relevance | path

Searched defs:Multinomial (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmultinomial_op_test.cc27 static Graph* Multinomial(int batch_size, int num_classes, int num_samples) { in Multinomial() function
/aosp_15_r20/external/pytorch/torch/distributions/
H A Dmultinomial.py13 class Multinomial(Distribution): class
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DMultinomial.cpp60 Multinomial::Multinomial(const Operation& operation, RunTimeOperandInfo* operands) { in Multinomial() function in android::nn::Multinomial
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/distributions/
H A Dmultinomial.py52 class Multinomial(distribution.Distribution): class
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py4482 class Multinomial(torch.nn.Module): class
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go27005 func Multinomial(scope *Scope, logits tf.Output, num_samples tf.Output, optional ...MultinomialAttr… func