Lines Matching full:compositeimplicitautograd

277 CompositeImplicitAutograd[alias]: impl_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
330 CompositeImplicitAutograd[alias]: default_def_name_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
374 CompositeImplicitAutograd[alias]: impl_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
406 CompositeImplicitAutograd[alias]: default_def_name_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
449 CompositeImplicitAutograd[alias]: default_def_name_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
480                 lambda m: m.impl_t_t("foo", "CompositeImplicitAutograd"),
491 CompositeImplicitAutograd[alias]: impl_t_t :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
525                     "foo", "CompositeImplicitAutograd", debug="fn_math"
538 CompositeImplicitAutograd[alias]: fn_math :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
599     # Now that catchAll maps to CompositeImplicitAutograd, registering to both
600     # catchAll and CompositeImplicitAutograd breaks commutativity.
613                     "foo", "CompositeImplicitAutograd", debug="fn_math"
627 CompositeImplicitAutograd[alias]: fn_math :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
658                     "foo", "CompositeImplicitAutograd", debug="fn_math"
673 CompositeImplicitAutograd[alias]: fn_math :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
805                     "foo", "CompositeImplicitAutograd", debug="fn_math"
823 CompositeImplicitAutograd[alias]: fn_math :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
939 CompositeImplicitAutograd[alias]: fn2 :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
940 CompositeImplicitAutograd[alias] (inactive): fn1 :: (Tensor _0) -> Tensor _0 [ boxed unboxed ]
990         dispatcher.register(["CPU", "XLA", "Lazy", "CompositeImplicitAutograd"])
1012             ["CPU", "XLA", "Lazy", "CompositeImplicitAutograd", "AutogradCPU"]
1042 CompositeImplicitAutograd[alias] fn_CompositeImplicitAutograd
1086         dispatcher.register(["CPU", "FPGA", "CompositeImplicitAutograd"])
1114 CompositeImplicitAutograd[alias] fn_CompositeImplicitAutograd
1129 …    r"Registration to both CompositeImplicitAutograd and CompositeExplicitAutograd is not allowed",
1132                 ["CompositeExplicitAutograd", "CompositeImplicitAutograd"]