BVh+xdZddlZddlZddlmZddlmZddl m Z dddd d d Z d Z d ZdZGddZdZy)zGA utility class to generate the report HTML based on a common template.N)toco_conversion_log_pb2)file_io)resource_loaderz&z"z'z>z<)&"'>zhtml_escape.."s;"&&q!,;s ")join)texts r html_escaper!s ;d; ;;c|jd}|jd}||dz|}|jd}d}tt|D]"}|dzdk(r|dz }n|d z }|||z }|d z }$|d d S) aParses op_signature and returns a string denoting the input tensor type. Args: op_signature: a string specifying the signature of a particular operator. The signature of an operator contains the input tensor's shape and type, output tensor's shape and type, operator's name and its version. It has the following schema: INPUT:input_1_shape::input_1_type::input_2_shape::input_2_type::.. ::OUTPUT:output_1_shape::output_1_type::output_2_shape::output_2_type:: ..::NAME:operator_name ::VERSION:operator_version An example of an operator signature is: INPUT:[1,73,73,160]::float::[64,1,1,160]::float::[64]::float:: OUTPUT:[1,73,73,64]::float::NAME:Conv::VERSION:1 Returns: A string denoting the input tensors' type. In the form of shape/type separated by comma. For example: shape:[1,73,73,160],type:float,shape:[64,1,1,160],type:float,shape:[64], type:float :z::OUTPUTz::r rzshape:ztype:,N)findsplitrangelen) op_signaturestartendinputslstout_stris rget_input_type_from_signaturer)%s,   C %*%#  # && T# ' S?a1uz g g s1vG sNG  "rc@||jvry||jvryy)NzBUILT-INz CUSTOM OPz SELECT OP) built_in_ops custom_ops)op_nameconversion_logs rget_operator_typer/Js' +++ .+++  rc"eZdZdZdZ ddZy) HTMLGeneratorz)Utility class to generate an HTML report.cTtj|stdj|tj|d5}|j |_dddtjtjj|||_ y#1swYCxYw)alReads the HTML template content. Args: html_template_path: A string, path to the template HTML file. export_report_path: A string, path to the generated HTML report. This path should point to a '.html' file with date and time in its name. e.g. 2019-01-01-10:05.toco_report.html. Raises: IOError: File doesn't exist. zFile '{0}' does not exist.rN) _file_io file_existsIOErrorformatFileIOread html_templaterecursive_create_dirospathdirnameexport_report_path)selfhtml_template_pathr?fs r__init__zHTMLGenerator.__init__Vs    2 3 0778JK LL +S 1$Q668d$ !!"''//2D"EF0D $$s BB'ci}|rdnd|d<t|j|d<t|j|d<tt|jj |d<tt|j j |d<tt|j j |d<|rd nd |d <d } d } tt|jD]Y} | d |j| zdzz } | t|jkr"| dt|j| zdzz } U| dz } [|jD]} t| |} | d | zdz| zdzz } !| |d<| |d<||d<||d<|rt||d<nd|z}t||d<|j}|D]:}|j|||d}|j!|dk7s.t#d|zt%j&|j(d5}|j+|dddy#1swYyxYw)a Generates the HTML report and writes it to local directory. This function uses the fields in `toco_conversion_log_before` and `toco_conversion_log_after` to populate the HTML content. Certain markers (placeholders) in the HTML template are then substituted with the fields from the protos. Once finished it will write the HTML file to the specified local file path. Args: toco_conversion_log_before: A `TocoConversionLog` protobuf generated before the model is converted by TOCO. toco_conversion_log_after: A `TocoConversionLog` protobuf generated after the model is converted by TOCO. post_training_quant_enabled: A boolean, whether post-training quantization is enabled. dot_before: A string, the dot representation of the model before the conversion. dot_after: A string, the dot representation of the model after the conversion. toco_err_log: A string, the logs emitted by TOCO during conversion. Caller need to ensure that this string is properly anonymized (any kind of user data should be eliminated). tflite_graph_path: A string, the filepath to the converted TFLite model. Raises: RuntimeError: When error occurs while generating the template. z,Failz0Successzzzzzziszisn'tz"r zzzz zz zzzzzzLTFLite graph conversion successful. You can preview the converted model at: rrz!Could not populate marker text %rwN)str model_sizesumr+values select_opsr,r r!op_list op_signaturesr)r/rr:replacer RuntimeErrorr4r8r?write)r@toco_conversion_log_beforetoco_conversion_log_afterpost_training_quant_enabled dot_before dot_after toco_err_logtflite_graph_path html_dictpre_op_profilepost_op_profiler(opsupported_type success_infotemplatemarkerrBs rgeneratezHTMLGenerator.generateksFI  8N()47"--4/I/025!,,3.I./-0 % 2 2 9 9 ;<.>I)*+. % 0 0 7 7 9:,GHH I 00# 6!gghs !H<<IN)r r )__name__ __module__ __qualname____doc__rCr`rrr1r1Ss116!#^rr1c ztjd}tjj |st dj |tjj|d}tjj|d}tjj|d}tjj|d}tjj |st dj |tjj |st dj |tjj |st dj |tjj |st dj |t|tjj|d}tj} tj} t|d5} | j| jd d d t|d5} | j| jd d d tj|d d 5} | jj} d d d tj|d d 5} | jj} d d d |j!| | |  | j"|y #1swYxYw#1swYxYw#1swYxYw#1swYOxYw) aGenerates an HTML report about the conversion process. Args: conversion_log_dir: A string specifying the file directory of the conversion logs. It's required that before calling this function, the `conversion_log_dir` already contains the following files: `toco_log_before.pb`, `toco_log_after.pb`, `toco_tf_graph.dot`, `toco_tflite_graph.dot`. quantization_enabled: A boolean, passed from the tflite converter to indicate whether post-training quantization is enabled during conversion. tflite_graph_path: A string, the filepath to the converted TFLite model. Raises: IOError: When any of the required files doesn't exist. z template.htmlz2Failed to generate HTML: file '{0}' doesn't exist.ztoco_log_before.pbztoco_log_after.pbztoco_tf_graph.dotztoco_tflite_graph.dotztoco_conversion_summary.htmlrbNr3zutf-8)encoding)_resource_loaderget_path_to_datafiler<r=existsr6r7rr1_toco_conversion_log_pb2TocoConversionLogopenParseFromStringr9iorstripr` toco_err_logs)conversion_log_dirquantization_enabledrWtemplate_filenametoco_log_before_pathtoco_log_after_pathdot_before_pathdot_after_pathhtml_generatorrQrRrBrTrUs rgen_conversion_log_htmlr{s$';;OL ) * FMM &8:NO %79LMGGLL!35HI/77<< 24KL. , - FMM  + , FMM   ( FMM   ' FMM !ggll%'EFH. 8IIK6HHJ $'91..qvvx89 &8!--affh78 wwg6#!"J# ww~sW5"!I"46O. I3AA+-9988##""s0 L  LL%L1 LL"%L.1L:)rdrpr<tensorflow.lite.toco.loggingrrltensorflow.python.lib.iorr4tensorflow.python.platformrrirrr)r/r1r{rerrrsUN \8J      <"Jvvr=-r