Skip to content

Commit fe4ab87

Browse files
author
Ashutosh Parkhi
committed
Lint fix: added a missing docstring
Change-Id: If1fb8bb09c538c04e333ccab65a20cff247a504d
1 parent 4d77595 commit fe4ab87

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/tvm/relay/testing/tflite.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,8 @@ def __init__(self, dtype):
3636
self.dtype_dict = {}
3737

3838
def create_conv2d_single(self, kernel_shape, strides, padding, dilation, activation):
39+
"""Returns tf.function that creates TFLite Conv2d layer"""
40+
3941
@tf.function
4042
def conv2d_single_function(ifm_tensor):
4143
"""Returns TFLite Conv2d layer"""

0 commit comments

Comments
 (0)