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/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)

• vector

Included By

• File _functions.h

• File autograd.h

Namespaces

• Namespace torch

• Namespace torch::autograd

Classes

• Struct AutogradContext

• Template Struct CppNode

• Struct ExtractVariables

• Template Struct Function

• Struct VariableInfo

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<c10::optional<Variable>>&)

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

Typedefs

• Typedef torch::autograd::_jvp_fn_t

• Typedef torch::autograd::forward_t

• Typedef torch::autograd::optional_variable_list