| /external/pytorch/aten/src/ATen/core/ |
| D | PythonFallbackKernel.cpp | 96 void pythonDispatcherFallback(const c10::OperatorHandle& op, c10::DispatchKeySet dispatch_keys, tor… in pythonDispatcherFallback() argument
99 (*state)->python_dispatcher(op, dispatch_keys.remove(c10::DispatchKey::PythonDispatcher), stack); in pythonDispatcherFallback()
102 void pythonTLSSnapshotFallback(const c10::OperatorHandle &op, c10::DispatchKeySet dispatch_keys, to… in pythonTLSSnapshotFallback() argument
108 …op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(c10::DispatchKeySet::FULL_AFTER, c10::Dispa… in pythonTLSSnapshotFallback()
117 void preDispatchFallback(const c10::OperatorHandle& op, c10::DispatchKeySet dispatch_keys, torch::j… in preDispatchFallback() argument
118 …op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(c10::DispatchKeySet::FULL_AFTER, c10::Dispa… in preDispatchFallback()
|
| D | VariableFallbackKernel.cpp | 47 c10::DispatchKeySet dispatch_keys,
54 op.redispatchBoxed(dispatch_keys & c10::after_autograd_keyset, stack);
57 at::impl::GetVariableHooks()->basic_autograd_not_implemented_fallback(op, dispatch_keys, stack);
|
| D | MetaFallbackKernel.cpp | 9 c10::DispatchKeySet dispatch_keys, in metaFallback() argument
|
| D | VariableHooksInterface.h | 69 c10::DispatchKeySet dispatch_keys,
|
| /external/pytorch/torch/csrc/autograd/ |
| D | autograd_not_implemented_fallback.cpp | 100 c10::DispatchKeySet dispatch_keys, in basicAutogradNotImplementedFallbackImpl() argument
109 op.redispatchBoxed(dispatch_keys & c10::after_autograd_keyset, stack); in basicAutogradNotImplementedFallbackImpl()
147 op.redispatchBoxed(dispatch_keys & c10::after_autograd_keyset, stack); in basicAutogradNotImplementedFallbackImpl()
227 c10::DispatchKeySet dispatch_keys, in basic_autograd_not_implemented_fallback() argument
229 basicAutogradNotImplementedFallbackImpl(op, dispatch_keys, stack); in basic_autograd_not_implemented_fallback()
234 c10::DispatchKeySet dispatch_keys, in autogradNotImplementedFallbackImpl() argument
356 op.redispatchBoxed(dispatch_keys & c10::after_autograd_keyset, stack); in autogradNotImplementedFallbackImpl()
361 dispatch_keys & c10::after_ADInplaceOrView_keyset, stack); in autogradNotImplementedFallbackImpl()
480 c10::DispatchKeySet dispatch_keys, in autogradNotImplementedInplaceOrViewFallbackImpl() argument
557 dispatch_keys & c10::after_ADInplaceOrView_keyset, stack); in autogradNotImplementedInplaceOrViewFallbackImpl()
|
| D | variable.h | 937 c10::DispatchKeySet dispatch_keys,
|
| /external/pytorch/aten/src/ATen/ |
| D | ZeroTensorFallback.cpp | 14 …static void zeroTensorFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch:… in zeroTensorFallback() argument
40 …op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::Ze… in zeroTensorFallback()
82 …op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, DispatchKey::Ze… in zeroTensorFallback()
|
| D | ConjugateFallback.cpp | 12 static void conjugateFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch::j… in conjugateFallback() argument
14 object.fallback_impl(op, dispatch_keys, stack); in conjugateFallback()
|
| /external/pytorch/aten/src/ATen/core/op_registration/ |
| D | op_registration.cpp | 93 std::unordered_set<DispatchKey> dispatch_keys; in checkNoDuplicateKernels_() local
99 0 == dispatch_keys.count(*kernel.dispatch_key), in checkNoDuplicateKernels_()
104 dispatch_keys.insert(*kernel.dispatch_key); in checkNoDuplicateKernels_()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | MathBitsFallback.h | 27 …void fallback_impl(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch::jit::Stack*… in fallback_impl()
84 … op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, key), stack); in fallback_impl()
130 op.redispatchBoxed(dispatch_keys & c10::DispatchKeySet(DispatchKeySet::FULL_AFTER, key), stack); in fallback_impl()
|
| D | NegateFallback.cpp | 13 static void negationFallback(const c10::OperatorHandle& op, DispatchKeySet dispatch_keys, torch::ji… in negationFallback() argument
15 object.fallback_impl(op, dispatch_keys, stack); in negationFallback()
|
| D | README.md | 311 Available backend options can be found by searching `dispatch_keys` in
|
| /external/pytorch/torchgen/ |
| D | gen.py | 1441 dispatch_keys = set()
1446 dispatch_keys.add(dispatch_key)
1729 dispatch_keys: Sequence[DispatchKey],
1806 for dispatch_key in dispatch_keys:
1850 dispatch_keys: Sequence[DispatchKey],
1958 for dispatch_key in dispatch_keys:
2046 dispatch_keys: Sequence[DispatchKey],
2061 dispatch_keys=dispatch_keys,
2075 dispatch_keys=dispatch_keys,
2185 dispatch_keys: Sequence[DispatchKey],
[all …]
|
| D | model.py | 274 dispatch_keys = [ variable
742 assert dispatch_key in dispatch_keys, (
|
| /external/pytorch/test/ |
| D | test_dispatch.py | 44 def extract_dispatch_table_with_keys(table, dispatch_keys): argument
48 for k in dispatch_keys:
|