File custom_function.h

↰ Parent directory (torch/csrc/autograd)

Definition (torch/csrc/autograd/custom_function.h)

• Program Listing for File custom_function.h

Includes

• ATen/core/ivalue.h (File ivalue.h)

• c10/core/SymInt.h

• c10/util/flat_hash_map.h

• c10/util/irange.h

• torch/csrc/autograd/function.h (File function.h)

• torch/csrc/autograd/variable.h (File variable.h)

• torch/csrc/autograd/variable_info.h

• torch/csrc/dynamo/compiled_autograd.h

• vector

Included By

• File autograd.h

• File _functions.h

Namespaces

• Namespace torch

• Namespace torch::autograd

Classes

• Struct AutogradContext

• Template Struct CppNode

• Struct ExtractVariables

• Template Struct Function

Functions

• Function torch::autograd::_wrap_outputs

• Function torch::autograd::check_variable_result

• Template Function torch::autograd::extract_vars

• Function torch::autograd::to_optional(variable_list&)

• Function torch::autograd::to_optional(Variable&)

• Template Function torch::autograd::to_output_type(std::vector<std::optional<Variable>>&)

• Template Function torch::autograd::to_output_type(std::vector<std::optional<Variable>>&)

Typedefs

• Typedef torch::autograd::_jvp_fn_t

• Typedef torch::autograd::_view_as_self_fn_t

• Typedef torch::autograd::forward_t

• Typedef torch::autograd::optional_variable_list