<?xml version="1.0" encoding="UTF-8"?>
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">
<html>
<head>
<!-- Generated by HsColour, http://code.haskell.org/~malcolm/hscolour/ -->
<title>src/TensorFlow/OpGen.hs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
<pre><a name="line-1"></a><span class='hs-comment'>-- Copyright 2016 TensorFlow authors.</span>
<a name="line-2"></a><span class='hs-comment'>--</span>
<a name="line-3"></a><span class='hs-comment'>-- Licensed under the Apache License, Version 2.0 (the "License");</span>
<a name="line-4"></a><span class='hs-comment'>-- you may not use this file except in compliance with the License.</span>
<a name="line-5"></a><span class='hs-comment'>-- You may obtain a copy of the License at</span>
<a name="line-6"></a><span class='hs-comment'>--</span>
<a name="line-7"></a><span class='hs-comment'>--     <a href="http://www.apache.org/licenses/LICENSE-2.0">http://www.apache.org/licenses/LICENSE-2.0</a></span>
<a name="line-8"></a><span class='hs-comment'>--</span>
<a name="line-9"></a><span class='hs-comment'>-- Unless required by applicable law or agreed to in writing, software</span>
<a name="line-10"></a><span class='hs-comment'>-- distributed under the License is distributed on an "AS IS" BASIS,</span>
<a name="line-11"></a><span class='hs-comment'>-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span>
<a name="line-12"></a><span class='hs-comment'>-- See the License for the specific language governing permissions and</span>
<a name="line-13"></a><span class='hs-comment'>-- limitations under the License.</span>
<a name="line-14"></a>
<a name="line-15"></a><span class='hs-comment'>{-# LANGUAGE FlexibleContexts #-}</span>
<a name="line-16"></a><span class='hs-comment'>{-# LANGUAGE OverloadedStrings #-}</span>
<a name="line-17"></a><span class='hs-comment'>{-# LANGUAGE TypeFamilies #-}</span>
<a name="line-18"></a><span class='hs-comment'>-- | Rendering of TensorFlow operations as Haskell functions.</span>
<a name="line-19"></a>
<a name="line-20"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>TensorFlow</span><span class='hs-varop'>.</span><span class='hs-conid'>OpGen</span>
<a name="line-21"></a>  <span class='hs-layout'>(</span> <span class='hs-conid'>OpGenFlags</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-22"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>docOpList</span>
<a name="line-23"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>flagParser</span><span class='hs-layout'>)</span>
<a name="line-24"></a>  <span class='hs-keyword'>where</span>
<a name="line-25"></a>
<a name="line-26"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Prelude</span> <span class='hs-varid'>hiding</span> <span class='hs-layout'>(</span><span class='hs-varid'>head</span><span class='hs-layout'>,</span> <span class='hs-varid'>tail</span><span class='hs-layout'>)</span>
<a name="line-27"></a>
<a name="line-28"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Control</span><span class='hs-varop'>.</span><span class='hs-conid'>Monad</span> <span class='hs-layout'>(</span><span class='hs-varid'>guard</span><span class='hs-layout'>)</span>
<a name="line-29"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Char</span> <span class='hs-layout'>(</span><span class='hs-varid'>toLower</span><span class='hs-layout'>,</span> <span class='hs-varid'>toUpper</span><span class='hs-layout'>)</span>
<a name="line-30"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Foldable</span> <span class='hs-layout'>(</span><span class='hs-varid'>toList</span><span class='hs-layout'>)</span>
<a name="line-31"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Maybe</span> <span class='hs-layout'>(</span><span class='hs-varid'>fromMaybe</span><span class='hs-layout'>,</span> <span class='hs-varid'>maybeToList</span><span class='hs-layout'>)</span>
<a name="line-32"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>ProtoLens</span> <span class='hs-layout'>(</span><span class='hs-varid'>def</span><span class='hs-layout'>,</span> <span class='hs-varid'>showMessage</span><span class='hs-layout'>)</span>
<a name="line-33"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>List</span><span class='hs-varop'>.</span><span class='hs-conid'>NonEmpty</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonEmpty</span><span class='hs-layout'>(</span><span class='hs-layout'>(</span><span class='hs-conop'>:|</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>head</span><span class='hs-layout'>)</span>
<a name="line-34"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>List</span><span class='hs-varop'>.</span><span class='hs-conid'>NonEmpty</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>NE</span>
<a name="line-35"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Lens</span><span class='hs-varop'>.</span><span class='hs-conid'>Family2</span> <span class='hs-layout'>(</span><span class='hs-layout'>(</span><span class='hs-varop'>^.</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>.~</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>&amp;</span><span class='hs-layout'>)</span><span class='hs-layout'>,</span> <span class='hs-varid'>view</span><span class='hs-layout'>)</span>
<a name="line-36"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Options</span><span class='hs-varop'>.</span><span class='hs-conid'>Applicative</span> <span class='hs-layout'>(</span><span class='hs-conid'>Parser</span><span class='hs-layout'>,</span> <span class='hs-varid'>help</span><span class='hs-layout'>,</span> <span class='hs-varid'>long</span><span class='hs-layout'>,</span> <span class='hs-varid'>strOption</span><span class='hs-layout'>,</span> <span class='hs-varid'>value</span><span class='hs-layout'>)</span>
<a name="line-37"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Proto</span><span class='hs-varop'>.</span><span class='hs-conid'>Tensorflow</span><span class='hs-varop'>.</span><span class='hs-conid'>Core</span><span class='hs-varop'>.</span><span class='hs-conid'>Framework</span><span class='hs-varop'>.</span><span class='hs-conid'>OpDef</span>
<a name="line-38"></a>  <span class='hs-layout'>(</span> <span class='hs-conid'>OpList</span>
<a name="line-39"></a>  <span class='hs-layout'>,</span> <span class='hs-conid'>OpDef</span>
<a name="line-40"></a>  <span class='hs-layout'>,</span> <span class='hs-conid'>OpDef'ArgDef</span>
<a name="line-41"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>attr</span>
<a name="line-42"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>description</span>
<a name="line-43"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>inputArg</span>
<a name="line-44"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>name</span>
<a name="line-45"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>numberAttr</span>
<a name="line-46"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>op</span>
<a name="line-47"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>outputArg</span>
<a name="line-48"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>summary</span>
<a name="line-49"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>type'</span>
<a name="line-50"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>typeAttr</span>
<a name="line-51"></a>  <span class='hs-layout'>)</span>
<a name="line-52"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Proto</span><span class='hs-varop'>.</span><span class='hs-conid'>Tensorflow</span><span class='hs-varop'>.</span><span class='hs-conid'>Core</span><span class='hs-varop'>.</span><span class='hs-conid'>Framework</span><span class='hs-varop'>.</span><span class='hs-conid'>Types</span> <span class='hs-layout'>(</span><span class='hs-conid'>DataType</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-53"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>System</span><span class='hs-varop'>.</span><span class='hs-conid'>FilePath</span> <span class='hs-layout'>(</span><span class='hs-varid'>takeBaseName</span><span class='hs-layout'>)</span>
<a name="line-54"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>TensorFlow</span><span class='hs-varop'>.</span><span class='hs-conid'>OpGen</span><span class='hs-varop'>.</span><span class='hs-conid'>AttrVal</span>
<a name="line-55"></a>  <span class='hs-layout'>(</span><span class='hs-conid'>AttrDef</span>
<a name="line-56"></a>  <span class='hs-layout'>,</span> <span class='hs-conid'>AttrCase</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-57"></a>  <span class='hs-layout'>,</span> <span class='hs-conid'>AttrTemplate</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-58"></a>  <span class='hs-layout'>,</span> <span class='hs-conid'>Template</span>
<a name="line-59"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>attrDef</span>
<a name="line-60"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>attrOriginal</span>
<a name="line-61"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>attrTemplate</span>
<a name="line-62"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>templateDefault</span>
<a name="line-63"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>templateRestrictions</span>
<a name="line-64"></a>  <span class='hs-layout'>)</span>
<a name="line-65"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>PrettyPrint</span><span class='hs-varop'>.</span><span class='hs-conid'>Mainland</span>
<a name="line-66"></a>  <span class='hs-layout'>(</span> <span class='hs-conid'>Doc</span>
<a name="line-67"></a>  <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>&lt;&gt;</span><span class='hs-layout'>)</span>
<a name="line-68"></a>  <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>&lt;+&gt;</span><span class='hs-layout'>)</span>
<a name="line-69"></a>  <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>&lt;/&gt;</span><span class='hs-layout'>)</span>
<a name="line-70"></a>  <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'>&lt;+/&gt;</span><span class='hs-layout'>)</span>
<a name="line-71"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>brackets</span>
<a name="line-72"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>comma</span>
<a name="line-73"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>commasep</span>
<a name="line-74"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>dquotes</span>
<a name="line-75"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-76"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>enclose</span>
<a name="line-77"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>flatten</span>
<a name="line-78"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>folddoc</span>
<a name="line-79"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>hang</span>
<a name="line-80"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>indent</span>
<a name="line-81"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>int</span>
<a name="line-82"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>parens</span>
<a name="line-83"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>sep</span>
<a name="line-84"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>stack</span>
<a name="line-85"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>strictText</span>
<a name="line-86"></a>  <span class='hs-layout'>,</span> <span class='hs-varid'>tuple</span>
<a name="line-87"></a>  <span class='hs-layout'>)</span>
<a name="line-88"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-conid'>Strict</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>Map</span>
<a name="line-89"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Set</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>Set</span>
<a name="line-90"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>Text</span>
<a name="line-91"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Semigroup</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>Semigroup</span>
<a name="line-92"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-layout'>)</span>
<a name="line-93"></a>
<a name="line-94"></a><a name="OpGenFlags"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>OpGenFlags</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>OpGenFlags</span>
<a name="line-95"></a>     <span class='hs-layout'>{</span> <span class='hs-varid'>outputFile</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span>
<a name="line-96"></a>     <span class='hs-layout'>,</span> <span class='hs-varid'>prefix</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span>
<a name="line-97"></a>     <span class='hs-layout'>,</span> <span class='hs-varid'>excludeList</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>String</span>
<a name="line-98"></a>     <span class='hs-layout'>}</span>
<a name="line-99"></a>
<a name="line-100"></a><a name="flagParser"></a><span class='hs-definition'>flagParser</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Parser</span> <span class='hs-conid'>OpGenFlags</span>
<a name="line-101"></a><span class='hs-definition'>flagParser</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>OpGenFlags</span>
<a name="line-102"></a>     <span class='hs-varop'>&lt;$&gt;</span> <span class='hs-varid'>strOption</span> <span class='hs-layout'>(</span><span class='hs-varid'>mconcat</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>long</span> <span class='hs-str'>"output"</span>
<a name="line-103"></a>                            <span class='hs-layout'>,</span> <span class='hs-varid'>help</span> <span class='hs-str'>"File to write."</span>
<a name="line-104"></a>                            <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-105"></a>     <span class='hs-varop'>&lt;*&gt;</span> <span class='hs-varid'>strOption</span> <span class='hs-layout'>(</span><span class='hs-varid'>mconcat</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>long</span> <span class='hs-str'>"prefix"</span>
<a name="line-106"></a>                            <span class='hs-layout'>,</span> <span class='hs-varid'>help</span> <span class='hs-str'>"Haskell package prefix to use"</span>
<a name="line-107"></a>                            <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-108"></a>     <span class='hs-varop'>&lt;*&gt;</span> <span class='hs-varid'>strOption</span> <span class='hs-layout'>(</span><span class='hs-varid'>mconcat</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>long</span> <span class='hs-str'>"exclude_list"</span>
<a name="line-109"></a>                            <span class='hs-layout'>,</span> <span class='hs-varid'>value</span> <span class='hs-str'>""</span>
<a name="line-110"></a>                            <span class='hs-layout'>,</span> <span class='hs-varid'>help</span> <span class='hs-str'>"Comma separated Ops names to ignore"</span>
<a name="line-111"></a>                            <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-112"></a>
<a name="line-113"></a>
<a name="line-114"></a><a name="docOpList"></a><span class='hs-definition'>docOpList</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpGenFlags</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>OpList</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-115"></a><span class='hs-definition'>docOpList</span> <span class='hs-varid'>flags</span> <span class='hs-varid'>opList</span> <span class='hs-keyglyph'>=</span>
<a name="line-116"></a>  <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"{-# LANGUAGE ConstraintKinds #-}"</span>
<a name="line-117"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE DataKinds #-}"</span>
<a name="line-118"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE FlexibleInstances #-}"</span>
<a name="line-119"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE OverloadedStrings #-}"</span>
<a name="line-120"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE RankNTypes #-}"</span>
<a name="line-121"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE ScopedTypeVariables #-}"</span>
<a name="line-122"></a>        <span class='hs-layout'>,</span> <span class='hs-str'>"module"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>moduleName</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"where"</span>
<a name="line-123"></a>        <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-124"></a>        <span class='hs-layout'>,</span> <span class='hs-varid'>imports</span>
<a name="line-125"></a>        <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-126"></a>        <span class='hs-layout'>,</span> <span class='hs-varid'>folddoc</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>x</span> <span class='hs-varid'>y</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>x</span> <span class='hs-varop'>&lt;/&gt;</span> <span class='hs-varid'>empty</span> <span class='hs-varop'>&lt;/&gt;</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span>
<a name="line-127"></a>                  <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>renderDef</span> <span class='hs-varop'>$</span>
<a name="line-128"></a>                   <span class='hs-varid'>filter</span> <span class='hs-layout'>(</span><span class='hs-varid'>not</span> <span class='hs-varop'>.</span> <span class='hs-varid'>flip</span> <span class='hs-varid'>elem</span> <span class='hs-varid'>exclusions</span> <span class='hs-varop'>.</span> <span class='hs-varid'>view</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-varop'>$</span>
<a name="line-129"></a>                   <span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-varid'>opList</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>op</span><span class='hs-layout'>)</span>
<a name="line-130"></a>        <span class='hs-keyglyph'>]</span>
<a name="line-131"></a>  <span class='hs-keyword'>where</span> <span class='hs-varid'>moduleName</span> <span class='hs-keyglyph'>=</span>
<a name="line-132"></a>            <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>pack</span> <span class='hs-layout'>(</span><span class='hs-varid'>prefix</span> <span class='hs-varid'>flags</span><span class='hs-layout'>)</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-str'>"."</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>camelCase</span>
<a name="line-133"></a>             <span class='hs-comment'>-- Discards the optional trailing _op_lib</span>
<a name="line-134"></a>            <span class='hs-layout'>(</span><span class='hs-varid'>fromMaybe</span> <span class='hs-varid'>shortName</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>stripSuffix</span> <span class='hs-str'>"_op_lib"</span> <span class='hs-varid'>shortName</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-135"></a>        <span class='hs-varid'>shortName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>pack</span> <span class='hs-layout'>(</span><span class='hs-varid'>takeBaseName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>outputFile</span> <span class='hs-varid'>flags</span><span class='hs-layout'>)</span>
<a name="line-136"></a>        <span class='hs-varid'>exclusions</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>splitOn</span> <span class='hs-str'>","</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>pack</span> <span class='hs-varop'>$</span> <span class='hs-varid'>excludeList</span> <span class='hs-varid'>flags</span>
<a name="line-137"></a>
<a name="line-138"></a><a name="camelCase"></a><span class='hs-definition'>camelCase</span> <span class='hs-varid'>s</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>concat</span> <span class='hs-varop'>$</span> <span class='hs-varid'>map</span> <span class='hs-varid'>upCase</span>
<a name="line-139"></a>                          <span class='hs-varop'>$</span> <span class='hs-varid'>filter</span> <span class='hs-layout'>(</span><span class='hs-varop'>/=</span> <span class='hs-str'>"ops"</span><span class='hs-layout'>)</span>
<a name="line-140"></a>                          <span class='hs-varop'>$</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>splitOn</span> <span class='hs-str'>"_"</span> <span class='hs-varid'>s</span>
<a name="line-141"></a>
<a name="line-142"></a><a name="upCase"></a><span class='hs-comment'>-- | Upper-case the given text.</span>
<a name="line-143"></a><span class='hs-definition'>upCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span>
<a name="line-144"></a><span class='hs-definition'>upCase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>forceCase</span> <span class='hs-varid'>toUpper</span>
<a name="line-145"></a>
<a name="line-146"></a><a name="lowCase"></a><span class='hs-comment'>-- | Lower-case the given name, and prevent it from overlapping with a reserved</span>
<a name="line-147"></a><span class='hs-comment'>-- Haskell name.</span>
<a name="line-148"></a><span class='hs-definition'>lowCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span>
<a name="line-149"></a><span class='hs-definition'>lowCase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>replaceReservedName</span> <span class='hs-varop'>.</span> <span class='hs-varid'>forceCase</span> <span class='hs-varid'>toLower</span>
<a name="line-150"></a>
<a name="line-151"></a><a name="forceCase"></a><span class='hs-definition'>forceCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Char</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Char</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span>
<a name="line-152"></a><span class='hs-definition'>forceCase</span> <span class='hs-varid'>convert</span> <span class='hs-varid'>s</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>maybe</span> <span class='hs-str'>""</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-layout'>(</span><span class='hs-varid'>c</span><span class='hs-layout'>,</span> <span class='hs-varid'>cs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>cons</span> <span class='hs-layout'>(</span><span class='hs-varid'>convert</span> <span class='hs-varid'>c</span><span class='hs-layout'>)</span> <span class='hs-varid'>cs</span><span class='hs-layout'>)</span>
<a name="line-153"></a>                      <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>uncons</span> <span class='hs-varid'>s</span><span class='hs-layout'>)</span>
<a name="line-154"></a>
<a name="line-155"></a><a name="imports"></a><span class='hs-definition'>imports</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span>
<a name="line-156"></a>      <span class='hs-str'>"import Data.ByteString (ByteString)"</span>
<a name="line-157"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Complex (Complex)"</span>
<a name="line-158"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Int (Int8, Int16, Int32, Int64)"</span>
<a name="line-159"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Word (Word8, Word16)"</span>
<a name="line-160"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import Lens.Family2 ((.~), (&amp;))"</span>
<a name="line-161"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Build"</span>
<a name="line-162"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.BuildOp"</span>
<a name="line-163"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Tensor"</span>
<a name="line-164"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Types"</span>
<a name="line-165"></a>      <span class='hs-keyglyph'>]</span>
<a name="line-166"></a>
<a name="line-167"></a><a name="renderDef"></a><span class='hs-definition'>renderDef</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-168"></a><span class='hs-definition'>renderDef</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span>
<a name="line-169"></a>  <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span>
<a name="line-170"></a>      <span class='hs-varid'>haddocks</span>
<a name="line-171"></a>    <span class='hs-layout'>,</span> <span class='hs-varid'>n</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"::"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-layout'>(</span><span class='hs-varid'>typeSig</span> <span class='hs-varid'>d</span><span class='hs-layout'>)</span>
<a name="line-172"></a>    <span class='hs-layout'>,</span> <span class='hs-varid'>n</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-varid'>args</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"|"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>funcGuard</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"="</span> <span class='hs-varop'>&lt;/&gt;</span>  <span class='hs-comment'>-- args are indented</span>
<a name="line-173"></a>            <span class='hs-comment'>-- the body needs to be indented wrt the name</span>
<a name="line-174"></a>            <span class='hs-varid'>indent</span> <span class='hs-varid'>indentation</span> <span class='hs-varid'>functionBody</span>
<a name="line-175"></a>    <span class='hs-layout'>,</span> <span class='hs-varid'>extras</span>  <span class='hs-comment'>-- just for debug</span>
<a name="line-176"></a>    <span class='hs-keyglyph'>]</span>
<a name="line-177"></a>  <span class='hs-keyword'>where</span>
<a name="line-178"></a>    <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span> <span class='hs-varid'>fixOpName</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span>
<a name="line-179"></a>    <span class='hs-varid'>args</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>sep</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>hsName</span> <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-varid'>hsName</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>mandatoryAttrs</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span> <span class='hs-varid'>tensorArgs</span>
<a name="line-180"></a>    <span class='hs-varid'>tensorArgs</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span> <span class='hs-varid'>lowCase</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span><span class='hs-keyglyph'>]</span>
<a name="line-181"></a>    <span class='hs-varid'>fixOpName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>lowCase</span>
<a name="line-182"></a>    <span class='hs-varid'>funcGuard</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"eqLengthGuard"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>brackets</span> <span class='hs-layout'>(</span><span class='hs-varid'>commasep</span> <span class='hs-varid'>entries</span><span class='hs-layout'>)</span>
<a name="line-183"></a>      <span class='hs-keyword'>where</span>
<a name="line-184"></a>        <span class='hs-varid'>entries</span> <span class='hs-keyglyph'>=</span>
<a name="line-185"></a>            <span class='hs-keyglyph'>[</span> <span class='hs-varid'>parens</span> <span class='hs-varop'>$</span> <span class='hs-varid'>quotedText</span> <span class='hs-varid'>nAttr</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>comma</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-186"></a>              <span class='hs-varid'>brackets</span> <span class='hs-layout'>(</span><span class='hs-varid'>commasep</span> <span class='hs-varop'>$</span> <span class='hs-varid'>toList</span> <span class='hs-varop'>$</span>
<a name="line-187"></a>              <span class='hs-conid'>NE</span><span class='hs-varop'>.</span><span class='hs-varid'>map</span> <span class='hs-varid'>renderTensorName</span> <span class='hs-varid'>tensorNames</span><span class='hs-layout'>)</span>
<a name="line-188"></a>            <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>nAttr</span><span class='hs-layout'>,</span> <span class='hs-varid'>tensorNames</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-varid'>numberAttrMap</span> <span class='hs-varid'>d</span>
<a name="line-189"></a>            <span class='hs-keyglyph'>]</span>
<a name="line-190"></a>        <span class='hs-varid'>renderTensorName</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>parens</span> <span class='hs-varop'>$</span> <span class='hs-varid'>quotedText</span> <span class='hs-varid'>x</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>comma</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-191"></a>                             <span class='hs-str'>"length"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>x</span>
<a name="line-192"></a>    <span class='hs-comment'>-- Uses hang 0 to align the argument vertically on multiple lines.</span>
<a name="line-193"></a>    <span class='hs-varid'>functionBody</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>buildFunction</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>parens</span> <span class='hs-layout'>(</span><span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-layout'>(</span><span class='hs-varid'>stack</span> <span class='hs-varid'>buildOpParts</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-194"></a>                                 <span class='hs-varop'>&lt;/&gt;</span> <span class='hs-varid'>indent</span> <span class='hs-varid'>indentation</span> <span class='hs-layout'>(</span><span class='hs-varid'>sep</span> <span class='hs-varid'>tensorArgs</span><span class='hs-layout'>)</span>
<a name="line-195"></a>    <span class='hs-varid'>buildFunction</span>
<a name="line-196"></a>        <span class='hs-keyglyph'>|</span> <span class='hs-varid'>null</span> <span class='hs-varid'>outputListsSizes</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"buildOp"</span>
<a name="line-197"></a>        <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"buildListOp"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>brackets</span> <span class='hs-layout'>(</span><span class='hs-varid'>commasep</span> <span class='hs-varid'>outputListsSizes</span><span class='hs-layout'>)</span>
<a name="line-198"></a>    <span class='hs-varid'>outputListsSizes</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>numberAttrName</span>
<a name="line-199"></a>                       <span class='hs-keyglyph'>|</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>outputArg</span>
<a name="line-200"></a>                       <span class='hs-layout'>,</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>numberAttrName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>numberAttr</span>
<a name="line-201"></a>                       <span class='hs-layout'>,</span> <span class='hs-varid'>not</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>numberAttrName</span><span class='hs-layout'>)</span> <span class='hs-varop'>&amp;&amp;</span>
<a name="line-202"></a>                         <span class='hs-varid'>numberAttrName</span> <span class='hs-varop'>`</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>member</span><span class='hs-varop'>`</span> <span class='hs-varid'>mandatoryAttrMap</span> <span class='hs-varid'>d</span>
<a name="line-203"></a>                       <span class='hs-keyglyph'>]</span>
<a name="line-204"></a>    <span class='hs-varid'>buildOpParts</span> <span class='hs-keyglyph'>=</span>
<a name="line-205"></a>        <span class='hs-str'>"opDef"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>quotedText</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-conop'>:</span>
<a name="line-206"></a>        <span class='hs-comment'>-- Renders tensor arguments.</span>
<a name="line-207"></a>        <span class='hs-keyglyph'>[</span> <span class='hs-str'>"&amp; opAttr"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>quotedText</span> <span class='hs-varid'>tfName</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-208"></a>          <span class='hs-str'>".~ tensorType (undefined ::"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>hsName</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-str'>")"</span>
<a name="line-209"></a>        <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>tfName</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>hsName</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varid'>typeMap</span>
<a name="line-210"></a>        <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span>
<a name="line-211"></a>        <span class='hs-comment'>-- Renders mandatory attributes as function parameters.</span>
<a name="line-212"></a>        <span class='hs-keyglyph'>[</span> <span class='hs-str'>"&amp; opAttr"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>dquotes</span> <span class='hs-varid'>tfName</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>".~"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>hsName</span>
<a name="line-213"></a>        <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>tfName</span><span class='hs-layout'>,</span> <span class='hs-varid'>hsName</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>mandatoryAttrs</span>
<a name="line-214"></a>        <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span>
<a name="line-215"></a>        <span class='hs-comment'>-- Renders sizes of tensor list types having number_attr.</span>
<a name="line-216"></a>        <span class='hs-keyglyph'>[</span> <span class='hs-str'>"&amp; opAttr"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>quotedText</span> <span class='hs-varid'>nAttr</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>".~"</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-217"></a>          <span class='hs-str'>"(fromIntegral (length"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-layout'>(</span><span class='hs-varid'>head</span> <span class='hs-varid'>tensorNames</span><span class='hs-layout'>)</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-str'>") :: Int64)"</span>
<a name="line-218"></a>        <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>nAttr</span><span class='hs-layout'>,</span> <span class='hs-varid'>tensorNames</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-varid'>numberAttrMap</span> <span class='hs-varid'>d</span>
<a name="line-219"></a>        <span class='hs-keyglyph'>]</span>
<a name="line-220"></a>    <span class='hs-varid'>mandatoryAttrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-varid'>strictText</span> <span class='hs-varid'>tf</span><span class='hs-layout'>,</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>hs</span><span class='hs-layout'>)</span>
<a name="line-221"></a>                     <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>tf</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>hs</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-keyword'>_</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-layout'>(</span><span class='hs-varid'>mandatoryAttrMap</span> <span class='hs-varid'>d</span><span class='hs-layout'>)</span>
<a name="line-222"></a>                     <span class='hs-keyglyph'>]</span>
<a name="line-223"></a>    <span class='hs-varid'>haddocks</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"-- |"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>multilineComment</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>summary</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>description</span><span class='hs-layout'>)</span>
<a name="line-224"></a>    <span class='hs-varid'>extras</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>enclose</span> <span class='hs-str'>"{-\n"</span> <span class='hs-str'>"\n-}"</span> <span class='hs-varop'>$</span>
<a name="line-225"></a>             <span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>pack</span> <span class='hs-varop'>$</span>
<a name="line-226"></a>             <span class='hs-varid'>showMessage</span> <span class='hs-layout'>(</span><span class='hs-layout'>(</span><span class='hs-varid'>def</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span><span class='hs-layout'>)</span>
<a name="line-227"></a>                          <span class='hs-varop'>&amp;</span> <span class='hs-varid'>inputArg</span> <span class='hs-varop'>.~</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span><span class='hs-layout'>)</span>
<a name="line-228"></a>                          <span class='hs-varop'>&amp;</span> <span class='hs-varid'>outputArg</span> <span class='hs-varop'>.~</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>outputArg</span><span class='hs-layout'>)</span>
<a name="line-229"></a>                          <span class='hs-varop'>&amp;</span> <span class='hs-varid'>attr</span> <span class='hs-varop'>.~</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attr</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-230"></a>    <span class='hs-varid'>typeMap</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>opDefTypeMap</span> <span class='hs-varid'>d</span>
<a name="line-231"></a>
<a name="line-232"></a><a name="quotedText"></a><span class='hs-comment'>-- | Makes a quoted string doc out of the given text value.</span>
<a name="line-233"></a><span class='hs-definition'>quotedText</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-234"></a><span class='hs-definition'>quotedText</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dquotes</span> <span class='hs-varop'>.</span> <span class='hs-varid'>strictText</span>
<a name="line-235"></a>
<a name="line-236"></a><a name="typeSig"></a><span class='hs-comment'>-- | typeSig renders the type signature of the given OpDef.</span>
<a name="line-237"></a><span class='hs-definition'>typeSig</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-238"></a><span class='hs-definition'>typeSig</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span>
<a name="line-239"></a>    <span class='hs-varid'>foralls</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>constraints</span> <span class='hs-varop'>&lt;+/&gt;</span>
<a name="line-240"></a>    <span class='hs-varid'>signatureFold</span> <span class='hs-layout'>(</span><span class='hs-varid'>mandatoryAttrInputs</span> <span class='hs-varop'>++</span> <span class='hs-varid'>tensorInputs</span> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>outputs</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-241"></a>  <span class='hs-keyword'>where</span>
<a name="line-242"></a>    <span class='hs-varid'>foralls</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>typeMap</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>empty</span>
<a name="line-243"></a>            <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span>
<a name="line-244"></a>              <span class='hs-str'>"forall"</span>
<a name="line-245"></a>              <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>sep</span> <span class='hs-layout'>(</span><span class='hs-varid'>refTypes</span> <span class='hs-varop'>++</span> <span class='hs-varid'>map</span> <span class='hs-layout'>(</span><span class='hs-varid'>strictText</span> <span class='hs-varop'>.</span> <span class='hs-varid'>fst</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>elems</span> <span class='hs-varid'>typeMap</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-246"></a>              <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"."</span>
<a name="line-247"></a>    <span class='hs-varid'>constraints</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>typeMap</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>empty</span>
<a name="line-248"></a>                <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span>
<a name="line-249"></a>                  <span class='hs-varid'>tuple</span> <span class='hs-layout'>(</span><span class='hs-varid'>concatMap</span>
<a name="line-250"></a>                         <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-layout'>(</span><span class='hs-varid'>t</span><span class='hs-layout'>,</span> <span class='hs-varid'>aDef</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span>
<a name="line-251"></a>                           <span class='hs-str'>"TensorType"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>t</span>
<a name="line-252"></a>                           <span class='hs-conop'>:</span> <span class='hs-varid'>maybeToList</span> <span class='hs-layout'>(</span><span class='hs-varid'>oneOfRestrictions</span> <span class='hs-varid'>aDef</span> <span class='hs-varid'>t</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-253"></a>                         <span class='hs-layout'>(</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>elems</span> <span class='hs-varid'>typeMap</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"=&gt;"</span>
<a name="line-254"></a>    <span class='hs-varid'>tensorInputs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>zipWith</span> <span class='hs-varid'>tensorArg</span> <span class='hs-varid'>refTypes</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span><span class='hs-layout'>)</span>
<a name="line-255"></a>    <span class='hs-varid'>refTypes</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>map</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>x</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-str'>"v"</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>int</span> <span class='hs-varid'>x</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>[</span><span class='hs-num'>1</span><span class='hs-keyglyph'>..</span><span class='hs-varid'>length</span> <span class='hs-layout'>(</span><span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span>
<a name="line-256"></a>    <span class='hs-varid'>tensorArg</span> <span class='hs-varid'>refType</span> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>wrapArg</span> <span class='hs-varid'>refType</span> <span class='hs-varid'>arg</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-257"></a>                            <span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-layout'>(</span><span class='hs-str'>"-- ^"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>arg</span><span class='hs-layout'>)</span>
<a name="line-258"></a>    <span class='hs-comment'>-- Argument type is a list of tensors if number_attr is set;</span>
<a name="line-259"></a>    <span class='hs-comment'>-- otherwise it's a single Tensor.</span>
<a name="line-260"></a>    <span class='hs-varid'>wrapArg</span> <span class='hs-varid'>refType</span> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'>=</span>
<a name="line-261"></a>        <span class='hs-keyword'>if</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>numberAttr</span><span class='hs-layout'>)</span> <span class='hs-keyword'>then</span> <span class='hs-varid'>typ</span> <span class='hs-keyword'>else</span> <span class='hs-varid'>brackets</span> <span class='hs-varid'>typ</span>
<a name="line-262"></a>      <span class='hs-keyword'>where</span> <span class='hs-varid'>typ</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tensorType</span> <span class='hs-varid'>refType</span> <span class='hs-varid'>arg</span>
<a name="line-263"></a>    <span class='hs-varid'>tensorType</span> <span class='hs-varid'>refType</span> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'>=</span>
<a name="line-264"></a>      <span class='hs-str'>"Tensor"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>refType</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>maybe</span> <span class='hs-varid'>directType</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>indirectType</span>
<a name="line-265"></a>      <span class='hs-keyword'>where</span>
<a name="line-266"></a>        <span class='hs-varid'>indirectType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fmap</span> <span class='hs-varid'>fst</span> <span class='hs-layout'>(</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>lookup</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>typeAttr</span><span class='hs-layout'>)</span> <span class='hs-varid'>typeMap</span><span class='hs-layout'>)</span>
<a name="line-267"></a>        <span class='hs-varid'>directType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dtTypeToDoc</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span><span class='hs-layout'>)</span>
<a name="line-268"></a>    <span class='hs-varid'>outputs</span> <span class='hs-keyglyph'>=</span>
<a name="line-269"></a>      <span class='hs-keyword'>case</span> <span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>outputArg</span> <span class='hs-keyword'>of</span>
<a name="line-270"></a>        <span class='hs-conid'>[]</span>  <span class='hs-keyglyph'>-&gt;</span> <span class='hs-str'>"ControlNode"</span>
<a name="line-271"></a>        <span class='hs-keyglyph'>[</span><span class='hs-varid'>o</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>wrappedOutput</span> <span class='hs-varid'>o</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"-- ^"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>o</span>
<a name="line-272"></a>        <span class='hs-varid'>os</span>  <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>renderTupleResult</span> <span class='hs-varid'>os</span>
<a name="line-273"></a>    <span class='hs-varid'>wrappedOutput</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>wrapArg</span> <span class='hs-str'>"Value"</span>
<a name="line-274"></a>    <span class='hs-comment'>-- Tuple result case is rendered differently to give</span>
<a name="line-275"></a>    <span class='hs-comment'>-- individual elements their own comments.</span>
<a name="line-276"></a>    <span class='hs-varid'>renderTupleResult</span> <span class='hs-varid'>os</span> <span class='hs-keyglyph'>=</span>
<a name="line-277"></a>        <span class='hs-varid'>stack</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>tuple</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>wrappedOutput</span> <span class='hs-varid'>os</span><span class='hs-layout'>)</span>
<a name="line-278"></a>                <span class='hs-layout'>,</span> <span class='hs-varid'>flatten</span> <span class='hs-varid'>commentSummary</span>
<a name="line-279"></a>                <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span> <span class='hs-varid'>map</span> <span class='hs-varid'>commentDetails</span> <span class='hs-varid'>os</span>
<a name="line-280"></a>      <span class='hs-keyword'>where</span>
<a name="line-281"></a>        <span class='hs-varid'>commentSummary</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"-- ^"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>tuple</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>bold</span> <span class='hs-layout'>(</span><span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>os</span><span class='hs-keyglyph'>]</span>
<a name="line-282"></a>        <span class='hs-varid'>commentDetails</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>=</span>
<a name="line-283"></a>          <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"--"</span>
<a name="line-284"></a>                <span class='hs-layout'>,</span> <span class='hs-str'>"-- *"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>o</span>
<a name="line-285"></a>                <span class='hs-keyglyph'>]</span>
<a name="line-286"></a>    <span class='hs-varid'>signatureFold</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>folddoc</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>x</span> <span class='hs-varid'>y</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>x</span> <span class='hs-varop'>&lt;/&gt;</span> <span class='hs-str'>"-&gt;"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span>
<a name="line-287"></a>    <span class='hs-varid'>mandatoryAttrInputs</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span>
<a name="line-288"></a>      <span class='hs-varid'>dtTypeToDoc</span> <span class='hs-varid'>dtType</span> <span class='hs-varop'>&lt;+&gt;</span>
<a name="line-289"></a>          <span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-layout'>(</span><span class='hs-str'>"-- ^"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>argComment'</span> <span class='hs-varid'>tfName</span> <span class='hs-varid'>descr</span><span class='hs-layout'>)</span>
<a name="line-290"></a>      <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>tfName</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-keyword'>_</span><span class='hs-layout'>,</span> <span class='hs-varid'>dtType</span><span class='hs-layout'>,</span> <span class='hs-varid'>descr</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-varid'>mandatoryAttrMap</span> <span class='hs-varid'>d</span><span class='hs-keyglyph'>]</span>
<a name="line-291"></a>    <span class='hs-varid'>typeMap</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>opDefTypeMap</span> <span class='hs-varid'>d</span>
<a name="line-292"></a>
<a name="line-293"></a><a name="oneOfRestrictions"></a><span class='hs-comment'>-- | Returns the type restriction for the given tensor type if the</span>
<a name="line-294"></a><span class='hs-comment'>-- set of allowed types is not empty (i.e. restricted).</span>
<a name="line-295"></a><span class='hs-definition'>oneOfRestrictions</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AttrDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Maybe</span> <span class='hs-conid'>Doc</span>
<a name="line-296"></a><span class='hs-definition'>oneOfRestrictions</span> <span class='hs-varid'>aDef</span> <span class='hs-varid'>tName</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>do</span>
<a name="line-297"></a>    <span class='hs-varid'>typs</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>onAttrType</span> <span class='hs-layout'>(</span><span class='hs-varop'>^.</span> <span class='hs-varid'>templateRestrictions</span><span class='hs-layout'>)</span> <span class='hs-varid'>aDef</span>
<a name="line-298"></a>    <span class='hs-varid'>guard</span> <span class='hs-varop'>$</span> <span class='hs-varid'>not</span> <span class='hs-varop'>$</span> <span class='hs-varid'>null</span> <span class='hs-varid'>typs</span>
<a name="line-299"></a>    <span class='hs-keyword'>let</span> <span class='hs-varid'>typeList</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>commasep</span> <span class='hs-varop'>$</span> <span class='hs-varid'>map</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span>
<a name="line-300"></a>                   <span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-varid'>fromList</span> <span class='hs-varop'>$</span>
<a name="line-301"></a>                   <span class='hs-varid'>map</span> <span class='hs-varid'>dtTypeToHaskell</span> <span class='hs-varid'>typs</span>
<a name="line-302"></a>    <span class='hs-varid'>return</span> <span class='hs-varop'>$</span> <span class='hs-str'>"OneOf"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-str'>"'"</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>brackets</span> <span class='hs-varid'>typeList</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>tName</span>
<a name="line-303"></a>
<a name="line-304"></a><a name="numberAttrMap"></a><span class='hs-comment'>-- | Identifies the attributes used as tensor cardinalities. In such</span>
<a name="line-305"></a><span class='hs-comment'>-- cases a list of tensors is supplied as an input_arg. The number of</span>
<a name="line-306"></a><span class='hs-comment'>-- such inputs is communicated as a separate opAttr.</span>
<a name="line-307"></a><span class='hs-comment'>-- The result key is TensorFlow attribute name and the value is the</span>
<a name="line-308"></a><span class='hs-comment'>-- tensor names which have number_attr set to the result key.</span>
<a name="line-309"></a><span class='hs-definition'>numberAttrMap</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonEmpty</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span><span class='hs-layout'>)</span>
<a name="line-310"></a><span class='hs-definition'>numberAttrMap</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>fromListWith</span> <span class='hs-layout'>(</span><span class='hs-conid'>Semigroup</span><span class='hs-varop'>.&lt;&gt;</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>[</span>
<a name="line-311"></a>    <span class='hs-layout'>(</span><span class='hs-varid'>nAttr</span><span class='hs-layout'>,</span> <span class='hs-varid'>replaceReservedName</span> <span class='hs-layout'>(</span><span class='hs-varid'>inp</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-conop'>:|</span> <span class='hs-conid'>[]</span><span class='hs-layout'>)</span>
<a name="line-312"></a>    <span class='hs-keyglyph'>|</span> <span class='hs-varid'>inp</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span>
<a name="line-313"></a>    <span class='hs-layout'>,</span> <span class='hs-varid'>nAttr</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>inp</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>numberAttr</span><span class='hs-keyglyph'>]</span>
<a name="line-314"></a>    <span class='hs-layout'>,</span> <span class='hs-varid'>not</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>null</span> <span class='hs-varid'>nAttr</span><span class='hs-layout'>)</span>
<a name="line-315"></a>    <span class='hs-keyglyph'>]</span>
<a name="line-316"></a>
<a name="line-317"></a><a name="argComment"></a><span class='hs-definition'>argComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'ArgDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-318"></a><span class='hs-definition'>argComment</span> <span class='hs-varid'>arg</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>argComment'</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>arg</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>description</span><span class='hs-layout'>)</span>
<a name="line-319"></a>
<a name="line-320"></a><a name="argComment'"></a><span class='hs-definition'>argComment'</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-321"></a><span class='hs-definition'>argComment'</span> <span class='hs-varid'>argName</span> <span class='hs-varid'>argDesc</span> <span class='hs-keyglyph'>=</span>
<a name="line-322"></a>    <span class='hs-varid'>bold</span> <span class='hs-varid'>argName</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>splitMultilineText</span> <span class='hs-layout'>(</span><span class='hs-str'>":"</span> <span class='hs-varop'>&lt;+&gt;</span><span class='hs-layout'>)</span> <span class='hs-varid'>argDesc</span>
<a name="line-323"></a>
<a name="line-324"></a><a name="bold"></a><span class='hs-definition'>bold</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-325"></a><span class='hs-definition'>bold</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span> <span class='hs-layout'>(</span><span class='hs-str'>"__"</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-varid'>n</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-str'>"__"</span><span class='hs-layout'>)</span>
<a name="line-326"></a>
<a name="line-327"></a><a name="opDefTypeMap"></a><span class='hs-definition'>opDefTypeMap</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span><span class='hs-layout'>,</span> <span class='hs-conid'>AttrDef</span><span class='hs-layout'>)</span>
<a name="line-328"></a><span class='hs-definition'>opDefTypeMap</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span>
<a name="line-329"></a>    <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>fromList</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>lowCase</span> <span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>attrList</span> <span class='hs-varid'>d</span><span class='hs-layout'>,</span> <span class='hs-varid'>isType</span> <span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span>
<a name="line-330"></a>
<a name="line-331"></a><a name="attrList"></a><span class='hs-definition'>attrList</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span><span class='hs-layout'>,</span> <span class='hs-conid'>AttrDef</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span>
<a name="line-332"></a><span class='hs-definition'>attrList</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span><span class='hs-layout'>,</span> <span class='hs-varid'>attrDef</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>d</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attr</span><span class='hs-keyglyph'>]</span>
<a name="line-333"></a>
<a name="line-334"></a><a name="isType"></a><span class='hs-definition'>isType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AttrDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Bool</span>
<a name="line-335"></a><span class='hs-definition'>isType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>fromMaybe</span> <span class='hs-conid'>False</span> <span class='hs-varop'>.</span> <span class='hs-varid'>onAttrType</span> <span class='hs-layout'>(</span><span class='hs-varid'>const</span> <span class='hs-conid'>True</span><span class='hs-layout'>)</span>
<a name="line-336"></a>
<a name="line-337"></a><a name="onAttrType"></a><span class='hs-comment'>-- | Applies the given function to the data type. Is this a Prism?</span>
<a name="line-338"></a><span class='hs-definition'>onAttrType</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Template</span> <span class='hs-conid'>DataType</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>AttrDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Maybe</span> <span class='hs-varid'>a</span>
<a name="line-339"></a><span class='hs-definition'>onAttrType</span> <span class='hs-varid'>f</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>x</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attrTemplate</span> <span class='hs-keyword'>of</span>
<a name="line-340"></a>    <span class='hs-conid'>AttrSingle</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrType</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Just</span> <span class='hs-layout'>(</span><span class='hs-varid'>f</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-341"></a>    <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Nothing</span>
<a name="line-342"></a>
<a name="line-343"></a><a name="mandatoryAttrMap"></a><span class='hs-comment'>-- | mandatoryAttrMap contains the attributes chosen by</span>
<a name="line-344"></a><span class='hs-comment'>-- isMandatoryAttr, excluding those which are derived from list of</span>
<a name="line-345"></a><span class='hs-comment'>-- tensor arguments. The key is the TF name of the attribute. The</span>
<a name="line-346"></a><span class='hs-comment'>-- value tuple is (haskell name, TF type, attribute comment).</span>
<a name="line-347"></a><span class='hs-definition'>mandatoryAttrMap</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-layout'>(</span><span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span><span class='hs-layout'>,</span> <span class='hs-conid'>DataType</span><span class='hs-layout'>,</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span><span class='hs-layout'>)</span>
<a name="line-348"></a><span class='hs-definition'>mandatoryAttrMap</span> <span class='hs-varid'>d</span> <span class='hs-keyglyph'>=</span>
<a name="line-349"></a>    <span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>fromList</span> <span class='hs-keyglyph'>[</span> <span class='hs-layout'>(</span><span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varid'>lowCase</span> <span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-varid'>dtType</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attrOriginal</span><span class='hs-varop'>.</span><span class='hs-varid'>description</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-350"></a>                 <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>n</span><span class='hs-layout'>,</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-varid'>attrList</span> <span class='hs-varid'>d</span>
<a name="line-351"></a>                 <span class='hs-layout'>,</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>dtType</span> <span class='hs-keyglyph'>&lt;-</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>isMandatoryAttr</span> <span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span>
<a name="line-352"></a>                 <span class='hs-comment'>-- Excludes the attributes rendered as list lengths.</span>
<a name="line-353"></a>                 <span class='hs-layout'>,</span> <span class='hs-varid'>n</span> <span class='hs-varop'>`</span><span class='hs-conid'>Map</span><span class='hs-varop'>.</span><span class='hs-varid'>notMember</span><span class='hs-varop'>`</span> <span class='hs-varid'>numberAttrMap</span> <span class='hs-varid'>d</span>
<a name="line-354"></a>                 <span class='hs-keyglyph'>]</span>
<a name="line-355"></a>
<a name="line-356"></a><a name="isMandatoryAttr"></a><span class='hs-comment'>-- | Inspects the attribute and if it is one of the implemented</span>
<a name="line-357"></a><span class='hs-comment'>-- non-tensor values lacking default, then returns Just the TF type.</span>
<a name="line-358"></a><span class='hs-definition'>isMandatoryAttr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AttrDef</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Maybe</span> <span class='hs-conid'>DataType</span>
<a name="line-359"></a><span class='hs-definition'>isMandatoryAttr</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span>
<a name="line-360"></a>   <span class='hs-keyword'>case</span> <span class='hs-varid'>x</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attrTemplate</span> <span class='hs-keyword'>of</span>
<a name="line-361"></a>     <span class='hs-conid'>AttrSingle</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrBool</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span>  <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>noDefault</span> <span class='hs-conid'>DT_BOOL</span> <span class='hs-varid'>y</span>
<a name="line-362"></a>     <span class='hs-conid'>AttrSingle</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrInt64</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>noDefault</span> <span class='hs-conid'>DT_INT64</span> <span class='hs-varid'>y</span>
<a name="line-363"></a>     <span class='hs-conid'>AttrSingle</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrFloat</span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>noDefault</span> <span class='hs-conid'>DT_FLOAT</span> <span class='hs-varid'>y</span>
<a name="line-364"></a>     <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Nothing</span>
<a name="line-365"></a>   <span class='hs-keyword'>where</span>
<a name="line-366"></a>     <span class='hs-varid'>noDefault</span> <span class='hs-varid'>typ</span> <span class='hs-varid'>y</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>maybe</span> <span class='hs-layout'>(</span><span class='hs-conid'>Just</span> <span class='hs-varid'>typ</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>const</span> <span class='hs-conid'>Nothing</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>y</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>templateDefault</span><span class='hs-layout'>)</span>
<a name="line-367"></a>
<a name="line-368"></a><a name="dtTypeToDoc"></a><span class='hs-definition'>dtTypeToDoc</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataType</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-369"></a><span class='hs-definition'>dtTypeToDoc</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>.</span> <span class='hs-varid'>dtTypeToHaskell</span>
<a name="line-370"></a>
<a name="line-371"></a><a name="dtTypeToHaskell"></a><span class='hs-comment'>-- NOTE: The cases of this function should be kept in sync with</span>
<a name="line-372"></a><span class='hs-comment'>-- TensorFlow.Types.AllTensorTypes.</span>
<a name="line-373"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataType</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span>
<a name="line-374"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_BOOL</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Bool"</span>
<a name="line-375"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_BFLOAT16</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word16"</span>
<a name="line-376"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_COMPLEX128</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"(Data.Complex.Complex Double)"</span>
<a name="line-377"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_COMPLEX64</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"(Data.Complex.Complex Float)"</span>
<a name="line-378"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_DOUBLE</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Double"</span>
<a name="line-379"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_FLOAT</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Float"</span>
<a name="line-380"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_INT16</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int16"</span>
<a name="line-381"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_INT32</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int32"</span>
<a name="line-382"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_INT64</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int64"</span>
<a name="line-383"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_INT8</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int8"</span>
<a name="line-384"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_QINT32</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int32"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-385"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_QINT8</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word8"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-386"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_QINT16</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Int.Int16"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-387"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_QUINT16</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word16"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-388"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_QUINT8</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word8"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-389"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_STRING</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.ByteString.ByteString"</span>
<a name="line-390"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_UINT16</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word16"</span>
<a name="line-391"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_HALF</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word16"</span>  <span class='hs-comment'>-- TODO(gnezdo): make unique</span>
<a name="line-392"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_UINT8</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Data.Word.Word8"</span>
<a name="line-393"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span>
<a name="line-394"></a>    <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>pack</span> <span class='hs-varop'>$</span> <span class='hs-str'>"Unsupported type in dtTypeToHaskell: "</span> <span class='hs-varop'>++</span> <span class='hs-varid'>show</span> <span class='hs-varid'>x</span>
<a name="line-395"></a>
<a name="line-396"></a><a name="haddockComment"></a><span class='hs-comment'>-- | haddockComment escapes TensorFlow doc strings into haddock.</span>
<a name="line-397"></a><span class='hs-comment'>-- TODO(gnezdo): deal with the markup.</span>
<a name="line-398"></a><span class='hs-definition'>haddockComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-399"></a><span class='hs-definition'>haddockComment</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span>
<a name="line-400"></a>
<a name="line-401"></a><a name="multilineComment"></a><span class='hs-definition'>multilineComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-402"></a><span class='hs-definition'>multilineComment</span> <span class='hs-varid'>summary'</span> <span class='hs-varid'>detail</span> <span class='hs-keyglyph'>=</span>
<a name="line-403"></a>    <span class='hs-varid'>haddockComment</span> <span class='hs-varid'>summary'</span> <span class='hs-varop'>&lt;/&gt;</span>
<a name="line-404"></a>    <span class='hs-varid'>splitMultilineText</span> <span class='hs-varid'>insertParagraphAndComment</span> <span class='hs-varid'>detail</span>
<a name="line-405"></a>  <span class='hs-keyword'>where</span> <span class='hs-varid'>insertParagraphAndComment</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"--"</span> <span class='hs-varop'>&lt;/&gt;</span> <span class='hs-str'>"--"</span> <span class='hs-varop'>&lt;+&gt;</span> <span class='hs-varid'>x</span>
<a name="line-406"></a>
<a name="line-407"></a><a name="splitMultilineText"></a><span class='hs-comment'>-- | Converts the given multi-line detail string into</span>
<a name="line-408"></a><span class='hs-comment'>-- a multi-line haddock. Applies the given lead to the</span>
<a name="line-409"></a><span class='hs-comment'>-- first line. Returns an empty document for empty detail.</span>
<a name="line-410"></a><span class='hs-definition'>splitMultilineText</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Doc</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Doc</span>
<a name="line-411"></a><span class='hs-definition'>splitMultilineText</span> <span class='hs-varid'>lead</span> <span class='hs-varid'>detail</span> <span class='hs-keyglyph'>=</span>
<a name="line-412"></a>  <span class='hs-keyword'>case</span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-varid'>lines</span> <span class='hs-varid'>detail</span> <span class='hs-keyword'>of</span>
<a name="line-413"></a>    <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>empty</span>
<a name="line-414"></a>    <span class='hs-layout'>(</span><span class='hs-varid'>l</span> <span class='hs-conop'>:</span> <span class='hs-varid'>ls</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-varid'>stack</span> <span class='hs-varop'>$</span> <span class='hs-varid'>lead</span> <span class='hs-layout'>(</span><span class='hs-varid'>haddockComment</span> <span class='hs-varid'>l</span><span class='hs-layout'>)</span>
<a name="line-415"></a>                      <span class='hs-conop'>:</span> <span class='hs-varid'>map</span> <span class='hs-layout'>(</span><span class='hs-layout'>(</span><span class='hs-str'>"--"</span> <span class='hs-varop'>&lt;+&gt;</span><span class='hs-layout'>)</span> <span class='hs-varop'>.</span> <span class='hs-varid'>haddockComment</span><span class='hs-layout'>)</span> <span class='hs-varid'>ls</span>
<a name="line-416"></a>
<a name="line-417"></a><a name="replaceReservedName"></a><span class='hs-definition'>replaceReservedName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-&gt;</span> <span class='hs-conid'>Text</span>
<a name="line-418"></a><span class='hs-definition'>replaceReservedName</span> <span class='hs-varid'>n</span>
<a name="line-419"></a>    <span class='hs-keyglyph'>|</span> <span class='hs-varid'>n</span> <span class='hs-varop'>`</span><span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-varid'>member</span><span class='hs-varop'>`</span> <span class='hs-varid'>reservedKeywords</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>n</span> <span class='hs-varop'>&lt;&gt;</span> <span class='hs-str'>"'"</span>
<a name="line-420"></a>    <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>n</span>
<a name="line-421"></a>
<a name="line-422"></a><a name="indentation"></a><span class='hs-definition'>indentation</span> <span class='hs-keyglyph'>=</span> <span class='hs-num'>4</span>
<a name="line-423"></a>
<a name="line-424"></a><a name="reservedKeywords"></a><span class='hs-definition'>reservedKeywords</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-conid'>Set</span> <span class='hs-conid'>Text</span>
<a name="line-425"></a><span class='hs-definition'>reservedKeywords</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-varid'>fromList</span> <span class='hs-varop'>$</span>
<a name="line-426"></a>    <span class='hs-comment'>-- Haskell2010 keywords:</span>
<a name="line-427"></a>    <span class='hs-comment'>-- https://www.haskell.org/onlinereport/haskell2010/haskellch2.html#x7-180002.4</span>
<a name="line-428"></a>    <span class='hs-comment'>-- We don't include keywords that are allowed to be variable names,</span>
<a name="line-429"></a>    <span class='hs-comment'>-- in particular: "as", "forall", and "hiding".</span>
<a name="line-430"></a>    <span class='hs-keyglyph'>[</span> <span class='hs-str'>"case"</span>
<a name="line-431"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"class"</span>
<a name="line-432"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"data"</span>
<a name="line-433"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"default"</span>
<a name="line-434"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"deriving"</span>
<a name="line-435"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"do"</span>
<a name="line-436"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"else"</span>
<a name="line-437"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"foreign"</span>
<a name="line-438"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"if"</span>
<a name="line-439"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"import"</span>
<a name="line-440"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"in"</span>
<a name="line-441"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"infix"</span>
<a name="line-442"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"infixl"</span>
<a name="line-443"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"infixr"</span>
<a name="line-444"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"instance"</span>
<a name="line-445"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"let"</span>
<a name="line-446"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"module"</span>
<a name="line-447"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"newtype"</span>
<a name="line-448"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"of"</span>
<a name="line-449"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"then"</span>
<a name="line-450"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"type"</span>
<a name="line-451"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"where"</span>
<a name="line-452"></a>    <span class='hs-keyglyph'>]</span>
<a name="line-453"></a>    <span class='hs-varop'>++</span>  <span class='hs-comment'>-- Nonstandard extensions</span>
<a name="line-454"></a>    <span class='hs-keyglyph'>[</span> <span class='hs-str'>"mdo"</span>   <span class='hs-comment'>-- RecursiveDo</span>
<a name="line-455"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"rec"</span>   <span class='hs-comment'>-- Arrows, RecursiveDo</span>
<a name="line-456"></a>    <span class='hs-layout'>,</span> <span class='hs-str'>"proc"</span>  <span class='hs-comment'>-- Arrows</span>
<a name="line-457"></a>    <span class='hs-keyglyph'>]</span>
</pre></body>
</html>