Searched refs:check_randomness (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesRandomness.cpp | 29 check_randomness(randomness); in random_batching_rule()
45 check_randomness(randomness); in random_inplace_batching_rule()
71 check_randomness(randomness, other_bdim.has_value()); in bernoulli_inplace_Tensor_batching_rule()
112 check_randomness(randomness); in randperm_batching_rule()
135 check_randomness(randomness, tensor_bdim.has_value()); in unary_pointwise_random_batch_rule()
157 check_randomness(randomness); in tensor_like_random_batch_rule()
186 check_randomness(randomness); // if we are in eval mode, we don't use about randomness in native_dropout_batching_rule()
225 check_randomness(randomness, self_bdim.has_value()); in multinomial_batching_rule()
|
| D | BatchRulesHelper.cpp | 86 void check_randomness(RandomnessType randomness, bool any_tensor_batched) { in check_randomness() function
100 void check_randomness(RandomnessType randomness) { in check_randomness() function
101 …check_randomness(randomness, false); // for ops that don't take in any tensors, don't hit same err… in check_randomness()
|
| D | BatchRulesHelper.h | 44 void check_randomness(RandomnessType randomness);
45 void check_randomness(RandomnessType randomness, bool any_tensor_bdim);
|
| D | BatchRulesLinearAlgebra.cpp | 502 …check_randomness(randomness, query_bdim.has_value() || key_bdim.has_value() || value_bdim.has_valu… in _scaled_dot_product_flash_attention_batch_rule()
551 …check_randomness(randomness, query_bdim.has_value() || key_bdim.has_value() || value_bdim.has_valu… in _scaled_dot_product_efficient_attention_batch_rule()
593 …check_randomness(randomness, query_bdim.has_value() || key_bdim.has_value() || value_bdim.has_valu… in _scaled_dot_product_cudnn_attention_batch_rule()
|
| D | BatchRulesBinaryOps.cpp | 67 check_randomness(randomness, (tensor_bdim || other_bdim)); in apply()
|