| /external/pytorch/aten/src/ATen/functorch/ |
| D | BatchRulesDynamic.cpp | 27 m.impl(#op, torch::CppFunction::makeFromBoxedFunction<&unsupportedDynamicOp>());
73 …m.impl("_local_scalar_dense", torch::CppFunction::makeFromBoxedFunction<&unsupportedLocalScalarDen… in TORCH_LIBRARY_IMPL()
74 m.impl("item", torch::CppFunction::makeFromBoxedFunction<&unsupportedItem>()); in TORCH_LIBRARY_IMPL()
75 m.impl("is_nonzero", torch::CppFunction::makeFromBoxedFunction<&unsupportedIsNonzero>()); in TORCH_LIBRARY_IMPL()
76 m.impl("allclose", torch::CppFunction::makeFromBoxedFunction<&unsupportedAllclose>()); in TORCH_LIBRARY_IMPL()
|
| D | VmapModeRegistrations.cpp | 37 m.impl(#op, torch::CppFunction::makeFromBoxedFunction<&unsupportedRandomOp>());
40 m.impl(#op"."#overload, torch::CppFunction::makeFromBoxedFunction<&unsupportedRandomOp>());
43 m.impl(#op, torch::CppFunction::makeFromBoxedFunction<&nyiRandomOp>());
46 m.impl(#op"."#overload, torch::CppFunction::makeFromBoxedFunction<&nyiRandomOp>());
|
| D | BatchRulesHelper.h | 190 …m.impl(#op, torch::CppFunction::makeFromBoxedFunction<boxed_tensor_inputs_batch_rule<decltype(&han…
193 …m.impl(#op "." #overload, torch::CppFunction::makeFromBoxedFunction<boxed_tensor_inputs_batch_rule…
202 …m.impl(#op, torch::CppFunction::makeFromBoxedFunction<boxed_tensor_inputs_batch_rule<decltype(&han…
293 m.impl(#op, torch::CppFunction::makeFromBoxedFunction<boxed_existing_bdim_all_batch_rule>());
372 …m.impl(#op, torch::CppFunction::makeFromBoxedFunction<boxed_all_tensors_have_optional_bdim<feature…
376 torch::CppFunction::makeFromBoxedFunction<\
|
| D | DynamicLayer.cpp | 494 m.fallback(torch::CppFunction::makeFromBoxedFunction<&dynamicLayerFrontFallback>()); in TORCH_LIBRARY_IMPL()
498 m.fallback(torch::CppFunction::makeFromBoxedFunction<&dynamicLayerBackFallback>()); in TORCH_LIBRARY_IMPL()
503 m.impl(#op, torch::CppFunction::makeFromBoxedFunction<&dynamicLayerBackGradSpecialCase>());
|
| D | TensorWrapper.cpp | 208 m.fallback(torch::CppFunction::makeFromBoxedFunction<&dead_tensor_wrapper_fallback>()); in TORCH_LIBRARY_IMPL()
|
| D | LegacyBatchingRegistrations.cpp | 732 m.fallback(torch::CppFunction::makeFromBoxedFunction<&batchedTensorForLoopFallback>()); in TORCH_LIBRARY_IMPL()
759 m.fallback(torch::CppFunction::makeFromBoxedFunction<&batchedNestedTensorForLoopFallback>()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/mps/ |
| D | MPSFallback.mm | 78 m.fallback(torch::CppFunction::makeFromBoxedFunction<&mps_error_fallback>());
80 m.fallback(torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
88 m.impl("embedding_renorm_", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
89 m.impl("linalg_svd", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
90 m.impl("linalg_svd.U", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
91 …m.impl("im2col", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>()); // Used in preproce…
92 m.impl("col2im", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
94 m.impl("upsample_nearest3d.vec", torch::CppFunction::makeFromBoxedFunction<&mps_fallback>());
|
| /external/pytorch/aten/src/ATen/core/boxing/ |
| D | KernelFunction_test.cpp | 378 KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_with_return>(); in TEST()
383 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_without_return>(); in TEST()
388 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_with_multi_return… in TEST()
395 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_for_inplace_op>(); in TEST()
400 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_for_outofplace_op… in TEST()
405 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_for_outofplace_mu… in TEST()
412 KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_with_return>(); in TEST()
417 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_without_return>(); in TEST()
422 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_with_multi_return… in TEST()
429 …KernelFunction func = KernelFunction::makeFromBoxedFunction<&kernels::boxed_func_for_inplace_op>(); in TEST()
[all …]
|
| D | KernelFunction.h | 154 static KernelFunction makeFromBoxedFunction();
161 static KernelFunction makeFromBoxedFunction();
|
| D | KernelFunction_impl.h | 122 inline KernelFunction KernelFunction::makeFromBoxedFunction() { in makeFromBoxedFunction() function
128 inline KernelFunction KernelFunction::makeFromBoxedFunction() { in makeFromBoxedFunction() function
|
| /external/pytorch/aten/src/ATen/core/ |
| D | PythonFallbackKernel.cpp | 154 m.fallback(torch::CppFunction::makeFromBoxedFunction<&pythonFallback>()); in TORCH_LIBRARY_IMPL()
158 m.fallback(torch::CppFunction::makeFromBoxedFunction<&pythonDispatcherFallback>()); in TORCH_LIBRARY_IMPL()
162 m.fallback(torch::CppFunction::makeFromBoxedFunction<&pythonTLSSnapshotFallback>()); in TORCH_LIBRARY_IMPL()
166 m.fallback(torch::CppFunction::makeFromBoxedFunction<&preDispatchFallback>()); in TORCH_LIBRARY_IMPL()
|
| D | MetaFallbackKernel.cpp | 24 m.fallback(torch::CppFunction::makeFromBoxedFunction<&metaFallback>()); in TORCH_LIBRARY_IMPL()
|
| D | VariableFallbackKernel.cpp | 60 #define AUTOGRAD_FALLBACK torch::CppFunction::makeFromBoxedFunction<&autograd_fallback>()
|
| /external/pytorch/aten/src/ATen/core/dispatch/ |
| D | backend_fallback_test.cpp | 87 m.fallback(torch::CppFunction::makeFromBoxedFunction<&generic_mode_fallback>()); in TEST()
99 m.fallback(torch::CppFunction::makeFromBoxedFunction<&generic_wrapper_fallback>()); in TEST()
109 m.impl("mul.Tensor", torch::CppFunction::makeFromBoxedFunction<&generic_mode_fallback>()); in TEST()
|
| /external/pytorch/aten/src/ATen/core/boxing/impl/ |
| D | kernel_stackbased_test.cpp | 108 …m_cpu.impl("my_op", DispatchKey::CPU, torch::CppFunction::makeFromBoxedFunction<incrementKernel>()… in TEST()
111 …m_cuda.impl("my_op", DispatchKey::CUDA, torch::CppFunction::makeFromBoxedFunction<decrementKernel>… in TEST()
185 m_cpu.fallback(torch::CppFunction::makeFromBoxedFunction<&incrementKernel>()); in TEST()
187 …m_testing.fallback(torch::CppFunction::makeFromBoxedFunction<&redispatchingKernel_with_DispatchKey… in TEST()
|
| /external/pytorch/test/cpp_extensions/ |
| D | open_registration_extension.cpp | 529 m.impl("sub.Tensor", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()
530 m.impl("_foreach_add.List", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()
531 m.impl("_fused_adamw_", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()
532 m.impl("index.Tensor", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()
533 m.impl("triu_indices", torch::CppFunction::makeFromBoxedFunction<&custom_cpu_fallback>()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | NegateFallback.cpp | 19 m.fallback(torch::CppFunction::makeFromBoxedFunction<&negationFallback>()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/torch/csrc/autograd/ |
| D | autograd_not_implemented_fallback.cpp | 221 return torch::CppFunction::makeFromBoxedFunction< in basicAutogradNotImplementedFallback()
474 return torch::CppFunction::makeFromBoxedFunction< in autogradNotImplementedFallback()
628 return torch::CppFunction::makeFromBoxedFunction< in autogradNotImplementedInplaceOrViewFallback()
|
| D | TraceTypeManual.cpp | 295 m.fallback(CppFunction::makeFromBoxedFunction<&general_trace_function>()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/ |
| D | ZeroTensorFallback.cpp | 87 m.fallback(torch::CppFunction::makeFromBoxedFunction<&zeroTensorFallback>()); in TORCH_LIBRARY_IMPL()
|
| D | ConjugateFallback.cpp | 18 m.fallback(torch::CppFunction::makeFromBoxedFunction<&conjugateFallback>()); in TORCH_LIBRARY_IMPL()
|
| /external/pytorch/aten/src/ATen/core/op_registration/ |
| D | op_registration.h | 73 …return std::move(*this).kernel(dispatch_key, KernelFunction::makeFromBoxedFunction<kernel_func>(),… in kernel()
79 …return std::move(*this).kernel(std::nullopt, KernelFunction::makeFromBoxedFunction<kernel_func>(),… in catchAllKernel()
|
| D | op_registration_test.cpp | 424 …on().registerFallback(c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_f… in TEST()
434 …n().registerFallback(c10::DispatchKey::CUDA, c10::KernelFunction::makeFromBoxedFunction<&backend_f… in TEST()
447 …on().registerFallback(c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_f… in TEST()
462 …on().registerFallback(c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_f… in TEST()
478 …on().registerFallback(c10::DispatchKey::CPU, c10::KernelFunction::makeFromBoxedFunction<&backend_f… in TEST()
1352 m1.fallback(CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
1932 m.fallback(CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
1988 m.def("fn6(Tensor x) -> Tensor", CppFunction::makeFromBoxedFunction<&backend_fallback_kernel>()); in TEST()
|
| /external/pytorch/torch/ |
| D | library.h | 245 static CppFunction makeFromBoxedFunction() { in makeFromBoxedFunction() function
253 static CppFunction makeFromBoxedFunction() { in makeFromBoxedFunction() function
|
| /external/pytorch/torch/csrc/lazy/ts_backend/ |
| D | ts_eager_fallback.cpp | 204 m.fallback(torch::CppFunction::makeFromBoxedFunction<<c_eager_fallback>()); in register_ts_ltc_eager_fallback()
|