Home
last modified time | relevance | path

Searched full:compositeexplicitautogradnonfunctional (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torchgen/dest/
H A Dregister_dispatch_key.py99 DispatchKey.CompositeExplicitAutogradNonFunctional,
152 if backend_index.dispatch_key == DispatchKey.CompositeExplicitAutogradNonFunctional:
372 == DispatchKey.CompositeExplicitAutogradNonFunctional
616 DispatchKey.CompositeExplicitAutogradNonFunctional,
648 DispatchKey.CompositeExplicitAutogradNonFunctional,
719 == DispatchKey.CompositeExplicitAutogradNonFunctional
756 … # we generate CompositeExplicitAutogradNonFunctional implementations of functional and inplace
768 == DispatchKey.CompositeExplicitAutogradNonFunctional
829 is DispatchKey.CompositeExplicitAutogradNonFunctional
922 == DispatchKey.CompositeExplicitAutogradNonFunctional
H A Dlazy_ir.py535 # view_copy ops always have a CompositeExplicitAutogradNonFunctional kernel
537 dispatch_ns = "compositeexplicitautogradnonfunctional"
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.cpp12 // See Note [CompositeExplicitAutogradNonFunctional Key]
15 // `CompositeExplicitAutogradNonFunctional key` if: (1) It's an out-of-place op
80 case DispatchKey::CompositeExplicitAutogradNonFunctional: in getRuntimeDispatchKeySet()
101 case DispatchKey::CompositeExplicitAutogradNonFunctional: in runtimeDispatchKeySetHas()
H A DDispatchKey.cpp206 case DispatchKey::CompositeExplicitAutogradNonFunctional: in toString()
207 return "CompositeExplicitAutogradNonFunctional"; in toString()
378 {"CompositeExplicitAutogradNonFunctional", in parseDispatchKey()
379 c10::DispatchKey::CompositeExplicitAutogradNonFunctional}, in parseDispatchKey()
H A DDispatchKey.h478 // See Note [CompositeExplicitAutogradNonFunctional Key]
479 CompositeExplicitAutogradNonFunctional, // registered at enumerator
485 EndOfAliasKeys = CompositeExplicitAutogradNonFunctional, //
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DOperatorEntry.cpp269 // (2.1) Use kernel from DispatchKey::CompositeExplicitAutogradNonFunctional if available. in registerKernel()
303 // 2.1 Use CompositeExplicitAutogradNonFunctional kernel if available. in registerKernel()
305 …ndefined || isIncludedInAlias(dispatch_key, DispatchKey::CompositeExplicitAutogradNonFunctional)) { in registerKernel()
306 …kend_registration = getKernelForDispatchKey(DispatchKey::CompositeExplicitAutogradNonFunctional)) { in registerKernel()
414 …// Registration to CompositeExplicitAutogradNonFunctional, CompositeExplicitAutograd and Composite… in registerKernel()
418 || dispatch_key == DispatchKey::CompositeExplicitAutogradNonFunctional) { in registerKernel()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A Dnative_functions.yaml949 CompositeExplicitAutogradNonFunctional: as_strided__symint
1126 CompositeExplicitAutogradNonFunctional: bernoulli
1754 CompositeExplicitAutogradNonFunctional: copy
2060 CompositeExplicitAutogradNonFunctional: diag_embed
2394 CompositeExplicitAutogradNonFunctional: new_empty_strided_symint
3724 CompositeExplicitAutogradNonFunctional: logsumexp_out
4267 CompositeExplicitAutogradNonFunctional: narrow_copy_dense_symint
4509 CompositeExplicitAutogradNonFunctional: math_pixel_shuffle
4516 CompositeExplicitAutogradNonFunctional: math_pixel_unshuffle
5140 CompositeExplicitAutogradNonFunctional: select_backward_symint
[all …]
H A DREADME.md326 - `CompositeExplicitAutogradNonFunctional`:
/aosp_15_r20/external/pytorch/torchgen/
H A Dgen.py331 DispatchKey.CompositeExplicitAutogradNonFunctional,
347 return DispatchKey.CompositeExplicitAutogradNonFunctional
453 …return f"return {ns}::{DispatchKey.CompositeExplicitAutogradNonFunctional.lower()}::{name}({exprs}…
2228 # to the Meta + CompositeExplicitAutogradNonFunctional keys).
2231 DispatchKey.CompositeExplicitAutogradNonFunctional,
2240 == DispatchKey.CompositeExplicitAutogradNonFunctional
2254 if dispatch_key == DispatchKey.CompositeExplicitAutogradNonFunctional:
2641 # (1) A CompositeExplicitAutogradNonFunctional kernel for every {view}_copy operator.
2680 DispatchKey.CompositeExplicitAutogradNonFunctional
2899 DispatchKey.CompositeExplicitAutogradNonFunctional,
H A Dgen_aoti_c_shim.py332 elif backend_indices[DispatchKey.CompositeExplicitAutogradNonFunctional].has_kernel(
335 # We need to create C shim wrappers for CompositeExplicitAutogradNonFunctional kernels
337 DispatchKey.CompositeExplicitAutogradNonFunctional
H A Dmodel.py131 CompositeExplicitAutogradNonFunctional = auto() variable in DispatchKey
289 DispatchKey.CompositeExplicitAutogradNonFunctional,
308 DispatchKey.CompositeExplicitAutogradNonFunctional,
812 or d == DispatchKey.CompositeExplicitAutogradNonFunctional
820 DispatchKey.CompositeExplicitAutogradNonFunctional
828 …annot specify more than one of CompositeExplicitAutograd, CompositeExplicitAutogradNonFunctional, "
903 DispatchKey.CompositeExplicitAutogradNonFunctional in dispatch
/aosp_15_r20/external/pytorch/torch/csrc/lazy/core/
H A Dshape_inference.cpp1288 auto out_meta = at::compositeexplicitautogradnonfunctional::select_scatter( in compute_shape_select_scatter()
1313 auto out_meta = at::compositeexplicitautogradnonfunctional::diagonal_scatter( in compute_shape_diagonal_scatter()
1340 at::compositeexplicitautogradnonfunctional::slice_scatter_symint( in compute_shape_slice_scatter_symint()
1366 at::compositeexplicitautogradnonfunctional::as_strided_scatter_symint( in compute_shape_as_strided_scatter_symint()
/aosp_15_r20/external/pytorch/torch/
H A D_ops.py181 # 2.1 Use CompositeExplicitAutogradNonFunctional kernel if available
182 cand = DispatchKey.CompositeExplicitAutogradNonFunctional
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp704 DEF_ONE(CompositeExplicitAutogradNonFunctional) in initDispatchBindings()
/aosp_15_r20/external/pytorch/
H A Dbuckbuild.bzl290 "CompositeExplicitAutogradNonFunctional",