TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
model.LocationLayer Class Reference
Inheritance diagram for model.LocationLayer:
Collaboration diagram for model.LocationLayer:

Public Member Functions

def __init__ (self, attention_n_filters, attention_kernel_size, attention_dim)
 
def forward (self, attention_weights_cat)
 

Public Attributes

 location_conv
 
 location_dense
 

Constructor & Destructor Documentation

◆ __init__()

def model.LocationLayer.__init__ (   self,
  attention_n_filters,
  attention_kernel_size,
  attention_dim 
)

Member Function Documentation

◆ forward()

def model.LocationLayer.forward (   self,
  attention_weights_cat 
)

Member Data Documentation

◆ location_conv

model.LocationLayer.location_conv

◆ location_dense

model.LocationLayer.location_dense

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