Searched defs:uniform_ (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/_ops/ |
| H A D | init.py | 13 def uniform_(types, args=(), kwargs=None, pg=None): function
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | unary_test.py | 63 def uniform_(input): function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Random.cpp | 17 static Tensor& uniform_( in uniform_() function
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/ |
| H A D | init.cpp | 175 Tensor uniform_(Tensor tensor, double low, double high) { in uniform_() function
|
| /aosp_15_r20/external/pytorch/torch/nn/ |
| H A D | init.py | 142 def uniform_( function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | cpu_rng_test.cpp | 91 Tensor& uniform_(Tensor& self, double from, double to, std::optional<Generator> generator) { in uniform_() function
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Distributions.cpp | 250 Tensor& uniform_(Tensor& self, double from, double to, std::optional<Generator> gen) { in uniform_() function
|
| /aosp_15_r20/external/pytorch/torch/_decomp/ |
| H A D | decompositions.py | 2725 def uniform_(self, low=0, high=1, generator=None): function
|