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 test_trt.py.

---

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

• caffe2/python/trt/test_trt.py