SYMBOL INDEX (1092 symbols across 198 files)

FILE: examples/python/custom_lp.py
  class MYDATASET (line 13) | class MYDATASET(LinkPredictionDataset):
    method __init__ (line 14) | def __init__(self, output_directory: Path, spark=False):
    method download (line 20) | def download(self, overwrite=False):
    method preprocess (line 35) | def preprocess(self, remap_ids=True, splits=None):
  function init_model (line 51) | def init_model(embedding_dim, num_nodes, num_relations, device, dtype):
  function train_epoch (line 87) | def train_epoch(model, dataloader):
  function eval_epoch (line 104) | def eval_epoch(model, dataloader):

FILE: examples/python/custom_nc_graphsage.py
  function switch_to_num (line 17) | def switch_to_num(row):
  class MYDATASET (line 36) | class MYDATASET(NodeClassificationDataset):
    method __init__ (line 37) | def __init__(self, output_directory: Path, spark=False):
    method download (line 43) | def download(self, overwrite=False):
    method preprocess (line 103) | def preprocess(
  function init_model (line 164) | def init_model(feature_dim, num_classes, device):
  function train_epoch (line 202) | def train_epoch(model, dataloader):
  function eval_epoch (line 218) | def eval_epoch(model, dataloader):

FILE: examples/python/fb15k_237.py
  function init_model (line 11) | def init_model(embedding_dim, num_nodes, num_relations, device, dtype):
  function train_epoch (line 47) | def train_epoch(model, dataloader):
  function eval_epoch (line 64) | def eval_epoch(model, dataloader):

FILE: examples/python/fb15k_237_gpu.py
  function init_model (line 11) | def init_model(embedding_dim, num_nodes, num_relations, device, dtype):
  function train_epoch (line 47) | def train_epoch(model, dataloader):
  function eval_epoch (line 64) | def eval_epoch(model, dataloader):

FILE: examples/python/ogbn_arxiv_nc.py
  function init_model (line 11) | def init_model(feature_dim, num_classes, device):
  function train_epoch (line 49) | def train_epoch(model, dataloader):
  function eval_epoch (line 65) | def eval_epoch(model, dataloader):

FILE: setup.py
  class CMakeExtension (line 10) | class CMakeExtension(Extension):
    method __init__ (line 11) | def __init__(self, name, sourcedir=""):
  class CMakeBuild (line 16) | class CMakeBuild(build_ext):
    method run (line 17) | def run(self):
    method build_extension (line 32) | def build_extension(self, ext):

FILE: src/cpp/include/common/datatypes.h
  function class (line 32) | class DummyCudaEvent {
  function class (line 45) | class DummyCudaStream {
  function class (line 52) | class DummyCudaStreamGuard {
  type at (line 65) | typedef at::cuda::CUDAEvent CudaEvent;
  type at (line 66) | typedef at::cuda::CUDAStream CudaStream;
  type at (line 67) | typedef at::cuda::CUDAStreamGuard CudaStreamGuard;
  type DummyCudaEvent (line 72) | typedef DummyCudaEvent CudaEvent;
  type DummyCudaStream (line 73) | typedef DummyCudaStream CudaStream;
  type DummyCudaStreamGuard (line 74) | typedef DummyCudaStreamGuard CudaStreamGuard;
  type torch (line 91) | typedef torch::Tensor EdgeList;
  type torch (line 94) | typedef torch::Tensor Indices;
  type torch (line 97) | typedef torch::Tensor Gradients;
  type torch (line 100) | typedef torch::Tensor OptimizerState;
  type std (line 102) | typedef std::chrono::time_point<std::chrono::steady_clock> Timestamp;

FILE: src/cpp/include/common/exception.h
  type MariusRuntimeException (line 12) | struct MariusRuntimeException
  function MariusRuntimeException (line 17) | struct UndefinedTensorException : public MariusRuntimeException {
  function MariusRuntimeException (line 22) | struct NANTensorException : public MariusRuntimeException {
  function MariusRuntimeException (line 27) | struct OOMTensorException : public MariusRuntimeException {
  function MariusRuntimeException (line 32) | struct TensorSizeMismatchException : public MariusRuntimeException {
  function MariusRuntimeException (line 39) | struct UnexpectedNullPtrException : public MariusRuntimeException {

FILE: src/cpp/include/common/util.h
  function class (line 10) | class Timer {
  function start (line 29) | void start() {
  function stop (line 36) | void stop() {

FILE: src/cpp/include/configuration/config.h
  type NeighborSamplingConfig (line 16) | struct NeighborSamplingConfig {
  type OptimizerConfig (line 22) | struct OptimizerConfig {
  type InitConfig (line 27) | struct InitConfig {
  type LossConfig (line 35) | struct LossConfig {
  type LayerConfig (line 40) | struct LayerConfig {
  type EncoderConfig (line 52) | struct EncoderConfig {
  type DecoderConfig (line 60) | struct DecoderConfig {
  type StorageBackendConfig (line 66) | struct StorageBackendConfig {
  type DatasetConfig (line 71) | struct DatasetConfig {
  type NegativeSamplingConfig (line 84) | struct NegativeSamplingConfig {
  type PipelineConfig (line 92) | struct PipelineConfig {
  type CheckpointConfig (line 108) | struct CheckpointConfig {
  type ModelConfig (line 115) | struct ModelConfig {
  type StorageConfig (line 125) | struct StorageConfig {
  type TrainingConfig (line 142) | struct TrainingConfig {
  type EvaluationConfig (line 155) | struct EvaluationConfig {
  type MariusConfig (line 164) | struct MariusConfig {

FILE: src/cpp/include/configuration/constants.h
  function namespace (line 16) | namespace PathConstants {

FILE: src/cpp/include/configuration/options.h
  type class (line 12) | enum class
  type class (line 16) | enum class
  type class (line 20) | enum class
  type class (line 24) | enum class
  type class (line 28) | enum class
  type class (line 32) | enum class
  type class (line 36) | enum class
  type class (line 44) | enum class
  type class (line 48) | enum class
  type class (line 52) | enum class
  type class (line 56) | enum class
  type class (line 60) | enum class
  type class (line 64) | enum class
  type class (line 68) | enum class
  type class (line 72) | enum class
  type class (line 76) | enum class
  type class (line 80) | enum class
  type class (line 84) | enum class
  type InitOptions (line 92) | struct InitOptions {
  function InitOptions (line 96) | struct ConstantInitOptions : InitOptions {
  function InitOptions (line 102) | struct UniformInitOptions : InitOptions {
  function InitOptions (line 108) | struct NormalInitOptions : InitOptions {
  type LossOptions (line 115) | struct LossOptions {
  function LossOptions (line 121) | struct RankingLossOptions : LossOptions {
  type OptimizerOptions (line 126) | struct OptimizerOptions {
  function OptimizerOptions (line 132) | struct AdagradOptions : OptimizerOptions {
  function OptimizerOptions (line 139) | struct AdamOptions : OptimizerOptions {
  type LayerOptions (line 147) | struct LayerOptions {
  function LayerOptions (line 151) | struct EmbeddingLayerOptions : LayerOptions {}
  function LayerOptions (line 153) | struct FeatureLayerOptions : LayerOptions {}
  function LayerOptions (line 155) | struct DenseLayerOptions : LayerOptions {
  function LayerOptions (line 159) | struct ReductionLayerOptions : LayerOptions {
  function LayerOptions (line 163) | struct GNNLayerOptions : LayerOptions {
  function GNNLayerOptions (line 168) | struct GraphSageLayerOptions : GNNLayerOptions {
  function GNNLayerOptions (line 172) | struct GATLayerOptions : GNNLayerOptions {
  type DecoderOptions (line 180) | struct DecoderOptions {
  function DecoderOptions (line 184) | struct EdgeDecoderOptions : DecoderOptions {
  type StorageOptions (line 190) | struct StorageOptions {
  function StorageOptions (line 195) | struct PartitionBufferOptions : StorageOptions {
  type NeighborSamplingOptions (line 206) | struct NeighborSamplingOptions {
  function NeighborSamplingOptions (line 210) | struct UniformSamplingOptions : NeighborSamplingOptions {
  function NeighborSamplingOptions (line 214) | struct DropoutSamplingOptions : NeighborSamplingOptions {

FILE: src/cpp/include/data/batch.h
  type class (line 17) | enum class
  function class (line 32) | class Batch {

FILE: src/cpp/include/data/dataloader.h
  function class (line 22) | class DataLoader {
  function getNumEdges (line 180) | int64_t getNumEdges() { return graph_storage_->getNumEdges(); }
  function getEpochsProcessed (line 182) | int64_t getEpochsProcessed() { return epochs_processed_; }
  function getBatchesProcessed (line 184) | int64_t getBatchesProcessed() { return batches_processed_; }
  function isTrain (line 186) | bool isTrain() { return train_; }
  function setTrainSet (line 191) | void setTrainSet() {
  function setValidationSet (line 207) | void setValidationSet() {
  function setTestSet (line 220) | void setTestSet() {
  function setEncode (line 233) | void setEncode() {

FILE: src/cpp/include/data/graph.h
  function class (line 16) | class MariusGraph {
  function class (line 108) | class DENSEGraph : public MariusGraph {

FILE: src/cpp/include/data/samplers/edge.h
  function class (line 13) | class EdgeSampler {
  function class (line 27) | class RandomEdgeSampler : public EdgeSampler {

FILE: src/cpp/include/data/samplers/negative.h
  function class (line 31) | class NegativeSampler {
  function class (line 45) | class CorruptNodeNegativeSampler : public NegativeSampler {
  function class (line 59) | class CorruptRelNegativeSampler : public NegativeSampler {
  function class (line 70) | class NegativeEdgeSampler : public NegativeSampler {

FILE: src/cpp/include/data/samplers/neighbor.h
  function class (line 32) | class NeighborSampler {
  function class (line 47) | class LayeredNeighborSampler : public NeighborSampler {

FILE: src/cpp/include/nn/decoders/decoder.h
  function class (line 12) | class Decoder {

FILE: src/cpp/include/nn/decoders/edge/comparators.h
  function class (line 13) | class Comparator {
  function class (line 19) | class L2Compare : public Comparator {
  function class (line 26) | class CosineCompare : public Comparator {
  function class (line 33) | class DotCompare : public Comparator {

FILE: src/cpp/include/nn/decoders/edge/edge_decoder.h
  function class (line 13) | class EdgeDecoder : public Decoder {

FILE: src/cpp/include/nn/decoders/edge/relation_operators.h
  function class (line 11) | class RelationOperator {
  function class (line 17) | class HadamardOperator : public RelationOperator {
  function class (line 22) | class ComplexHadamardOperator : public RelationOperator {
  function class (line 27) | class TranslationOperator : public RelationOperator {
  function class (line 32) | class NoOp : public RelationOperator {

FILE: src/cpp/include/nn/decoders/node/node_decoder.h
  function class (line 10) | class NodeDecoder : public Decoder {

FILE: src/cpp/include/nn/encoders/encoder.h
  function class (line 11) | class GeneralEncoder : public torch::nn::Cloneable<GeneralEncoder> {

FILE: src/cpp/include/nn/layers/embedding/embedding.h
  function class (line 12) | class EmbeddingLayer : public Layer {

FILE: src/cpp/include/nn/layers/feature/feature.h
  function class (line 11) | class FeatureLayer : public Layer {

FILE: src/cpp/include/nn/layers/gnn/gat_layer.h
  function class (line 10) | class GATLayer : public GNNLayer {

FILE: src/cpp/include/nn/layers/gnn/gcn_layer.h
  function class (line 10) | class GCNLayer : public GNNLayer {

FILE: src/cpp/include/nn/layers/gnn/gnn_layer.h
  function class (line 14) | class GNNLayer : public Layer {

FILE: src/cpp/include/nn/layers/gnn/graph_sage_layer.h
  function class (line 10) | class GraphSageLayer : public GNNLayer {

FILE: src/cpp/include/nn/layers/gnn/rgcn_layer.h
  function class (line 10) | class RGCNLayer : public GNNLayer {

FILE: src/cpp/include/nn/layers/layer.h
  function virtual (line 22) | virtual ~Layer(){}

FILE: src/cpp/include/nn/layers/reduction/concat.h
  function class (line 11) | class ConcatReduction : public ReductionLayer {

FILE: src/cpp/include/nn/layers/reduction/linear.h
  function class (line 11) | class LinearReduction : public ReductionLayer {

FILE: src/cpp/include/nn/layers/reduction/reduction_layer.h
  function class (line 15) | class ReductionLayer : public Layer {

FILE: src/cpp/include/nn/loss.h
  function class (line 21) | class LossFunction {
  function class (line 33) | class SoftmaxCrossEntropy : public LossFunction {
  function class (line 43) | class RankingLoss : public LossFunction {
  function class (line 57) | class CrossEntropyLoss : public LossFunction {
  function class (line 67) | class BCEAfterSigmoidLoss : public LossFunction {
  function class (line 77) | class BCEWithLogitsLoss : public LossFunction {
  function class (line 87) | class MSELoss : public LossFunction {
  function class (line 97) | class SoftPlusLoss : public LossFunction {

FILE: src/cpp/include/nn/optim.h
  function class (line 11) | class Optimizer {
  function class (line 33) | class SGDOptimizer : public Optimizer {
  function class (line 52) | class AdagradOptimizer : public Optimizer {
  function class (line 79) | class AdamOptimizer : public Optimizer {

FILE: src/cpp/include/nn/regularizer.h
  function class (line 10) | class Regularizer {
  function class (line 17) | class NormRegularizer : public Regularizer {

FILE: src/cpp/include/pipeline/evaluator.h
  function class (line 17) | class Evaluator {
  function class (line 30) | class PipelineEvaluator : public Evaluator {
  function class (line 39) | class SynchronousEvaluator : public Evaluator {

FILE: src/cpp/include/pipeline/graph_encoder.h
  function class (line 12) | class GraphEncoder {
  function class (line 25) | class PipelineGraphEncoder : public GraphEncoder {
  function class (line 34) | class SynchronousGraphEncoder : public GraphEncoder {

FILE: src/cpp/include/pipeline/pipeline.h
  function class (line 18) | class Worker {
  function class (line 57) | class UpdateBatchWorker : public Worker {
  function class (line 64) | class WriteNodesWorker : public Worker {
  function class (line 71) | class Pipeline {

FILE: src/cpp/include/pipeline/pipeline_cpu.h
  function class (line 11) | class ComputeWorkerCPU : public Worker {
  function class (line 18) | class EncodeNodesWorkerCPU : public Worker {
  function class (line 27) | class PipelineCPU : public Pipeline {

FILE: src/cpp/include/pipeline/pipeline_gpu.h
  function class (line 11) | class BatchToDeviceWorker : public Worker {
  function class (line 45) | class PipelineGPU : public Pipeline {

FILE: src/cpp/include/pipeline/queue.h
  function unlock (line 83) | void unlock() { mutex_->unlock(); }
  function flush (line 85) | void flush() {
  function size (line 91) | int size() { return queue_.size(); }
  function isFull (line 93) | bool isFull() { return queue_.size() == max_size_; }
  function isEmpty (line 95) | bool isEmpty() { return queue_.size() == 0; }
  function getMaxSize (line 97) | int getMaxSize() { return max_size_; }
  type typename (line 99) | typedef typename std::deque<T> queue_type;
  type typename (line 101) | typedef typename queue_type::iterator iterator;
  type typename (line 102) | typedef typename queue_type::const_iterator const_iterator;
  function iterator (line 104) | inline iterator begin() noexcept { return queue_.begin(); }
  function iterator (line 108) | inline iterator end() noexcept { return queue_.end(); }

FILE: src/cpp/include/pipeline/trainer.h
  function class (line 14) | class Trainer {
  function class (line 28) | class PipelineTrainer : public Trainer {
  function class (line 37) | class SynchronousTrainer : public Trainer {

FILE: src/cpp/include/reporting/logger.h
  function class (line 18) | class MariusLogger {

FILE: src/cpp/include/reporting/reporting.h
  function class (line 10) | class Metric {
  function class (line 18) | class RankingMetric : public Metric {
  function class (line 23) | class HitskMetric : public RankingMetric {
  function class (line 32) | class MeanRankMetric : public RankingMetric {
  function class (line 39) | class MeanReciprocalRankMetric : public RankingMetric {
  function class (line 46) | class ClassificationMetric : public Metric {
  function class (line 51) | class CategoricalAccuracyMetric : public ClassificationMetric {
  function class (line 58) | class Reporter {
  function report (line 119) | void report() override;

FILE: src/cpp/include/storage/buffer.h
  function class (line 11) | class Partition {
  function class (line 41) | class PartitionedFile {
  function class (line 62) | class LookaheadBlock {
  function class (line 89) | class AsyncWriteBlock {
  function class (line 116) | class PartitionBuffer {

FILE: src/cpp/include/storage/checkpointer.h
  type CheckpointMeta (line 12) | struct CheckpointMeta {
  function class (line 23) | class Checkpointer {

FILE: src/cpp/include/storage/graph_storage.h
  type GraphModelStoragePtrs (line 12) | struct GraphModelStoragePtrs {
  type InMemorySubgraphState (line 32) | struct InMemorySubgraphState {
  function class (line 43) | class GraphModelStorage {

FILE: src/cpp/include/storage/storage.h
  function class (line 35) | class Storage {
  function class (line 89) | class PartitionBufferStorage : public Storage {
  function load (line 168) | void load() override;

FILE: src/cpp/python_bindings/configuration/config_wrap.cpp
  function init_config (line 4) | void init_config(py::module &m) {

FILE: src/cpp/python_bindings/configuration/options_wrap.cpp
  function init_options (line 4) | void init_options(py::module &m) {

FILE: src/cpp/python_bindings/configuration/wrap.cpp
  function PYBIND11_MODULE (line 7) | PYBIND11_MODULE(_config, m) {

FILE: src/cpp/python_bindings/manager/marius_wrap.cpp
  function init_marius (line 8) | void init_marius(py::module &m) {

FILE: src/cpp/python_bindings/manager/wrap.cpp
  function PYBIND11_MODULE (line 5) | PYBIND11_MODULE(_manager, m) {

FILE: src/cpp/python_bindings/nn/activation_wrap.cpp
  function init_activation (line 6) | void init_activation(py::module &m) { m.def("apply_activation", &apply_a...

FILE: src/cpp/python_bindings/nn/decoders/decoder_wrap.cpp
  class PyDecoder (line 8) | class PyDecoder : Decoder {
  function init_decoder (line 12) | void init_decoder(py::module &m) { py::class_<Decoder, PyDecoder, shared...

FILE: src/cpp/python_bindings/nn/decoders/edge/comparators_wrap.cpp
  class PyComparator (line 8) | class PyComparator : Comparator {
  function init_comparators (line 16) | void init_comparators(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/edge/complex_wrap.cpp
  function init_complex (line 8) | void init_complex(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/edge/distmult_wrap.cpp
  function init_distmult (line 5) | void init_distmult(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/edge/edge_decoder_wrap.cpp
  function init_edge_decoder (line 8) | void init_edge_decoder(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/edge/relation_operators_wrap.cpp
  class PyRelationOperator (line 9) | class PyRelationOperator : RelationOperator {
  function init_relation_operators (line 17) | void init_relation_operators(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/edge/transe_wrap.cpp
  function init_transe (line 8) | void init_transe(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/node/node_decoder_wrap.cpp
  function init_node_decoder (line 9) | void init_node_decoder(py::module &m) {

FILE: src/cpp/python_bindings/nn/decoders/node/noop_node_decoder.cpp
  function init_noop_node_decoder (line 9) | void init_noop_node_decoder(py::module &m) {

FILE: src/cpp/python_bindings/nn/encoders/encoder_wrap.cpp
  function init_encoder (line 5) | void init_encoder(py::module &m) {

FILE: src/cpp/python_bindings/nn/initialization_wrap.cpp
  function init_initialization (line 4) | void init_initialization(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/embedding/embedding_wrap.cpp
  function init_embedding_layer (line 4) | void init_embedding_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/feature/feature_wrap.cpp
  function init_feature_layer (line 8) | void init_feature_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/gat_layer_wrap.cpp
  function init_gat_layer (line 8) | void init_gat_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/gcn_layer_wrap.cpp
  function init_gcn_layer (line 8) | void init_gcn_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/gnn_layer_wrap.cpp
  class PyGNNLayer (line 8) | class PyGNNLayer : GNNLayer {
    method forward (line 11) | torch::Tensor forward(torch::Tensor inputs, DENSEGraph dense_graph, bo...
  function init_gnn_layer (line 16) | void init_gnn_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/graph_sage_layer_wrap.cpp
  function init_graph_sage_layer (line 8) | void init_graph_sage_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/layer_helpers_wrap.cpp
  function init_layer_helpers (line 5) | void init_layer_helpers(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/gnn/rgcn_layer_wrap.cpp
  function init_rgcn_layer (line 8) | void init_rgcn_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/layer_wrap.cpp
  class PyLayer (line 10) | class PyLayer : Layer {
  function init_layer (line 15) | void init_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/reduction/concat_wrap.cpp
  function init_concat_reduction_layer (line 8) | void init_concat_reduction_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/reduction/linear_wrap.cpp
  function init_linear_reduction_layer (line 8) | void init_linear_reduction_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/layers/reduction/reduction_layer_wrap.cpp
  class PyReductionLayer (line 8) | class PyReductionLayer : ReductionLayer {
    method forward (line 11) | torch::Tensor forward(std::vector<torch::Tensor> inputs) override { PY...
  function init_reduction_layer (line 14) | void init_reduction_layer(py::module &m) {

FILE: src/cpp/python_bindings/nn/loss_wrap.cpp
  class PyLossFunction (line 6) | class PyLossFunction : LossFunction {
  function init_loss (line 14) | void init_loss(py::module &m) {

FILE: src/cpp/python_bindings/nn/model_wrap.cpp
  class PyModel (line 10) | class PyModel : Model {
  function init_model (line 15) | void init_model(py::module &m) {

FILE: src/cpp/python_bindings/nn/optim_wrap.cpp
  function PYBIND11_OVERRIDE_PURE_NAME (line 12) | PYBIND11_OVERRIDE_PURE_NAME(void, Optimizer, "step", step); }

FILE: src/cpp/python_bindings/nn/regularizer_wrap.cpp
  class PyRegularizer (line 6) | class PyRegularizer : Regularizer {
  function init_regularizer (line 14) | void init_regularizer(py::module &m) {

FILE: src/cpp/python_bindings/nn/wrap.cpp
  function PYBIND11_MODULE (line 57) | PYBIND11_MODULE(_nn, m) {

FILE: src/cpp/python_bindings/pipeline/evaluator_wrap.cpp
  class PyEvaluator (line 7) | class PyEvaluator : Evaluator {
    method evaluate (line 10) | void evaluate(bool validation) override { PYBIND11_OVERRIDE_PURE(void,...
  function init_evaluator (line 13) | void init_evaluator(py::module &m) {

FILE: src/cpp/python_bindings/pipeline/graph_encoder_wrap.cpp
  class PyGraphEncoder (line 7) | class PyGraphEncoder : GraphEncoder {
    method encode (line 10) | void encode(bool separate_layers) override { PYBIND11_OVERRIDE_PURE(vo...
  function init_graph_encoder (line 13) | void init_graph_encoder(py::module &m) {

FILE: src/cpp/python_bindings/pipeline/trainer_wrap.cpp
  class PyTrainer (line 7) | class PyTrainer : Trainer {
    method train (line 10) | void train(int num_epochs = 1) override { PYBIND11_OVERRIDE_PURE(void,...
  function init_trainer (line 13) | void init_trainer(py::module &m) {

FILE: src/cpp/python_bindings/pipeline/wrap.cpp
  function PYBIND11_MODULE (line 8) | PYBIND11_MODULE(_pipeline, m) {

FILE: src/cpp/python_bindings/reporting/reporting_wrap.cpp
  function PYBIND11_OVERRIDE_PURE_NAME (line 7) | PYBIND11_OVERRIDE_PURE_NAME(void, Reporter, "report", report); }

FILE: src/cpp/python_bindings/reporting/wrap.cpp
  function PYBIND11_MODULE (line 6) | PYBIND11_MODULE(_report, m) {

FILE: src/cpp/python_bindings/storage/graph_storage_wrap.cpp
  function init_graph_storage (line 4) | void init_graph_storage(py::module &m) {

FILE: src/cpp/python_bindings/storage/io_wrap.cpp
  function init_io (line 7) | void init_io(py::module &m) {

FILE: src/cpp/python_bindings/storage/storage_wrap.cpp
  class PyStorage (line 7) | class PyStorage : Storage {
    method indexRead (line 11) | torch::Tensor indexRead(torch::Tensor indices) override { PYBIND11_OVE...
    method indexAdd (line 13) | void indexAdd(torch::Tensor indices, torch::Tensor values) override { ...
    method range (line 15) | torch::Tensor range(int64_t offset, int64_t n) override { PYBIND11_OVE...
    method indexPut (line 17) | void indexPut(torch::Tensor indices, torch::Tensor values) override { ...
    method rangePut (line 19) | void rangePut(int64_t offset, int64_t n, torch::Tensor values) overrid...
    method load (line 21) | void load() override { PYBIND11_OVERRIDE_PURE(void, Storage, load); }
    method write (line 23) | void write() override { PYBIND11_OVERRIDE_PURE(void, Storage, write); }
    method unload (line 25) | void unload(bool write) override { PYBIND11_OVERRIDE_PURE(void, Storag...
    method shuffle (line 27) | void shuffle() override { PYBIND11_OVERRIDE_PURE(void, Storage, shuffl...
    method sort (line 29) | void sort(bool src) override { PYBIND11_OVERRIDE_PURE(void, Storage, s...
  function init_storage (line 32) | void init_storage(py::module &m) {

FILE: src/cpp/python_bindings/storage/wrap.cpp
  function PYBIND11_MODULE (line 8) | PYBIND11_MODULE(_storage, m) {

FILE: src/cpp/src/common/util.cpp
  function assert_no_nans (line 14) | void assert_no_nans(torch::Tensor values) {
  function assert_no_neg (line 20) | void assert_no_neg(torch::Tensor values) {
  function assert_in_range (line 26) | void assert_in_range(torch::Tensor values, int64_t start, int64_t end) {
  function process_mem_usage (line 32) | void process_mem_usage() {
  function pread_wrapper (line 89) | int64_t pread_wrapper(int fd, void *buf, int64_t count, int64_t offset) {
  function pwrite_wrapper (line 109) | int64_t pwrite_wrapper(int fd, const void *buf, int64_t count, int64_t o...
  function transfer_tensor (line 129) | torch::Tensor transfer_tensor(torch::Tensor input, torch::Device device,...
  function get_dtype_size_wrapper (line 147) | int64_t get_dtype_size_wrapper(torch::Dtype dtype_) {
  function get_directory (line 168) | std::string get_directory(std::string filename) {
  function map_tensors (line 180) | std::tuple<torch::Tensor, std::vector<torch::Tensor>> map_tensors(std::v...
  function apply_tensor_map (line 209) | std::vector<torch::Tensor> apply_tensor_map(torch::Tensor map, std::vect...

FILE: src/cpp/src/configuration/config.cpp
  function check_missing (line 10) | bool check_missing(pyobj python_object) {
  function T (line 25) | T cast_helper(pyobj python_object) {
  function initNeighborSamplingConfig (line 36) | shared_ptr<NeighborSamplingConfig> initNeighborSamplingConfig(pyobj pyth...
  function initInitConfig (line 61) | shared_ptr<InitConfig> initInitConfig(pyobj python_object) {
  function initOptimizerConfig (line 89) | shared_ptr<OptimizerConfig> initOptimizerConfig(pyobj python_config) {
  function initDatasetConfig (line 129) | shared_ptr<DatasetConfig> initDatasetConfig(pyobj python_config) {
  function initLayerConfig (line 146) | shared_ptr<LayerConfig> initLayerConfig(pyobj python_config) {
  function initEncoderConfig (line 231) | shared_ptr<EncoderConfig> initEncoderConfig(pyobj python_config) {
  function initDecoderConfig (line 277) | shared_ptr<DecoderConfig> initDecoderConfig(pyobj python_config) {
  function initLossConfig (line 301) | shared_ptr<LossConfig> initLossConfig(pyobj python_config) {
  function initStorageBackendConfig (line 326) | shared_ptr<StorageBackendConfig> initStorageBackendConfig(pyobj python_c...
  function initNegativeSamplingConfig (line 358) | shared_ptr<NegativeSamplingConfig> initNegativeSamplingConfig(pyobj pyth...
  function initPipelineConfig (line 381) | shared_ptr<PipelineConfig> initPipelineConfig(pyobj python_config) {
  function initCheckpointConfig (line 403) | shared_ptr<CheckpointConfig> initCheckpointConfig(pyobj python_config) {
  function initModelConfig (line 416) | shared_ptr<ModelConfig> initModelConfig(pyobj python_config) {
  function initStorageConfig (line 430) | shared_ptr<StorageConfig> initStorageConfig(pyobj python_config) {
  function initTrainingConfig (line 464) | shared_ptr<TrainingConfig> initTrainingConfig(pyobj python_config) {
  function initEvaluationConfig (line 480) | shared_ptr<EvaluationConfig> initEvaluationConfig(pyobj python_config) {
  function initMariusConfig (line 491) | shared_ptr<MariusConfig> initMariusConfig(pyobj python_config) {
  function loadConfig (line 502) | shared_ptr<MariusConfig> loadConfig(string config_path, bool save) {

FILE: src/cpp/src/configuration/options.cpp
  function LearningTask (line 7) | LearningTask getLearningTask(std::string string_val) {
  function InitDistribution (line 19) | InitDistribution getInitDistribution(std::string string_val) {
  function LossFunctionType (line 41) | LossFunctionType getLossFunctionType(std::string string_val) {
  function LossReduction (line 63) | LossReduction getLossReduction(std::string string_val) {
  function ActivationFunction (line 75) | ActivationFunction getActivationFunction(std::string string_val) {
  function OptimizerType (line 89) | OptimizerType getOptimizerType(std::string string_val) {
  function ReductionLayerType (line 105) | ReductionLayerType getReductionLayerType(std::string string_val) {
  function DenseLayerType (line 119) | DenseLayerType getDenseLayerType(std::string string_val) {
  function LayerType (line 133) | LayerType getLayerType(std::string string_val) {
  function GNNLayerType (line 153) | GNNLayerType getGNNLayerType(std::string string_val) {
  function GraphSageAggregator (line 171) | GraphSageAggregator getGraphSageAggregator(std::string string_val) {
  function DecoderType (line 183) | DecoderType getDecoderType(std::string string_val) {
  function EdgeDecoderMethod (line 199) | EdgeDecoderMethod getEdgeDecoderMethod(std::string string_val) {
  function StorageBackend (line 219) | StorageBackend getStorageBackend(std::string string_val) {
  function EdgeBucketOrdering (line 235) | EdgeBucketOrdering getEdgeBucketOrderingEnum(std::string string_val) {
  function NodePartitionOrdering (line 253) | NodePartitionOrdering getNodePartitionOrderingEnum(std::string string_va...
  function NeighborSamplingLayer (line 267) | NeighborSamplingLayer getNeighborSamplingLayer(std::string string_val) {
  function LocalFilterMode (line 281) | LocalFilterMode getLocalFilterMode(std::string string_val) {
  function getDtype (line 293) | torch::Dtype getDtype(std::string string_val) {
  function getLogLevel (line 309) | spdlog::level::level_enum getLogLevel(std::string string_val) {

FILE: src/cpp/src/configuration/util.cpp
  function devices_from_config (line 7) | std::vector<torch::Device> devices_from_config(std::shared_ptr<StorageCo...

FILE: src/cpp/src/data/graph.cpp
  function Indices (line 56) | Indices MariusGraph::getNodeIDs() { return node_ids_; }
  function Indices (line 58) | Indices MariusGraph::getEdges(bool incoming) {
  function Indices (line 66) | Indices MariusGraph::getRelationIDs(bool incoming) {
  function Indices (line 78) | Indices MariusGraph::getNeighborOffsets(bool incoming) {
  function Indices (line 86) | Indices MariusGraph::getNumNeighbors(bool incoming) {
  function Indices (line 322) | Indices DENSEGraph::getNeighborIDs(bool incoming, bool global_ids) {

FILE: src/cpp/src/data/ordering.cpp
  function getEdgeBucketOrdering (line 12) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getEdgeBucketOr...
  function getNodePartitionOrdering (line 36) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getNodePartitio...
  function convertEdgeBucketOrderToTensors (line 55) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> convertEdgeBuck...
  function getBetaOrderingHelper (line 78) | vector<vector<int>> getBetaOrderingHelper(int num_partitions, int buffer...
  function greedyAssignEdgeBucketsToBuffers (line 128) | vector<vector<std::pair<int, int>>> greedyAssignEdgeBucketsToBuffers(vec...
  function randomlyAssignEdgeBucketsToBuffers (line 150) | vector<vector<std::pair<int, int>>> randomlyAssignEdgeBucketsToBuffers(v...
  function getTwoLevelBetaOrdering (line 241) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getTwoLevelBeta...
  function getDispersedNodePartitionOrdering (line 294) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getDispersedNod...
  function getSequentialNodePartitionOrdering (line 389) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getSequentialNo...
  function getCustomNodePartitionOrdering (line 412) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getCustomNodePa...
  function getCustomEdgeBucketOrdering (line 418) | std::tuple<vector<torch::Tensor>, vector<torch::Tensor>> getCustomEdgeBu...

FILE: src/cpp/src/data/samplers/edge.cpp
  function EdgeList (line 12) | EdgeList RandomEdgeSampler::getEdges(shared_ptr<Batch> batch) {

FILE: src/cpp/src/data/samplers/negative.cpp
  function batch_sample (line 7) | std::tuple<torch::Tensor, torch::Tensor> batch_sample(torch::Tensor edge...
  function deg_negative_local_filter (line 21) | torch::Tensor deg_negative_local_filter(torch::Tensor deg_sample_indices...
  function compute_filter_corruption (line 41) | torch::Tensor compute_filter_corruption(shared_ptr<MariusGraph> graph, t...
  function compute_filter_corruption_cpu (line 50) | torch::Tensor compute_filter_corruption_cpu(shared_ptr<MariusGraph> grap...
  function compute_filter_corruption_gpu (line 197) | torch::Tensor compute_filter_corruption_gpu(shared_ptr<MariusGraph> grap...
  function apply_score_filter (line 306) | torch::Tensor apply_score_filter(torch::Tensor scores, torch::Tensor fil...

FILE: src/cpp/src/data/samplers/neighbor.cpp
  function sample_all_gpu (line 9) | std::tuple<torch::Tensor, torch::Tensor> sample_all_gpu(torch::Tensor ed...
  function sample_all_cpu (line 19) | std::tuple<torch::Tensor, torch::Tensor> sample_all_cpu(torch::Tensor ed...
  function sample_uniform_gpu (line 80) | std::tuple<torch::Tensor, torch::Tensor> sample_uniform_gpu(torch::Tenso...
  function sample_uniform_cpu (line 104) | std::tuple<torch::Tensor, torch::Tensor> sample_uniform_cpu(torch::Tenso...
  function sample_dropout_gpu (line 236) | std::tuple<torch::Tensor, torch::Tensor> sample_dropout_gpu(torch::Tenso...
  function sample_dropout_cpu (line 255) | std::tuple<torch::Tensor, torch::Tensor> sample_dropout_cpu(torch::Tenso...
  function DENSEGraph (line 402) | DENSEGraph LayeredNeighborSampler::getNeighbors(torch::Tensor node_ids, ...

FILE: src/cpp/src/marius.cpp
  function encode_and_export (line 13) | void encode_and_export(shared_ptr<DataLoader> dataloader, shared_ptr<Mod...
  function marius_init (line 38) | std::tuple<shared_ptr<Model>, shared_ptr<GraphModelStorage>, shared_ptr<...
  function marius_train (line 105) | void marius_train(shared_ptr<MariusConfig> marius_config) {
  function marius_eval (line 165) | void marius_eval(shared_ptr<MariusConfig> marius_config) {
  function marius (line 187) | void marius(int argc, char *argv[]) {
  function main (line 207) | int main(int argc, char *argv[]) { marius(argc, argv); }

FILE: src/cpp/src/nn/activation.cpp
  function apply_activation (line 7) | torch::Tensor apply_activation(ActivationFunction activation_function, t...

FILE: src/cpp/src/nn/decoders/edge/comparators.cpp
  function pad_and_reshape (line 7) | torch::Tensor pad_and_reshape(torch::Tensor input, int num_chunks) {

FILE: src/cpp/src/nn/decoders/edge/decoder_methods.cpp
  function only_pos_forward (line 7) | std::tuple<torch::Tensor, torch::Tensor> only_pos_forward(shared_ptr<Edg...
  function neg_and_pos_forward (line 44) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> n...
  function node_corrupt_forward (line 57) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> n...
  function rel_corrupt_forward (line 116) | std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> r...

FILE: src/cpp/src/nn/encoders/encoder.cpp
  function device_ (line 29) | device_(torch::kCPU) {

FILE: src/cpp/src/nn/initialization.cpp
  function compute_fans (line 7) | std::tuple<int64_t, int64_t> compute_fans(std::vector<int64_t> shape) {
  function glorot_uniform (line 26) | torch::Tensor glorot_uniform(std::vector<int64_t> shape, std::tuple<int6...
  function glorot_normal (line 43) | torch::Tensor glorot_normal(std::vector<int64_t> shape, std::tuple<int64...
  function uniform_init (line 58) | torch::Tensor uniform_init(float scale_factor, std::vector<int64_t> shap...
  function normal_init (line 61) | torch::Tensor normal_init(float mean, float std, std::vector<int64_t> sh...
  function constant_init (line 65) | torch::Tensor constant_init(float constant, std::vector<int64_t> shape, ...
  function initialize_tensor (line 67) | torch::Tensor initialize_tensor(shared_ptr<InitConfig> init_config, std:...
  function initialize_subtensor (line 100) | torch::Tensor initialize_subtensor(shared_ptr<InitConfig> init_config, s...

FILE: src/cpp/src/nn/layers/gnn/layer_helpers.cpp
  function segment_ids_from_offsets (line 11) | torch::Tensor segment_ids_from_offsets(torch::Tensor segment_offsets, in...
  function segmented_sum (line 19) | torch::Tensor segmented_sum(torch::Tensor tensor, torch::Tensor segment_...
  function segmented_sum_with_offsets (line 27) | torch::Tensor segmented_sum_with_offsets(torch::Tensor tensor, torch::Te...
  function segmented_max_with_offsets (line 32) | torch::Tensor segmented_max_with_offsets(torch::Tensor tensor, torch::Te...
  function attention_softmax (line 44) | std::tuple<torch::Tensor, torch::Tensor> attention_softmax(torch::Tensor...

FILE: src/cpp/src/nn/loss.cpp
  function check_score_shapes (line 7) | void check_score_shapes(torch::Tensor pos_scores, torch::Tensor neg_scor...
  function to_one_hot (line 31) | torch::Tensor to_one_hot(torch::Tensor labels, int num_classes) {
  function scores_to_labels (line 37) | std::tuple<torch::Tensor, torch::Tensor> scores_to_labels(torch::Tensor ...
  function getLossFunction (line 177) | std::shared_ptr<LossFunction> getLossFunction(shared_ptr<LossConfig> con...

FILE: src/cpp/src/nn/model.cpp
  function initModelFromConfig (line 381) | shared_ptr<Model> initModelFromConfig(shared_ptr<ModelConfig> model_conf...

FILE: src/cpp/src/storage/buffer.cpp
  function Indices (line 457) | Indices PartitionBuffer::getRandomIds(int64_t size) { return torch::rand...

FILE: src/cpp/src/storage/checkpointer.cpp
  function CheckpointMeta (line 75) | CheckpointMeta Checkpointer::loadMetadata(string directory) {

FILE: src/cpp/src/storage/graph_storage.cpp
  function EdgeList (line 163) | EdgeList GraphModelStorage::getEdges(Indices indices) {
  function EdgeList (line 171) | EdgeList GraphModelStorage::getEdgesRange(int64_t start, int64_t size) {
  function Indices (line 181) | Indices GraphModelStorage::getRandomNodeIds(int64_t size) {
  function Indices (line 198) | Indices GraphModelStorage::getNodeIdsRange(int64_t start, int64_t size) {
  function OptimizerState (line 297) | OptimizerState GraphModelStorage::getNodeEmbeddingState(Indices indices) {
  function OptimizerState (line 305) | OptimizerState GraphModelStorage::getNodeEmbeddingStateRange(int64_t sta...
  function EdgeList (line 737) | EdgeList GraphModelStorage::merge_sorted_edge_buckets(EdgeList edges, to...

FILE: src/cpp/src/storage/io.cpp
  function initializeEdges (line 12) | std::tuple<shared_ptr<Storage>, shared_ptr<Storage>, shared_ptr<Storage>...
  function initializeNodeEmbeddings (line 154) | std::tuple<shared_ptr<Storage>, shared_ptr<Storage>> initializeNodeEmbed...
  function initializeNodeIds (line 226) | std::tuple<shared_ptr<Storage>, shared_ptr<Storage>, shared_ptr<Storage>...
  function initializeRelationFeatures (line 294) | shared_ptr<Storage> initializeRelationFeatures(shared_ptr<Model> model, ...
  function initializeNodeFeatures (line 311) | shared_ptr<Storage> initializeNodeFeatures(shared_ptr<Model> model, shar...
  function initializeNodeLabels (line 347) | shared_ptr<Storage> initializeNodeLabels(shared_ptr<Model> model, shared...
  function initializeStorageLinkPrediction (line 378) | shared_ptr<GraphModelStorage> initializeStorageLinkPrediction(shared_ptr...
  function initializeStorageNodeClassification (line 402) | shared_ptr<GraphModelStorage> initializeStorageNodeClassification(shared...
  function initializeStorage (line 433) | shared_ptr<GraphModelStorage> initializeStorage(shared_ptr<Model> model,...

FILE: src/cpp/src/storage/storage.cpp
  function renameFile (line 19) | void renameFile(string old_filename, string new_filename) {
  function copyFile (line 27) | void copyFile(string src_file, string dst_file) {
  function fileExists (line 40) | bool fileExists(string filename) {
  function createDir (line 49) | void createDir(string path, bool exist_ok) {

FILE: src/cuda/third_party/pytorch_scatter/segment_max.cpp
  function list2vec (line 4) | inline std::vector<int64_t> list2vec(const c10::List<int64_t> list) {
  class SegmentMaxCSR (line 16) | class SegmentMaxCSR : public torch::autograd::Function<SegmentMaxCSR> {
    method variable_list (line 18) | static variable_list forward(AutogradContext *ctx, Variable src,
    method variable_list (line 32) | static variable_list backward(AutogradContext *ctx, variable_list grad...
  function segment_max_csr (line 47) | std::tuple<torch::Tensor, torch::Tensor>

FILE: src/python/console_scripts/marius_eval.py
  function main (line 7) | def main():

FILE: src/python/console_scripts/marius_train.py
  function main (line 7) | def main():

FILE: src/python/distribution/generate_stubs.py
  function generate_stubs (line 6) | def generate_stubs(output_dir, module_name):
  function gen_all_stubs (line 23) | def gen_all_stubs(output_dir):

FILE: src/python/distribution/marius_env_info.py
  class MyDumper (line 10) | class MyDumper(yaml.Dumper):
    method increase_indent (line 11) | def increase_indent(self, flow=False, indentless=False):
  function get_os_info (line 15) | def get_os_info():
  function get_cpu_info (line 20) | def get_cpu_info():
  function get_gpu_info (line 33) | def get_gpu_info():
  function get_python_info (line 49) | def get_python_info():
  function get_cuda_info (line 75) | def get_cuda_info():
  function get_openmp_info (line 87) | def get_openmp_info():
  function get_pytorch_info (line 99) | def get_pytorch_info():
  function get_marius_info (line 112) | def get_marius_info():
  function get_pybind_info (line 132) | def get_pybind_info():
  function get_cmake_info (line 146) | def get_cmake_info():
  function main (line 158) | def main():

FILE: src/python/tools/configuration/constants.py
  class PathConstants (line 5) | class PathConstants:

FILE: src/python/tools/configuration/datatypes.py
  class InitOptions (line 8) | class InitOptions:
  class UniformInitOptions (line 13) | class UniformInitOptions(InitOptions):
    method __post_init__ (line 16) | def __post_init__(self):
  class NormalInitOptions (line 22) | class NormalInitOptions(InitOptions):
    method __post_init__ (line 26) | def __post_init__(self):
  class ConstantInitOptions (line 32) | class ConstantInitOptions(InitOptions):
  class LossOptions (line 37) | class LossOptions:
  class RankingLossOptions (line 42) | class RankingLossOptions(LossOptions):
  class OptimizerOptions (line 47) | class OptimizerOptions:
    method __post_init__ (line 50) | def __post_init__(self):
  class AdagradOptions (line 56) | class AdagradOptions(OptimizerOptions):
    method __post_init__ (line 63) | def __post_init__(self):
  class AdamOptions (line 74) | class AdamOptions(OptimizerOptions):
    method __post_init__ (line 82) | def __post_init__(self):
  class LayerOptions (line 93) | class LayerOptions:
  class EmbeddingLayerOptions (line 98) | class EmbeddingLayerOptions(LayerOptions):
  class FeatureLayerOptions (line 103) | class FeatureLayerOptions(LayerOptions):
  class DenseLayerOptions (line 108) | class DenseLayerOptions(LayerOptions):
  class ReductionLayerOptions (line 113) | class ReductionLayerOptions(LayerOptions):
  class GNNLayerOptions (line 118) | class GNNLayerOptions(LayerOptions):
  class GraphSageLayerOptions (line 124) | class GraphSageLayerOptions(GNNLayerOptions):
  class GATLayerOptions (line 130) | class GATLayerOptions(GNNLayerOptions):
    method __post_init__ (line 138) | def __post_init__(self):
  class DecoderOptions (line 144) | class DecoderOptions:
  class EdgeDecoderOptions (line 149) | class EdgeDecoderOptions(DecoderOptions):
  class StorageOptions (line 156) | class StorageOptions:
  class PartitionBufferOptions (line 161) | class PartitionBufferOptions(StorageOptions):
    method __post_init__ (line 171) | def __post_init__(self):
  class NeighborSamplingOptions (line 187) | class NeighborSamplingOptions:
  class UniformSamplingOptions (line 192) | class UniformSamplingOptions(NeighborSamplingOptions):
    method __post_init__ (line 195) | def __post_init__(self):
  class DropoutSamplingOptions (line 201) | class DropoutSamplingOptions(NeighborSamplingOptions):
    method __post_init__ (line 204) | def __post_init__(self):

FILE: src/python/tools/configuration/marius_config.py
  function get_model_dir_path (line 47) | def get_model_dir_path(dataset_dir):
  class NeighborSamplingConfig (line 60) | class NeighborSamplingConfig:
    method merge (line 65) | def merge(self, input_config: DictConfig):
  class OptimizerConfig (line 95) | class OptimizerConfig:
    method merge (line 99) | def merge(self, input_config: DictConfig):
  class InitConfig (line 129) | class InitConfig:
    method merge (line 133) | def merge(self, input_config: DictConfig):
  class LossConfig (line 162) | class LossConfig:
    method merge (line 166) | def merge(self, input_config: DictConfig):
  class LayerConfig (line 190) | class LayerConfig:
    method merge (line 201) | def merge(self, input_config: DictConfig):
  class EncoderConfig (line 258) | class EncoderConfig:
    method merge (line 266) | def merge(self, input_config: DictConfig):
  class DecoderConfig (line 313) | class DecoderConfig:
    method merge (line 318) | def merge(self, input_config: DictConfig):
  class ModelConfig (line 345) | class ModelConfig:
    method __post_init__ (line 354) | def __post_init__(self):
    method merge (line 358) | def merge(self, input_config: DictConfig):
  class StorageBackendConfig (line 396) | class StorageBackendConfig:
    method merge (line 400) | def merge(self, input_config: DictConfig):
  class DatasetConfig (line 424) | class DatasetConfig:
    method __post_init__ (line 437) | def __post_init__(self):
    method populate_dataset_stats (line 470) | def populate_dataset_stats(self):
    method merge (line 495) | def merge(self, input_config: DictConfig):
  class StorageConfig (line 514) | class StorageConfig:
    method __post_init__ (line 534) | def __post_init__(self):
    method merge (line 546) | def merge(self, input_config: DictConfig):
  class NegativeSamplingConfig (line 607) | class NegativeSamplingConfig:
    method __post_init__ (line 614) | def __post_init__(self):
    method merge (line 623) | def merge(self, input_config: DictConfig):
  class CheckpointConfig (line 649) | class CheckpointConfig:
    method merge (line 654) | def merge(self, input_config: DictConfig):
  class PipelineConfig (line 672) | class PipelineConfig:
    method __post_init__ (line 687) | def __post_init__(self):
    method merge (line 709) | def merge(self, input_config: DictConfig):
  class TrainingConfig (line 725) | class TrainingConfig:
    method __post_init__ (line 737) | def __post_init__(self):
    method merge (line 747) | def merge(self, input_config: DictConfig):
  class EvaluationConfig (line 777) | class EvaluationConfig:
    method __post_init__ (line 784) | def __post_init__(self):
    method merge (line 788) | def merge(self, input_config: DictConfig):
  class MariusConfig (line 813) | class MariusConfig:
    method __init__ (line 820) | def __init__(self):
    method __post_init__ (line 826) | def __post_init__(self):
  function type_safe_merge (line 836) | def type_safe_merge(base_config: MariusConfig, input_config: DictConfig):
  function initialize_model_dir (line 861) | def initialize_model_dir(output_config):
  function infer_model_dir (line 875) | def infer_model_dir(output_config):
  function load_config (line 899) | def load_config(input_config_path, save=False):

FILE: src/python/tools/configuration/validation.py
  function get_lines_in_file (line 10) | def get_lines_in_file(filepath):
  function validate_dataset_config (line 14) | def validate_dataset_config(output_config):
  function validate_storage_config (line 134) | def validate_storage_config(output_config):
  function check_encoder_layer_dimensions (line 186) | def check_encoder_layer_dimensions(output_config):
  function check_gnn_layers_alignment (line 257) | def check_gnn_layers_alignment(output_config):
  function retrieve_memory_info (line 276) | def retrieve_memory_info():
  function get_storage_overheads (line 282) | def get_storage_overheads(output_config):
  function check_full_graph_evaluation (line 305) | def check_full_graph_evaluation(output_config):

FILE: src/python/tools/db2graph/marius_db2graph.py
  function set_args (line 21) | def set_args():
  function config_parser_fn (line 48) | def config_parser_fn(config_name):
  function connect_to_db (line 137) | def connect_to_db(db_server, db_name, db_user, db_password, db_host):
  function validation_check_edge_entity_entity_queries (line 173) | def validation_check_edge_entity_entity_queries(edges_queries_list):
  function clean_token (line 231) | def clean_token(token):
  function get_init_fetch_size (line 243) | def get_init_fetch_size():
  function get_fetch_size (line 256) | def get_fetch_size(fetch_size, limit_fetch_size, mem_copy_used):
  function get_cursor (line 278) | def get_cursor(cnx, db_server, cursor_name):
  function post_processing (line 295) | def post_processing(output_dir, cnx, edges_queries_list, edge_rel_list, ...
  function main (line 380) | def main():

FILE: src/python/tools/marius_config_generator.py
  function output_config (line 12) | def output_config(config_dict, output_dir):
  function read_template (line 50) | def read_template(file):
  function set_up_files (line 70) | def set_up_files(output_directory):
  function update_dataset_stats (line 80) | def update_dataset_stats(dataset, arg_dict, config_dict):
  function update_stats (line 92) | def update_stats(stats, arg_dict, config_dict, opt="stats"):
  function update_data_path (line 112) | def update_data_path(dir, arg_dict):
  function set_args (line 161) | def set_args():
  function parse_args (line 272) | def parse_args(args, config_dict):
  function main (line 302) | def main():

FILE: src/python/tools/marius_postprocess.py
  function set_args (line 10) | def set_args():
  function main (line 47) | def main():

FILE: src/python/tools/marius_predict.py
  function str2bool (line 25) | def str2bool(v):
  function set_args (line 36) | def set_args():
  function get_metrics (line 228) | def get_metrics(config, args):
  function get_dtype (line 274) | def get_dtype(storage_backend, args):
  function get_columns (line 294) | def get_columns(config, args):
  function infer_input_shape (line 317) | def infer_input_shape(config, args):
  function get_nbrs_config (line 362) | def get_nbrs_config(config, args):
  function get_neg_config (line 390) | def get_neg_config(config, args):
  function preprocess_input_file (line 401) | def preprocess_input_file(config, args):
  function get_input_file_storage (line 490) | def get_input_file_storage(config, args):
  function run_predict (line 520) | def run_predict(args):
  function main (line 582) | def main():

FILE: src/python/tools/marius_preprocess.py
  function set_args (line 24) | def set_args():
  function main (line 133) | def main():

FILE: src/python/tools/postprocess/in_memory_exporter.py
  function get_ordered_raw_ids (line 15) | def get_ordered_raw_ids(mapping_path):
  function save_df (line 28) | def save_df(output_df: pd.DataFrame, output_dir: Path, name: str, fmt: s...
  class InMemoryExporter (line 45) | class InMemoryExporter(object):
    method __init__ (line 46) | def __init__(self, model_dir: Path, fmt: str = "CSV", delim: str = ","...
    method export_node_embeddings (line 61) | def export_node_embeddings(self, output_dir: Path):
    method export_rel_embeddings (line 103) | def export_rel_embeddings(self, output_dir: Path):
    method export_model (line 143) | def export_model(self, output_dir: Path):
    method copy_model (line 161) | def copy_model(self, output_dir: Path):
    method export (line 171) | def export(self, output_dir: Path):

FILE: src/python/tools/prediction/link_prediction.py
  function infer_lp (line 4) | def infer_lp(

FILE: src/python/tools/prediction/node_classification.py
  function infer_nc (line 4) | def infer_nc(

FILE: src/python/tools/preprocess/converters/partitioners/partitioner.py
  class Partitioner (line 4) | class Partitioner(ABC):
    method __init__ (line 5) | def __init__(self):

FILE: src/python/tools/preprocess/converters/partitioners/spark_partitioner.py
  function get_partition_size (line 11) | def get_partition_size(nodes, num_partitions):
  function get_edge_buckets (line 16) | def get_edge_buckets(edges_df: DataFrame, partition_size):
  class SparkPartitioner (line 23) | class SparkPartitioner(Partitioner):
    method __init__ (line 24) | def __init__(self, spark, partitioned_evaluation):
    method partition_edges (line 30) | def partition_edges(self, train_edges_df, valid_edges_df, test_edges_d...

FILE: src/python/tools/preprocess/converters/partitioners/torch_partitioner.py
  function dataframe_to_tensor (line 8) | def dataframe_to_tensor(df):
  function partition_edges (line 12) | def partition_edges(edges, num_nodes, num_partitions, edge_weights=None):
  class TorchPartitioner (line 49) | class TorchPartitioner(Partitioner):
    method __init__ (line 50) | def __init__(self, partitioned_evaluation):
    method partition_edges (line 55) | def partition_edges(

FILE: src/python/tools/preprocess/converters/readers/pandas_readers.py
  class PandasDelimitedFileReader (line 9) | class PandasDelimitedFileReader(Reader):
    method __init__ (line 10) | def __init__(
    method read_single_file (line 44) | def read_single_file(self, file_path):
    method read (line 77) | def read(self):

FILE: src/python/tools/preprocess/converters/readers/reader.py
  class Reader (line 4) | class Reader(ABC):
    method __init__ (line 5) | def __init__(self):
    method read (line 9) | def read(self):

FILE: src/python/tools/preprocess/converters/readers/spark_readers.py
  class SparkDelimitedFileReader (line 9) | class SparkDelimitedFileReader(Reader):
    method __init__ (line 10) | def __init__(
    method read (line 67) | def read(self):

FILE: src/python/tools/preprocess/converters/spark_converter.py
  function remap_columns (line 29) | def remap_columns(df, has_rels):
  function get_nodes_df (line 36) | def get_nodes_df(edges_df):
  function get_relations_df (line 49) | def get_relations_df(edges_df):
  function assign_ids (line 62) | def assign_ids(df, col_id):
  function remap_edges (line 68) | def remap_edges(edges_df, nodes, rels):
  function write_df_to_csv (line 94) | def write_df_to_csv(df, output_filename):
  class SparkEdgeListConverter (line 101) | class SparkEdgeListConverter(object):
    method __init__ (line 102) | def __init__(
    method convert (line 169) | def convert(self):

FILE: src/python/tools/preprocess/converters/torch_constants.py
  class TorchConverterColumnKeys (line 5) | class TorchConverterColumnKeys(Enum):
    method __hash__ (line 11) | def __hash__(self) -> int:

FILE: src/python/tools/preprocess/converters/torch_converter.py
  function dataframe_to_tensor (line 19) | def dataframe_to_tensor(df):
  function apply_mapping_edges (line 23) | def apply_mapping_edges(input_edges, node_mapping_df, rel_mapping_df=None):
  function apply_mapping1d (line 69) | def apply_mapping1d(input_ids, mapping_df):
  function extract_tensors_from_df (line 80) | def extract_tensors_from_df(df, column_mappings):
  function map_edge_list_dfs (line 97) | def map_edge_list_dfs(
  function extract_tensor_from_tens (line 174) | def extract_tensor_from_tens(edges_tensor, column_mappings):
  function map_edge_lists (line 191) | def map_edge_lists(
  function split_edges (line 374) | def split_edges(edges, edges_weights, splits):
  class TorchEdgeListConverter (line 428) | class TorchEdgeListConverter(object):
    method __init__ (line 429) | def __init__(
    method convert (line 626) | def convert(self):

FILE: src/python/tools/preprocess/converters/writers/spark_writer.py
  function convert_to_binary (line 26) | def convert_to_binary(input_filename, output_filename):
  function merge_csvs (line 37) | def merge_csvs(input_directory, output_file):
  function write_df_to_csv (line 50) | def write_df_to_csv(df, output_filename):
  function write_partitioned_df_to_csv (line 56) | def write_partitioned_df_to_csv(partition_triples, num_partitions, outpu...
  class SparkWriter (line 111) | class SparkWriter(object):
    method __init__ (line 112) | def __init__(self, spark, output_dir, partitioned_evaluation):
    method write_to_csv (line 119) | def write_to_csv(self, train_edges_df, valid_edges_df, test_edges_df, ...
    method write_to_binary (line 193) | def write_to_binary(self, train_edges_df, valid_edges_df, test_edges_d...

FILE: src/python/tools/preprocess/converters/writers/torch_writer.py
  class TorchWriter (line 10) | class TorchWriter(object):
    method __init__ (line 11) | def __init__(self, output_dir, partitioned_evaluation):
    method write_to_binary (line 17) | def write_to_binary(

FILE: src/python/tools/preprocess/custom.py
  class CustomLinkPredictionDataset (line 14) | class CustomLinkPredictionDataset(LinkPredictionDataset):
    method __init__ (line 15) | def __init__(
    method download (line 36) | def download(self, overwrite=False):
    method preprocess (line 39) | def preprocess(

FILE: src/python/tools/preprocess/dataset.py
  class Dataset (line 9) | class Dataset(ABC):
    method __init__ (line 32) | def __init__(self, output_directory, spark=False):
    method download (line 45) | def download(self, overwrite=False):
    method preprocess (line 49) | def preprocess(self) -> DatasetConfig:
  class NodeClassificationDataset (line 53) | class NodeClassificationDataset(Dataset):
    method __init__ (line 54) | def __init__(self, output_directory, spark):
  class LinkPredictionDataset (line 64) | class LinkPredictionDataset(Dataset):
    method __init__ (line 65) | def __init__(self, output_directory, spark):
  class GraphClassificationDataset (line 78) | class GraphClassificationDataset(Dataset):

FILE: src/python/tools/preprocess/datasets/dataset_helpers.py
  function remap_nodes (line 4) | def remap_nodes(node_mapping, train_nodes, valid_nodes, test_nodes, feat...

FILE: src/python/tools/preprocess/datasets/fb15k.py
  class FB15K (line 8) | class FB15K(LinkPredictionDataset):
    method __init__ (line 17) | def __init__(self, output_directory: Path, spark=False):
    method download (line 23) | def download(self, overwrite=False):
    method preprocess (line 45) | def preprocess(

FILE: src/python/tools/preprocess/datasets/fb15k_237.py
  class FB15K237 (line 8) | class FB15K237(LinkPredictionDataset):
    method __init__ (line 19) | def __init__(self, output_directory: Path, spark=False):
    method download (line 25) | def download(self, overwrite=False):
    method preprocess (line 47) | def preprocess(

FILE: src/python/tools/preprocess/datasets/freebase86m.py
  class Freebase86m (line 8) | class Freebase86m(LinkPredictionDataset):
    method __init__ (line 15) | def __init__(self, output_directory: Path, spark=False):
    method download (line 21) | def download(self, overwrite=False):
    method preprocess (line 43) | def preprocess(

FILE: src/python/tools/preprocess/datasets/friendster.py
  class Friendster (line 8) | class Friendster(LinkPredictionDataset):
    method __init__ (line 21) | def __init__(self, output_directory: Path, spark=False):
    method download (line 27) | def download(self, overwrite=False):
    method preprocess (line 35) | def preprocess(

FILE: src/python/tools/preprocess/datasets/livejournal.py
  class Livejournal (line 8) | class Livejournal(LinkPredictionDataset):
    method __init__ (line 20) | def __init__(self, output_directory: Path, spark=False):
    method download (line 26) | def download(self, overwrite=False):
    method preprocess (line 34) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogb_mag240m.py
  class OGBMag240M (line 16) | class OGBMag240M(NodeClassificationDataset):
    method __init__ (line 29) | def __init__(self, output_directory: Path, spark=False):
    method download (line 35) | def download(self, overwrite=False):
    method preprocess (line 66) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogb_wikikg90mv2.py
  class OGBWikiKG90Mv2 (line 12) | class OGBWikiKG90Mv2(LinkPredictionDataset):
    method __init__ (line 22) | def __init__(self, output_directory: Path, spark=False):
    method download (line 28) | def download(self, overwrite=False):
    method preprocess (line 53) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbl_citation2.py
  class OGBLCitation2 (line 12) | class OGBLCitation2(LinkPredictionDataset):
    method __init__ (line 23) | def __init__(self, output_directory: Path, spark=False):
    method download (line 29) | def download(self, overwrite=False):
    method preprocess (line 49) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbl_collab.py
  class OGBLCollab (line 12) | class OGBLCollab(LinkPredictionDataset):
    method __init__ (line 25) | def __init__(self, output_directory: Path, spark=False, include_edge_t...
    method download (line 34) | def download(self, overwrite=False):
    method preprocess (line 61) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbl_ppa.py
  class OGBLPpa (line 10) | class OGBLPpa(LinkPredictionDataset):
    method __init__ (line 21) | def __init__(self, output_directory: Path, spark=False):
    method download (line 27) | def download(self, overwrite=False, remap_ids=True):
    method preprocess (line 47) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbl_wikikg2.py
  class OGBLWikiKG2 (line 12) | class OGBLWikiKG2(LinkPredictionDataset):
    method __init__ (line 23) | def __init__(self, output_directory: Path, spark=False):
    method download (line 29) | def download(self, overwrite=False):
    method preprocess (line 49) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbn_arxiv.py
  class OGBNArxiv (line 13) | class OGBNArxiv(NodeClassificationDataset):
    method __init__ (line 27) | def __init__(self, output_directory: Path, spark=False):
    method download (line 33) | def download(self, overwrite=False):
    method preprocess (line 73) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbn_papers100m.py
  class OGBNPapers100M (line 15) | class OGBNPapers100M(NodeClassificationDataset):
    method __init__ (line 25) | def __init__(self, output_directory: Path, spark=False):
    method download (line 31) | def download(self, overwrite=False):
    method preprocess (line 66) | def preprocess(

FILE: src/python/tools/preprocess/datasets/ogbn_products.py
  class OGBNProducts (line 13) | class OGBNProducts(NodeClassificationDataset):
    method __init__ (line 26) | def __init__(self, output_directory: Path, spark=False):
    method download (line 32) | def download(self, overwrite=False):
    method preprocess (line 72) | def preprocess(

FILE: src/python/tools/preprocess/datasets/twitter.py
  class Twitter (line 8) | class Twitter(LinkPredictionDataset):
    method __init__ (line 18) | def __init__(self, output_directory: Path, spark=False):
    method download (line 24) | def download(self, overwrite=False):
    method preprocess (line 31) | def preprocess(

FILE: src/python/tools/preprocess/utils.py
  function get_df_count (line 12) | def get_df_count(df, col):
  function download_url (line 18) | def download_url(url, output_dir, overwrite):
  function extract_file (line 37) | def extract_file(filepath, remove_input=True):
  function strip_header (line 78) | def strip_header(filepath, num_lines):

FILE: test/cpp/end_to_end/main.cpp
  function main (line 7) | int main(int argc, char **argv) {

FILE: test/cpp/end_to_end/test_main.cpp
  function TEST (line 13) | TEST(TestMain, TestLinkPred) {

FILE: test/cpp/integration/main.cpp
  function main (line 7) | int main(int argc, char **argv) {

FILE: test/cpp/performance/main.cpp
  function main (line 7) | int main(int argc, char **argv) {

FILE: test/cpp/unit/main.cpp
  function main (line 7) | int main(int argc, char **argv) {

FILE: test/cpp/unit/nn/test_activation.cpp
  function TEST (line 15) | TEST(TestActivation, TestRelu) {
  function TEST (line 33) | TEST(TestActivation, TestSigmoid) {
  function TEST (line 51) | TEST(TestActivation, TestNone) {

FILE: test/cpp/unit/nn/test_initialization.cpp
  function TEST (line 16) | TEST(TestInitialization, TestUniform) {
  function TEST (line 52) | TEST(TestInitialization, TestNormal) {
  function TEST (line 97) | TEST(TestInitialization, TestConstant) {
  function TEST (line 132) | TEST(TestInitialization, TestComputeFans) {
  function TEST (line 186) | TEST(TestInitialization, TestGlorotUniform) {
  function TEST (line 283) | TEST(TestInitialization, TestGlorotNormal) {
  function TEST (line 313) | TEST(TestInitialization, TestTensorInit) {
  function TEST (line 375) | TEST(TestInitialization, TestSubtensorInit) {

FILE: test/cpp/unit/nn/test_loss.cpp
  function TEST (line 30) | TEST(TestLoss, TestShapeMismatch) {
  function TEST (line 55) | TEST(TestLoss, TestSoftmaxCrossEntropy) {
  function TEST (line 90) | TEST(TestLoss, TestRankingLoss) {
  function TEST (line 151) | TEST(TestLoss, TestCrossEntropyLoss) {
  function TEST (line 186) | TEST(TestLoss, TestBCEAfterSigmoid) {
  function TEST (line 221) | TEST(TestLoss, TestBCEWithLogits) {
  function TEST (line 256) | TEST(TestLoss, TestMSE) {
  function TEST (line 291) | TEST(TestLoss, TestSoftPlus) {
  function TEST (line 326) | TEST(TestLoss, TestGetLossFunction) {

FILE: test/cpp/unit/nn/test_model.cpp
  function TEST (line 17) | TEST(TestModel, TestInitModelFromConfigLP) {
  function TEST (line 125) | TEST(TestModel, TestInitModelFromConfigNC) {

FILE: test/cpp/unit/test_buffer.cpp
  class PartitionBufferTest (line 19) | class PartitionBufferTest : public ::testing::Test {
    method PartitionBufferTest (line 37) | PartitionBufferTest() {
    method SetUp (line 52) | void SetUp() override {
    method initializePartitionBuffer (line 90) | void initializePartitionBuffer(bool prefetch) {
    method TearDown (line 96) | void TearDown() override {
  class PartitionedFileTest (line 103) | class PartitionedFileTest : public ::testing::Test {
    method PartitionedFileTest (line 116) | PartitionedFileTest() {
    method SetUp (line 127) | void SetUp() override {
    method TearDown (line 136) | void TearDown() override {
  class LookaheadBlockTest (line 143) | class LookaheadBlockTest : public ::testing::Test {
    method LookaheadBlockTest (line 160) | LookaheadBlockTest() {
    method SetUp (line 173) | void SetUp() override {
    method TearDown (line 182) | void TearDown() override {
  class AsyncWriteBlockTest (line 193) | class AsyncWriteBlockTest : public ::testing::Test {
    method AsyncWriteBlockTest (line 209) | AsyncWriteBlockTest() {
    method SetUp (line 222) | void SetUp() override {
    method TearDown (line 231) | void TearDown() override {
  function TEST_F (line 241) | TEST_F(PartitionBufferTest, TestPartitionBufferOrdering) {
  function TEST_F (line 259) | TEST_F(PartitionBufferTest, TestPartitionBufferPrefetch) {
  function TEST_F (line 275) | TEST_F(PartitionBufferTest, TestPartitionBufferIndexRead) {
  function TEST_F (line 285) | TEST_F(PartitionBufferTest, TestPartitionBufferIndexAdd) {
  function TEST_F (line 299) | TEST_F(PartitionBufferTest, TestPartitionBufferSync) {
  function TEST_F (line 310) | TEST_F(PartitionBufferTest, TestPartitionBufferGlobalMap) {
  function TEST_F (line 320) | TEST_F(PartitionedFileTest, TestReadPartition) {
  function TEST_F (line 336) | TEST_F(PartitionedFileTest, TestWritePartition) {
  function TEST_F (line 363) | TEST_F(LookaheadBlockTest, TestMoveToBuffer) {
  function TEST_F (line 410) | TEST_F(AsyncWriteBlockTest, TestAsyncWrite) {

FILE: test/cpp/unit/test_storage.cpp
  class StorageTest (line 17) | class StorageTest : public ::testing::Test {
    method StorageTest (line 27) | StorageTest() {
    method SetUp (line 36) | void SetUp() override {
    method TearDown (line 46) | void TearDown() override {
  class FlatFileTest (line 54) | class FlatFileTest : public StorageTest {
    method FlatFileTest (line 61) | FlatFileTest() {
  class InMemoryTest (line 72) | class InMemoryTest : public StorageTest {
    method InMemoryTest (line 79) | InMemoryTest() {
  class PartitionBufferStorageTest (line 90) | class PartitionBufferStorageTest : public StorageTest {
    method PartitionBufferStorageTest (line 99) | PartitionBufferStorageTest() {
  function TEST_F (line 141) | TEST_F(FlatFileTest, TestFlatFileWrite) {
  function TEST_F (line 168) | TEST_F(FlatFileTest, TestFlatFileCopy) {
  function TEST_F (line 189) | TEST_F(FlatFileTest, TestFlatFileShuffle) {
  function TEST_F (line 205) | TEST_F(FlatFileTest, TestFlatFileSort) {
  function TEST_F (line 222) | TEST_F(FlatFileTest, TestFlatFileSortEdges) {
  function TEST_F (line 260) | TEST_F(InMemoryTest, TestIndexRead) {
  function TEST_F (line 274) | TEST_F(InMemoryTest, TestIndexAdd) {
  function TEST_F (line 294) | TEST_F(InMemoryTest, TestIndexPut) {
  function TEST_F (line 318) | TEST_F(InMemoryTest, TestInMemoryShuffle) {
  function TEST_F (line 331) | TEST_F(InMemoryTest, TestInMemorySort) {
  function TEST_F (line 345) | TEST_F(InMemoryTest, TestInMemorySortEdges) {
  function TEST_F (line 381) | TEST_F(PartitionBufferStorageTest, TestBufferOrdering) {
  function TEST_F (line 402) | TEST_F(PartitionBufferStorageTest, TestIndexRead) {
  function TEST_F (line 417) | TEST_F(PartitionBufferStorageTest, TestRangePut) {
  function TEST_F (line 440) | TEST_F(PartitionBufferStorageTest, TestIndexAdd) {

FILE: test/cpp/unit/testing_util.cpp
  function createTmpFile (line 6) | int createTmpFile(std::string &filename) { return open(filename.c_str(),...
  function getRandTensor (line 8) | torch::Tensor getRandTensor(int dim0_size, int dim1_size, torch::Dtype d...
  function genRandTensorAndWriteToFile (line 15) | int genRandTensorAndWriteToFile(torch::Tensor &rand_tensor, int total_em...
  function checkPermOf2dTensor (line 21) | bool checkPermOf2dTensor(torch::Tensor &a, torch::Tensor &b) {
  function sortWithinEdgeBuckets (line 36) | void sortWithinEdgeBuckets(torch::Tensor &rand_tensor, vector<int64_t> &...
  function sortEdgesSrcDest (line 47) | bool sortEdgesSrcDest(vector<int> &edge1, vector<int> &edge2) {
  function partitionEdges (line 53) | vector<int64_t> partitionEdges(torch::Tensor &edges, int num_partitions,...

FILE: test/db2graph/test_postgres.py
  class TestConnector (line 10) | class TestConnector:
    method fill_db (line 40) | def fill_db(self):
    method test_connect_to_db (line 175) | def test_connect_to_db(self):
    method test_edges_entity_entity (line 229) | def test_edges_entity_entity(self):

FILE: test/python/bindings/end_to_end/test_fb15k_acc.py
  class TestFB15K (line 10) | class TestFB15K(unittest.TestCase):
    method setUp (line 12) | def setUp(self):
    method tearDown (line 17) | def tearDown(self):
    method test_one_epoch (line 23) | def test_one_epoch(self):

FILE: test/python/bindings/end_to_end/test_interval_checkpointing.py
  function replace_string_in_file (line 12) | def replace_string_in_file(filepath, before, after):
  function get_line_in_file (line 16) | def get_line_in_file(filepath, line_num):
  function run_config (line 20) | def run_config(config_file, enable_checkpointing, checkpoint_interval, s...
  class TestIntervalCheckpointing (line 29) | class TestIntervalCheckpointing(unittest.TestCase):
    method setUp (line 34) | def setUp(self):
    method tearDown (line 40) | def tearDown(self):
    method init_dataset_dir (line 44) | def init_dataset_dir(self, name):
    method test_checkpointing_with_state (line 71) | def test_checkpointing_with_state(self):
    method test_checkpointing_wo_state (line 118) | def test_checkpointing_wo_state(self):

FILE: test/python/bindings/end_to_end/test_lp_basic.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False):
  class TestLP (line 28) | class TestLP(unittest.TestCase):
    method setUp (line 32) | def setUp(self):
    method tearDown (line 51) | def tearDown(self):
    method test_dm (line 56) | def test_dm(self):
    method test_gs (line 72) | def test_gs(self):
    method test_gs_uniform (line 88) | def test_gs_uniform(self):
    method test_gat (line 104) | def test_gat(self):
    method test_sync_training (line 120) | def test_sync_training(self):
    method test_async_training (line 137) | def test_async_training(self):
    method test_sync_eval (line 153) | def test_sync_eval(self):
    method test_async_eval (line 170) | def test_async_eval(self):
  class TestLPNoRelations (line 186) | class TestLPNoRelations(unittest.TestCase):
    method setUp (line 190) | def setUp(self):
    method tearDown (line 209) | def tearDown(self):
    method test_dm (line 214) | def test_dm(self):
    method test_gs (line 230) | def test_gs(self):
    method test_gs_uniform (line 246) | def test_gs_uniform(self):
    method test_gat (line 262) | def test_gat(self):
    method test_sync_training (line 278) | def test_sync_training(self):
    method test_async_training (line 295) | def test_async_training(self):
    method test_sync_eval (line 311) | def test_sync_eval(self):
    method test_async_eval (line 328) | def test_async_eval(self):

FILE: test/python/bindings/end_to_end/test_lp_buffer.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False):
  class TestLPBuffer (line 29) | class TestLPBuffer(unittest.TestCase):
    method setUp (line 33) | def setUp(self):
    method tearDown (line 53) | def tearDown(self):
    method test_dm (line 58) | def test_dm(self):
    method test_gs (line 74) | def test_gs(self):
    method test_gs_uniform (line 90) | def test_gs_uniform(self):
    method test_gat (line 106) | def test_gat(self):
    method test_sync_training (line 123) | def test_sync_training(self):
    method test_async_training (line 140) | def test_async_training(self):
    method test_sync_eval (line 156) | def test_sync_eval(self):
    method test_async_eval (line 173) | def test_async_eval(self):
    method test_partitioned_eval (line 189) | def test_partitioned_eval(self):
  class TestLPBufferNoRelations (line 218) | class TestLPBufferNoRelations(unittest.TestCase):
    method setUp (line 222) | def setUp(self):
    method tearDown (line 242) | def tearDown(self):
    method test_dm (line 247) | def test_dm(self):
    method test_gs (line 264) | def test_gs(self):
    method test_gs_uniform (line 281) | def test_gs_uniform(self):
    method test_gat (line 297) | def test_gat(self):
    method test_sync_training (line 314) | def test_sync_training(self):
    method test_async_training (line 331) | def test_async_training(self):
    method test_sync_eval (line 347) | def test_sync_eval(self):
    method test_async_eval (line 364) | def test_async_eval(self):
    method test_partitioned_eval (line 380) | def test_partitioned_eval(self):

FILE: test/python/bindings/end_to_end/test_lp_storage.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False):
  class TestLPStorage (line 28) | class TestLPStorage(unittest.TestCase):
    method setUp (line 32) | def setUp(self):
    method tearDown (line 37) | def tearDown(self):
    method test_no_valid (line 42) | def test_no_valid(self):
    method test_only_train (line 69) | def test_only_train(self):
    method test_no_valid_no_relations (line 95) | def test_no_valid_no_relations(self):
    method test_only_train_no_relations (line 122) | def test_only_train_no_relations(self):
    method test_no_valid_buffer (line 148) | def test_no_valid_buffer(self):
    method test_only_train_buffer (line 177) | def test_only_train_buffer(self):
    method test_no_valid_buffer_no_relations (line 204) | def test_no_valid_buffer_no_relations(self):
    method test_only_train_buffer_no_relations (line 233) | def test_only_train_buffer_no_relations(self):

FILE: test/python/bindings/end_to_end/test_model_dir.py
  function run_configs (line 14) | def run_configs(directory, model_dir=None, partitioned_eval=False, seque...
  function has_model_params (line 50) | def has_model_params(model_dir_path, task="lp", has_embeddings=False, ha...
  class TestLP (line 83) | class TestLP(unittest.TestCase):
    method setUp (line 87) | def setUp(self):
    method tearDown (line 106) | def tearDown(self):
    method test_dm (line 111) | def test_dm(self):
  class TestNC (line 152) | class TestNC(unittest.TestCase):
    method setUp (line 156) | def setUp(self):
    method tearDown (line 176) | def tearDown(self):
    method test_gs (line 181) | def test_gs(self):
    method test_async (line 203) | def test_async(self):
    method test_emb (line 224) | def test_emb(self):
  class TestLPBufferNoRelations (line 245) | class TestLPBufferNoRelations(unittest.TestCase):
    method setUp (line 249) | def setUp(self):
    method tearDown (line 269) | def tearDown(self):
    method test_dm (line 274) | def test_dm(self):
    method test_partitioned_eval (line 295) | def test_partitioned_eval(self):
  class TestNCBuffer (line 329) | class TestNCBuffer(unittest.TestCase):
    method setUp (line 333) | def setUp(self):
    method tearDown (line 354) | def tearDown(self):
    method test_gs (line 359) | def test_gs(self):
    method test_async (line 381) | def test_async(self):
    method test_emb (line 402) | def test_emb(self):
    method test_partitioned_eval (line 423) | def test_partitioned_eval(self):
    method test_sequential (line 459) | def test_sequential(self):

FILE: test/python/bindings/end_to_end/test_nc_basic.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False):
  class TestNC (line 28) | class TestNC(unittest.TestCase):
    method setUp (line 32) | def setUp(self):
    method tearDown (line 52) | def tearDown(self):
    method test_gs (line 57) | def test_gs(self):
    method test_gs_uniform (line 73) | def test_gs_uniform(self):
    method test_gat (line 89) | def test_gat(self):
    method test_async (line 106) | def test_async(self):
    method test_emb (line 122) | def test_emb(self):
  class TestNCNoRelations (line 138) | class TestNCNoRelations(unittest.TestCase):
    method setUp (line 142) | def setUp(self):
    method tearDown (line 162) | def tearDown(self):
    method test_gs (line 167) | def test_gs(self):
    method test_gs_uniform (line 183) | def test_gs_uniform(self):
    method test_gat (line 199) | def test_gat(self):
    method test_async (line 216) | def test_async(self):
    method test_emb (line 232) | def test_emb(self):

FILE: test/python/bindings/end_to_end/test_nc_buffer.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False, sequential_train_node...
  class TestNCBuffer (line 32) | class TestNCBuffer(unittest.TestCase):
    method setUp (line 36) | def setUp(self):
    method tearDown (line 57) | def tearDown(self):
    method test_gs (line 62) | def test_gs(self):
    method test_gs_uniform (line 78) | def test_gs_uniform(self):
    method test_gat (line 94) | def test_gat(self):
    method test_async (line 111) | def test_async(self):
    method test_emb (line 127) | def test_emb(self):
    method test_partitioned_eval (line 143) | def test_partitioned_eval(self):
    method test_sequential (line 174) | def test_sequential(self):
  class TestNCBufferNoRelations (line 206) | class TestNCBufferNoRelations(unittest.TestCase):
    method setUp (line 210) | def setUp(self):
    method tearDown (line 231) | def tearDown(self):
    method test_gs (line 236) | def test_gs(self):
    method test_gs_uniform (line 252) | def test_gs_uniform(self):
    method test_gat (line 268) | def test_gat(self):
    method test_async (line 285) | def test_async(self):
    method test_emb (line 301) | def test_emb(self):
    method test_partitioned_eval (line 317) | def test_partitioned_eval(self):
    method test_sequential (line 348) | def test_sequential(self):

FILE: test/python/bindings/end_to_end/test_nc_storage.py
  function run_configs (line 14) | def run_configs(directory, partitioned_eval=False):
  class TestNCStorage (line 28) | class TestNCStorage(unittest.TestCase):
    method setUp (line 32) | def setUp(self):
    method tearDown (line 37) | def tearDown(self):
    method test_no_valid (line 42) | def test_no_valid(self):
    method test_only_train (line 70) | def test_only_train(self):
    method test_no_valid_no_relations (line 97) | def test_no_valid_no_relations(self):
    method test_only_train_no_relations (line 125) | def test_only_train_no_relations(self):
    method test_no_valid_buffer (line 152) | def test_no_valid_buffer(self):
    method test_only_train_buffer (line 182) | def test_only_train_buffer(self):
    method test_no_valid_buffer_no_relations (line 210) | def test_no_valid_buffer_no_relations(self):
    method test_only_train_buffer_no_relations (line 240) | def test_only_train_buffer_no_relations(self):

FILE: test/python/bindings/end_to_end/test_resume_training.py
  function replace_string_in_file (line 12) | def replace_string_in_file(filepath, before, after):
  function get_line_in_file (line 16) | def get_line_in_file(filepath, line_num):
  function run_config (line 20) | def run_config(config_file):
  class TestResumeTraining (line 25) | class TestResumeTraining(unittest.TestCase):
    method setUp (line 30) | def setUp(self):
    method tearDown (line 36) | def tearDown(self):
    method init_dataset_dir (line 40) | def init_dataset_dir(self, name):
    method test_resume_training_model_dir (line 69) | def test_resume_training_model_dir(self):
    method test_resume_training_checkpoint_dir (line 111) | def test_resume_training_checkpoint_dir(self):

FILE: test/python/bindings/integration/test_config.py
  class TestConfig (line 14) | class TestConfig(unittest.TestCase):
    method setUp (line 32) | def setUp(self):
    method tearDown (line 39) | def tearDown(self):
    method test_missing_config (line 43) | def test_missing_config(self):
    method test_missing_dataset_yaml (line 50) | def test_missing_dataset_yaml(self):
    method test_load_config (line 93) | def test_load_config(self):

FILE: test/python/bindings/integration/test_data.py
  class TestBatch (line 9) | class TestBatch(unittest.TestCase):
    method test_construction (line 14) | def test_construction(self):
    method test_accumulate_gradients (line 34) | def test_accumulate_gradients(self):
    method test_clear (line 49) | def test_clear(self):
  class TestDataloader (line 67) | class TestDataloader(unittest.TestCase):
    method test_lp_only_edges (line 68) | def test_lp_only_edges(self):
    method test_lp_negs (line 108) | def test_lp_negs(self):
    method test_lp_negs_nbrs (line 164) | def test_lp_negs_nbrs(self):
    method test_lp_nbrs (line 222) | def test_lp_nbrs(self):
    method test_nc_nbrs (line 264) | def test_nc_nbrs(self):
    method test_nc_no_nbrs (line 307) | def test_nc_no_nbrs(self):

FILE: test/python/bindings/integration/test_nn.py
  function get_test_model_lp (line 35) | def get_test_model_lp():
  function get_test_model_lp_neg (line 59) | def get_test_model_lp_neg():
  function get_test_model_nc (line 83) | def get_test_model_nc():
  class CustomModelBasic (line 99) | class CustomModelBasic(Model):
    method __init__ (line 100) | def __init__(self, encoder, decoder):
  class CustomModelOverrideForward (line 113) | class CustomModelOverrideForward(Model):
    method __init__ (line 114) | def __init__(self, encoder, decoder):
    method forward_lp (line 124) | def forward_lp(self, batch, train):
  class TestModel (line 130) | class TestModel(unittest.TestCase):
    method test_construction_lp (line 135) | def test_construction_lp(self):
    method test_construction_nc (line 138) | def test_construction_nc(self):
    method test_forward_nc (line 141) | def test_forward_nc(self):
    method test_forward_lp (line 148) | def test_forward_lp(self):
    method test_train_batch (line 162) | def test_train_batch(self):
    method test_clear_grad (line 185) | def test_clear_grad(self):
    method test_step (line 197) | def test_step(self):
    method test_save (line 209) | def test_save(self):
    method test_load (line 212) | def test_load(self):
    method test_custom_model_basic (line 215) | def test_custom_model_basic(self):
    method test_custom_model_forward_override (line 230) | def test_custom_model_forward_override(self):
    method test_init_from_config (line 246) | def test_init_from_config(self):

FILE: test/python/helpers.py
  function dataset_generator (line 5) | def dataset_generator(

FILE: test/python/postprocessing/test_in_memory_exporter.py
  function check_output (line 16) | def check_output(output_dir, fmt, has_rels=False):
  class TestLP (line 90) | class TestLP(unittest.TestCase):
    method setUp (line 94) | def setUp(self):
    method tearDown (line 134) | def tearDown(self):
    method test_export_csv (line 138) | def test_export_csv(self):
    method test_export_binary (line 148) | def test_export_binary(self):
    method test_export_parquet (line 163) | def test_export_parquet(self):
    method test_export_no_model (line 179) | def test_export_no_model(self):
    method test_export_overwrite (line 186) | def test_export_overwrite(self):
  class TestNC (line 203) | class TestNC(unittest.TestCase):
    method setUp (line 207) | def setUp(self):
    method tearDown (line 248) | def tearDown(self):
    method test_export_csv (line 252) | def test_export_csv(self):
    method test_export_binary (line 262) | def test_export_binary(self):
    method test_export_parquet (line 277) | def test_export_parquet(self):
    method test_export_no_model (line 293) | def test_export_no_model(self):
    method test_export_overwrite (line 300) | def test_export_overwrite(self):

FILE: test/python/predict/test_predict.py
  function validate_metrics (line 15) | def validate_metrics(config, metrics, num_items, output_dir=None):
  function validate_scores (line 57) | def validate_scores(config, num_edges, save_scores, save_ranks, output_d...
  function validate_labels (line 82) | def validate_labels(config, num_nodes, output_dir=None):
  class TestPredictLP (line 97) | class TestPredictLP(unittest.TestCase):
    method setUp (line 101) | def setUp(self):
    method tearDown (line 138) | def tearDown(self):
    method test_basic_lp (line 142) | def test_basic_lp(self):
    method test_lp_metrics (line 150) | def test_lp_metrics(self):
    method test_predict_model_dir (line 177) | def test_predict_model_dir(self):
    method test_lp_save_ranks (line 272) | def test_lp_save_ranks(self):
    method test_lp_save_scores (line 275) | def test_lp_save_scores(self):

FILE: test/python/preprocessing/test_spark_converter.py
  class TestSparkConverter (line 23) | class TestSparkConverter(unittest.TestCase):
    method setUp (line 25) | def setUp(self):
    method tearDown (line 34) | def tearDown(self):
    method make_directory_tree (line 39) | def make_directory_tree(self, dir_path):
    method test_delimited_defaults (line 48) | def test_delimited_defaults(self):
    method test_delimited_str_ids (line 66) | def test_delimited_str_ids(self):
    method test_columns (line 94) | def test_columns(self):
    method test_header (line 112) | def test_header(self):
    method test_delim (line 138) | def test_delim(self):
    method test_partitions (line 159) | def test_partitions(self):

FILE: test/python/preprocessing/test_torch_converter.py
  function validate_partitioned_output_dir (line 20) | def validate_partitioned_output_dir(
  function validate_output_dir (line 77) | def validate_output_dir(
  class TestTorchConverter (line 155) | class TestTorchConverter(unittest.TestCase):
    method setUp (line 161) | def setUp(self):
    method tearDown (line 170) | def tearDown(self):
    method test_delimited_defaults (line 174) | def test_delimited_defaults(self):
    method test_delimited_str_ids (line 198) | def test_delimited_str_ids(self):
    method test_numpy_defaults (line 230) | def test_numpy_defaults(self):
    method test_pytorch_defaults (line 258) | def test_pytorch_defaults(self):
    method test_splits (line 286) | def test_splits(self):
    method test_columns (line 313) | def test_columns(self):
    method test_header (line 336) | def test_header(self):
    method test_delim (line 366) | def test_delim(self):
    method test_dtype (line 393) | def test_dtype(self):
    method test_partitions (line 424) | def test_partitions(self):
    method test_no_remap (line 467) | def test_no_remap(self):
    method test_torch_no_relation_no_remap (line 494) | def test_torch_no_relation_no_remap(self):
    method test_pandas_no_relation_no_remap (line 524) | def test_pandas_no_relation_no_remap(self):
    method test_torch_no_relation_remap (line 551) | def test_torch_no_relation_remap(self):
    method test_pandas_no_relation_remap (line 582) | def test_pandas_no_relation_remap(self):
    method test_torch_only_weights_no_remap (line 609) | def test_torch_only_weights_no_remap(self):
    method test_pandas_only_weights_no_remap (line 643) | def test_pandas_only_weights_no_remap(self):
    method test_torch_only_weights_remap (line 673) | def test_torch_only_weights_remap(self):
    method test_pandas_only_weights_remap (line 707) | def test_pandas_only_weights_remap(self):
    method test_torch_relationship_weights_no_remap (line 737) | def test_torch_relationship_weights_no_remap(self):
    method test_pandas_relationship_weights_no_remap (line 773) | def test_pandas_relationship_weights_no_remap(self):
    method test_torch_relationship_weights_remap (line 805) | def test_torch_relationship_weights_remap(self):
    method test_pandas_relationship_weights_remap (line 840) | def test_pandas_relationship_weights_remap(self):
    method test_torch_relationship_weights_remap_partioned (line 872) | def test_torch_relationship_weights_remap_partioned(self):
    method test_pandas_relationship_weights_remap_partioned (line 910) | def test_pandas_relationship_weights_remap_partioned(self):

FILE: test/test_configs/generate_test_configs.py
  function get_config (line 11) | def get_config(model_config_path, storage_config_path, train_config_path...
  function set_dataset_config (line 22) | def set_dataset_config(base_config, dataset_dir):
  function config_from_sub_configs (line 38) | def config_from_sub_configs(model_config, storage_config, train_config, ...
  function get_cartesian_product_of_configs (line 49) | def get_cartesian_product_of_configs(config_directory, model_names, stor...
  function get_all_configs_for_dataset (line 106) | def get_all_configs_for_dataset(
  function generate_configs_for_dataset (line 122) | def generate_configs_for_dataset(

FILE: test/test_data/generate.py
  function get_random_graph (line 12) | def get_random_graph(num_nodes, num_edges, num_rels=1):
  function generate_features (line 25) | def generate_features(num_nodes, feature_dim):
  function generate_labels (line 29) | def generate_labels(num_nodes, num_classes):
  function shuffle_with_map (line 33) | def shuffle_with_map(values, node_mapping):
  function apply_mapping (line 39) | def apply_mapping(values, node_mapping):
  function remap_nc (line 44) | def remap_nc(output_dir, train_nodes, labels, num_nodes, valid_nodes=Non...
  function remap_lp (line 66) | def remap_lp(output_dir, features=None):
  function generate_random_dataset_nc (line 73) | def generate_random_dataset_nc(
  function generate_random_dataset_lp (line 186) | def generate_random_dataset_lp(
  function generate_random_dataset (line 244) | def generate_random_dataset(