Skip to content

T
tic
Commit b05dc12f by 雾雨魔理沙 Committed by Yizhi Liu
Options

[Relay][Parser] simplify build script, remove python 2 support (#3419) 

* simplify build script, remove python 2 support

* remove py2 file

* update py3
parent 1e9d014b
Showing with 20 additions and 3203 deletions
cmake/modules/ANTLR.cmake
......@@ -16,44 +16,19 @@
# under the License.
if(USE_ANTLR)
file(GLOB_RECURSE ANTLR4
/usr/local/lib/antlr-*-complete.jar
/usr/local/Cellar/*antlr-*-complete.jar)
set(RELAY_PARSER_DIR
${CMAKE_CURRENT_SOURCE_DIR}/python/tvm/relay/grammar)
if(DEFINED ANTLR4)
# Get the first element of the list of antlr jars.
# Sort and reverse the list so the item selected is the highest
# version in lib or else in Cellar if no lib installation exists.
list(SORT ANTLR4)
list(REVERSE ANTLR4)
list(GET ANTLR4 0 ANTLR4)
set(RELAY_PARSER_DIR
${CMAKE_CURRENT_SOURCE_DIR}/python/tvm/relay/grammar)
set(RELAY_PARSER
${RELAY_PARSER_DIR}/py3/RelayVisitor.py
${RELAY_PARSER_DIR}/py3/RelayParser.py
${RELAY_PARSER_DIR}/py3/RelayLexer.py)
set(RELAY_PARSER
${RELAY_PARSER_DIR}/py2/RelayVisitor.py
${RELAY_PARSER_DIR}/py2/RelayParser.py
${RELAY_PARSER_DIR}/py2/RelayLexer.py
# Generate ANTLR grammar for parsing.
add_custom_command(OUTPUT ${RELAY_PARSER}
COMMAND antlr4 -visitor -no-listener -Dlanguage=Python3 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py3
DEPENDS ${RELAY_PARSER_DIR}/Relay.g4
WORKING_DIRECTORY ${RELAY_PARSER_DIR})
${RELAY_PARSER_DIR}/py3/RelayVisitor.py
${RELAY_PARSER_DIR}/py3/RelayParser.py
${RELAY_PARSER_DIR}/py3/RelayLexer.py)
set(JAVA_HOME $ENV{JAVA_HOME})
if (NOT DEFINED JAVA_HOME)
# Hack to get system to search for Java itself.
set(JAVA_HOME "/usr")
endif()
# Generate ANTLR grammar for parsing.
add_custom_command(OUTPUT ${RELAY_PARSER}
COMMAND ${JAVA_HOME}/bin/java -jar ${ANTLR4} -visitor -no-listener -Dlanguage=Python2 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py2
COMMAND ${JAVA_HOME}/bin/java -jar ${ANTLR4} -visitor -no-listener -Dlanguage=Python3 ${RELAY_PARSER_DIR}/Relay.g4 -o ${RELAY_PARSER_DIR}/py3
DEPENDS ${RELAY_PARSER_DIR}/Relay.g4
WORKING_DIRECTORY ${RELAY_PARSER_DIR})
add_custom_target(relay_parser ALL DEPENDS ${RELAY_PARSER})
else()
message(FATAL_ERROR "Can't find ANTLR4: ANTLR4=" ${ANTLR4})
endif()
add_custom_target(relay_parser ALL DEPENDS ${RELAY_PARSER})
endif(USE_ANTLR)
python/tvm/relay/_parser.py
......@@ -43,14 +43,9 @@ class ParseError(Exception):
PYTHON_VERSION = sys.version_info.major
try:
if PYTHON_VERSION == 2:
from .grammar.py2.RelayVisitor import RelayVisitor
from .grammar.py2.RelayParser import RelayParser
from .grammar.py2.RelayLexer import RelayLexer
else:
from .grammar.py3.RelayVisitor import RelayVisitor
from .grammar.py3.RelayParser import RelayParser
from .grammar.py3.RelayLexer import RelayLexer
from .grammar.py3.RelayVisitor import RelayVisitor
from .grammar.py3.RelayParser import RelayParser
from .grammar.py3.RelayLexer import RelayLexer
except ImportError:
raise ParseError("Couldn't find ANTLR parser. Try building with USE_ANTLR=ON.")
......
python/tvm/relay/grammar/py2/.gitattributes deleted 100644 → 0
Relay* binary
Relay* linguist-generated=true
Relay* linguist-detectable=false
\ No newline at end of file
python/tvm/relay/grammar/py2/Relay.interp deleted 100644 → 0
token literal names:
null
'('
')'
','
'['
']'
'if'
'else'
'let'
'='
';'
'{'
'}'
'fn'
'->'
'def'
':'
'Tensor'
'_'
'v0.0.2'
null
null
null
'*'
'/'
'+'
'-'
'<'
'>'
'<='
'>='
'=='
'!='
null
null
null
'mut'
null
null
null
null
token symbolic names:
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
SEMVER
WS
LINE_COMMENT
COMMENT
MUL
DIV
ADD
SUB
LT
GT
LE
GE
EQ
NE
GLOBAL_VAR
LOCAL_VAR
GRAPH_VAR
MUT
BOOL_LIT
FLOAT
NAT
CNAME
rule names:
opIdent
prog
expr
func
defn
argList
varList
var
attrList
attr
typeParamSeq
type_
shapeSeq
shape
typeIdent
body
scalar
ident
atn:
[3, 24715, 42794, 33075, 47597, 16764, 15335, 30598, 22884, 3, 42, 332, 4, 2, 9, 2, 4, 3, 9, 3, 4, 4, 9, 4, 4, 5, 9, 5, 4, 6, 9, 6, 4, 7, 9, 7, 4, 8, 9, 8, 4, 9, 9, 9, 4, 10, 9, 10, 4, 11, 9, 11, 4, 12, 9, 12, 4, 13, 9, 13, 4, 14, 9, 14, 4, 15, 9, 15, 4, 16, 9, 16, 4, 17, 9, 17, 4, 18, 9, 18, 4, 19, 9, 19, 3, 2, 3, 2, 3, 3, 3, 3, 7, 3, 43, 10, 3, 12, 3, 14, 3, 46, 11, 3, 3, 3, 5, 3, 49, 10, 3, 3, 3, 3, 3, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 6, 4, 72, 10, 4, 13, 4, 14, 4, 73, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 7, 4, 82, 10, 4, 12, 4, 14, 4, 85, 11, 4, 5, 4, 87, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 100, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 110, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 5, 4, 128, 10, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 3, 4, 7, 4, 150, 10, 4, 12, 4, 14, 4, 153, 11, 4, 5, 4, 155, 10, 4, 3, 4, 7, 4, 158, 10, 4, 12, 4, 14, 4, 161, 11, 4, 3, 5, 3, 5, 5, 5, 165, 10, 5, 3, 5, 3, 5, 3, 5, 3, 5, 3, 5, 5, 5, 172, 10, 5, 3, 5, 3, 5, 3, 6, 3, 6, 3, 6, 5, 6, 179, 10, 6, 3, 6, 3, 6, 3, 6, 3, 6, 3, 6, 5, 6, 186, 10, 6, 3, 6, 3, 6, 3, 7, 3, 7, 3, 7, 3, 7, 3, 7, 3, 7, 5, 7, 196, 10, 7, 3, 8, 3, 8, 3, 8, 7, 8, 201, 10, 8, 12, 8, 14, 8, 204, 11, 8, 5, 8, 206, 10, 8, 3, 9, 3, 9, 3, 9, 5, 9, 211, 10, 9, 3, 10, 3, 10, 3, 10, 7, 10, 216, 10, 10, 12, 10, 14, 10, 219, 11, 10, 5, 10, 221, 10, 10, 3, 11, 3, 11, 3, 11, 3, 11, 3, 12, 3, 12, 3, 12, 3, 12, 3, 12, 3, 12, 7, 12, 233, 10, 12, 12, 12, 14, 12, 236, 11, 12, 3, 12, 3, 12, 5, 12, 240, 10, 12, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 6, 13, 253, 10, 13, 13, 13, 14, 13, 254, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 5, 13, 269, 10, 13, 3, 13, 3, 13, 3, 13, 3, 13, 7, 13, 275, 10, 13, 12, 13, 14, 13, 278, 11, 13, 5, 13, 280, 10, 13, 3, 13, 3, 13, 3, 13, 3, 13, 3, 13, 5, 13, 287, 10, 13, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 3, 14, 6, 14, 300, 10, 14, 13, 14, 14, 14, 301, 3, 14, 3, 14, 5, 14, 306, 10, 14, 3, 15, 3, 15, 3, 15, 3, 15, 3, 15, 5, 15, 313, 10, 15, 3, 16, 3, 16, 3, 17, 3, 17, 3, 17, 3, 17, 3, 18, 3, 18, 3, 18, 5, 18, 324, 10, 18, 3, 19, 3, 19, 3, 19, 3, 19, 5, 19, 330, 10, 19, 3, 19, 2, 3, 6, 20, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26, 28, 30, 32, 34, 36, 2, 6, 3, 2, 25, 26, 3, 2, 27, 28, 3, 2, 29, 32, 3, 2, 33, 34, 2, 373, 2, 38, 3, 2, 2, 2, 4, 40, 3, 2, 2, 2, 6, 127, 3, 2, 2, 2, 8, 162, 3, 2, 2, 2, 10, 175, 3, 2, 2, 2, 12, 195, 3, 2, 2, 2, 14, 205, 3, 2, 2, 2, 16, 207, 3, 2, 2, 2, 18, 220, 3, 2, 2, 2, 20, 222, 3, 2, 2, 2, 22, 239, 3, 2, 2, 2, 24, 286, 3, 2, 2, 2, 26, 305, 3, 2, 2, 2, 28, 312, 3, 2, 2, 2, 30, 314, 3, 2, 2, 2, 32, 316, 3, 2, 2, 2, 34, 323, 3, 2, 2, 2, 36, 329, 3, 2, 2, 2, 38, 39, 7, 42, 2, 2, 39, 3, 3, 2, 2, 2, 40, 48, 7, 21, 2, 2, 41, 43, 5, 10, 6, 2, 42, 41, 3, 2, 2, 2, 43, 46, 3, 2, 2, 2, 44, 42, 3, 2, 2, 2, 44, 45, 3, 2, 2, 2, 45, 49, 3, 2, 2, 2, 46, 44, 3, 2, 2, 2, 47, 49, 5, 6, 4, 2, 48, 44, 3, 2, 2, 2, 48, 47, 3, 2, 2, 2, 49, 50, 3, 2, 2, 2, 50, 51, 7, 2, 2, 3, 51, 5, 3, 2, 2, 2, 52, 53, 8, 4, 1, 2, 53, 54, 7, 3, 2, 2, 54, 55, 5, 6, 4, 2, 55, 56, 7, 4, 2, 2, 56, 128, 3, 2, 2, 2, 57, 58, 7, 28, 2, 2, 58, 128, 5, 6, 4, 19, 59, 128, 5, 8, 5, 2, 60, 61, 7, 3, 2, 2, 61, 128, 7, 4, 2, 2, 62, 63, 7, 3, 2, 2, 63, 64, 5, 6, 4, 2, 64, 65, 7, 5, 2, 2, 65, 66, 7, 4, 2, 2, 66, 128, 3, 2, 2, 2, 67, 68, 7, 3, 2, 2, 68, 71, 5, 6, 4, 2, 69, 70, 7, 5, 2, 2, 70, 72, 5, 6, 4, 2, 71, 69, 3, 2, 2, 2, 72, 73, 3, 2, 2, 2, 73, 71, 3, 2, 2, 2, 73, 74, 3, 2, 2, 2, 74, 75, 3, 2, 2, 2, 75, 76, 7, 4, 2, 2, 76, 128, 3, 2, 2, 2, 77, 86, 7, 6, 2, 2, 78, 83, 5, 6, 4, 2, 79, 80, 7, 5, 2, 2, 80, 82, 5, 6, 4, 2, 81, 79, 3, 2, 2, 2, 82, 85, 3, 2, 2, 2, 83, 81, 3, 2, 2, 2, 83, 84, 3, 2, 2, 2, 84, 87, 3, 2, 2, 2, 85, 83, 3, 2, 2, 2, 86, 78, 3, 2, 2, 2, 86, 87, 3, 2, 2, 2, 87, 88, 3, 2, 2, 2, 88, 128, 7, 7, 2, 2, 89, 90, 7, 8, 2, 2, 90, 91, 7, 3, 2, 2, 91, 92, 5, 6, 4, 2, 92, 93, 7, 4, 2, 2, 93, 94, 5, 32, 17, 2, 94, 95, 7, 9, 2, 2, 95, 96, 5, 32, 17, 2, 96, 128, 3, 2, 2, 2, 97, 99, 7, 10, 2, 2, 98, 100, 7, 38, 2, 2, 99, 98, 3, 2, 2, 2, 99, 100, 3, 2, 2, 2, 100, 101, 3, 2, 2, 2, 101, 102, 5, 16, 9, 2, 102, 103, 7, 11, 2, 2, 103, 104, 5, 6, 4, 2, 104, 105, 7, 12, 2, 2, 105, 106, 5, 6, 4, 8, 106, 128, 3, 2, 2, 2, 107, 109, 7, 10, 2, 2, 108, 110, 7, 38, 2, 2, 109, 108, 3, 2, 2, 2, 109, 110, 3, 2, 2, 2, 110, 111, 3, 2, 2, 2, 111, 112, 5, 16, 9, 2, 112, 113, 7, 11, 2, 2, 113, 114, 7, 13, 2, 2, 114, 115, 5, 6, 4, 2, 115, 116, 7, 14, 2, 2, 116, 117, 7, 12, 2, 2, 117, 118, 5, 6, 4, 7, 118, 128, 3, 2, 2, 2, 119, 120, 5, 36, 19, 2, 120, 121, 7, 11, 2, 2, 121, 122, 5, 6, 4, 2, 122, 123, 7, 12, 2, 2, 123, 124, 5, 6, 4, 5, 124, 128, 3, 2, 2, 2, 125, 128, 5, 36, 19, 2, 126, 128, 5, 34, 18, 2, 127, 52, 3, 2, 2, 2, 127, 57, 3, 2, 2, 2, 127, 59, 3, 2, 2, 2, 127, 60, 3, 2, 2, 2, 127, 62, 3, 2, 2, 2, 127, 67, 3, 2, 2, 2, 127, 77, 3, 2, 2, 2, 127, 89, 3, 2, 2, 2, 127, 97, 3, 2, 2, 2, 127, 107, 3, 2, 2, 2, 127, 119, 3, 2, 2, 2, 127, 125, 3, 2, 2, 2, 127, 126, 3, 2, 2, 2, 128, 159, 3, 2, 2, 2, 129, 130, 12, 18, 2, 2, 130, 131, 9, 2, 2, 2, 131, 158, 5, 6, 4, 19, 132, 133, 12, 17, 2, 2, 133, 134, 9, 3, 2, 2, 134, 158, 5, 6, 4, 18, 135, 136, 12, 16, 2, 2, 136, 137, 9, 4, 2, 2, 137, 158, 5, 6, 4, 17, 138, 139, 12, 15, 2, 2, 139, 140, 9, 5, 2, 2, 140, 158, 5, 6, 4, 16, 141, 142, 12, 6, 2, 2, 142, 143, 7, 12, 2, 2, 143, 158, 5, 6, 4, 7, 144, 145, 12, 20, 2, 2, 145, 154, 7, 3, 2, 2, 146, 151, 5, 6, 4, 2, 147, 148, 7, 5, 2, 2, 148, 150, 5, 6, 4, 2, 149, 147, 3, 2, 2, 2, 150, 153, 3, 2, 2, 2, 151, 149, 3, 2, 2, 2, 151, 152, 3, 2, 2, 2, 152, 155, 3, 2, 2, 2, 153, 151, 3, 2, 2, 2, 154, 146, 3, 2, 2, 2, 154, 155, 3, 2, 2, 2, 155, 156, 3, 2, 2, 2, 156, 158, 7, 4, 2, 2, 157, 129, 3, 2, 2, 2, 157, 132, 3, 2, 2, 2, 157, 135, 3, 2, 2, 2, 157, 138, 3, 2, 2, 2, 157, 141, 3, 2, 2, 2, 157, 144, 3, 2, 2, 2, 158, 161, 3, 2, 2, 2, 159, 157, 3, 2, 2, 2, 159, 160, 3, 2, 2, 2, 160, 7, 3, 2, 2, 2, 161, 159, 3, 2, 2, 2, 162, 164, 7, 15, 2, 2, 163, 165, 5, 22, 12, 2, 164, 163, 3, 2, 2, 2, 164, 165, 3, 2, 2, 2, 165, 166, 3, 2, 2, 2, 166, 167, 7, 3, 2, 2, 167, 168, 5, 12, 7, 2, 168, 171, 7, 4, 2, 2, 169, 170, 7, 16, 2, 2, 170, 172, 5, 24, 13, 2, 171, 169, 3, 2, 2, 2, 171, 172, 3, 2, 2, 2, 172, 173, 3, 2, 2, 2, 173, 174, 5, 32, 17, 2, 174, 9, 3, 2, 2, 2, 175, 176, 7, 17, 2, 2, 176, 178, 5, 36, 19, 2, 177, 179, 5, 22, 12, 2, 178, 177, 3, 2, 2, 2, 178, 179, 3, 2, 2, 2, 179, 180, 3, 2, 2, 2, 180, 181, 7, 3, 2, 2, 181, 182, 5, 12, 7, 2, 182, 185, 7, 4, 2, 2, 183, 184, 7, 16, 2, 2, 184, 186, 5, 24, 13, 2, 185, 183, 3, 2, 2, 2, 185, 186, 3, 2, 2, 2, 186, 187, 3, 2, 2, 2, 187, 188, 5, 32, 17, 2, 188, 11, 3, 2, 2, 2, 189, 196, 5, 14, 8, 2, 190, 196, 5, 18, 10, 2, 191, 192, 5, 14, 8, 2, 192, 193, 7, 5, 2, 2, 193, 194, 5, 18, 10, 2, 194, 196, 3, 2, 2, 2, 195, 189, 3, 2, 2, 2, 195, 190, 3, 2, 2, 2, 195, 191, 3, 2, 2, 2, 196, 13, 3, 2, 2, 2, 197, 202, 5, 16, 9, 2, 198, 199, 7, 5, 2, 2, 199, 201, 5, 16, 9, 2, 200, 198, 3, 2, 2, 2, 201, 204, 3, 2, 2, 2, 202, 200, 3, 2, 2, 2, 202, 203, 3, 2, 2, 2, 203, 206, 3, 2, 2, 2, 204, 202, 3, 2, 2, 2, 205, 197, 3, 2, 2, 2, 205, 206, 3, 2, 2, 2, 206, 15, 3, 2, 2, 2, 207, 210, 5, 36, 19, 2, 208, 209, 7, 18, 2, 2, 209, 211, 5, 24, 13, 2, 210, 208, 3, 2, 2, 2, 210, 211, 3, 2, 2, 2, 211, 17, 3, 2, 2, 2, 212, 217, 5, 20, 11, 2, 213, 214, 7, 5, 2, 2, 214, 216, 5, 20, 11, 2, 215, 213, 3, 2, 2, 2, 216, 219, 3, 2, 2, 2, 217, 215, 3, 2, 2, 2, 217, 218, 3, 2, 2, 2, 218, 221, 3, 2, 2, 2, 219, 217, 3, 2, 2, 2, 220, 212, 3, 2, 2, 2, 220, 221, 3, 2, 2, 2, 221, 19, 3, 2, 2, 2, 222, 223, 7, 42, 2, 2, 223, 224, 7, 11, 2, 2, 224, 225, 5, 6, 4, 2, 225, 21, 3, 2, 2, 2, 226, 227, 7, 6, 2, 2, 227, 240, 7, 7, 2, 2, 228, 229, 7, 6, 2, 2, 229, 234, 5, 36, 19, 2, 230, 231, 7, 5, 2, 2, 231, 233, 5, 36, 19, 2, 232, 230, 3, 2, 2, 2, 233, 236, 3, 2, 2, 2, 234, 232, 3, 2, 2, 2, 234, 235, 3, 2, 2, 2, 235, 237, 3, 2, 2, 2, 236, 234, 3, 2, 2, 2, 237, 238, 7, 7, 2, 2, 238, 240, 3, 2, 2, 2, 239, 226, 3, 2, 2, 2, 239, 228, 3, 2, 2, 2, 240, 23, 3, 2, 2, 2, 241, 242, 7, 3, 2, 2, 242, 287, 7, 4, 2, 2, 243, 244, 7, 3, 2, 2, 244, 245, 5, 24, 13, 2, 245, 246, 7, 5, 2, 2, 246, 247, 7, 4, 2, 2, 247, 287, 3, 2, 2, 2, 248, 249, 7, 3, 2, 2, 249, 252, 5, 24, 13, 2, 250, 251, 7, 5, 2, 2, 251, 253, 5, 24, 13, 2, 252, 250, 3, 2, 2, 2, 253, 254, 3, 2, 2, 2, 254, 252, 3, 2, 2, 2, 254, 255, 3, 2, 2, 2, 255, 256, 3, 2, 2, 2, 256, 257, 7, 4, 2, 2, 257, 287, 3, 2, 2, 2, 258, 287, 5, 30, 16, 2, 259, 260, 7, 19, 2, 2, 260, 261, 7, 6, 2, 2, 261, 262, 5, 26, 14, 2, 262, 263, 7, 5, 2, 2, 263, 264, 5, 24, 13, 2, 264, 265, 7, 7, 2, 2, 265, 287, 3, 2, 2, 2, 266, 268, 7, 15, 2, 2, 267, 269, 5, 22, 12, 2, 268, 267, 3, 2, 2, 2, 268, 269, 3, 2, 2, 2, 269, 270, 3, 2, 2, 2, 270, 279, 7, 3, 2, 2, 271, 276, 5, 24, 13, 2, 272, 273, 7, 5, 2, 2, 273, 275, 5, 24, 13, 2, 274, 272, 3, 2, 2, 2, 275, 278, 3, 2, 2, 2, 276, 274, 3, 2, 2, 2, 276, 277, 3, 2, 2, 2, 277, 280, 3, 2, 2, 2, 278, 276, 3, 2, 2, 2, 279, 271, 3, 2, 2, 2, 279, 280, 3, 2, 2, 2, 280, 281, 3, 2, 2, 2, 281, 282, 7, 4, 2, 2, 282, 283, 7, 16, 2, 2, 283, 287, 5, 24, 13, 2, 284, 287, 7, 20, 2, 2, 285, 287, 7, 41, 2, 2, 286, 241, 3, 2, 2, 2, 286, 243, 3, 2, 2, 2, 286, 248, 3, 2, 2, 2, 286, 258, 3, 2, 2, 2, 286, 259, 3, 2, 2, 2, 286, 266, 3, 2, 2, 2, 286, 284, 3, 2, 2, 2, 286, 285, 3, 2, 2, 2, 287, 25, 3, 2, 2, 2, 288, 289, 7, 3, 2, 2, 289, 306, 7, 4, 2, 2, 290, 291, 7, 3, 2, 2, 291, 292, 5, 28, 15, 2, 292, 293, 7, 5, 2, 2, 293, 294, 7, 4, 2, 2, 294, 306, 3, 2, 2, 2, 295, 296, 7, 3, 2, 2, 296, 299, 5, 28, 15, 2, 297, 298, 7, 5, 2, 2, 298, 300, 5, 28, 15, 2, 299, 297, 3, 2, 2, 2, 300, 301, 3, 2, 2, 2, 301, 299, 3, 2, 2, 2, 301, 302, 3, 2, 2, 2, 302, 303, 3, 2, 2, 2, 303, 304, 7, 4, 2, 2, 304, 306, 3, 2, 2, 2, 305, 288, 3, 2, 2, 2, 305, 290, 3, 2, 2, 2, 305, 295, 3, 2, 2, 2, 306, 27, 3, 2, 2, 2, 307, 308, 7, 3, 2, 2, 308, 309, 5, 28, 15, 2, 309, 310, 7, 4, 2, 2, 310, 313, 3, 2, 2, 2, 311, 313, 7, 41, 2, 2, 312, 307, 3, 2, 2, 2, 312, 311, 3, 2, 2, 2, 313, 29, 3, 2, 2, 2, 314, 315, 7, 42, 2, 2, 315, 31, 3, 2, 2, 2, 316, 317, 7, 13, 2, 2, 317, 318, 5, 6, 4, 2, 318, 319, 7, 14, 2, 2, 319, 33, 3, 2, 2, 2, 320, 324, 7, 40, 2, 2, 321, 324, 7, 41, 2, 2, 322, 324, 7, 39, 2, 2, 323, 320, 3, 2, 2, 2, 323, 321, 3, 2, 2, 2, 323, 322, 3, 2, 2, 2, 324, 35, 3, 2, 2, 2, 325, 330, 5, 2, 2, 2, 326, 330, 7, 35, 2, 2, 327, 330, 7, 36, 2, 2, 328, 330, 7, 37, 2, 2, 329, 325, 3, 2, 2, 2, 329, 326, 3, 2, 2, 2, 329, 327, 3, 2, 2, 2, 329, 328, 3, 2, 2, 2, 330, 37, 3, 2, 2, 2, 36, 44, 48, 73, 83, 86, 99, 109, 127, 151, 154, 157, 159, 164, 171, 178, 185, 195, 202, 205, 210, 217, 220, 234, 239, 254, 268, 276, 279, 286, 301, 305, 312, 323, 329]
\ No newline at end of file
python/tvm/relay/grammar/py2/Relay.tokens deleted 100644 → 0
T__0=1
T__1=2
T__2=3
T__3=4
T__4=5
T__5=6
T__6=7
T__7=8
T__8=9
T__9=10
T__10=11
T__11=12
T__12=13
T__13=14
T__14=15
T__15=16
T__16=17
T__17=18
SEMVER=19
WS=20
LINE_COMMENT=21
COMMENT=22
MUL=23
DIV=24
ADD=25
SUB=26
LT=27
GT=28
LE=29
GE=30
EQ=31
NE=32
GLOBAL_VAR=33
LOCAL_VAR=34
GRAPH_VAR=35
MUT=36
BOOL_LIT=37
FLOAT=38
NAT=39
CNAME=40
'('=1
')'=2
','=3
'['=4
']'=5
'if'=6
'else'=7
'let'=8
'='=9
';'=10
'{'=11
'}'=12
'fn'=13
'->'=14
'def'=15
':'=16
'Tensor'=17
'_'=18
'v0.0.2'=19
'*'=23
'/'=24
'+'=25
'-'=26
'<'=27
'>'=28
'<='=29
'>='=30
'=='=31
'!='=32
'mut'=36
python/tvm/relay/grammar/py2/RelayLexer.interp deleted 100644 → 0
token literal names:
null
'('
')'
','
'['
']'
'if'
'else'
'let'
'='
';'
'{'
'}'
'fn'
'->'
'def'
':'
'Tensor'
'_'
'v0.0.2'
null
null
null
'*'
'/'
'+'
'-'
'<'
'>'
'<='
'>='
'=='
'!='
null
null
null
'mut'
null
null
null
null
token symbolic names:
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
null
SEMVER
WS
LINE_COMMENT
COMMENT
MUL
DIV
ADD
SUB
LT
GT
LE
GE
EQ
NE
GLOBAL_VAR
LOCAL_VAR
GRAPH_VAR
MUT
BOOL_LIT
FLOAT
NAT
CNAME
rule names:
T__0
T__1
T__2
T__3
T__4
T__5
T__6
T__7
T__8
T__9
T__10
T__11
T__12
T__13
T__14
T__15
T__16
T__17
SEMVER
WS
LINE_COMMENT
COMMENT
MUL
DIV
ADD
SUB
LT
GT
LE
GE
EQ
NE
GLOBAL_VAR
LOCAL_VAR
GRAPH_VAR
MUT
BOOL_LIT
FLOAT
NAT
EXP
CNAME
LETTER
DIGIT
channel names:
DEFAULT_TOKEN_CHANNEL
HIDDEN
mode names:
DEFAULT_MODE
atn:
[3, 24715, 42794, 33075, 47597, 16764, 15335, 30598, 22884, 2, 42, 267, 8, 1, 4, 2, 9, 2, 4, 3, 9, 3, 4, 4, 9, 4, 4, 5, 9, 5, 4, 6, 9, 6, 4, 7, 9, 7, 4, 8, 9, 8, 4, 9, 9, 9, 4, 10, 9, 10, 4, 11, 9, 11, 4, 12, 9, 12, 4, 13, 9, 13, 4, 14, 9, 14, 4, 15, 9, 15, 4, 16, 9, 16, 4, 17, 9, 17, 4, 18, 9, 18, 4, 19, 9, 19, 4, 20, 9, 20, 4, 21, 9, 21, 4, 22, 9, 22, 4, 23, 9, 23, 4, 24, 9, 24, 4, 25, 9, 25, 4, 26, 9, 26, 4, 27, 9, 27, 4, 28, 9, 28, 4, 29, 9, 29, 4, 30, 9, 30, 4, 31, 9, 31, 4, 32, 9, 32, 4, 33, 9, 33, 4, 34, 9, 34, 4, 35, 9, 35, 4, 36, 9, 36, 4, 37, 9, 37, 4, 38, 9, 38, 4, 39, 9, 39, 4, 40, 9, 40, 4, 41, 9, 41, 4, 42, 9, 42, 4, 43, 9, 43, 4, 44, 9, 44, 3, 2, 3, 2, 3, 3, 3, 3, 3, 4, 3, 4, 3, 5, 3, 5, 3, 6, 3, 6, 3, 7, 3, 7, 3, 7, 3, 8, 3, 8, 3, 8, 3, 8, 3, 8, 3, 9, 3, 9, 3, 9, 3, 9, 3, 10, 3, 10, 3, 11, 3, 11, 3, 12, 3, 12, 3, 13, 3, 13, 3, 14, 3, 14, 3, 14, 3, 15, 3, 15, 3, 15, 3, 16, 3, 16, 3, 16, 3, 16, 3, 17, 3, 17, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 18, 3, 19, 3, 19, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 20, 3, 21, 6, 21, 149, 10, 21, 13, 21, 14, 21, 150, 3, 21, 3, 21, 3, 22, 3, 22, 3, 22, 3, 22, 7, 22, 159, 10, 22, 12, 22, 14, 22, 162, 11, 22, 3, 22, 3, 22, 3, 22, 3, 22, 3, 23, 3, 23, 3, 23, 3, 23, 7, 23, 172, 10, 23, 12, 23, 14, 23, 175, 11, 23, 3, 23, 3, 23, 3, 23, 3, 23, 3, 23, 3, 24, 3, 24, 3, 25, 3, 25, 3, 26, 3, 26, 3, 27, 3, 27, 3, 28, 3, 28, 3, 29, 3, 29, 3, 30, 3, 30, 3, 30, 3, 31, 3, 31, 3, 31, 3, 32, 3, 32, 3, 32, 3, 33, 3, 33, 3, 33, 3, 34, 3, 34, 3, 34, 3, 35, 3, 35, 3, 35, 3, 36, 3, 36, 3, 36, 3, 37, 3, 37, 3, 37, 3, 37, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 3, 38, 5, 38, 228, 10, 38, 3, 39, 3, 39, 3, 39, 3, 39, 5, 39, 234, 10, 39, 3, 39, 3, 39, 3, 39, 5, 39, 239, 10, 39, 3, 40, 6, 40, 242, 10, 40, 13, 40, 14, 40, 243, 3, 41, 3, 41, 5, 41, 248, 10, 41, 3, 41, 3, 41, 3, 42, 3, 42, 5, 42, 254, 10, 42, 3, 42, 3, 42, 3, 42, 7, 42, 259, 10, 42, 12, 42, 14, 42, 262, 11, 42, 3, 43, 3, 43, 3, 44, 3, 44, 4, 160, 173, 2, 45, 3, 3, 5, 4, 7, 5, 9, 6, 11, 7, 13, 8, 15, 9, 17, 10, 19, 11, 21, 12, 23, 13, 25, 14, 27, 15, 29, 16, 31, 17, 33, 18, 35, 19, 37, 20, 39, 21, 41, 22, 43, 23, 45, 24, 47, 25, 49, 26, 51, 27, 53, 28, 55, 29, 57, 30, 59, 31, 61, 32, 63, 33, 65, 34, 67, 35, 69, 36, 71, 37, 73, 38, 75, 39, 77, 40, 79, 41, 81, 2, 83, 42, 85, 2, 87, 2, 3, 2, 7, 5, 2, 11, 12, 15, 15, 34, 34, 4, 2, 71, 71, 103, 103, 4, 2, 45, 45, 47, 47, 4, 2, 67, 92, 99, 124, 3, 2, 50, 59, 2, 275, 2, 3, 3, 2, 2, 2, 2, 5, 3, 2, 2, 2, 2, 7, 3, 2, 2, 2, 2, 9, 3, 2, 2, 2, 2, 11, 3, 2, 2, 2, 2, 13, 3, 2, 2, 2, 2, 15, 3, 2, 2, 2, 2, 17, 3, 2, 2, 2, 2, 19, 3, 2, 2, 2, 2, 21, 3, 2, 2, 2, 2, 23, 3, 2, 2, 2, 2, 25, 3, 2, 2, 2, 2, 27, 3, 2, 2, 2, 2, 29, 3, 2, 2, 2, 2, 31, 3, 2, 2, 2, 2, 33, 3, 2, 2, 2, 2, 35, 3, 2, 2, 2, 2, 37, 3, 2, 2, 2, 2, 39, 3, 2, 2, 2, 2, 41, 3, 2, 2, 2, 2, 43, 3, 2, 2, 2, 2, 45, 3, 2, 2, 2, 2, 47, 3, 2, 2, 2, 2, 49, 3, 2, 2, 2, 2, 51, 3, 2, 2, 2, 2, 53, 3, 2, 2, 2, 2, 55, 3, 2, 2, 2, 2, 57, 3, 2, 2, 2, 2, 59, 3, 2, 2, 2, 2, 61, 3, 2, 2, 2, 2, 63, 3, 2, 2, 2, 2, 65, 3, 2, 2, 2, 2, 67, 3, 2, 2, 2, 2, 69, 3, 2, 2, 2, 2, 71, 3, 2, 2, 2, 2, 73, 3, 2, 2, 2, 2, 75, 3, 2, 2, 2, 2, 77, 3, 2, 2, 2, 2, 79, 3, 2, 2, 2, 2, 83, 3, 2, 2, 2, 3, 89, 3, 2, 2, 2, 5, 91, 3, 2, 2, 2, 7, 93, 3, 2, 2, 2, 9, 95, 3, 2, 2, 2, 11, 97, 3, 2, 2, 2, 13, 99, 3, 2, 2, 2, 15, 102, 3, 2, 2, 2, 17, 107, 3, 2, 2, 2, 19, 111, 3, 2, 2, 2, 21, 113, 3, 2, 2, 2, 23, 115, 3, 2, 2, 2, 25, 117, 3, 2, 2, 2, 27, 119, 3, 2, 2, 2, 29, 122, 3, 2, 2, 2, 31, 125, 3, 2, 2, 2, 33, 129, 3, 2, 2, 2, 35, 131, 3, 2, 2, 2, 37, 138, 3, 2, 2, 2, 39, 140, 3, 2, 2, 2, 41, 148, 3, 2, 2, 2, 43, 154, 3, 2, 2, 2, 45, 167, 3, 2, 2, 2, 47, 181, 3, 2, 2, 2, 49, 183, 3, 2, 2, 2, 51, 185, 3, 2, 2, 2, 53, 187, 3, 2, 2, 2, 55, 189, 3, 2, 2, 2, 57, 191, 3, 2, 2, 2, 59, 193, 3, 2, 2, 2, 61, 196, 3, 2, 2, 2, 63, 199, 3, 2, 2, 2, 65, 202, 3, 2, 2, 2, 67, 205, 3, 2, 2, 2, 69, 208, 3, 2, 2, 2, 71, 211, 3, 2, 2, 2, 73, 214, 3, 2, 2, 2, 75, 227, 3, 2, 2, 2, 77, 238, 3, 2, 2, 2, 79, 241, 3, 2, 2, 2, 81, 245, 3, 2, 2, 2, 83, 253, 3, 2, 2, 2, 85, 263, 3, 2, 2, 2, 87, 265, 3, 2, 2, 2, 89, 90, 7, 42, 2, 2, 90, 4, 3, 2, 2, 2, 91, 92, 7, 43, 2, 2, 92, 6, 3, 2, 2, 2, 93, 94, 7, 46, 2, 2, 94, 8, 3, 2, 2, 2, 95, 96, 7, 93, 2, 2, 96, 10, 3, 2, 2, 2, 97, 98, 7, 95, 2, 2, 98, 12, 3, 2, 2, 2, 99, 100, 7, 107, 2, 2, 100, 101, 7, 104, 2, 2, 101, 14, 3, 2, 2, 2, 102, 103, 7, 103, 2, 2, 103, 104, 7, 110, 2, 2, 104, 105, 7, 117, 2, 2, 105, 106, 7, 103, 2, 2, 106, 16, 3, 2, 2, 2, 107, 108, 7, 110, 2, 2, 108, 109, 7, 103, 2, 2, 109, 110, 7, 118, 2, 2, 110, 18, 3, 2, 2, 2, 111, 112, 7, 63, 2, 2, 112, 20, 3, 2, 2, 2, 113, 114, 7, 61, 2, 2, 114, 22, 3, 2, 2, 2, 115, 116, 7, 125, 2, 2, 116, 24, 3, 2, 2, 2, 117, 118, 7, 127, 2, 2, 118, 26, 3, 2, 2, 2, 119, 120, 7, 104, 2, 2, 120, 121, 7, 112, 2, 2, 121, 28, 3, 2, 2, 2, 122, 123, 7, 47, 2, 2, 123, 124, 7, 64, 2, 2, 124, 30, 3, 2, 2, 2, 125, 126, 7, 102, 2, 2, 126, 127, 7, 103, 2, 2, 127, 128, 7, 104, 2, 2, 128, 32, 3, 2, 2, 2, 129, 130, 7, 60, 2, 2, 130, 34, 3, 2, 2, 2, 131, 132, 7, 86, 2, 2, 132, 133, 7, 103, 2, 2, 133, 134, 7, 112, 2, 2, 134, 135, 7, 117, 2, 2, 135, 136, 7, 113, 2, 2, 136, 137, 7, 116, 2, 2, 137, 36, 3, 2, 2, 2, 138, 139, 7, 97, 2, 2, 139, 38, 3, 2, 2, 2, 140, 141, 7, 120, 2, 2, 141, 142, 7, 50, 2, 2, 142, 143, 7, 48, 2, 2, 143, 144, 7, 50, 2, 2, 144, 145, 7, 48, 2, 2, 145, 146, 7, 52, 2, 2, 146, 40, 3, 2, 2, 2, 147, 149, 9, 2, 2, 2, 148, 147, 3, 2, 2, 2, 149, 150, 3, 2, 2, 2, 150, 148, 3, 2, 2, 2, 150, 151, 3, 2, 2, 2, 151, 152, 3, 2, 2, 2, 152, 153, 8, 21, 2, 2, 153, 42, 3, 2, 2, 2, 154, 155, 7, 49, 2, 2, 155, 156, 7, 49, 2, 2, 156, 160, 3, 2, 2, 2, 157, 159, 11, 2, 2, 2, 158, 157, 3, 2, 2, 2, 159, 162, 3, 2, 2, 2, 160, 161, 3, 2, 2, 2, 160, 158, 3, 2, 2, 2, 161, 163, 3, 2, 2, 2, 162, 160, 3, 2, 2, 2, 163, 164, 7, 12, 2, 2, 164, 165, 3, 2, 2, 2, 165, 166, 8, 22, 2, 2, 166, 44, 3, 2, 2, 2, 167, 168, 7, 49, 2, 2, 168, 169, 7, 44, 2, 2, 169, 173, 3, 2, 2, 2, 170, 172, 11, 2, 2, 2, 171, 170, 3, 2, 2, 2, 172, 175, 3, 2, 2, 2, 173, 174, 3, 2, 2, 2, 173, 171, 3, 2, 2, 2, 174, 176, 3, 2, 2, 2, 175, 173, 3, 2, 2, 2, 176, 177, 7, 44, 2, 2, 177, 178, 7, 49, 2, 2, 178, 179, 3, 2, 2, 2, 179, 180, 8, 23, 2, 2, 180, 46, 3, 2, 2, 2, 181, 182, 7, 44, 2, 2, 182, 48, 3, 2, 2, 2, 183, 184, 7, 49, 2, 2, 184, 50, 3, 2, 2, 2, 185, 186, 7, 45, 2, 2, 186, 52, 3, 2, 2, 2, 187, 188, 7, 47, 2, 2, 188, 54, 3, 2, 2, 2, 189, 190, 7, 62, 2, 2, 190, 56, 3, 2, 2, 2, 191, 192, 7, 64, 2, 2, 192, 58, 3, 2, 2, 2, 193, 194, 7, 62, 2, 2, 194, 195, 7, 63, 2, 2, 195, 60, 3, 2, 2, 2, 196, 197, 7, 64, 2, 2, 197, 198, 7, 63, 2, 2, 198, 62, 3, 2, 2, 2, 199, 200, 7, 63, 2, 2, 200, 201, 7, 63, 2, 2, 201, 64, 3, 2, 2, 2, 202, 203, 7, 35, 2, 2, 203, 204, 7, 63, 2, 2, 204, 66, 3, 2, 2, 2, 205, 206, 7, 66, 2, 2, 206, 207, 5, 83, 42, 2, 207, 68, 3, 2, 2, 2, 208, 209, 7, 39, 2, 2, 209, 210, 5, 83, 42, 2, 210, 70, 3, 2, 2, 2, 211, 212, 7, 39, 2, 2, 212, 213, 5, 79, 40, 2, 213, 72, 3, 2, 2, 2, 214, 215, 7, 111, 2, 2, 215, 216, 7, 119, 2, 2, 216, 217, 7, 118, 2, 2, 217, 74, 3, 2, 2, 2, 218, 219, 7, 86, 2, 2, 219, 220, 7, 116, 2, 2, 220, 221, 7, 119, 2, 2, 221, 228, 7, 103, 2, 2, 222, 223, 7, 72, 2, 2, 223, 224, 7, 99, 2, 2, 224, 225, 7, 110, 2, 2, 225, 226, 7, 117, 2, 2, 226, 228, 7, 103, 2, 2, 227, 218, 3, 2, 2, 2, 227, 222, 3, 2, 2, 2, 228, 76, 3, 2, 2, 2, 229, 230, 5, 79, 40, 2, 230, 231, 7, 48, 2, 2, 231, 233, 5, 79, 40, 2, 232, 234, 5, 81, 41, 2, 233, 232, 3, 2, 2, 2, 233, 234, 3, 2, 2, 2, 234, 239, 3, 2, 2, 2, 235, 236, 5, 79, 40, 2, 236, 237, 5, 81, 41, 2, 237, 239, 3, 2, 2, 2, 238, 229, 3, 2, 2, 2, 238, 235, 3, 2, 2, 2, 239, 78, 3, 2, 2, 2, 240, 242, 5, 87, 44, 2, 241, 240, 3, 2, 2, 2, 242, 243, 3, 2, 2, 2, 243, 241, 3, 2, 2, 2, 243, 244, 3, 2, 2, 2, 244, 80, 3, 2, 2, 2, 245, 247, 9, 3, 2, 2, 246, 248, 9, 4, 2, 2, 247, 246, 3, 2, 2, 2, 247, 248, 3, 2, 2, 2, 248, 249, 3, 2, 2, 2, 249, 250, 5, 79, 40, 2, 250, 82, 3, 2, 2, 2, 251, 254, 7, 97, 2, 2, 252, 254, 5, 85, 43, 2, 253, 251, 3, 2, 2, 2, 253, 252, 3, 2, 2, 2, 254, 260, 3, 2, 2, 2, 255, 259, 7, 97, 2, 2, 256, 259, 5, 85, 43, 2, 257, 259, 5, 87, 44, 2, 258, 255, 3, 2, 2, 2, 258, 256, 3, 2, 2, 2, 258, 257, 3, 2, 2, 2, 259, 262, 3, 2, 2, 2, 260, 258, 3, 2, 2, 2, 260, 261, 3, 2, 2, 2, 261, 84, 3, 2, 2, 2, 262, 260, 3, 2, 2, 2, 263, 264, 9, 5, 2, 2, 264, 86, 3, 2, 2, 2, 265, 266, 9, 6, 2, 2, 266, 88, 3, 2, 2, 2, 14, 2, 150, 160, 173, 227, 233, 238, 243, 247, 253, 258, 260, 3, 8, 2, 2]
\ No newline at end of file
python/tvm/relay/grammar/py2/RelayLexer.py deleted 100644 → 0
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
# encoding: utf-8
from __future__ import print_function
from antlr4 import *
from io import StringIO
import sys
def serializedATN():
with StringIO() as buf:
buf.write(u"\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\2")
buf.write(u"*\u010b\b\1\4\2\t\2\4\3\t\3\4\4\t\4\4\5\t\5\4\6\t\6\4")
buf.write(u"\7\t\7\4\b\t\b\4\t\t\t\4\n\t\n\4\13\t\13\4\f\t\f\4\r")
buf.write(u"\t\r\4\16\t\16\4\17\t\17\4\20\t\20\4\21\t\21\4\22\t\22")
buf.write(u"\4\23\t\23\4\24\t\24\4\25\t\25\4\26\t\26\4\27\t\27\4")
buf.write(u"\30\t\30\4\31\t\31\4\32\t\32\4\33\t\33\4\34\t\34\4\35")
buf.write(u"\t\35\4\36\t\36\4\37\t\37\4 \t \4!\t!\4\"\t\"\4#\t#\4")
buf.write(u"$\t$\4%\t%\4&\t&\4\'\t\'\4(\t(\4)\t)\4*\t*\4+\t+\4,\t")
buf.write(u",\3\2\3\2\3\3\3\3\3\4\3\4\3\5\3\5\3\6\3\6\3\7\3\7\3\7")
buf.write(u"\3\b\3\b\3\b\3\b\3\b\3\t\3\t\3\t\3\t\3\n\3\n\3\13\3\13")
buf.write(u"\3\f\3\f\3\r\3\r\3\16\3\16\3\16\3\17\3\17\3\17\3\20\3")
buf.write(u"\20\3\20\3\20\3\21\3\21\3\22\3\22\3\22\3\22\3\22\3\22")
buf.write(u"\3\22\3\23\3\23\3\24\3\24\3\24\3\24\3\24\3\24\3\24\3")
buf.write(u"\25\6\25\u0095\n\25\r\25\16\25\u0096\3\25\3\25\3\26\3")
buf.write(u"\26\3\26\3\26\7\26\u009f\n\26\f\26\16\26\u00a2\13\26")
buf.write(u"\3\26\3\26\3\26\3\26\3\27\3\27\3\27\3\27\7\27\u00ac\n")
buf.write(u"\27\f\27\16\27\u00af\13\27\3\27\3\27\3\27\3\27\3\27\3")
buf.write(u"\30\3\30\3\31\3\31\3\32\3\32\3\33\3\33\3\34\3\34\3\35")
buf.write(u"\3\35\3\36\3\36\3\36\3\37\3\37\3\37\3 \3 \3 \3!\3!\3")
buf.write(u"!\3\"\3\"\3\"\3#\3#\3#\3$\3$\3$\3%\3%\3%\3%\3&\3&\3&")
buf.write(u"\3&\3&\3&\3&\3&\3&\5&\u00e4\n&\3\'\3\'\3\'\3\'\5\'\u00ea")
buf.write(u"\n\'\3\'\3\'\3\'\5\'\u00ef\n\'\3(\6(\u00f2\n(\r(\16(")
buf.write(u"\u00f3\3)\3)\5)\u00f8\n)\3)\3)\3*\3*\5*\u00fe\n*\3*\3")
buf.write(u"*\3*\7*\u0103\n*\f*\16*\u0106\13*\3+\3+\3,\3,\4\u00a0")
buf.write(u"\u00ad\2-\3\3\5\4\7\5\t\6\13\7\r\b\17\t\21\n\23\13\25")
buf.write(u"\f\27\r\31\16\33\17\35\20\37\21!\22#\23%\24\'\25)\26")
buf.write(u"+\27-\30/\31\61\32\63\33\65\34\67\359\36;\37= ?!A\"C")
buf.write(u"#E$G%I&K\'M(O)Q\2S*U\2W\2\3\2\7\5\2\13\f\17\17\"\"\4")
buf.write(u"\2GGgg\4\2--//\4\2C\\c|\3\2\62;\2\u0113\2\3\3\2\2\2\2")
buf.write(u"\5\3\2\2\2\2\7\3\2\2\2\2\t\3\2\2\2\2\13\3\2\2\2\2\r\3")
buf.write(u"\2\2\2\2\17\3\2\2\2\2\21\3\2\2\2\2\23\3\2\2\2\2\25\3")
buf.write(u"\2\2\2\2\27\3\2\2\2\2\31\3\2\2\2\2\33\3\2\2\2\2\35\3")
buf.write(u"\2\2\2\2\37\3\2\2\2\2!\3\2\2\2\2#\3\2\2\2\2%\3\2\2\2")
buf.write(u"\2\'\3\2\2\2\2)\3\2\2\2\2+\3\2\2\2\2-\3\2\2\2\2/\3\2")
buf.write(u"\2\2\2\61\3\2\2\2\2\63\3\2\2\2\2\65\3\2\2\2\2\67\3\2")
buf.write(u"\2\2\29\3\2\2\2\2;\3\2\2\2\2=\3\2\2\2\2?\3\2\2\2\2A\3")
buf.write(u"\2\2\2\2C\3\2\2\2\2E\3\2\2\2\2G\3\2\2\2\2I\3\2\2\2\2")
buf.write(u"K\3\2\2\2\2M\3\2\2\2\2O\3\2\2\2\2S\3\2\2\2\3Y\3\2\2\2")
buf.write(u"\5[\3\2\2\2\7]\3\2\2\2\t_\3\2\2\2\13a\3\2\2\2\rc\3\2")
buf.write(u"\2\2\17f\3\2\2\2\21k\3\2\2\2\23o\3\2\2\2\25q\3\2\2\2")
buf.write(u"\27s\3\2\2\2\31u\3\2\2\2\33w\3\2\2\2\35z\3\2\2\2\37}")
buf.write(u"\3\2\2\2!\u0081\3\2\2\2#\u0083\3\2\2\2%\u008a\3\2\2\2")
buf.write(u"\'\u008c\3\2\2\2)\u0094\3\2\2\2+\u009a\3\2\2\2-\u00a7")
buf.write(u"\3\2\2\2/\u00b5\3\2\2\2\61\u00b7\3\2\2\2\63\u00b9\3\2")
buf.write(u"\2\2\65\u00bb\3\2\2\2\67\u00bd\3\2\2\29\u00bf\3\2\2\2")
buf.write(u";\u00c1\3\2\2\2=\u00c4\3\2\2\2?\u00c7\3\2\2\2A\u00ca")
buf.write(u"\3\2\2\2C\u00cd\3\2\2\2E\u00d0\3\2\2\2G\u00d3\3\2\2\2")
buf.write(u"I\u00d6\3\2\2\2K\u00e3\3\2\2\2M\u00ee\3\2\2\2O\u00f1")
buf.write(u"\3\2\2\2Q\u00f5\3\2\2\2S\u00fd\3\2\2\2U\u0107\3\2\2\2")
buf.write(u"W\u0109\3\2\2\2YZ\7*\2\2Z\4\3\2\2\2[\\\7+\2\2\\\6\3\2")
buf.write(u"\2\2]^\7.\2\2^\b\3\2\2\2_`\7]\2\2`\n\3\2\2\2ab\7_\2\2")
buf.write(u"b\f\3\2\2\2cd\7k\2\2de\7h\2\2e\16\3\2\2\2fg\7g\2\2gh")
buf.write(u"\7n\2\2hi\7u\2\2ij\7g\2\2j\20\3\2\2\2kl\7n\2\2lm\7g\2")
buf.write(u"\2mn\7v\2\2n\22\3\2\2\2op\7?\2\2p\24\3\2\2\2qr\7=\2\2")
buf.write(u"r\26\3\2\2\2st\7}\2\2t\30\3\2\2\2uv\7\177\2\2v\32\3\2")
buf.write(u"\2\2wx\7h\2\2xy\7p\2\2y\34\3\2\2\2z{\7/\2\2{|\7@\2\2")
buf.write(u"|\36\3\2\2\2}~\7f\2\2~\177\7g\2\2\177\u0080\7h\2\2\u0080")
buf.write(u" \3\2\2\2\u0081\u0082\7<\2\2\u0082\"\3\2\2\2\u0083\u0084")
buf.write(u"\7V\2\2\u0084\u0085\7g\2\2\u0085\u0086\7p\2\2\u0086\u0087")
buf.write(u"\7u\2\2\u0087\u0088\7q\2\2\u0088\u0089\7t\2\2\u0089$")
buf.write(u"\3\2\2\2\u008a\u008b\7a\2\2\u008b&\3\2\2\2\u008c\u008d")
buf.write(u"\7x\2\2\u008d\u008e\7\62\2\2\u008e\u008f\7\60\2\2\u008f")
buf.write(u"\u0090\7\62\2\2\u0090\u0091\7\60\2\2\u0091\u0092\7\64")
buf.write(u"\2\2\u0092(\3\2\2\2\u0093\u0095\t\2\2\2\u0094\u0093\3")
buf.write(u"\2\2\2\u0095\u0096\3\2\2\2\u0096\u0094\3\2\2\2\u0096")
buf.write(u"\u0097\3\2\2\2\u0097\u0098\3\2\2\2\u0098\u0099\b\25\2")
buf.write(u"\2\u0099*\3\2\2\2\u009a\u009b\7\61\2\2\u009b\u009c\7")
buf.write(u"\61\2\2\u009c\u00a0\3\2\2\2\u009d\u009f\13\2\2\2\u009e")
buf.write(u"\u009d\3\2\2\2\u009f\u00a2\3\2\2\2\u00a0\u00a1\3\2\2")
buf.write(u"\2\u00a0\u009e\3\2\2\2\u00a1\u00a3\3\2\2\2\u00a2\u00a0")
buf.write(u"\3\2\2\2\u00a3\u00a4\7\f\2\2\u00a4\u00a5\3\2\2\2\u00a5")
buf.write(u"\u00a6\b\26\2\2\u00a6,\3\2\2\2\u00a7\u00a8\7\61\2\2\u00a8")
buf.write(u"\u00a9\7,\2\2\u00a9\u00ad\3\2\2\2\u00aa\u00ac\13\2\2")
buf.write(u"\2\u00ab\u00aa\3\2\2\2\u00ac\u00af\3\2\2\2\u00ad\u00ae")
buf.write(u"\3\2\2\2\u00ad\u00ab\3\2\2\2\u00ae\u00b0\3\2\2\2\u00af")
buf.write(u"\u00ad\3\2\2\2\u00b0\u00b1\7,\2\2\u00b1\u00b2\7\61\2")
buf.write(u"\2\u00b2\u00b3\3\2\2\2\u00b3\u00b4\b\27\2\2\u00b4.\3")
buf.write(u"\2\2\2\u00b5\u00b6\7,\2\2\u00b6\60\3\2\2\2\u00b7\u00b8")
buf.write(u"\7\61\2\2\u00b8\62\3\2\2\2\u00b9\u00ba\7-\2\2\u00ba\64")
buf.write(u"\3\2\2\2\u00bb\u00bc\7/\2\2\u00bc\66\3\2\2\2\u00bd\u00be")
buf.write(u"\7>\2\2\u00be8\3\2\2\2\u00bf\u00c0\7@\2\2\u00c0:\3\2")
buf.write(u"\2\2\u00c1\u00c2\7>\2\2\u00c2\u00c3\7?\2\2\u00c3<\3\2")
buf.write(u"\2\2\u00c4\u00c5\7@\2\2\u00c5\u00c6\7?\2\2\u00c6>\3\2")
buf.write(u"\2\2\u00c7\u00c8\7?\2\2\u00c8\u00c9\7?\2\2\u00c9@\3\2")
buf.write(u"\2\2\u00ca\u00cb\7#\2\2\u00cb\u00cc\7?\2\2\u00ccB\3\2")
buf.write(u"\2\2\u00cd\u00ce\7B\2\2\u00ce\u00cf\5S*\2\u00cfD\3\2")
buf.write(u"\2\2\u00d0\u00d1\7\'\2\2\u00d1\u00d2\5S*\2\u00d2F\3\2")
buf.write(u"\2\2\u00d3\u00d4\7\'\2\2\u00d4\u00d5\5O(\2\u00d5H\3\2")
buf.write(u"\2\2\u00d6\u00d7\7o\2\2\u00d7\u00d8\7w\2\2\u00d8\u00d9")
buf.write(u"\7v\2\2\u00d9J\3\2\2\2\u00da\u00db\7V\2\2\u00db\u00dc")
buf.write(u"\7t\2\2\u00dc\u00dd\7w\2\2\u00dd\u00e4\7g\2\2\u00de\u00df")
buf.write(u"\7H\2\2\u00df\u00e0\7c\2\2\u00e0\u00e1\7n\2\2\u00e1\u00e2")
buf.write(u"\7u\2\2\u00e2\u00e4\7g\2\2\u00e3\u00da\3\2\2\2\u00e3")
buf.write(u"\u00de\3\2\2\2\u00e4L\3\2\2\2\u00e5\u00e6\5O(\2\u00e6")
buf.write(u"\u00e7\7\60\2\2\u00e7\u00e9\5O(\2\u00e8\u00ea\5Q)\2\u00e9")
buf.write(u"\u00e8\3\2\2\2\u00e9\u00ea\3\2\2\2\u00ea\u00ef\3\2\2")
buf.write(u"\2\u00eb\u00ec\5O(\2\u00ec\u00ed\5Q)\2\u00ed\u00ef\3")
buf.write(u"\2\2\2\u00ee\u00e5\3\2\2\2\u00ee\u00eb\3\2\2\2\u00ef")
buf.write(u"N\3\2\2\2\u00f0\u00f2\5W,\2\u00f1\u00f0\3\2\2\2\u00f2")
buf.write(u"\u00f3\3\2\2\2\u00f3\u00f1\3\2\2\2\u00f3\u00f4\3\2\2")
buf.write(u"\2\u00f4P\3\2\2\2\u00f5\u00f7\t\3\2\2\u00f6\u00f8\t\4")
buf.write(u"\2\2\u00f7\u00f6\3\2\2\2\u00f7\u00f8\3\2\2\2\u00f8\u00f9")
buf.write(u"\3\2\2\2\u00f9\u00fa\5O(\2\u00faR\3\2\2\2\u00fb\u00fe")
buf.write(u"\7a\2\2\u00fc\u00fe\5U+\2\u00fd\u00fb\3\2\2\2\u00fd\u00fc")
buf.write(u"\3\2\2\2\u00fe\u0104\3\2\2\2\u00ff\u0103\7a\2\2\u0100")
buf.write(u"\u0103\5U+\2\u0101\u0103\5W,\2\u0102\u00ff\3\2\2\2\u0102")
buf.write(u"\u0100\3\2\2\2\u0102\u0101\3\2\2\2\u0103\u0106\3\2\2")
buf.write(u"\2\u0104\u0102\3\2\2\2\u0104\u0105\3\2\2\2\u0105T\3\2")
buf.write(u"\2\2\u0106\u0104\3\2\2\2\u0107\u0108\t\5\2\2\u0108V\3")
buf.write(u"\2\2\2\u0109\u010a\t\6\2\2\u010aX\3\2\2\2\16\2\u0096")
buf.write(u"\u00a0\u00ad\u00e3\u00e9\u00ee\u00f3\u00f7\u00fd\u0102")
buf.write(u"\u0104\3\b\2\2")
return buf.getvalue()
class RelayLexer(Lexer):
atn = ATNDeserializer().deserialize(serializedATN())
decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ]
T__0 = 1
T__1 = 2
T__2 = 3
T__3 = 4
T__4 = 5
T__5 = 6
T__6 = 7
T__7 = 8
T__8 = 9
T__9 = 10
T__10 = 11
T__11 = 12
T__12 = 13
T__13 = 14
T__14 = 15
T__15 = 16
T__16 = 17
T__17 = 18
SEMVER = 19
WS = 20
LINE_COMMENT = 21
COMMENT = 22
MUL = 23
DIV = 24
ADD = 25
SUB = 26
LT = 27
GT = 28
LE = 29
GE = 30
EQ = 31
NE = 32
GLOBAL_VAR = 33
LOCAL_VAR = 34
GRAPH_VAR = 35
MUT = 36
BOOL_LIT = 37
FLOAT = 38
NAT = 39
CNAME = 40
channelNames = [ u"DEFAULT_TOKEN_CHANNEL", u"HIDDEN" ]
modeNames = [ u"DEFAULT_MODE" ]
literalNames = [ u"<INVALID>",
u"'('", u"')'", u"','", u"'['", u"']'", u"'if'", u"'else'",
u"'let'", u"'='", u"';'", u"'{'", u"'}'", u"'fn'", u"'->'",
u"'def'", u"':'", u"'Tensor'", u"'_'", u"'v0.0.2'", u"'*'",
u"'/'", u"'+'", u"'-'", u"'<'", u"'>'", u"'<='", u"'>='", u"'=='",
u"'!='", u"'mut'" ]
symbolicNames = [ u"<INVALID>",
u"SEMVER", u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL", u"DIV",
u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ", u"NE", u"GLOBAL_VAR",
u"LOCAL_VAR", u"GRAPH_VAR", u"MUT", u"BOOL_LIT", u"FLOAT", u"NAT",
u"CNAME" ]
ruleNames = [ u"T__0", u"T__1", u"T__2", u"T__3", u"T__4", u"T__5",
u"T__6", u"T__7", u"T__8", u"T__9", u"T__10", u"T__11",
u"T__12", u"T__13", u"T__14", u"T__15", u"T__16", u"T__17",
u"SEMVER", u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL",
u"DIV", u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ",
u"NE", u"GLOBAL_VAR", u"LOCAL_VAR", u"GRAPH_VAR", u"MUT",
u"BOOL_LIT", u"FLOAT", u"NAT", u"EXP", u"CNAME", u"LETTER",
u"DIGIT" ]
grammarFileName = u"Relay.g4"
def __init__(self, input=None, output=sys.stdout):
super(RelayLexer, self).__init__(input, output=output)
self.checkVersion("4.7.2")
self._interp = LexerATNSimulator(self, self.atn, self.decisionsToDFA, PredictionContextCache())
self._actions = None
self._predicates = None
python/tvm/relay/grammar/py2/RelayLexer.tokens deleted 100644 → 0
T__0=1
T__1=2
T__2=3
T__3=4
T__4=5
T__5=6
T__6=7
T__7=8
T__8=9
T__9=10
T__10=11
T__11=12
T__12=13
T__13=14
T__14=15
T__15=16
T__16=17
T__17=18
SEMVER=19
WS=20
LINE_COMMENT=21
COMMENT=22
MUL=23
DIV=24
ADD=25
SUB=26
LT=27
GT=28
LE=29
GE=30
EQ=31
NE=32
GLOBAL_VAR=33
LOCAL_VAR=34
GRAPH_VAR=35
MUT=36
BOOL_LIT=37
FLOAT=38
NAT=39
CNAME=40
'('=1
')'=2
','=3
'['=4
']'=5
'if'=6
'else'=7
'let'=8
'='=9
';'=10
'{'=11
'}'=12
'fn'=13
'->'=14
'def'=15
':'=16
'Tensor'=17
'_'=18
'v0.0.2'=19
'*'=23
'/'=24
'+'=25
'-'=26
'<'=27
'>'=28
'<='=29
'>='=30
'=='=31
'!='=32
'mut'=36
python/tvm/relay/grammar/py2/RelayParser.py deleted 100644 → 0
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
# encoding: utf-8
from __future__ import print_function
from antlr4 import *
from io import StringIO
import sys
def serializedATN():
with StringIO() as buf:
buf.write(u"\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\3")
buf.write(u"*\u014c\4\2\t\2\4\3\t\3\4\4\t\4\4\5\t\5\4\6\t\6\4\7\t")
buf.write(u"\7\4\b\t\b\4\t\t\t\4\n\t\n\4\13\t\13\4\f\t\f\4\r\t\r")
buf.write(u"\4\16\t\16\4\17\t\17\4\20\t\20\4\21\t\21\4\22\t\22\4")
buf.write(u"\23\t\23\3\2\3\2\3\3\3\3\7\3+\n\3\f\3\16\3.\13\3\3\3")
buf.write(u"\5\3\61\n\3\3\3\3\3\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3")
buf.write(u"\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\6\4H\n\4\r")
buf.write(u"\4\16\4I\3\4\3\4\3\4\3\4\3\4\3\4\7\4R\n\4\f\4\16\4U\13")
buf.write(u"\4\5\4W\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3")
buf.write(u"\4\5\4d\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\5\4n\n\4")
buf.write(u"\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4")
buf.write(u"\3\4\3\4\3\4\5\4\u0080\n\4\3\4\3\4\3\4\3\4\3\4\3\4\3")
buf.write(u"\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3\4\3")
buf.write(u"\4\7\4\u0096\n\4\f\4\16\4\u0099\13\4\5\4\u009b\n\4\3")
buf.write(u"\4\7\4\u009e\n\4\f\4\16\4\u00a1\13\4\3\5\3\5\5\5\u00a5")
buf.write(u"\n\5\3\5\3\5\3\5\3\5\3\5\5\5\u00ac\n\5\3\5\3\5\3\6\3")
buf.write(u"\6\3\6\5\6\u00b3\n\6\3\6\3\6\3\6\3\6\3\6\5\6\u00ba\n")
buf.write(u"\6\3\6\3\6\3\7\3\7\3\7\3\7\3\7\3\7\5\7\u00c4\n\7\3\b")
buf.write(u"\3\b\3\b\7\b\u00c9\n\b\f\b\16\b\u00cc\13\b\5\b\u00ce")
buf.write(u"\n\b\3\t\3\t\3\t\5\t\u00d3\n\t\3\n\3\n\3\n\7\n\u00d8")
buf.write(u"\n\n\f\n\16\n\u00db\13\n\5\n\u00dd\n\n\3\13\3\13\3\13")
buf.write(u"\3\13\3\f\3\f\3\f\3\f\3\f\3\f\7\f\u00e9\n\f\f\f\16\f")
buf.write(u"\u00ec\13\f\3\f\3\f\5\f\u00f0\n\f\3\r\3\r\3\r\3\r\3\r")
buf.write(u"\3\r\3\r\3\r\3\r\3\r\3\r\6\r\u00fd\n\r\r\r\16\r\u00fe")
buf.write(u"\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\3\r\5\r")
buf.write(u"\u010d\n\r\3\r\3\r\3\r\3\r\7\r\u0113\n\r\f\r\16\r\u0116")
buf.write(u"\13\r\5\r\u0118\n\r\3\r\3\r\3\r\3\r\3\r\5\r\u011f\n\r")
buf.write(u"\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3\16\3")
buf.write(u"\16\6\16\u012c\n\16\r\16\16\16\u012d\3\16\3\16\5\16\u0132")
buf.write(u"\n\16\3\17\3\17\3\17\3\17\3\17\5\17\u0139\n\17\3\20\3")
buf.write(u"\20\3\21\3\21\3\21\3\21\3\22\3\22\3\22\5\22\u0144\n\22")
buf.write(u"\3\23\3\23\3\23\3\23\5\23\u014a\n\23\3\23\2\3\6\24\2")
buf.write(u"\4\6\b\n\f\16\20\22\24\26\30\32\34\36 \"$\2\6\3\2\31")
buf.write(u"\32\3\2\33\34\3\2\35 \3\2!\"\2\u0175\2&\3\2\2\2\4(\3")
buf.write(u"\2\2\2\6\177\3\2\2\2\b\u00a2\3\2\2\2\n\u00af\3\2\2\2")
buf.write(u"\f\u00c3\3\2\2\2\16\u00cd\3\2\2\2\20\u00cf\3\2\2\2\22")
buf.write(u"\u00dc\3\2\2\2\24\u00de\3\2\2\2\26\u00ef\3\2\2\2\30\u011e")
buf.write(u"\3\2\2\2\32\u0131\3\2\2\2\34\u0138\3\2\2\2\36\u013a\3")
buf.write(u"\2\2\2 \u013c\3\2\2\2\"\u0143\3\2\2\2$\u0149\3\2\2\2")
buf.write(u"&\'\7*\2\2\'\3\3\2\2\2(\60\7\25\2\2)+\5\n\6\2*)\3\2\2")
buf.write(u"\2+.\3\2\2\2,*\3\2\2\2,-\3\2\2\2-\61\3\2\2\2.,\3\2\2")
buf.write(u"\2/\61\5\6\4\2\60,\3\2\2\2\60/\3\2\2\2\61\62\3\2\2\2")
buf.write(u"\62\63\7\2\2\3\63\5\3\2\2\2\64\65\b\4\1\2\65\66\7\3\2")
buf.write(u"\2\66\67\5\6\4\2\678\7\4\2\28\u0080\3\2\2\29:\7\34\2")
buf.write(u"\2:\u0080\5\6\4\23;\u0080\5\b\5\2<=\7\3\2\2=\u0080\7")
buf.write(u"\4\2\2>?\7\3\2\2?@\5\6\4\2@A\7\5\2\2AB\7\4\2\2B\u0080")
buf.write(u"\3\2\2\2CD\7\3\2\2DG\5\6\4\2EF\7\5\2\2FH\5\6\4\2GE\3")
buf.write(u"\2\2\2HI\3\2\2\2IG\3\2\2\2IJ\3\2\2\2JK\3\2\2\2KL\7\4")
buf.write(u"\2\2L\u0080\3\2\2\2MV\7\6\2\2NS\5\6\4\2OP\7\5\2\2PR\5")
buf.write(u"\6\4\2QO\3\2\2\2RU\3\2\2\2SQ\3\2\2\2ST\3\2\2\2TW\3\2")
buf.write(u"\2\2US\3\2\2\2VN\3\2\2\2VW\3\2\2\2WX\3\2\2\2X\u0080\7")
buf.write(u"\7\2\2YZ\7\b\2\2Z[\7\3\2\2[\\\5\6\4\2\\]\7\4\2\2]^\5")
buf.write(u" \21\2^_\7\t\2\2_`\5 \21\2`\u0080\3\2\2\2ac\7\n\2\2b")
buf.write(u"d\7&\2\2cb\3\2\2\2cd\3\2\2\2de\3\2\2\2ef\5\20\t\2fg\7")
buf.write(u"\13\2\2gh\5\6\4\2hi\7\f\2\2ij\5\6\4\bj\u0080\3\2\2\2")
buf.write(u"km\7\n\2\2ln\7&\2\2ml\3\2\2\2mn\3\2\2\2no\3\2\2\2op\5")
buf.write(u"\20\t\2pq\7\13\2\2qr\7\r\2\2rs\5\6\4\2st\7\16\2\2tu\7")
buf.write(u"\f\2\2uv\5\6\4\7v\u0080\3\2\2\2wx\5$\23\2xy\7\13\2\2")
buf.write(u"yz\5\6\4\2z{\7\f\2\2{|\5\6\4\5|\u0080\3\2\2\2}\u0080")
buf.write(u"\5$\23\2~\u0080\5\"\22\2\177\64\3\2\2\2\1779\3\2\2\2")
buf.write(u"\177;\3\2\2\2\177<\3\2\2\2\177>\3\2\2\2\177C\3\2\2\2")
buf.write(u"\177M\3\2\2\2\177Y\3\2\2\2\177a\3\2\2\2\177k\3\2\2\2")
buf.write(u"\177w\3\2\2\2\177}\3\2\2\2\177~\3\2\2\2\u0080\u009f\3")
buf.write(u"\2\2\2\u0081\u0082\f\22\2\2\u0082\u0083\t\2\2\2\u0083")
buf.write(u"\u009e\5\6\4\23\u0084\u0085\f\21\2\2\u0085\u0086\t\3")
buf.write(u"\2\2\u0086\u009e\5\6\4\22\u0087\u0088\f\20\2\2\u0088")
buf.write(u"\u0089\t\4\2\2\u0089\u009e\5\6\4\21\u008a\u008b\f\17")
buf.write(u"\2\2\u008b\u008c\t\5\2\2\u008c\u009e\5\6\4\20\u008d\u008e")
buf.write(u"\f\6\2\2\u008e\u008f\7\f\2\2\u008f\u009e\5\6\4\7\u0090")
buf.write(u"\u0091\f\24\2\2\u0091\u009a\7\3\2\2\u0092\u0097\5\6\4")
buf.write(u"\2\u0093\u0094\7\5\2\2\u0094\u0096\5\6\4\2\u0095\u0093")
buf.write(u"\3\2\2\2\u0096\u0099\3\2\2\2\u0097\u0095\3\2\2\2\u0097")
buf.write(u"\u0098\3\2\2\2\u0098\u009b\3\2\2\2\u0099\u0097\3\2\2")
buf.write(u"\2\u009a\u0092\3\2\2\2\u009a\u009b\3\2\2\2\u009b\u009c")
buf.write(u"\3\2\2\2\u009c\u009e\7\4\2\2\u009d\u0081\3\2\2\2\u009d")
buf.write(u"\u0084\3\2\2\2\u009d\u0087\3\2\2\2\u009d\u008a\3\2\2")
buf.write(u"\2\u009d\u008d\3\2\2\2\u009d\u0090\3\2\2\2\u009e\u00a1")
buf.write(u"\3\2\2\2\u009f\u009d\3\2\2\2\u009f\u00a0\3\2\2\2\u00a0")
buf.write(u"\7\3\2\2\2\u00a1\u009f\3\2\2\2\u00a2\u00a4\7\17\2\2\u00a3")
buf.write(u"\u00a5\5\26\f\2\u00a4\u00a3\3\2\2\2\u00a4\u00a5\3\2\2")
buf.write(u"\2\u00a5\u00a6\3\2\2\2\u00a6\u00a7\7\3\2\2\u00a7\u00a8")
buf.write(u"\5\f\7\2\u00a8\u00ab\7\4\2\2\u00a9\u00aa\7\20\2\2\u00aa")
buf.write(u"\u00ac\5\30\r\2\u00ab\u00a9\3\2\2\2\u00ab\u00ac\3\2\2")
buf.write(u"\2\u00ac\u00ad\3\2\2\2\u00ad\u00ae\5 \21\2\u00ae\t\3")
buf.write(u"\2\2\2\u00af\u00b0\7\21\2\2\u00b0\u00b2\5$\23\2\u00b1")
buf.write(u"\u00b3\5\26\f\2\u00b2\u00b1\3\2\2\2\u00b2\u00b3\3\2\2")
buf.write(u"\2\u00b3\u00b4\3\2\2\2\u00b4\u00b5\7\3\2\2\u00b5\u00b6")
buf.write(u"\5\f\7\2\u00b6\u00b9\7\4\2\2\u00b7\u00b8\7\20\2\2\u00b8")
buf.write(u"\u00ba\5\30\r\2\u00b9\u00b7\3\2\2\2\u00b9\u00ba\3\2\2")
buf.write(u"\2\u00ba\u00bb\3\2\2\2\u00bb\u00bc\5 \21\2\u00bc\13\3")
buf.write(u"\2\2\2\u00bd\u00c4\5\16\b\2\u00be\u00c4\5\22\n\2\u00bf")
buf.write(u"\u00c0\5\16\b\2\u00c0\u00c1\7\5\2\2\u00c1\u00c2\5\22")
buf.write(u"\n\2\u00c2\u00c4\3\2\2\2\u00c3\u00bd\3\2\2\2\u00c3\u00be")
buf.write(u"\3\2\2\2\u00c3\u00bf\3\2\2\2\u00c4\r\3\2\2\2\u00c5\u00ca")
buf.write(u"\5\20\t\2\u00c6\u00c7\7\5\2\2\u00c7\u00c9\5\20\t\2\u00c8")
buf.write(u"\u00c6\3\2\2\2\u00c9\u00cc\3\2\2\2\u00ca\u00c8\3\2\2")
buf.write(u"\2\u00ca\u00cb\3\2\2\2\u00cb\u00ce\3\2\2\2\u00cc\u00ca")
buf.write(u"\3\2\2\2\u00cd\u00c5\3\2\2\2\u00cd\u00ce\3\2\2\2\u00ce")
buf.write(u"\17\3\2\2\2\u00cf\u00d2\5$\23\2\u00d0\u00d1\7\22\2\2")
buf.write(u"\u00d1\u00d3\5\30\r\2\u00d2\u00d0\3\2\2\2\u00d2\u00d3")
buf.write(u"\3\2\2\2\u00d3\21\3\2\2\2\u00d4\u00d9\5\24\13\2\u00d5")
buf.write(u"\u00d6\7\5\2\2\u00d6\u00d8\5\24\13\2\u00d7\u00d5\3\2")
buf.write(u"\2\2\u00d8\u00db\3\2\2\2\u00d9\u00d7\3\2\2\2\u00d9\u00da")
buf.write(u"\3\2\2\2\u00da\u00dd\3\2\2\2\u00db\u00d9\3\2\2\2\u00dc")
buf.write(u"\u00d4\3\2\2\2\u00dc\u00dd\3\2\2\2\u00dd\23\3\2\2\2\u00de")
buf.write(u"\u00df\7*\2\2\u00df\u00e0\7\13\2\2\u00e0\u00e1\5\6\4")
buf.write(u"\2\u00e1\25\3\2\2\2\u00e2\u00e3\7\6\2\2\u00e3\u00f0\7")
buf.write(u"\7\2\2\u00e4\u00e5\7\6\2\2\u00e5\u00ea\5$\23\2\u00e6")
buf.write(u"\u00e7\7\5\2\2\u00e7\u00e9\5$\23\2\u00e8\u00e6\3\2\2")
buf.write(u"\2\u00e9\u00ec\3\2\2\2\u00ea\u00e8\3\2\2\2\u00ea\u00eb")
buf.write(u"\3\2\2\2\u00eb\u00ed\3\2\2\2\u00ec\u00ea\3\2\2\2\u00ed")
buf.write(u"\u00ee\7\7\2\2\u00ee\u00f0\3\2\2\2\u00ef\u00e2\3\2\2")
buf.write(u"\2\u00ef\u00e4\3\2\2\2\u00f0\27\3\2\2\2\u00f1\u00f2\7")
buf.write(u"\3\2\2\u00f2\u011f\7\4\2\2\u00f3\u00f4\7\3\2\2\u00f4")
buf.write(u"\u00f5\5\30\r\2\u00f5\u00f6\7\5\2\2\u00f6\u00f7\7\4\2")
buf.write(u"\2\u00f7\u011f\3\2\2\2\u00f8\u00f9\7\3\2\2\u00f9\u00fc")
buf.write(u"\5\30\r\2\u00fa\u00fb\7\5\2\2\u00fb\u00fd\5\30\r\2\u00fc")
buf.write(u"\u00fa\3\2\2\2\u00fd\u00fe\3\2\2\2\u00fe\u00fc\3\2\2")
buf.write(u"\2\u00fe\u00ff\3\2\2\2\u00ff\u0100\3\2\2\2\u0100\u0101")
buf.write(u"\7\4\2\2\u0101\u011f\3\2\2\2\u0102\u011f\5\36\20\2\u0103")
buf.write(u"\u0104\7\23\2\2\u0104\u0105\7\6\2\2\u0105\u0106\5\32")
buf.write(u"\16\2\u0106\u0107\7\5\2\2\u0107\u0108\5\30\r\2\u0108")
buf.write(u"\u0109\7\7\2\2\u0109\u011f\3\2\2\2\u010a\u010c\7\17\2")
buf.write(u"\2\u010b\u010d\5\26\f\2\u010c\u010b\3\2\2\2\u010c\u010d")
buf.write(u"\3\2\2\2\u010d\u010e\3\2\2\2\u010e\u0117\7\3\2\2\u010f")
buf.write(u"\u0114\5\30\r\2\u0110\u0111\7\5\2\2\u0111\u0113\5\30")
buf.write(u"\r\2\u0112\u0110\3\2\2\2\u0113\u0116\3\2\2\2\u0114\u0112")
buf.write(u"\3\2\2\2\u0114\u0115\3\2\2\2\u0115\u0118\3\2\2\2\u0116")
buf.write(u"\u0114\3\2\2\2\u0117\u010f\3\2\2\2\u0117\u0118\3\2\2")
buf.write(u"\2\u0118\u0119\3\2\2\2\u0119\u011a\7\4\2\2\u011a\u011b")
buf.write(u"\7\20\2\2\u011b\u011f\5\30\r\2\u011c\u011f\7\24\2\2\u011d")
buf.write(u"\u011f\7)\2\2\u011e\u00f1\3\2\2\2\u011e\u00f3\3\2\2\2")
buf.write(u"\u011e\u00f8\3\2\2\2\u011e\u0102\3\2\2\2\u011e\u0103")
buf.write(u"\3\2\2\2\u011e\u010a\3\2\2\2\u011e\u011c\3\2\2\2\u011e")
buf.write(u"\u011d\3\2\2\2\u011f\31\3\2\2\2\u0120\u0121\7\3\2\2\u0121")
buf.write(u"\u0132\7\4\2\2\u0122\u0123\7\3\2\2\u0123\u0124\5\34\17")
buf.write(u"\2\u0124\u0125\7\5\2\2\u0125\u0126\7\4\2\2\u0126\u0132")
buf.write(u"\3\2\2\2\u0127\u0128\7\3\2\2\u0128\u012b\5\34\17\2\u0129")
buf.write(u"\u012a\7\5\2\2\u012a\u012c\5\34\17\2\u012b\u0129\3\2")
buf.write(u"\2\2\u012c\u012d\3\2\2\2\u012d\u012b\3\2\2\2\u012d\u012e")
buf.write(u"\3\2\2\2\u012e\u012f\3\2\2\2\u012f\u0130\7\4\2\2\u0130")
buf.write(u"\u0132\3\2\2\2\u0131\u0120\3\2\2\2\u0131\u0122\3\2\2")
buf.write(u"\2\u0131\u0127\3\2\2\2\u0132\33\3\2\2\2\u0133\u0134\7")
buf.write(u"\3\2\2\u0134\u0135\5\34\17\2\u0135\u0136\7\4\2\2\u0136")
buf.write(u"\u0139\3\2\2\2\u0137\u0139\7)\2\2\u0138\u0133\3\2\2\2")
buf.write(u"\u0138\u0137\3\2\2\2\u0139\35\3\2\2\2\u013a\u013b\7*")
buf.write(u"\2\2\u013b\37\3\2\2\2\u013c\u013d\7\r\2\2\u013d\u013e")
buf.write(u"\5\6\4\2\u013e\u013f\7\16\2\2\u013f!\3\2\2\2\u0140\u0144")
buf.write(u"\7(\2\2\u0141\u0144\7)\2\2\u0142\u0144\7\'\2\2\u0143")
buf.write(u"\u0140\3\2\2\2\u0143\u0141\3\2\2\2\u0143\u0142\3\2\2")
buf.write(u"\2\u0144#\3\2\2\2\u0145\u014a\5\2\2\2\u0146\u014a\7#")
buf.write(u"\2\2\u0147\u014a\7$\2\2\u0148\u014a\7%\2\2\u0149\u0145")
buf.write(u"\3\2\2\2\u0149\u0146\3\2\2\2\u0149\u0147\3\2\2\2\u0149")
buf.write(u"\u0148\3\2\2\2\u014a%\3\2\2\2$,\60ISVcm\177\u0097\u009a")
buf.write(u"\u009d\u009f\u00a4\u00ab\u00b2\u00b9\u00c3\u00ca\u00cd")
buf.write(u"\u00d2\u00d9\u00dc\u00ea\u00ef\u00fe\u010c\u0114\u0117")
buf.write(u"\u011e\u012d\u0131\u0138\u0143\u0149")
return buf.getvalue()
class RelayParser ( Parser ):
grammarFileName = "Relay.g4"
atn = ATNDeserializer().deserialize(serializedATN())
decisionsToDFA = [ DFA(ds, i) for i, ds in enumerate(atn.decisionToState) ]
sharedContextCache = PredictionContextCache()
literalNames = [ u"<INVALID>", u"'('", u"')'", u"','", u"'['", u"']'",
u"'if'", u"'else'", u"'let'", u"'='", u"';'", u"'{'",
u"'}'", u"'fn'", u"'->'", u"'def'", u"':'", u"'Tensor'",
u"'_'", u"'v0.0.2'", u"<INVALID>", u"<INVALID>", u"<INVALID>",
u"'*'", u"'/'", u"'+'", u"'-'", u"'<'", u"'>'", u"'<='",
u"'>='", u"'=='", u"'!='", u"<INVALID>", u"<INVALID>",
u"<INVALID>", u"'mut'" ]
symbolicNames = [ u"<INVALID>", u"<INVALID>", u"<INVALID>", u"<INVALID>",
u"<INVALID>", u"<INVALID>", u"<INVALID>", u"<INVALID>",
u"<INVALID>", u"<INVALID>", u"<INVALID>", u"<INVALID>",
u"<INVALID>", u"<INVALID>", u"<INVALID>", u"<INVALID>",
u"<INVALID>", u"<INVALID>", u"<INVALID>", u"SEMVER",
u"WS", u"LINE_COMMENT", u"COMMENT", u"MUL", u"DIV",
u"ADD", u"SUB", u"LT", u"GT", u"LE", u"GE", u"EQ",
u"NE", u"GLOBAL_VAR", u"LOCAL_VAR", u"GRAPH_VAR",
u"MUT", u"BOOL_LIT", u"FLOAT", u"NAT", u"CNAME" ]
RULE_opIdent = 0
RULE_prog = 1
RULE_expr = 2
RULE_func = 3
RULE_defn = 4
RULE_argList = 5
RULE_varList = 6
RULE_var = 7
RULE_attrList = 8
RULE_attr = 9
RULE_typeParamSeq = 10
RULE_type_ = 11
RULE_shapeSeq = 12
RULE_shape = 13
RULE_typeIdent = 14
RULE_body = 15
RULE_scalar = 16
RULE_ident = 17
ruleNames = [ u"opIdent", u"prog", u"expr", u"func", u"defn", u"argList",
u"varList", u"var", u"attrList", u"attr", u"typeParamSeq",
u"type_", u"shapeSeq", u"shape", u"typeIdent", u"body",
u"scalar", u"ident" ]
EOF = Token.EOF
T__0=1
T__1=2
T__2=3
T__3=4
T__4=5
T__5=6
T__6=7
T__7=8
T__8=9
T__9=10
T__10=11
T__11=12
T__12=13
T__13=14
T__14=15
T__15=16
T__16=17
T__17=18
SEMVER=19
WS=20
LINE_COMMENT=21
COMMENT=22
MUL=23
DIV=24
ADD=25
SUB=26
LT=27
GT=28
LE=29
GE=30
EQ=31
NE=32
GLOBAL_VAR=33
LOCAL_VAR=34
GRAPH_VAR=35
MUT=36
BOOL_LIT=37
FLOAT=38
NAT=39
CNAME=40
def __init__(self, input, output=sys.stdout):
super(RelayParser, self).__init__(input, output=output)
self.checkVersion("4.7.2")
self._interp = ParserATNSimulator(self, self.atn, self.decisionsToDFA, self.sharedContextCache)
self._predicates = None
class OpIdentContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.OpIdentContext, self).__init__(parent, invokingState)
self.parser = parser
def CNAME(self):
return self.getToken(RelayParser.CNAME, 0)
def getRuleIndex(self):
return RelayParser.RULE_opIdent
def accept(self, visitor):
if hasattr(visitor, "visitOpIdent"):
return visitor.visitOpIdent(self)
else:
return visitor.visitChildren(self)
def opIdent(self):
localctx = RelayParser.OpIdentContext(self, self._ctx, self.state)
self.enterRule(localctx, 0, self.RULE_opIdent)
try:
self.enterOuterAlt(localctx, 1)
self.state = 36
self.match(RelayParser.CNAME)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ProgContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ProgContext, self).__init__(parent, invokingState)
self.parser = parser
def SEMVER(self):
return self.getToken(RelayParser.SEMVER, 0)
def EOF(self):
return self.getToken(RelayParser.EOF, 0)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def defn(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.DefnContext)
else:
return self.getTypedRuleContext(RelayParser.DefnContext,i)
def getRuleIndex(self):
return RelayParser.RULE_prog
def accept(self, visitor):
if hasattr(visitor, "visitProg"):
return visitor.visitProg(self)
else:
return visitor.visitChildren(self)
def prog(self):
localctx = RelayParser.ProgContext(self, self._ctx, self.state)
self.enterRule(localctx, 2, self.RULE_prog)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 38
self.match(RelayParser.SEMVER)
self.state = 46
self._errHandler.sync(self)
token = self._input.LA(1)
if token in [RelayParser.EOF, RelayParser.T__14]:
self.state = 42
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__14:
self.state = 39
self.defn()
self.state = 44
self._errHandler.sync(self)
_la = self._input.LA(1)
pass
elif token in [RelayParser.T__0, RelayParser.T__3, RelayParser.T__5, RelayParser.T__7, RelayParser.T__12, RelayParser.SUB, RelayParser.GLOBAL_VAR, RelayParser.LOCAL_VAR, RelayParser.GRAPH_VAR, RelayParser.BOOL_LIT, RelayParser.FLOAT, RelayParser.NAT, RelayParser.CNAME]:
self.state = 45
self.expr(0)
pass
else:
raise NoViableAltException(self)
self.state = 48
self.match(RelayParser.EOF)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ExprContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ExprContext, self).__init__(parent, invokingState)
self.parser = parser
def getRuleIndex(self):
return RelayParser.RULE_expr
def copyFrom(self, ctx):
super(RelayParser.ExprContext, self).copyFrom(ctx)
class IdentExprContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.IdentExprContext, self).__init__(parser)
self.copyFrom(ctx)
def ident(self):
return self.getTypedRuleContext(RelayParser.IdentContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitIdentExpr"):
return visitor.visitIdentExpr(self)
else:
return visitor.visitChildren(self)
class CallContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.CallContext, self).__init__(parser)
self.copyFrom(ctx)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def accept(self, visitor):
if hasattr(visitor, "visitCall"):
return visitor.visitCall(self)
else:
return visitor.visitChildren(self)
class NegContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.NegContext, self).__init__(parser)
self.copyFrom(ctx)
def SUB(self):
return self.getToken(RelayParser.SUB, 0)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitNeg"):
return visitor.visitNeg(self)
else:
return visitor.visitChildren(self)
class TupleContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.TupleContext, self).__init__(parser)
self.copyFrom(ctx)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def accept(self, visitor):
if hasattr(visitor, "visitTuple"):
return visitor.visitTuple(self)
else:
return visitor.visitChildren(self)
class ParensContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.ParensContext, self).__init__(parser)
self.copyFrom(ctx)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitParens"):
return visitor.visitParens(self)
else:
return visitor.visitChildren(self)
class FuncExprContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.FuncExprContext, self).__init__(parser)
self.copyFrom(ctx)
def func(self):
return self.getTypedRuleContext(RelayParser.FuncContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitFuncExpr"):
return visitor.visitFuncExpr(self)
else:
return visitor.visitChildren(self)
class ScalarExprContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.ScalarExprContext, self).__init__(parser)
self.copyFrom(ctx)
def scalar(self):
return self.getTypedRuleContext(RelayParser.ScalarContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitScalarExpr"):
return visitor.visitScalarExpr(self)
else:
return visitor.visitChildren(self)
class LetContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.LetContext, self).__init__(parser)
self.copyFrom(ctx)
def var(self):
return self.getTypedRuleContext(RelayParser.VarContext,0)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def MUT(self):
return self.getToken(RelayParser.MUT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitLet"):
return visitor.visitLet(self)
else:
return visitor.visitChildren(self)
class TensorContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.TensorContext, self).__init__(parser)
self.copyFrom(ctx)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def accept(self, visitor):
if hasattr(visitor, "visitTensor"):
return visitor.visitTensor(self)
else:
return visitor.visitChildren(self)
class IfElseContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.IfElseContext, self).__init__(parser)
self.copyFrom(ctx)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def body(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.BodyContext)
else:
return self.getTypedRuleContext(RelayParser.BodyContext,i)
def accept(self, visitor):
if hasattr(visitor, "visitIfElse"):
return visitor.visitIfElse(self)
else:
return visitor.visitChildren(self)
class GraphContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.GraphContext, self).__init__(parser)
self.copyFrom(ctx)
def ident(self):
return self.getTypedRuleContext(RelayParser.IdentContext,0)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def accept(self, visitor):
if hasattr(visitor, "visitGraph"):
return visitor.visitGraph(self)
else:
return visitor.visitChildren(self)
class BinOpContext(ExprContext):
def __init__(self, parser, ctx): # actually a RelayParser.ExprContext)
super(RelayParser.BinOpContext, self).__init__(parser)
self.op = None # Token
self.copyFrom(ctx)
def expr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ExprContext)
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def MUL(self):
return self.getToken(RelayParser.MUL, 0)
def DIV(self):
return self.getToken(RelayParser.DIV, 0)
def ADD(self):
return self.getToken(RelayParser.ADD, 0)
def SUB(self):
return self.getToken(RelayParser.SUB, 0)
def LT(self):
return self.getToken(RelayParser.LT, 0)
def GT(self):
return self.getToken(RelayParser.GT, 0)
def LE(self):
return self.getToken(RelayParser.LE, 0)
def GE(self):
return self.getToken(RelayParser.GE, 0)
def EQ(self):
return self.getToken(RelayParser.EQ, 0)
def NE(self):
return self.getToken(RelayParser.NE, 0)
def accept(self, visitor):
if hasattr(visitor, "visitBinOp"):
return visitor.visitBinOp(self)
else:
return visitor.visitChildren(self)
def expr(self, _p=0):
_parentctx = self._ctx
_parentState = self.state
localctx = RelayParser.ExprContext(self, self._ctx, _parentState)
_prevctx = localctx
_startState = 4
self.enterRecursionRule(localctx, 4, self.RULE_expr, _p)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 125
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,7,self._ctx)
if la_ == 1:
localctx = RelayParser.ParensContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 51
self.match(RelayParser.T__0)
self.state = 52
self.expr(0)
self.state = 53
self.match(RelayParser.T__1)
pass
elif la_ == 2:
localctx = RelayParser.NegContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 55
self.match(RelayParser.SUB)
self.state = 56
self.expr(17)
pass
elif la_ == 3:
localctx = RelayParser.FuncExprContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 57
self.func()
pass
elif la_ == 4:
localctx = RelayParser.TupleContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 58
self.match(RelayParser.T__0)
self.state = 59
self.match(RelayParser.T__1)
pass
elif la_ == 5:
localctx = RelayParser.TupleContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 60
self.match(RelayParser.T__0)
self.state = 61
self.expr(0)
self.state = 62
self.match(RelayParser.T__2)
self.state = 63
self.match(RelayParser.T__1)
pass
elif la_ == 6:
localctx = RelayParser.TupleContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 65
self.match(RelayParser.T__0)
self.state = 66
self.expr(0)
self.state = 69
self._errHandler.sync(self)
_la = self._input.LA(1)
while True:
self.state = 67
self.match(RelayParser.T__2)
self.state = 68
self.expr(0)
self.state = 71
self._errHandler.sync(self)
_la = self._input.LA(1)
if not (_la==RelayParser.T__2):
break
self.state = 73
self.match(RelayParser.T__1)
pass
elif la_ == 7:
localctx = RelayParser.TensorContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 75
self.match(RelayParser.T__3)
self.state = 84
self._errHandler.sync(self)
_la = self._input.LA(1)
if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__3) | (1 << RelayParser.T__5) | (1 << RelayParser.T__7) | (1 << RelayParser.T__12) | (1 << RelayParser.SUB) | (1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.BOOL_LIT) | (1 << RelayParser.FLOAT) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0):
self.state = 76
self.expr(0)
self.state = 81
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__2:
self.state = 77
self.match(RelayParser.T__2)
self.state = 78
self.expr(0)
self.state = 83
self._errHandler.sync(self)
_la = self._input.LA(1)
self.state = 86
self.match(RelayParser.T__4)
pass
elif la_ == 8:
localctx = RelayParser.IfElseContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 87
self.match(RelayParser.T__5)
self.state = 88
self.match(RelayParser.T__0)
self.state = 89
self.expr(0)
self.state = 90
self.match(RelayParser.T__1)
self.state = 91
self.body()
self.state = 92
self.match(RelayParser.T__6)
self.state = 93
self.body()
pass
elif la_ == 9:
localctx = RelayParser.LetContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 95
self.match(RelayParser.T__7)
self.state = 97
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.MUT:
self.state = 96
self.match(RelayParser.MUT)
self.state = 99
self.var()
self.state = 100
self.match(RelayParser.T__8)
self.state = 101
self.expr(0)
self.state = 102
self.match(RelayParser.T__9)
self.state = 103
self.expr(6)
pass
elif la_ == 10:
localctx = RelayParser.LetContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 105
self.match(RelayParser.T__7)
self.state = 107
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.MUT:
self.state = 106
self.match(RelayParser.MUT)
self.state = 109
self.var()
self.state = 110
self.match(RelayParser.T__8)
self.state = 111
self.match(RelayParser.T__10)
self.state = 112
self.expr(0)
self.state = 113
self.match(RelayParser.T__11)
self.state = 114
self.match(RelayParser.T__9)
self.state = 115
self.expr(5)
pass
elif la_ == 11:
localctx = RelayParser.GraphContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 117
self.ident()
self.state = 118
self.match(RelayParser.T__8)
self.state = 119
self.expr(0)
self.state = 120
self.match(RelayParser.T__9)
self.state = 121
self.expr(3)
pass
elif la_ == 12:
localctx = RelayParser.IdentExprContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 123
self.ident()
pass
elif la_ == 13:
localctx = RelayParser.ScalarExprContext(self, localctx)
self._ctx = localctx
_prevctx = localctx
self.state = 124
self.scalar()
pass
self._ctx.stop = self._input.LT(-1)
self.state = 157
self._errHandler.sync(self)
_alt = self._interp.adaptivePredict(self._input,11,self._ctx)
while _alt!=2 and _alt!=ATN.INVALID_ALT_NUMBER:
if _alt==1:
if self._parseListeners is not None:
self.triggerExitRuleEvent()
_prevctx = localctx
self.state = 155
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,10,self._ctx)
if la_ == 1:
localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 127
if not self.precpred(self._ctx, 16):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 16)")
self.state = 128
localctx.op = self._input.LT(1)
_la = self._input.LA(1)
if not(_la==RelayParser.MUL or _la==RelayParser.DIV):
localctx.op = self._errHandler.recoverInline(self)
else:
self._errHandler.reportMatch(self)
self.consume()
self.state = 129
self.expr(17)
pass
elif la_ == 2:
localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 130
if not self.precpred(self._ctx, 15):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 15)")
self.state = 131
localctx.op = self._input.LT(1)
_la = self._input.LA(1)
if not(_la==RelayParser.ADD or _la==RelayParser.SUB):
localctx.op = self._errHandler.recoverInline(self)
else:
self._errHandler.reportMatch(self)
self.consume()
self.state = 132
self.expr(16)
pass
elif la_ == 3:
localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 133
if not self.precpred(self._ctx, 14):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 14)")
self.state = 134
localctx.op = self._input.LT(1)
_la = self._input.LA(1)
if not((((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.LT) | (1 << RelayParser.GT) | (1 << RelayParser.LE) | (1 << RelayParser.GE))) != 0)):
localctx.op = self._errHandler.recoverInline(self)
else:
self._errHandler.reportMatch(self)
self.consume()
self.state = 135
self.expr(15)
pass
elif la_ == 4:
localctx = RelayParser.BinOpContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 136
if not self.precpred(self._ctx, 13):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 13)")
self.state = 137
localctx.op = self._input.LT(1)
_la = self._input.LA(1)
if not(_la==RelayParser.EQ or _la==RelayParser.NE):
localctx.op = self._errHandler.recoverInline(self)
else:
self._errHandler.reportMatch(self)
self.consume()
self.state = 138
self.expr(14)
pass
elif la_ == 5:
localctx = RelayParser.LetContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 139
if not self.precpred(self._ctx, 4):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 4)")
self.state = 140
self.match(RelayParser.T__9)
self.state = 141
self.expr(5)
pass
elif la_ == 6:
localctx = RelayParser.CallContext(self, RelayParser.ExprContext(self, _parentctx, _parentState))
self.pushNewRecursionContext(localctx, _startState, self.RULE_expr)
self.state = 142
if not self.precpred(self._ctx, 18):
from antlr4.error.Errors import FailedPredicateException
raise FailedPredicateException(self, "self.precpred(self._ctx, 18)")
self.state = 143
self.match(RelayParser.T__0)
self.state = 152
self._errHandler.sync(self)
_la = self._input.LA(1)
if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__3) | (1 << RelayParser.T__5) | (1 << RelayParser.T__7) | (1 << RelayParser.T__12) | (1 << RelayParser.SUB) | (1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.BOOL_LIT) | (1 << RelayParser.FLOAT) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0):
self.state = 144
self.expr(0)
self.state = 149
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__2:
self.state = 145
self.match(RelayParser.T__2)
self.state = 146
self.expr(0)
self.state = 151
self._errHandler.sync(self)
_la = self._input.LA(1)
self.state = 154
self.match(RelayParser.T__1)
pass
self.state = 159
self._errHandler.sync(self)
_alt = self._interp.adaptivePredict(self._input,11,self._ctx)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.unrollRecursionContexts(_parentctx)
return localctx
class FuncContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.FuncContext, self).__init__(parent, invokingState)
self.parser = parser
def argList(self):
return self.getTypedRuleContext(RelayParser.ArgListContext,0)
def body(self):
return self.getTypedRuleContext(RelayParser.BodyContext,0)
def typeParamSeq(self):
return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0)
def type_(self):
return self.getTypedRuleContext(RelayParser.Type_Context,0)
def getRuleIndex(self):
return RelayParser.RULE_func
def accept(self, visitor):
if hasattr(visitor, "visitFunc"):
return visitor.visitFunc(self)
else:
return visitor.visitChildren(self)
def func(self):
localctx = RelayParser.FuncContext(self, self._ctx, self.state)
self.enterRule(localctx, 6, self.RULE_func)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 160
self.match(RelayParser.T__12)
self.state = 162
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__3:
self.state = 161
self.typeParamSeq()
self.state = 164
self.match(RelayParser.T__0)
self.state = 165
self.argList()
self.state = 166
self.match(RelayParser.T__1)
self.state = 169
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__13:
self.state = 167
self.match(RelayParser.T__13)
self.state = 168
self.type_()
self.state = 171
self.body()
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class DefnContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.DefnContext, self).__init__(parent, invokingState)
self.parser = parser
def ident(self):
return self.getTypedRuleContext(RelayParser.IdentContext,0)
def argList(self):
return self.getTypedRuleContext(RelayParser.ArgListContext,0)
def body(self):
return self.getTypedRuleContext(RelayParser.BodyContext,0)
def typeParamSeq(self):
return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0)
def type_(self):
return self.getTypedRuleContext(RelayParser.Type_Context,0)
def getRuleIndex(self):
return RelayParser.RULE_defn
def accept(self, visitor):
if hasattr(visitor, "visitDefn"):
return visitor.visitDefn(self)
else:
return visitor.visitChildren(self)
def defn(self):
localctx = RelayParser.DefnContext(self, self._ctx, self.state)
self.enterRule(localctx, 8, self.RULE_defn)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 173
self.match(RelayParser.T__14)
self.state = 174
self.ident()
self.state = 176
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__3:
self.state = 175
self.typeParamSeq()
self.state = 178
self.match(RelayParser.T__0)
self.state = 179
self.argList()
self.state = 180
self.match(RelayParser.T__1)
self.state = 183
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__13:
self.state = 181
self.match(RelayParser.T__13)
self.state = 182
self.type_()
self.state = 185
self.body()
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ArgListContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ArgListContext, self).__init__(parent, invokingState)
self.parser = parser
def varList(self):
return self.getTypedRuleContext(RelayParser.VarListContext,0)
def attrList(self):
return self.getTypedRuleContext(RelayParser.AttrListContext,0)
def getRuleIndex(self):
return RelayParser.RULE_argList
def accept(self, visitor):
if hasattr(visitor, "visitArgList"):
return visitor.visitArgList(self)
else:
return visitor.visitChildren(self)
def argList(self):
localctx = RelayParser.ArgListContext(self, self._ctx, self.state)
self.enterRule(localctx, 10, self.RULE_argList)
try:
self.state = 193
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,16,self._ctx)
if la_ == 1:
self.enterOuterAlt(localctx, 1)
self.state = 187
self.varList()
pass
elif la_ == 2:
self.enterOuterAlt(localctx, 2)
self.state = 188
self.attrList()
pass
elif la_ == 3:
self.enterOuterAlt(localctx, 3)
self.state = 189
self.varList()
self.state = 190
self.match(RelayParser.T__2)
self.state = 191
self.attrList()
pass
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class VarListContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.VarListContext, self).__init__(parent, invokingState)
self.parser = parser
def var(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.VarContext)
else:
return self.getTypedRuleContext(RelayParser.VarContext,i)
def getRuleIndex(self):
return RelayParser.RULE_varList
def accept(self, visitor):
if hasattr(visitor, "visitVarList"):
return visitor.visitVarList(self)
else:
return visitor.visitChildren(self)
def varList(self):
localctx = RelayParser.VarListContext(self, self._ctx, self.state)
self.enterRule(localctx, 12, self.RULE_varList)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 203
self._errHandler.sync(self)
_la = self._input.LA(1)
if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.GLOBAL_VAR) | (1 << RelayParser.LOCAL_VAR) | (1 << RelayParser.GRAPH_VAR) | (1 << RelayParser.CNAME))) != 0):
self.state = 195
self.var()
self.state = 200
self._errHandler.sync(self)
_alt = self._interp.adaptivePredict(self._input,17,self._ctx)
while _alt!=2 and _alt!=ATN.INVALID_ALT_NUMBER:
if _alt==1:
self.state = 196
self.match(RelayParser.T__2)
self.state = 197
self.var()
self.state = 202
self._errHandler.sync(self)
_alt = self._interp.adaptivePredict(self._input,17,self._ctx)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class VarContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.VarContext, self).__init__(parent, invokingState)
self.parser = parser
def ident(self):
return self.getTypedRuleContext(RelayParser.IdentContext,0)
def type_(self):
return self.getTypedRuleContext(RelayParser.Type_Context,0)
def getRuleIndex(self):
return RelayParser.RULE_var
def accept(self, visitor):
if hasattr(visitor, "visitVar"):
return visitor.visitVar(self)
else:
return visitor.visitChildren(self)
def var(self):
localctx = RelayParser.VarContext(self, self._ctx, self.state)
self.enterRule(localctx, 14, self.RULE_var)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 205
self.ident()
self.state = 208
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__15:
self.state = 206
self.match(RelayParser.T__15)
self.state = 207
self.type_()
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class AttrListContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.AttrListContext, self).__init__(parent, invokingState)
self.parser = parser
def attr(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.AttrContext)
else:
return self.getTypedRuleContext(RelayParser.AttrContext,i)
def getRuleIndex(self):
return RelayParser.RULE_attrList
def accept(self, visitor):
if hasattr(visitor, "visitAttrList"):
return visitor.visitAttrList(self)
else:
return visitor.visitChildren(self)
def attrList(self):
localctx = RelayParser.AttrListContext(self, self._ctx, self.state)
self.enterRule(localctx, 16, self.RULE_attrList)
self._la = 0 # Token type
try:
self.enterOuterAlt(localctx, 1)
self.state = 218
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.CNAME:
self.state = 210
self.attr()
self.state = 215
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__2:
self.state = 211
self.match(RelayParser.T__2)
self.state = 212
self.attr()
self.state = 217
self._errHandler.sync(self)
_la = self._input.LA(1)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class AttrContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.AttrContext, self).__init__(parent, invokingState)
self.parser = parser
def CNAME(self):
return self.getToken(RelayParser.CNAME, 0)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def getRuleIndex(self):
return RelayParser.RULE_attr
def accept(self, visitor):
if hasattr(visitor, "visitAttr"):
return visitor.visitAttr(self)
else:
return visitor.visitChildren(self)
def attr(self):
localctx = RelayParser.AttrContext(self, self._ctx, self.state)
self.enterRule(localctx, 18, self.RULE_attr)
try:
self.enterOuterAlt(localctx, 1)
self.state = 220
self.match(RelayParser.CNAME)
self.state = 221
self.match(RelayParser.T__8)
self.state = 222
self.expr(0)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class TypeParamSeqContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.TypeParamSeqContext, self).__init__(parent, invokingState)
self.parser = parser
def ident(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.IdentContext)
else:
return self.getTypedRuleContext(RelayParser.IdentContext,i)
def getRuleIndex(self):
return RelayParser.RULE_typeParamSeq
def accept(self, visitor):
if hasattr(visitor, "visitTypeParamSeq"):
return visitor.visitTypeParamSeq(self)
else:
return visitor.visitChildren(self)
def typeParamSeq(self):
localctx = RelayParser.TypeParamSeqContext(self, self._ctx, self.state)
self.enterRule(localctx, 20, self.RULE_typeParamSeq)
self._la = 0 # Token type
try:
self.state = 237
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,23,self._ctx)
if la_ == 1:
self.enterOuterAlt(localctx, 1)
self.state = 224
self.match(RelayParser.T__3)
self.state = 225
self.match(RelayParser.T__4)
pass
elif la_ == 2:
self.enterOuterAlt(localctx, 2)
self.state = 226
self.match(RelayParser.T__3)
self.state = 227
self.ident()
self.state = 232
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__2:
self.state = 228
self.match(RelayParser.T__2)
self.state = 229
self.ident()
self.state = 234
self._errHandler.sync(self)
_la = self._input.LA(1)
self.state = 235
self.match(RelayParser.T__4)
pass
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class Type_Context(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.Type_Context, self).__init__(parent, invokingState)
self.parser = parser
def getRuleIndex(self):
return RelayParser.RULE_type_
def copyFrom(self, ctx):
super(RelayParser.Type_Context, self).copyFrom(ctx)
class IntTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.IntTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def NAT(self):
return self.getToken(RelayParser.NAT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitIntType"):
return visitor.visitIntType(self)
else:
return visitor.visitChildren(self)
class TupleTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.TupleTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def type_(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.Type_Context)
else:
return self.getTypedRuleContext(RelayParser.Type_Context,i)
def accept(self, visitor):
if hasattr(visitor, "visitTupleType"):
return visitor.visitTupleType(self)
else:
return visitor.visitChildren(self)
class TypeIdentTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.TypeIdentTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def typeIdent(self):
return self.getTypedRuleContext(RelayParser.TypeIdentContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitTypeIdentType"):
return visitor.visitTypeIdentType(self)
else:
return visitor.visitChildren(self)
class IncompleteTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.IncompleteTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def accept(self, visitor):
if hasattr(visitor, "visitIncompleteType"):
return visitor.visitIncompleteType(self)
else:
return visitor.visitChildren(self)
class TensorTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.TensorTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def shapeSeq(self):
return self.getTypedRuleContext(RelayParser.ShapeSeqContext,0)
def type_(self):
return self.getTypedRuleContext(RelayParser.Type_Context,0)
def accept(self, visitor):
if hasattr(visitor, "visitTensorType"):
return visitor.visitTensorType(self)
else:
return visitor.visitChildren(self)
class FuncTypeContext(Type_Context):
def __init__(self, parser, ctx): # actually a RelayParser.Type_Context)
super(RelayParser.FuncTypeContext, self).__init__(parser)
self.copyFrom(ctx)
def type_(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.Type_Context)
else:
return self.getTypedRuleContext(RelayParser.Type_Context,i)
def typeParamSeq(self):
return self.getTypedRuleContext(RelayParser.TypeParamSeqContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitFuncType"):
return visitor.visitFuncType(self)
else:
return visitor.visitChildren(self)
def type_(self):
localctx = RelayParser.Type_Context(self, self._ctx, self.state)
self.enterRule(localctx, 22, self.RULE_type_)
self._la = 0 # Token type
try:
self.state = 284
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,28,self._ctx)
if la_ == 1:
localctx = RelayParser.TupleTypeContext(self, localctx)
self.enterOuterAlt(localctx, 1)
self.state = 239
self.match(RelayParser.T__0)
self.state = 240
self.match(RelayParser.T__1)
pass
elif la_ == 2:
localctx = RelayParser.TupleTypeContext(self, localctx)
self.enterOuterAlt(localctx, 2)
self.state = 241
self.match(RelayParser.T__0)
self.state = 242
self.type_()
self.state = 243
self.match(RelayParser.T__2)
self.state = 244
self.match(RelayParser.T__1)
pass
elif la_ == 3:
localctx = RelayParser.TupleTypeContext(self, localctx)
self.enterOuterAlt(localctx, 3)
self.state = 246
self.match(RelayParser.T__0)
self.state = 247
self.type_()
self.state = 250
self._errHandler.sync(self)
_la = self._input.LA(1)
while True:
self.state = 248
self.match(RelayParser.T__2)
self.state = 249
self.type_()
self.state = 252
self._errHandler.sync(self)
_la = self._input.LA(1)
if not (_la==RelayParser.T__2):
break
self.state = 254
self.match(RelayParser.T__1)
pass
elif la_ == 4:
localctx = RelayParser.TypeIdentTypeContext(self, localctx)
self.enterOuterAlt(localctx, 4)
self.state = 256
self.typeIdent()
pass
elif la_ == 5:
localctx = RelayParser.TensorTypeContext(self, localctx)
self.enterOuterAlt(localctx, 5)
self.state = 257
self.match(RelayParser.T__16)
self.state = 258
self.match(RelayParser.T__3)
self.state = 259
self.shapeSeq()
self.state = 260
self.match(RelayParser.T__2)
self.state = 261
self.type_()
self.state = 262
self.match(RelayParser.T__4)
pass
elif la_ == 6:
localctx = RelayParser.FuncTypeContext(self, localctx)
self.enterOuterAlt(localctx, 6)
self.state = 264
self.match(RelayParser.T__12)
self.state = 266
self._errHandler.sync(self)
_la = self._input.LA(1)
if _la==RelayParser.T__3:
self.state = 265
self.typeParamSeq()
self.state = 268
self.match(RelayParser.T__0)
self.state = 277
self._errHandler.sync(self)
_la = self._input.LA(1)
if (((_la) & ~0x3f) == 0 and ((1 << _la) & ((1 << RelayParser.T__0) | (1 << RelayParser.T__12) | (1 << RelayParser.T__16) | (1 << RelayParser.T__17) | (1 << RelayParser.NAT) | (1 << RelayParser.CNAME))) != 0):
self.state = 269
self.type_()
self.state = 274
self._errHandler.sync(self)
_la = self._input.LA(1)
while _la==RelayParser.T__2:
self.state = 270
self.match(RelayParser.T__2)
self.state = 271
self.type_()
self.state = 276
self._errHandler.sync(self)
_la = self._input.LA(1)
self.state = 279
self.match(RelayParser.T__1)
self.state = 280
self.match(RelayParser.T__13)
self.state = 281
self.type_()
pass
elif la_ == 7:
localctx = RelayParser.IncompleteTypeContext(self, localctx)
self.enterOuterAlt(localctx, 7)
self.state = 282
self.match(RelayParser.T__17)
pass
elif la_ == 8:
localctx = RelayParser.IntTypeContext(self, localctx)
self.enterOuterAlt(localctx, 8)
self.state = 283
self.match(RelayParser.NAT)
pass
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ShapeSeqContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ShapeSeqContext, self).__init__(parent, invokingState)
self.parser = parser
def shape(self, i=None):
if i is None:
return self.getTypedRuleContexts(RelayParser.ShapeContext)
else:
return self.getTypedRuleContext(RelayParser.ShapeContext,i)
def getRuleIndex(self):
return RelayParser.RULE_shapeSeq
def accept(self, visitor):
if hasattr(visitor, "visitShapeSeq"):
return visitor.visitShapeSeq(self)
else:
return visitor.visitChildren(self)
def shapeSeq(self):
localctx = RelayParser.ShapeSeqContext(self, self._ctx, self.state)
self.enterRule(localctx, 24, self.RULE_shapeSeq)
self._la = 0 # Token type
try:
self.state = 303
self._errHandler.sync(self)
la_ = self._interp.adaptivePredict(self._input,30,self._ctx)
if la_ == 1:
self.enterOuterAlt(localctx, 1)
self.state = 286
self.match(RelayParser.T__0)
self.state = 287
self.match(RelayParser.T__1)
pass
elif la_ == 2:
self.enterOuterAlt(localctx, 2)
self.state = 288
self.match(RelayParser.T__0)
self.state = 289
self.shape()
self.state = 290
self.match(RelayParser.T__2)
self.state = 291
self.match(RelayParser.T__1)
pass
elif la_ == 3:
self.enterOuterAlt(localctx, 3)
self.state = 293
self.match(RelayParser.T__0)
self.state = 294
self.shape()
self.state = 297
self._errHandler.sync(self)
_la = self._input.LA(1)
while True:
self.state = 295
self.match(RelayParser.T__2)
self.state = 296
self.shape()
self.state = 299
self._errHandler.sync(self)
_la = self._input.LA(1)
if not (_la==RelayParser.T__2):
break
self.state = 301
self.match(RelayParser.T__1)
pass
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ShapeContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ShapeContext, self).__init__(parent, invokingState)
self.parser = parser
def getRuleIndex(self):
return RelayParser.RULE_shape
def copyFrom(self, ctx):
super(RelayParser.ShapeContext, self).copyFrom(ctx)
class ParensShapeContext(ShapeContext):
def __init__(self, parser, ctx): # actually a RelayParser.ShapeContext)
super(RelayParser.ParensShapeContext, self).__init__(parser)
self.copyFrom(ctx)
def shape(self):
return self.getTypedRuleContext(RelayParser.ShapeContext,0)
def accept(self, visitor):
if hasattr(visitor, "visitParensShape"):
return visitor.visitParensShape(self)
else:
return visitor.visitChildren(self)
class IntShapeContext(ShapeContext):
def __init__(self, parser, ctx): # actually a RelayParser.ShapeContext)
super(RelayParser.IntShapeContext, self).__init__(parser)
self.copyFrom(ctx)
def NAT(self):
return self.getToken(RelayParser.NAT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitIntShape"):
return visitor.visitIntShape(self)
else:
return visitor.visitChildren(self)
def shape(self):
localctx = RelayParser.ShapeContext(self, self._ctx, self.state)
self.enterRule(localctx, 26, self.RULE_shape)
try:
self.state = 310
self._errHandler.sync(self)
token = self._input.LA(1)
if token in [RelayParser.T__0]:
localctx = RelayParser.ParensShapeContext(self, localctx)
self.enterOuterAlt(localctx, 1)
self.state = 305
self.match(RelayParser.T__0)
self.state = 306
self.shape()
self.state = 307
self.match(RelayParser.T__1)
pass
elif token in [RelayParser.NAT]:
localctx = RelayParser.IntShapeContext(self, localctx)
self.enterOuterAlt(localctx, 2)
self.state = 309
self.match(RelayParser.NAT)
pass
else:
raise NoViableAltException(self)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class TypeIdentContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.TypeIdentContext, self).__init__(parent, invokingState)
self.parser = parser
def CNAME(self):
return self.getToken(RelayParser.CNAME, 0)
def getRuleIndex(self):
return RelayParser.RULE_typeIdent
def accept(self, visitor):
if hasattr(visitor, "visitTypeIdent"):
return visitor.visitTypeIdent(self)
else:
return visitor.visitChildren(self)
def typeIdent(self):
localctx = RelayParser.TypeIdentContext(self, self._ctx, self.state)
self.enterRule(localctx, 28, self.RULE_typeIdent)
try:
self.enterOuterAlt(localctx, 1)
self.state = 312
self.match(RelayParser.CNAME)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class BodyContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.BodyContext, self).__init__(parent, invokingState)
self.parser = parser
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
def getRuleIndex(self):
return RelayParser.RULE_body
def accept(self, visitor):
if hasattr(visitor, "visitBody"):
return visitor.visitBody(self)
else:
return visitor.visitChildren(self)
def body(self):
localctx = RelayParser.BodyContext(self, self._ctx, self.state)
self.enterRule(localctx, 30, self.RULE_body)
try:
self.enterOuterAlt(localctx, 1)
self.state = 314
self.match(RelayParser.T__10)
self.state = 315
self.expr(0)
self.state = 316
self.match(RelayParser.T__11)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class ScalarContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.ScalarContext, self).__init__(parent, invokingState)
self.parser = parser
def getRuleIndex(self):
return RelayParser.RULE_scalar
def copyFrom(self, ctx):
super(RelayParser.ScalarContext, self).copyFrom(ctx)
class ScalarFloatContext(ScalarContext):
def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext)
super(RelayParser.ScalarFloatContext, self).__init__(parser)
self.copyFrom(ctx)
def FLOAT(self):
return self.getToken(RelayParser.FLOAT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitScalarFloat"):
return visitor.visitScalarFloat(self)
else:
return visitor.visitChildren(self)
class ScalarBoolContext(ScalarContext):
def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext)
super(RelayParser.ScalarBoolContext, self).__init__(parser)
self.copyFrom(ctx)
def BOOL_LIT(self):
return self.getToken(RelayParser.BOOL_LIT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitScalarBool"):
return visitor.visitScalarBool(self)
else:
return visitor.visitChildren(self)
class ScalarIntContext(ScalarContext):
def __init__(self, parser, ctx): # actually a RelayParser.ScalarContext)
super(RelayParser.ScalarIntContext, self).__init__(parser)
self.copyFrom(ctx)
def NAT(self):
return self.getToken(RelayParser.NAT, 0)
def accept(self, visitor):
if hasattr(visitor, "visitScalarInt"):
return visitor.visitScalarInt(self)
else:
return visitor.visitChildren(self)
def scalar(self):
localctx = RelayParser.ScalarContext(self, self._ctx, self.state)
self.enterRule(localctx, 32, self.RULE_scalar)
try:
self.state = 321
self._errHandler.sync(self)
token = self._input.LA(1)
if token in [RelayParser.FLOAT]:
localctx = RelayParser.ScalarFloatContext(self, localctx)
self.enterOuterAlt(localctx, 1)
self.state = 318
self.match(RelayParser.FLOAT)
pass
elif token in [RelayParser.NAT]:
localctx = RelayParser.ScalarIntContext(self, localctx)
self.enterOuterAlt(localctx, 2)
self.state = 319
self.match(RelayParser.NAT)
pass
elif token in [RelayParser.BOOL_LIT]:
localctx = RelayParser.ScalarBoolContext(self, localctx)
self.enterOuterAlt(localctx, 3)
self.state = 320
self.match(RelayParser.BOOL_LIT)
pass
else:
raise NoViableAltException(self)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
class IdentContext(ParserRuleContext):
def __init__(self, parser, parent=None, invokingState=-1):
super(RelayParser.IdentContext, self).__init__(parent, invokingState)
self.parser = parser
def opIdent(self):
return self.getTypedRuleContext(RelayParser.OpIdentContext,0)
def GLOBAL_VAR(self):
return self.getToken(RelayParser.GLOBAL_VAR, 0)
def LOCAL_VAR(self):
return self.getToken(RelayParser.LOCAL_VAR, 0)
def GRAPH_VAR(self):
return self.getToken(RelayParser.GRAPH_VAR, 0)
def getRuleIndex(self):
return RelayParser.RULE_ident
def accept(self, visitor):
if hasattr(visitor, "visitIdent"):
return visitor.visitIdent(self)
else:
return visitor.visitChildren(self)
def ident(self):
localctx = RelayParser.IdentContext(self, self._ctx, self.state)
self.enterRule(localctx, 34, self.RULE_ident)
try:
self.state = 327
self._errHandler.sync(self)
token = self._input.LA(1)
if token in [RelayParser.CNAME]:
self.enterOuterAlt(localctx, 1)
self.state = 323
self.opIdent()
pass
elif token in [RelayParser.GLOBAL_VAR]:
self.enterOuterAlt(localctx, 2)
self.state = 324
self.match(RelayParser.GLOBAL_VAR)
pass
elif token in [RelayParser.LOCAL_VAR]:
self.enterOuterAlt(localctx, 3)
self.state = 325
self.match(RelayParser.LOCAL_VAR)
pass
elif token in [RelayParser.GRAPH_VAR]:
self.enterOuterAlt(localctx, 4)
self.state = 326
self.match(RelayParser.GRAPH_VAR)
pass
else:
raise NoViableAltException(self)
except RecognitionException as re:
localctx.exception = re
self._errHandler.reportError(self, re)
self._errHandler.recover(self, re)
finally:
self.exitRule()
return localctx
def sempred(self, localctx, ruleIndex, predIndex):
if self._predicates == None:
self._predicates = dict()
self._predicates[2] = self.expr_sempred
pred = self._predicates.get(ruleIndex, None)
if pred is None:
raise Exception("No predicate with index:" + str(ruleIndex))
else:
return pred(localctx, predIndex)
def expr_sempred(self, localctx, predIndex):
if predIndex == 0:
return self.precpred(self._ctx, 16)
if predIndex == 1:
return self.precpred(self._ctx, 15)
if predIndex == 2:
return self.precpred(self._ctx, 14)
if predIndex == 3:
return self.precpred(self._ctx, 13)
if predIndex == 4:
return self.precpred(self._ctx, 4)
if predIndex == 5:
return self.precpred(self._ctx, 18)
python/tvm/relay/grammar/py2/RelayVisitor.py deleted 100644 → 0
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
from antlr4 import *
# This class defines a complete generic visitor for a parse tree produced by RelayParser.
class RelayVisitor(ParseTreeVisitor):
# Visit a parse tree produced by RelayParser#opIdent.
def visitOpIdent(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#prog.
def visitProg(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#identExpr.
def visitIdentExpr(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#call.
def visitCall(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#neg.
def visitNeg(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#tuple.
def visitTuple(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#parens.
def visitParens(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#funcExpr.
def visitFuncExpr(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#scalarExpr.
def visitScalarExpr(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#let.
def visitLet(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#tensor.
def visitTensor(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#ifElse.
def visitIfElse(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#graph.
def visitGraph(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#binOp.
def visitBinOp(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#func.
def visitFunc(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#defn.
def visitDefn(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#argList.
def visitArgList(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#varList.
def visitVarList(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#var.
def visitVar(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#attrList.
def visitAttrList(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#attr.
def visitAttr(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#typeParamSeq.
def visitTypeParamSeq(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#tupleType.
def visitTupleType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#typeIdentType.
def visitTypeIdentType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#tensorType.
def visitTensorType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#funcType.
def visitFuncType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#incompleteType.
def visitIncompleteType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#intType.
def visitIntType(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#shapeSeq.
def visitShapeSeq(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#parensShape.
def visitParensShape(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#intShape.
def visitIntShape(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#typeIdent.
def visitTypeIdent(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#body.
def visitBody(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#scalarFloat.
def visitScalarFloat(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#scalarInt.
def visitScalarInt(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#scalarBool.
def visitScalarBool(self, ctx):
return self.visitChildren(ctx)
# Visit a parse tree produced by RelayParser#ident.
def visitIdent(self, ctx):
return self.visitChildren(ctx)
python/tvm/relay/grammar/py3/.gitattributes deleted 100644 → 0
Relay* binary
Relay* linguist-generated=true
Relay* linguist-detectable=false
\ No newline at end of file
python/tvm/relay/grammar/py3/RelayLexer.py
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1
from antlr4 import *
from io import StringIO
from typing.io import TextIO
import sys
def serializedATN():
with StringIO() as buf:
buf.write("\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\2*")
......@@ -195,7 +194,7 @@ class RelayLexer(Lexer):
def __init__(self, input=None, output:TextIO = sys.stdout):
super().__init__(input, output)
self.checkVersion("4.7.2")
self.checkVersion("4.7.1")
self._interp = LexerATNSimulator(self, self.atn, self.decisionsToDFA, PredictionContextCache())
self._actions = None
self._predicates = None
......
python/tvm/relay/grammar/py3/RelayParser.py
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1
# encoding: utf-8
from antlr4 import *
from io import StringIO
from typing.io import TextIO
import sys
def serializedATN():
with StringIO() as buf:
buf.write("\3\u608b\ua72a\u8133\ub9ed\u417c\u3be7\u7786\u5964\3*")
......@@ -260,13 +259,12 @@ class RelayParser ( Parser ):
def __init__(self, input:TokenStream, output:TextIO = sys.stdout):
super().__init__(input, output)
self.checkVersion("4.7.2")
self.checkVersion("4.7.1")
self._interp = ParserATNSimulator(self, self.atn, self.decisionsToDFA, self.sharedContextCache)
self._predicates = None
class OpIdentContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -304,7 +302,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ProgContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -381,7 +378,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ExprContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -440,8 +436,6 @@ class RelayParser ( Parser ):
super().__init__(parser)
self.copyFrom(ctx)
def SUB(self):
return self.getToken(RelayParser.SUB, 0)
def expr(self):
return self.getTypedRuleContext(RelayParser.ExprContext,0)
......@@ -628,26 +622,6 @@ class RelayParser ( Parser ):
else:
return self.getTypedRuleContext(RelayParser.ExprContext,i)
def MUL(self):
return self.getToken(RelayParser.MUL, 0)
def DIV(self):
return self.getToken(RelayParser.DIV, 0)
def ADD(self):
return self.getToken(RelayParser.ADD, 0)
def SUB(self):
return self.getToken(RelayParser.SUB, 0)
def LT(self):
return self.getToken(RelayParser.LT, 0)
def GT(self):
return self.getToken(RelayParser.GT, 0)
def LE(self):
return self.getToken(RelayParser.LE, 0)
def GE(self):
return self.getToken(RelayParser.GE, 0)
def EQ(self):
return self.getToken(RelayParser.EQ, 0)
def NE(self):
return self.getToken(RelayParser.NE, 0)
def accept(self, visitor:ParseTreeVisitor):
if hasattr( visitor, "visitBinOp" ):
......@@ -1037,7 +1011,6 @@ class RelayParser ( Parser ):
self.unrollRecursionContexts(_parentctx)
return localctx
class FuncContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1115,7 +1088,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class DefnContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1199,7 +1171,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ArgListContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1265,7 +1236,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class VarListContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1327,7 +1297,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class VarContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1381,7 +1350,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class AttrListContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1442,7 +1410,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class AttrContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1488,7 +1455,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class TypeParamSeqContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1562,7 +1528,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class Type_Context(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1836,7 +1801,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ShapeSeqContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -1924,7 +1888,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ShapeContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -2010,7 +1973,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class TypeIdentContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -2048,7 +2010,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class BodyContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -2091,7 +2052,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class ScalarContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......@@ -2194,7 +2154,6 @@ class RelayParser ( Parser ):
self.exitRule()
return localctx
class IdentContext(ParserRuleContext):
def __init__(self, parser, parent:ParserRuleContext=None, invokingState:int=-1):
......
python/tvm/relay/grammar/py3/RelayVisitor.py
# Generated from /home/sslyu/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.2
# Generated from /home/marisa/Work/tvm/python/tvm/relay/grammar/Relay.g4 by ANTLR 4.7.1
from antlr4 import *
if __name__ is not None and "." in __name__:
from .RelayParser import RelayParser
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment