Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.trt.test_trt.TensorRTOpTest Class Reference
Inheritance diagram for caffe2.python.trt.test_trt.TensorRTOpTest:

Public Member Functions

def test_relu_graph_simple (self)
def test_relu_graph_big_batch (self)
def test_resnet50 (self)
def test_bvlc_alexnet (self)
def test_densenet121 (self)
def test_inception_v1 (self)
def test_inception_v2 (self)
def test_shufflenet (self)
def test_squeezenet (self)
def test_vgg16 (self)
def test_vgg19 (self)
- Public Member Functions inherited from caffe2.python.onnx.tests.test_utils.TestCase
def setUp (self)
def assertSameOutputs (self, outputs1, outputs2, decimal=7)
def add_test_case (self, name, test_func)

Detailed Description

Definition at line 82 of file

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