| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesRandomness.cpp | 28 RandomnessType randomness = maybe_layer->randomness(); in random_batching_rule() local
44 RandomnessType randomness = maybe_layer->randomness(); in random_inplace_batching_rule() local
65 RandomnessType randomness = maybe_layer->randomness(); in bernoulli_inplace_Tensor_batching_rule() local
111 RandomnessType randomness = maybe_layer->randomness(); in randperm_batching_rule() local
134 RandomnessType randomness = maybe_layer->randomness(); in unary_pointwise_random_batch_rule() local
156 RandomnessType randomness = maybe_layer->randomness(); in tensor_like_random_batch_rule() local
180 RandomnessType randomness = maybe_layer->randomness(); in native_dropout_batching_rule() local
224 RandomnessType randomness = maybe_layer->randomness(); in multinomial_batching_rule() local
|
| H A D | DynamicLayer.cpp | 31 std::optional<RandomnessType> randomness, in DynamicLayer()
72 RandomnessType DynamicLayer::randomness() const { in randomness() function in at::functorch::DynamicLayer
254 std::optional<RandomnessType> randomness, in initAndPushDynamicLayer()
|
| H A D | BatchRulesHelper.cpp | 86 void check_randomness(RandomnessType randomness, bool any_tensor_batched) { in check_randomness()
100 void check_randomness(RandomnessType randomness) { in check_randomness()
|
| H A D | BatchRulesLinearAlgebra.cpp | 501 RandomnessType randomness = maybe_layer->randomness(); in _scaled_dot_product_flash_attention_batch_rule() local
550 RandomnessType randomness = maybe_layer->randomness(); in _scaled_dot_product_efficient_attention_batch_rule() local
592 RandomnessType randomness = maybe_layer->randomness(); in _scaled_dot_product_cudnn_attention_batch_rule() local
|
| H A D | VmapInterpreter.h | 18 RandomnessType randomness() const { in randomness() function
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | vmap.py | 309 def vmap_impl(func, in_dims, out_dims, randomness, chunk_size, *args, **kwargs): argument
407 func, flat_in_dims, chunks_flat_args, args_spec, out_dims, randomness, **kwargs argument
456 def _check_randomness_arg(randomness): argument
464 def vmap_increment_nesting(batch_size, randomness): argument
473 func, batch_size, flat_in_dims, flat_args, args_spec, out_dims, randomness, **kwargs argument
503 def restore_vmap(func, in_dims, batch_size, randomness): argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_vmap.py | 3345 def test_chunk_vmap(self, in_dim, out_dim, randomness): argument
3371 def test_vmap_chunksize(self, in_dim, out_dim, randomness): argument
3465 def test_vmap_chunksize_error(self, in_dim, out_dim, randomness): argument
3501 def test_vmap_chunksize_composition(self, in_dim, out_dim, randomness): argument
3928 def test_randomness(self, device, randomness, backend): argument
5356 def test_factory_ops(self, device, randomness, use_generator): argument
5401 def test_randperm(self, device, randomness, use_generator): argument
5450 def test_dropout(self, device, randomness, batched_input): argument
5485 def test_alpha_dropout(self, device, randomness, batched_input): argument
5516 def test_feature_dropout(self, device, randomness, batched_input, dim): argument
[all …]
|
| /aosp_15_r20/external/tink/cc/keyderivation/internal/ |
| H A D | prf_based_deriver.cc | 48 std::unique_ptr<InputStream> randomness = (*streaming_prf)->ComputePrf("s"); in New() local
62 std::unique_ptr<InputStream> randomness = streaming_prf_->ComputePrf(salt); in DeriveKeyset() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/functorch/ |
| H A D | init.cpp | 228 RandomnessType get_randomness_enum(const std::string& randomness) { in get_randomness_enum()
274 const std::string& randomness) { in _vmap_increment_nesting()
|
| /aosp_15_r20/external/llvm-libc/src/search/ |
| H A D | hcreate_r.cpp | 22 uint64_t randomness = internal::randomness::next_random_seed(); variable
|
| H A D | hsearch.cpp | 22 uint64_t randomness = internal::randomness::next_random_seed(); variable
|
| H A D | hcreate.cpp | 24 uint64_t randomness = internal::randomness::next_random_seed(); variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.heap.operations/make.heap/ |
| H A D | make_heap.pass.cpp | 21 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.heap.operations/sort.heap/ |
| H A D | sort_heap.pass.cpp | 21 std::mt19937 randomness; variable
|
| H A D | sort_heap_comp.pass.cpp | 32 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.heap.operations/push.heap/ |
| H A D | push_heap.pass.cpp | 22 std::mt19937 randomness; variable
|
| H A D | push_heap_comp.pass.cpp | 33 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.heap.operations/pop.heap/ |
| H A D | pop_heap.pass.cpp | 21 std::mt19937 randomness; variable
|
| H A D | pop_heap_comp.pass.cpp | 33 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.nth.element/ |
| H A D | nth_element.pass.cpp | 22 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.sort/partial.sort/ |
| H A D | partial_sort.pass.cpp | 22 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/containers/sequences/forwardlist/forwardlist.ops/ |
| H A D | sort.pass.cpp | 23 std::mt19937 randomness; variable
|
| H A D | sort_pred.pass.cpp | 24 std::mt19937 randomness; variable
|
| /aosp_15_r20/external/libcxx/test/std/algorithms/alg.sorting/alg.min.max/ |
| H A D | min_element.pass.cpp | 23 std::mt19937 randomness; variable
|
| H A D | max_element.pass.cpp | 23 std::mt19937 randomness; variable
|