Searched refs:test_multinomial (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/rust/android-crates-io/crates/num-integer/src/ | ||
| D | lib.rs | 1315 fn test_multinomial() { in test_multinomial() function |
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/ | ||
| H A D | control_flow_ops_test.py | 924 def test_multinomial(self): member in StatelessRandomTest |
| /aosp_15_r20/external/pytorch/test/functorch/ | ||
| H A D | test_vmap.py | 5917 def test_multinomial( member in TestRandomness |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_torch.py | 5272 def test_multinomial(self, device, dtype): member in TestTorchDeviceType |
| H A D | test_mps.py | 8296 def test_multinomial(self): member in TestMPS |
| /aosp_15_r20/external/pytorch/test/onnx/ | ||
| H A D | test_pytorch_onnx_onnxruntime.py | 4481 def test_multinomial(self): member in TestONNXRuntime |