3e530f0fb84ba99e7cb6e84cc3547f1e1e5f01eb,coremltools/converters/nnssa/coreml/ssa_converter.py,SSAConverter,convert,#SSAConverter#,342

Before Change

            node = func.graph[node_name]
            op_type = node.op
            if op_type not in self.CONVERT_FUNCTION_MAP:
                raise NotImplementedError(
                    "[SSAConverter] Op type "{}" not implemented.".format(op_type))
            if builtins.is_tensor(node.datatype):
                print(
                    "[SSAConverter] [{}/{}] Converting op "{}" of type "{}", output shape: {}".format(
                        idx + 1, len(instruction_order), node_name, op_type, node.datatype.get_shape()))

After Change

                convert_func = self._convert_custom_layer
                conversion_message = " with custom layer"
            else:       
                raise NotImplementedError(
                    "[SSAConverter] Conversion for op %s not implemented, terminating..." %
                    (op_type))
            
            print("[SSAConverter] [{}/{}] Converting op type \"{}\", of name \"{}\" {} {}".format(
                  idx + 1, len(instruction_order), op_type, node_name, conversion_message,
                  (("(output shape: " + str(node.datatype.get_shape()) +")") if builtins.is_tensor(node.datatype) else "")))

Instances