We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d77595 commit fe4ab87Copy full SHA for fe4ab87
python/tvm/relay/testing/tflite.py
@@ -36,6 +36,8 @@ def __init__(self, dtype):
36
self.dtype_dict = {}
37
38
def create_conv2d_single(self, kernel_shape, strides, padding, dilation, activation):
39
+ """Returns tf.function that creates TFLite Conv2d layer"""
40
+
41
@tf.function
42
def conv2d_single_function(ifm_tensor):
43
"""Returns TFLite Conv2d layer"""
0 commit comments