TrivialModel Class Reference

Inheritance diagram for TrivialModel:

Collaboration diagram for TrivialModel:

Public Member Functions
	__init__ (self, forward_fn)
	forward (self, *args)

Public Attributes
	forward_fn = forward_fn
	forward pass function

Detailed Description

Wrapper class to create simple non-parametric models with multiple inputs and outputs for ONNX tests

Definition at line 40 of file write_test_files.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			forward_fn )

Initialize with forward pass function passed to the constructor

Definition at line 45 of file write_test_files.py.

    def __init__(self, forward_fn):
        """
        Initialize with forward pass function passed to the constructor
        """
        super().__init__()
 
        
        self.forward_fn = forward_fn
 

Member Function Documentation

forward()

forward	(		self,
		*	args )

Run the forward pass based on `forward_fn`

Definition at line 54 of file write_test_files.py.

    def forward(self, *args):
        """
        Run the forward pass based on `forward_fn`
        """
        return self.forward_fn(*args)
 
 

Member Data Documentation

forward_fn

forward_fn = forward_fn

forward pass function

Definition at line 52 of file write_test_files.py.

---

The documentation for this class was generated from the following file:

• mva/examples/onnx/write_test_files.py