CppFunction (reference) in projects: aosp_15_r20

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:CppFunction (Results 1 – 25 of 35) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A D	NamedRegistrations.cpp	5 using torch::CppFunction; 8 m.fallback(CppFunction::makeNamedNotSupported()); in TORCH_LIBRARY_IMPL() 12 m.impl("_cdist_forward", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 13 m.impl("_fused_dropout", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 14 m.impl("native_dropout", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 15 m.impl("_local_scalar_dense", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 16 m.impl("_sparse_log_softmax.Dimname", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 17 m.impl("_sparse_log_softmax.int", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 18 m.impl("_sparse_softmax.Dimname", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 19 m.impl("_sparse_softmax.int", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() [all …]
H A D	library.cpp	`47 CppFunction::CppFunction(c10::KernelFunction func, std::optional<c10::impl::CppSignature> cpp_signa… in CppFunction() function in torch::CppFunction 54 CppFunction::~CppFunction() = default; 163 …std::variant<c10::OperatorName, c10::FunctionSchema>&& name_or_schema, CppFunction&& f, const std:… in _def() 225 Library& Library::_impl(const char* name_str, CppFunction&& f, _RegisterOrVerify rv) & { in _impl() 265 Library& Library::_fallback(CppFunction&& f) & { in _fallback()`
H A 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()`
H A D	VariableFallbackKernel.cpp	`39 #define AUTOGRAD_FALLBACK torch::CppFunction::makeFallthrough() 60 #define AUTOGRAD_FALLBACK torch::CppFunction::makeFromBoxedFunction<&autograd_fallback>() 97 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()`
H A D	BackendSelectFallbackKernel.cpp	`4 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()`
H A D	MetaFallbackKernel.cpp	`24 m.fallback(torch::CppFunction::makeFromBoxedFunction<&metaFallback>()); in TORCH_LIBRARY_IMPL()`
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A D	MathBitFallThroughLists.h	6 m.impl("as_strided_", torch::CppFunction::makeFallthrough()); \ 7 m.impl("detach", torch::CppFunction::makeFallthrough()); \ 8 m.impl("detach_", torch::CppFunction::makeFallthrough()); \ 9 m.impl("diagonal", torch::CppFunction::makeFallthrough()); \ 10 m.impl("expand", torch::CppFunction::makeFallthrough()); \ 11 m.impl("expand_as", torch::CppFunction::makeFallthrough()); \ 12 m.impl("movedim.int", torch::CppFunction::makeFallthrough()); \ 13 m.impl("movedim.intlist", torch::CppFunction::makeFallthrough()); \ 14 m.impl("narrow", torch::CppFunction::makeFallthrough()); \ 15 m.impl("permute", torch::CppFunction::makeFallthrough()); \ [all …]
H A D	NegateFallback.cpp	19 m.fallback(torch::CppFunction::makeFromBoxedFunction<&negationFallback>()); in TORCH_LIBRARY_IMPL() 23 m.impl("set_.source_Storage_storage_offset", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 24 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 25 m.impl("set_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 26 m.impl("copy_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 27 m.impl("clone", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 28 m.impl("neg_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 29 m.impl("resolve_neg", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 30 m.impl("resolve_conj", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 31 m.impl("repeat_interleave.Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() [all …]
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A D	ConjugateFallback.cpp	18 m.fallback(torch::CppFunction::makeFromBoxedFunction<&conjugateFallback>()); in TORCH_LIBRARY_IMPL() 22 m.impl("set_.source_Storage_storage_offset", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 23 m.impl("set_.source_Tensor", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 24 m.impl("set_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 25 m.impl("copy_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 26 m.impl("clone", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 27 m.impl("_conj_physical", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 28 m.impl("conj_physical", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 29 m.impl("conj_physical_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 30 m.impl("resolve_conj", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() [all …]
H A D	ZeroTensorFallback.cpp	87 m.fallback(torch::CppFunction::makeFromBoxedFunction<&zeroTensorFallback>()); in TORCH_LIBRARY_IMPL() 91 m.impl("zeros_like", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 92 m.impl("mul.Scalar", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 93 m.impl("add.Scalar", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 94 m.impl("copy_", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 95 m.impl("clone", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 96 m.impl("dot", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 97 m.impl("vdot", torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()
H A D	autocast_mode.cpp	`166 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 210 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 322 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 464 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()`
H A D	VmapModeRegistrations.cpp	`4 using torch::CppFunction; 34 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL()`
/aosp_15_r20/external/pytorch/torch/
H A D	library.h	`107 class TORCH_API CppFunction final { 114 explicit CppFunction( 128 explicit CppFunction( 143 explicit CppFunction( 159 explicit CppFunction( 175 explicit CppFunction( 192 explicit CppFunction( 207 ~CppFunction(); 209 CppFunction(CppFunction&&) noexcept = default; 211 CppFunction& operator=(CppFunction&&) = default; [all …]`
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A D	TraceTypeManual.cpp	140 m.impl("_backward", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 141 m.impl("set_data", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 142 m.impl("data", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 143 m.impl("is_leaf", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 144 m.impl("output_nr", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 145 m.impl("_version", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 146 m.impl("requires_grad_", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 147 m.impl("retain_grad", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 148 m.impl("_fw_primal", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 149 m.impl("_make_dual", CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() [all …]
H A D	autograd_not_implemented_fallback.h	`9 TORCH_API torch::CppFunction autogradNotImplementedFallback(); 13 TORCH_API torch::CppFunction autogradNotImplementedInplaceOrViewFallback(); 17 TORCH_API torch::CppFunction basicAutogradNotImplementedFallback();`
H A D	autograd_not_implemented_fallback.cpp	220 torch::CppFunction basicAutogradNotImplementedFallback() { in basicAutogradNotImplementedFallback() 221 return torch::CppFunction::makeFromBoxedFunction< in basicAutogradNotImplementedFallback() 473 torch::CppFunction autogradNotImplementedFallback() { in autogradNotImplementedFallback() 474 return torch::CppFunction::makeFromBoxedFunction< in autogradNotImplementedFallback() 627 torch::CppFunction autogradNotImplementedInplaceOrViewFallback() { in autogradNotImplementedInplaceOrViewFallback() 628 return torch::CppFunction::makeFromBoxedFunction< in autogradNotImplementedInplaceOrViewFallback()
/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A 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>());
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A 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()
H A D	VmapModeRegistrations.cpp	`28 m.fallback(torch::CppFunction::makeFallthrough()); in TORCH_LIBRARY_IMPL() 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>());`
H A 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<\
H A 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>());`
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A 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() 112 gm.fallback(torch::CppFunction::makeFallthrough()); in TEST()`
/aosp_15_r20/external/pytorch/aten/src/ATen/core/boxing/impl/
H A 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()`
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A D	python_dispatch.cpp	66 inline torch::CppFunction dispatch_str(const char* key, Func&& raw_f) { in dispatch_str() 71 torch::CppFunction f(std::forward<Func>(raw_f)); in dispatch_str() 402 CppFunction::makeFromBoxedFunctor( in initDispatchBindings() 427 torch::dispatch(dispatch, CppFunction::makeFallthrough()), in initDispatchBindings() 434 CppFunction::makeFromBoxedFunctor( in initDispatchBindings() 473 dispatch_str(dispatch, CppFunction::makeFallthrough())); in initDispatchBindings() 490 torch::dispatch(dispatch, CppFunction::makeFallthrough())); in initDispatchBindings() 494 CppFunction::makeFromBoxedFunctor( in initDispatchBindings()
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A 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()