| /aosp_15_r20/external/pytorch/c10/core/impl/ |
| H A D | LocalDispatchKeySet.h | 100 class C10_API ExcludeDispatchKeyGuard {
102 ExcludeDispatchKeyGuard(DispatchKeySet);
103 ExcludeDispatchKeyGuard(DispatchKey k) in ExcludeDispatchKeyGuard() function
104 : ExcludeDispatchKeyGuard(DispatchKeySet(k)) {} in ExcludeDispatchKeyGuard()
105 ExcludeDispatchKeyGuard(const ExcludeDispatchKeyGuard&) = delete;
106 ExcludeDispatchKeyGuard operator=(const ExcludeDispatchKeyGuard&) = delete;
107 ExcludeDispatchKeyGuard(ExcludeDispatchKeyGuard&&) = delete;
108 ExcludeDispatchKeyGuard operator=(ExcludeDispatchKeyGuard&&) = delete;
109 ~ExcludeDispatchKeyGuard();
|
| H A D | LocalDispatchKeySet.cpp | 61 ExcludeDispatchKeyGuard::ExcludeDispatchKeyGuard(DispatchKeySet exclude) in ExcludeDispatchKeyGuard() function in c10::impl::ExcludeDispatchKeyGuard
68 ExcludeDispatchKeyGuard::~ExcludeDispatchKeyGuard() { in ~ExcludeDispatchKeyGuard()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | LegacyBatchingRegistrations.cpp | 104 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in squeeze_dims__batching_rule()
150 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in squeeze__batching_rule()
187 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in unsqueeze__batching_rule()
207 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in transpose__batching_rule()
239 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in split_batching_rule()
251 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in split_with_sizes_batching_rule()
263 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in split_with_sizes_copy_batching_rule()
275 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in unbind_batching_rule()
366 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in as_strided_batching_rule()
494 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in unwrap_and_call()
[all …]
|
| H A D | BatchedFallback.cpp | 270 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batchedTensorForLoopFallback()
356 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batchedTensorForLoopFallback()
364 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batchedTensorForLoopFallback()
381 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batchedTensorForLoopFallback()
410 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batchedNestedTensorForLoopFallback()
411 c10::impl::ExcludeDispatchKeyGuard nt_guard(DispatchKey::BatchedNestedTensor); in batchedNestedTensorForLoopFallback()
457 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::BatchedNestedTensor); in batchedNestedTensorForLoopFallback()
487 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::BatchedNestedTensor); in batchedNestedTensorForLoopFallback()
505 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::BatchedNestedTensor); in batchedNestedTensorForLoopFallback()
|
| H A D | BatchRulesLoss.cpp | 97 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_plumbing()
106 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_plumbing()
115 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_plumbing()
132 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_backward_plumbing()
143 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_backward_plumbing()
159 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in binary_cross_entropy_backward_plumbing()
|
| H A D | BatchRulesRandomness.cpp | 23 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in random_batching_rule()
39 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in random_inplace_batching_rule()
62 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in bernoulli_inplace_Tensor_batching_rule()
108 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in randperm_batching_rule()
127 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in unary_pointwise_random_batch_rule()
153 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in tensor_like_random_batch_rule()
177 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in native_dropout_batching_rule()
217 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchVmapMode); in multinomial_batching_rule()
|
| H A D | LegacyVmapTransforms.cpp | 145 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in logicalToPhysical()
150 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in logicalToPhysical()
200 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in logicalToPhysical()
205 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in logicalToPhysical()
|
| H A D | BatchRulesNorm.cpp | 272 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in batch_norm_backward_plumbing()
301 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_group_norm_plumbing()
314 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_group_norm_plumbing()
320 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_group_norm_plumbing()
388 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_group_norm_backward_plumbing()
424 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_group_norm_backward_plumbing()
590 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_layer_norm_backward_plumbing()
640 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in native_layer_norm_backward_plumbing()
734 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in apply()
764 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in apply()
[all …]
|
| H A D | BatchRulesConvolution.cpp | 371 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in convolution_backward_plumbing()
402 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in convolution_backward_plumbing()
425 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in convolution_backward_plumbing()
436 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in convolution_backward_plumbing()
|
| H A D | BatchRulesHelper.h | 129 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched);
246 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched);
302 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched);
|
| H A D | BatchRulesReduceOps.cpp | 111 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in boxed_reduction_batch_rule()
118 c10::impl::ExcludeDispatchKeyGuard guard_2(DispatchKey::FuncTorchBatched); in boxed_reduction_batch_rule()
|
| H A D | BatchRulesScatterOps.cpp | 321 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in index_plumbing()
497 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in index_put__plumbing()
537 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in _index_put_impl__plumbing()
651 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::FuncTorchBatched); in index_put_plumbing()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/ |
| H A D | LegacyTypeDispatch.h | 67 c10::impl::ExcludeDispatchKeyGuard autograd_guard_;
84 c10::impl::ExcludeDispatchKeyGuard autograd_guard_;
91 c10::impl::ExcludeDispatchKeyGuard dispatch_key_guard_;
109 c10::impl::ExcludeDispatchKeyGuard dispatch_key_guard_;
|
| H A D | PythonFallbackKernel.cpp | 52 c10::impl::ExcludeDispatchKeyGuard guard(after_Python_keyset); in pythonFallback()
|
| /aosp_15_r20/external/pytorch/torch/csrc/utils/ |
| H A D | disable_torch_function.h | 26 c10::impl::ExcludeDispatchKeyGuard guard_;
27 c10::impl::ExcludeDispatchKeyGuard guard_tls_snapshot_;
|
| H A D | tensor_new.cpp | 390 c10::impl::ExcludeDispatchKeyGuard torchdispatchmode_guard( in internal_new_from_data()
392 c10::impl::ExcludeDispatchKeyGuard torchdispatchmode_snapshot_guard( in internal_new_from_data()
396 c10::impl::ExcludeDispatchKeyGuard functorch_front_guard( in internal_new_from_data()
398 c10::impl::ExcludeDispatchKeyGuard functorch_back_guard( in internal_new_from_data()
402 c10::impl::ExcludeDispatchKeyGuard fake_and_deferred_init_guard( in internal_new_from_data()
408 c10::impl::ExcludeDispatchKeyGuard functionalize_guard( in internal_new_from_data()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | autocast_mode.h | 453 c10::impl::ExcludeDispatchKeyGuard no_autocast(
477 c10::impl::ExcludeDispatchKeyGuard no_autocast(
498 c10::impl::ExcludeDispatchKeyGuard no_autocast(
526 c10::impl::ExcludeDispatchKeyGuard no_autocast(
549 c10::impl::ExcludeDispatchKeyGuard no_autocast(
|
| H A D | TracerMode.h | 129 c10::impl::ExcludeDispatchKeyGuard guard_{at::DispatchKey::Tracer};
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/ |
| H A D | ConvPrepack.cpp | 62 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in create()
161 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_convolution_out()
205 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in run()
|
| H A D | Linear.cpp | 212 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_linear_pointwise()
304 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_linear_pointwise_binary()
369 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkl_linear()
|
| H A D | Conv.cpp | 310 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_convolution_pointwise()
399 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_convolution_pointwise_binary()
551 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_convolution_pointwise_binary_()
726 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in mkldnn_convolution_transpose_pointwise()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | frozen_ops_to_mkldnn.cpp | 241 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in createUnaryOp()
283 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in MKLDNNLayerNormOp()
361 c10::impl::ExcludeDispatchKeyGuard edkg(c10::autograd_dispatch_keyset); in BroadOp()
571 c10::impl::ExcludeDispatchKeyGuard edkg( in __anon6095a4571302()
589 c10::impl::ExcludeDispatchKeyGuard edkg( in __anon6095a4571402()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/miopen/ |
| H A D | AutocastRNN.cpp | 30 c10::impl::ExcludeDispatchKeyGuard no_autocast(DispatchKey::Autocast); in miopen_rnn()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/ |
| H A D | AutocastRNN.cpp | 39 c10::impl::ExcludeDispatchKeyGuard no_autocast(DispatchKey::Autocast); in _cudnn_rnn_cast_reflatten()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/ |
| H A D | backend_fallback_test.cpp | 28 c10::impl::ExcludeDispatchKeyGuard guard(DispatchKey::TESTING_ONLY_GenericMode); in generic_mode_fallback()
|