Searched refs:_register_external_op_helper (Results 1 – 2 of 2) sorted by relevance
40 def _register_external_op_helper(op_name, supported=True): function62 _register_external_op_helper("nn.batch_norm")63 _register_external_op_helper("nn.conv2d")64 _register_external_op_helper("nn.dense")65 _register_external_op_helper("nn.relu")66 _register_external_op_helper("add")67 _register_external_op_helper("subtract")68 _register_external_op_helper("multiply")
221 def _register_external_op_helper(op_name, supported=True): function229 _register_external_op_helper("reshape")