FM / autograd

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd

Name	Size	Mode	Actions
functions/	-	0755	rm
generated/	-	0755	rm
utils/	-	0755	rm
anomaly_mode.h	1525	0644	edit dl rm
autograd.h	5294	0644	edit dl rm
autograd_not_implemented_fallback.h	201	0644	edit dl rm
cpp_hook.h	518	0644	edit dl rm
custom_function.h	14470	0644	edit dl rm
edge.h	1627	0644	edit dl rm
engine.h	17237	0644	edit dl rm
forward_grad.h	8583	0644	edit dl rm
function.h	25046	0644	edit dl rm
FunctionsManual.h	19904	0644	edit dl rm
function_hook.h	641	0644	edit dl rm
grad_mode.h	207	0644	edit dl rm
InferenceMode.h	182	0644	edit dl rm
input_buffer.h	1597	0644	edit dl rm
input_metadata.h	1384	0644	edit dl rm
profiler.h	112	0644	edit dl rm
profiler_kineto.h	9111	0644	edit dl rm
profiler_legacy.h	16930	0644	edit dl rm
profiler_utils.h	439	0644	edit dl rm
python_anomaly_mode.h	1246	0644	edit dl rm
python_autograd.h	388	0644	edit dl rm
python_cpp_function.h	2456	0644	edit dl rm
python_engine.h	1262	0644	edit dl rm
python_fft_functions.h	128	0644	edit dl rm
python_function.h	4099	0644	edit dl rm
python_hook.h	700	0644	edit dl rm
python_legacy_variable.h	299	0644	edit dl rm
python_linalg_functions.h	131	0644	edit dl rm
python_mode.h	425	0644	edit dl rm
python_nn_functions.h	127	0644	edit dl rm
python_saved_variable_hooks.h	883	0644	edit dl rm
python_special_functions.h	132	0644	edit dl rm
python_torch_functions.h	671	0644	edit dl rm
python_variable.h	1645	0644	edit dl rm
python_variable_indexing.h	313	0644	edit dl rm
record_function_ops.h	587	0644	edit dl rm
saved_variable.h	4322	0644	edit dl rm
saved_variable_hooks.h	272	0644	edit dl rm
symbolic.h	330	0644	edit dl rm
variable.h	33411	0644	edit dl rm
VariableTypeUtils.h	15086	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/autograd/edge.h (1627B)

#pragma once

#include <cstdint>
#include <functional>
#include <memory>

#include <c10/util/hash.h>

namespace torch { namespace autograd {

struct Node;

/// Represents a particular input of a function.
struct Edge {
  Edge() noexcept : function(nullptr), input_nr(0) {}

Edge(std::shared_ptr<Node> function_, uint32_t input_nr_) noexcept
      : function(std::move(function_)), input_nr(input_nr_) {}

/// Convenience method to test if an edge is valid.
  bool is_valid() const noexcept {
    return function != nullptr;
  }

// Required for use in associative containers.
  bool operator==(const Edge& other) const noexcept {
    return this->function == other.function && this->input_nr == other.input_nr;
  }

bool operator!=(const Edge& other) const noexcept {
    return !(*this == other);
  }

/// The function this `Edge` points to.
  std::shared_ptr<Node> function;

/// The identifier of a particular input to the function.
  uint32_t input_nr;
};
}} // namespace torch::autograd

// The idiomatic way of enabling use of a custom type as the key of hash
// containers in C++11. This method removes the requirement of having to pass
// a custom hasher to std::unordered_{map, set}.
// See http://en.cppreference.com/w/cpp/utility/hash for more information.
namespace std {
template <>
struct hash<torch::autograd::Edge> {
  // These type aliases are required by the standard.
  using argument_type = torch::autograd::Edge;
  using return_type = size_t;
  return_type operator()(const argument_type& edge) const noexcept {
    return c10::get_hash(edge.function, edge.input_nr);
  }
};
} // namespace std

cmd: