Home
last modified time | relevance | path

Searched defs:maybe_layer (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchRulesRandomness.cpp24 auto maybe_layer = maybeCurrentDynamicLayer(); in random_batching_rule() local
40 auto maybe_layer = maybeCurrentDynamicLayer(); in random_inplace_batching_rule() local
63 auto maybe_layer = maybeCurrentDynamicLayer(); in bernoulli_inplace_Tensor_batching_rule() local
109 auto maybe_layer = maybeCurrentDynamicLayer(); in randperm_batching_rule() local
128 auto maybe_layer = maybeCurrentDynamicLayer(); in unary_pointwise_random_batch_rule() local
154 auto maybe_layer = maybeCurrentDynamicLayer(); in tensor_like_random_batch_rule() local
178 auto maybe_layer = maybeCurrentDynamicLayer(); in native_dropout_batching_rule() local
218 auto maybe_layer = maybeCurrentDynamicLayer(); in multinomial_batching_rule() local
H A DBatchRulesNorm.cpp224 auto maybe_layer = maybeCurrentDynamicLayer(); in batch_norm_backward_plumbing() local
296 auto maybe_layer = maybeCurrentDynamicLayer(); in native_group_norm_plumbing() local
383 auto maybe_layer = maybeCurrentDynamicLayer(); in native_group_norm_backward_plumbing() local
586 auto maybe_layer = maybeCurrentDynamicLayer(); in native_layer_norm_backward_plumbing() local
728 auto maybe_layer = maybeCurrentDynamicLayer(); in apply() local
758 auto maybe_layer = maybeCurrentDynamicLayer(); in apply() local
786 auto maybe_layer = maybeCurrentDynamicLayer(); in apply() local
H A DBatchRulesLoss.cpp91 auto maybe_layer = maybeCurrentDynamicLayer(); in binary_cross_entropy_plumbing() local
127 auto maybe_layer = maybeCurrentDynamicLayer(); in binary_cross_entropy_backward_plumbing() local
H A DBatchRulesScatterOps.cpp322 auto maybe_layer = maybeCurrentDynamicLayer(); in index_plumbing() local
498 auto maybe_layer = maybeCurrentDynamicLayer(); in index_put__plumbing() local
538 auto maybe_layer = maybeCurrentDynamicLayer(); in _index_put_impl__plumbing() local
652 auto maybe_layer = maybeCurrentDynamicLayer(); in index_put_plumbing() local
H A DBatchRulesLinearAlgebra.cpp500 auto maybe_layer = maybeCurrentDynamicLayer(); in _scaled_dot_product_flash_attention_batch_rule() local
549 auto maybe_layer = maybeCurrentDynamicLayer(); in _scaled_dot_product_efficient_attention_batch_rule() local
591 auto maybe_layer = maybeCurrentDynamicLayer(); in _scaled_dot_product_cudnn_attention_batch_rule() local
H A DBatchRulesReduceOps.cpp112 auto maybe_layer = maybeCurrentDynamicLayer(); in boxed_reduction_batch_rule() local
H A DBatchRulesBinaryOps.cpp59 auto maybe_layer = maybeCurrentDynamicLayer(); in apply() local
H A DBatchRulesConvolution.cpp366 const auto maybe_layer = maybeCurrentDynamicLayer(); in convolution_backward_plumbing() local
H A DBatchRulesViews.cpp157 auto maybe_layer = maybeCurrentDynamicLayer(); in resize__plumbing() local
/aosp_15_r20/external/pytorch/torch/csrc/functorch/
H A Dinit.cpp370 auto maybe_layer = maybeCurrentDynamicLayer(); in currentLevel() local
377 auto maybe_layer = maybeCurrentDynamicLayer(); in maybe_current_level() local