<?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 LambdaCase #-}</span>
<a name="line-17"></a><span class='hs-comment'>{-# LANGUAGE OverloadedStrings #-}</span>
<a name="line-18"></a><span class='hs-comment'>{-# LANGUAGE TypeFamilies #-}</span>
<a name="line-19"></a><span class='hs-comment'>{- | Rendering of TensorFlow operations as Haskell functions.
<a name="line-20"></a>
<a name="line-21"></a>The basic type signature generated for each op is:
<a name="line-22"></a>
<a name="line-23"></a>> {constraints} => {mandatory attrs} -> {input tensors} -> {output tensors}
<a name="line-24"></a>
<a name="line-25"></a>where:
<a name="line-26"></a>
<a name="line-27"></a>* @{mandatory attrs}@ is of the form @A_1 -> ... -> A_N@, where each @A@ is an
<a name="line-28"></a> op attribute that doesn't have a default and can't be inferred from other
<a name="line-29"></a> inputs.
<a name="line-30"></a>
<a name="line-31"></a>* @{constraints}@ restrict the type parameters of the input and output tensors
<a name="line-32"></a> (for example: 'TensorType' or 'OneOf').
<a name="line-33"></a>
<a name="line-34"></a>* @{input tensors}@ is of the form @T_1 -> ... -> T_N@, where each @T@ is of
<a name="line-35"></a>the form @Tensor Ref a@, @Tensor v a@ or @ResourceHandle a@ (or a list of one
<a name="line-36"></a>of those types), and @a@ is either a concrete type or a (constrained) type
<a name="line-37"></a>variable.
<a name="line-38"></a>
<a name="line-39"></a>* @{output tensors}@ is of the form @(T_1,...,T_N)@ for "pure" ops, and
<a name="line-40"></a>@Build (T_1,...,T_N)@ for "stateful" ops. An op is considered "stateful" if
<a name="line-41"></a>it takes a @Tensor Ref@ or @ResourceHandle@ as input, or if it's explicitly
<a name="line-42"></a>marked \"Stateful\" in its @REGISTER_OP@ definition. (If there are no outputs,
<a name="line-43"></a>it is either @ControlNode@ or @Build ControlNode@.)
<a name="line-44"></a>-}</span>
<a name="line-45"></a>
<a name="line-46"></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-47"></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-48"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>docOpList</span>
<a name="line-49"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>flagParser</span><span class='hs-layout'>)</span>
<a name="line-50"></a> <span class='hs-keyword'>where</span>
<a name="line-51"></a>
<a name="line-52"></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-53"></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>
<a name="line-54"></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-55"></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>
<a name="line-56"></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-57"></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'>&</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-58"></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-59"></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-60"></a> <span class='hs-layout'>(</span> <span class='hs-conid'>OpList</span>
<a name="line-61"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>OpDef</span>
<a name="line-62"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attr</span>
<a name="line-63"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>inputArg</span>
<a name="line-64"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>name</span>
<a name="line-65"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>op</span>
<a name="line-66"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>outputArg</span>
<a name="line-67"></a> <span class='hs-layout'>)</span>
<a name="line-68"></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-69"></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-70"></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'>ParsedOp</span>
<a name="line-71"></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-72"></a> <span class='hs-layout'>(</span> <span class='hs-conid'>Doc</span>
<a name="line-73"></a> <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'><></span><span class='hs-layout'>)</span>
<a name="line-74"></a> <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'><+></span><span class='hs-layout'>)</span>
<a name="line-75"></a> <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'></></span><span class='hs-layout'>)</span>
<a name="line-76"></a> <span class='hs-layout'>,</span> <span class='hs-layout'>(</span><span class='hs-varop'><+/></span><span class='hs-layout'>)</span>
<a name="line-77"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>brackets</span>
<a name="line-78"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>comma</span>
<a name="line-79"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>commasep</span>
<a name="line-80"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>dquotes</span>
<a name="line-81"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-82"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>enclose</span>
<a name="line-83"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>flatten</span>
<a name="line-84"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>folddoc</span>
<a name="line-85"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>hang</span>
<a name="line-86"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>indent</span>
<a name="line-87"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parens</span>
<a name="line-88"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>sep</span>
<a name="line-89"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>stack</span>
<a name="line-90"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>strictText</span>
<a name="line-91"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>tuple</span>
<a name="line-92"></a> <span class='hs-layout'>)</span>
<a name="line-93"></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-94"></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-95"></a>
<a name="line-96"></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-97"></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-98"></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-99"></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-100"></a> <span class='hs-layout'>}</span>
<a name="line-101"></a>
<a name="line-102"></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-103"></a><span class='hs-definition'>flagParser</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>OpGenFlags</span>
<a name="line-104"></a> <span class='hs-varop'><$></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-105"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>help</span> <span class='hs-str'>"File to write."</span>
<a name="line-106"></a> <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-107"></a> <span class='hs-varop'><*></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-108"></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-109"></a> <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-110"></a> <span class='hs-varop'><*></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-111"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>value</span> <span class='hs-str'>""</span>
<a name="line-112"></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-113"></a> <span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-114"></a>
<a name="line-115"></a>
<a name="line-116"></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'>-></span> <span class='hs-conid'>OpList</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-117"></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-118"></a> <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"{-# LANGUAGE ConstraintKinds #-}"</span>
<a name="line-119"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE DataKinds #-}"</span>
<a name="line-120"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE FlexibleInstances #-}"</span>
<a name="line-121"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE OverloadedStrings #-}"</span>
<a name="line-122"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# LANGUAGE ScopedTypeVariables #-}"</span>
<a name="line-123"></a> <span class='hs-comment'>-- Avoids reports about shadowing standard library names.</span>
<a name="line-124"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# OPTIONS_GHC -fno-warn-name-shadowing #-}"</span>
<a name="line-125"></a> <span class='hs-comment'>-- eqLengthGuard never returns false and dies instead.</span>
<a name="line-126"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"{-# OPTIONS_GHC -fno-warn-incomplete-patterns #-}"</span>
<a name="line-127"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"module"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>moduleName</span> <span class='hs-varop'><+></span> <span class='hs-str'>"where"</span>
<a name="line-128"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-129"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>imports</span>
<a name="line-130"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>empty</span>
<a name="line-131"></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'>-></span> <span class='hs-varid'>x</span> <span class='hs-varop'></></span> <span class='hs-varid'>empty</span> <span class='hs-varop'></></span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span>
<a name="line-132"></a> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>renderOpAndExtras</span> <span class='hs-varop'>$</span>
<a name="line-133"></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-134"></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-135"></a> <span class='hs-keyglyph'>]</span>
<a name="line-136"></a> <span class='hs-keyword'>where</span> <span class='hs-varid'>moduleName</span> <span class='hs-keyglyph'>=</span>
<a name="line-137"></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'><></span> <span class='hs-str'>"."</span> <span class='hs-varop'><></span> <span class='hs-varid'>camelCase</span>
<a name="line-138"></a> <span class='hs-comment'>-- Discards the optional trailing _ops_op_lib</span>
<a name="line-139"></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'>"_ops_op_lib"</span> <span class='hs-varid'>shortName</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-140"></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-141"></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-142"></a> <span class='hs-varid'>renderOpAndExtras</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderOp</span> <span class='hs-layout'>(</span><span class='hs-varid'>parseOp</span> <span class='hs-varid'>o</span><span class='hs-layout'>)</span> <span class='hs-varop'></></span> <span class='hs-varid'>extras</span> <span class='hs-varid'>o</span>
<a name="line-143"></a>
<a name="line-144"></a><a name="imports"></a><span class='hs-definition'>imports</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Doc</span>
<a name="line-145"></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-146"></a> <span class='hs-str'>"import Data.ByteString (ByteString)"</span>
<a name="line-147"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Complex (Complex)"</span>
<a name="line-148"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Int (Int8, Int16, Int32, Int64)"</span>
<a name="line-149"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import Data.Word (Word8, Word16)"</span>
<a name="line-150"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import Lens.Family2 ((.~), (&))"</span>
<a name="line-151"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Build"</span>
<a name="line-152"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.BuildOp"</span>
<a name="line-153"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Output (ResourceHandle)"</span>
<a name="line-154"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Tensor"</span>
<a name="line-155"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import TensorFlow.Types"</span>
<a name="line-156"></a> <span class='hs-keyglyph'>]</span>
<a name="line-157"></a>
<a name="line-158"></a><a name="renderHaskellName"></a><span class='hs-definition'>renderHaskellName</span><span class='hs-layout'>,</span> <span class='hs-varid'>renderTFName</span><span class='hs-layout'>,</span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-159"></a><span class='hs-definition'>renderHaskellName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>.</span> <span class='hs-varid'>unHaskellName</span> <span class='hs-varop'>.</span> <span class='hs-varid'>haskellName</span>
<a name="line-160"></a><a name="renderTFName"></a><span class='hs-definition'>renderTFName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>.</span> <span class='hs-varid'>unTFName</span> <span class='hs-varop'>.</span> <span class='hs-varid'>tfName</span>
<a name="line-161"></a><a name="renderQuotedTFName"></a><span class='hs-definition'>renderQuotedTFName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>dquotes</span> <span class='hs-varop'>.</span> <span class='hs-varid'>renderTFName</span>
<a name="line-162"></a>
<a name="line-163"></a>
<a name="line-164"></a><a name="renderOp"></a><span class='hs-comment'>-- | Generate the source code for a single op.</span>
<a name="line-165"></a><span class='hs-comment'>-- For example:</span>
<a name="line-166"></a><span class='hs-comment'>--</span>
<a name="line-167"></a><span class='hs-comment'>-- -- | {haddock comment}</span>
<a name="line-168"></a><span class='hs-comment'>-- foo :: {type sig}</span>
<a name="line-169"></a><span class='hs-comment'>-- foo attr1 attr2 input1 input2 | eqLengthGuard [...] = {function body}</span>
<a name="line-170"></a><span class='hs-definition'>renderOp</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedOp</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-171"></a><span class='hs-definition'>renderOp</span> <span class='hs-varid'>pOp</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>stack</span> <span class='hs-varop'>$</span>
<a name="line-172"></a> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>haddocks</span>
<a name="line-173"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>n</span> <span class='hs-varop'><+></span> <span class='hs-str'>"::"</span> <span class='hs-varop'><+></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'>pOp</span><span class='hs-layout'>)</span>
<a name="line-174"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>n</span> <span class='hs-varop'><+></span> <span class='hs-varid'>hang</span> <span class='hs-num'>0</span> <span class='hs-varid'>args</span> <span class='hs-varop'><+></span> <span class='hs-str'>"|"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>funcGuard</span> <span class='hs-varid'>listSizeAttrs</span>
<a name="line-175"></a> <span class='hs-varop'><+></span> <span class='hs-str'>"="</span> <span class='hs-varop'></></span> <span class='hs-comment'>-- args are indented</span>
<a name="line-176"></a> <span class='hs-comment'>-- the body needs to be indented wrt the name</span>
<a name="line-177"></a> <span class='hs-varid'>indent</span> <span class='hs-varid'>indentation</span> <span class='hs-layout'>(</span><span class='hs-varid'>functionBody</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-178"></a> <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span> <span class='hs-varid'>whereClause</span> <span class='hs-varid'>listSizeAttrs</span>
<a name="line-179"></a> <span class='hs-keyword'>where</span>
<a name="line-180"></a> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>parsedOpName</span> <span class='hs-varid'>pOp</span>
<a name="line-181"></a> <span class='hs-varid'>listSizeAttrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>inferredListSizeAttrs</span> <span class='hs-varid'>pOp</span>
<a name="line-182"></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-varid'>map</span> <span class='hs-varid'>renderHaskellName</span>
<a name="line-183"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>map</span> <span class='hs-varid'>attrName</span> <span class='hs-layout'>(</span><span class='hs-varid'>explicitInputAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-184"></a> <span class='hs-varop'>++</span> <span class='hs-varid'>map</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedInputs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-185"></a> <span class='hs-varid'>haddocks</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"-- |"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>multilineComment</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedOpSummary</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedOpDescription</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-186"></a>
<a name="line-187"></a><a name="funcGuard"></a><span class='hs-comment'>-- | A check that all lists of the given size have the given length.</span>
<a name="line-188"></a><span class='hs-comment'>-- For example:</span>
<a name="line-189"></a><span class='hs-comment'>-- eqLengthGuard [("N", [("input1", length input1), ("input2", length input2)])]</span>
<a name="line-190"></a><span class='hs-definition'>funcGuard</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Attr</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonEmpty</span> <span class='hs-conid'>Name</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-191"></a><span class='hs-definition'>funcGuard</span> <span class='hs-varid'>attrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"eqLengthGuard"</span> <span class='hs-varop'><+></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-192"></a> <span class='hs-keyword'>where</span>
<a name="line-193"></a> <span class='hs-varid'>entries</span> <span class='hs-keyglyph'>=</span>
<a name="line-194"></a> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>parens</span> <span class='hs-varop'>$</span> <span class='hs-varid'>nAttr</span> <span class='hs-varop'><></span> <span class='hs-varid'>comma</span> <span class='hs-varop'><+></span>
<a name="line-195"></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-196"></a> <span class='hs-varid'>map</span> <span class='hs-varid'>renderTensorName</span> <span class='hs-layout'>(</span><span class='hs-varid'>toList</span> <span class='hs-varop'>$</span> <span class='hs-varid'>attrInfo</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-197"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>attrs</span>
<a name="line-198"></a> <span class='hs-layout'>,</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>nAttr</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-layout'>(</span><span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-199"></a> <span class='hs-keyglyph'>]</span>
<a name="line-200"></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'>renderQuotedTFName</span> <span class='hs-varid'>x</span> <span class='hs-varop'><></span> <span class='hs-varid'>comma</span> <span class='hs-varop'><+></span>
<a name="line-201"></a> <span class='hs-str'>"length"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>x</span>
<a name="line-202"></a>
<a name="line-203"></a><a name="whereClause"></a><span class='hs-comment'>-- | Define the implicit list length attributes.</span>
<a name="line-204"></a><span class='hs-comment'>-- For example:</span>
<a name="line-205"></a><span class='hs-comment'>-- where</span>
<a name="line-206"></a><span class='hs-comment'>-- n1 = fromIntegral (length input1) :: Int64</span>
<a name="line-207"></a><span class='hs-comment'>-- n2 = fromIntegral (length input2) :: Int64</span>
<a name="line-208"></a><span class='hs-definition'>whereClause</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Attr</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonEmpty</span> <span class='hs-conid'>Name</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Doc</span><span class='hs-keyglyph'>]</span>
<a name="line-209"></a><span class='hs-definition'>whereClause</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>[]</span>
<a name="line-210"></a><span class='hs-definition'>whereClause</span> <span class='hs-keyword'>as</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>indent</span> <span class='hs-num'>2</span> <span class='hs-varop'>$</span> <span class='hs-str'>"where"</span> <span class='hs-varop'></></span> <span class='hs-varid'>indent</span> <span class='hs-num'>2</span> <span class='hs-layout'>(</span><span class='hs-varid'>stack</span> <span class='hs-varop'>$</span> <span class='hs-varid'>map</span> <span class='hs-varid'>defineLengthAttr</span> <span class='hs-keyword'>as</span><span class='hs-layout'>)</span><span class='hs-keyglyph'>]</span>
<a name="line-211"></a> <span class='hs-keyword'>where</span>
<a name="line-212"></a> <span class='hs-varid'>defineLengthAttr</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-layout'>(</span><span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-varop'><+></span> <span class='hs-str'>"="</span>
<a name="line-213"></a> <span class='hs-varop'><+></span> <span class='hs-str'>"fromIntegral (length"</span>
<a name="line-214"></a> <span class='hs-varop'><+></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-layout'>(</span><span class='hs-conid'>NE</span><span class='hs-varop'>.</span><span class='hs-varid'>head</span> <span class='hs-varop'>$</span> <span class='hs-varid'>attrInfo</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-215"></a> <span class='hs-varop'><></span> <span class='hs-str'>") :: Int64"</span>
<a name="line-216"></a>
<a name="line-217"></a><a name="functionBody"></a><span class='hs-definition'>functionBody</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedOp</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-218"></a><span class='hs-definition'>functionBody</span> <span class='hs-varid'>pOp</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>buildFunction</span> <span class='hs-varop'><+></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-219"></a> <span class='hs-varop'></></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-220"></a> <span class='hs-keyword'>where</span>
<a name="line-221"></a> <span class='hs-varid'>buildFunction</span>
<a name="line-222"></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-223"></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'><+></span>
<a name="line-224"></a> <span class='hs-varid'>brackets</span> <span class='hs-layout'>(</span><span class='hs-varid'>commasep</span> <span class='hs-varop'>$</span>
<a name="line-225"></a> <span class='hs-varid'>map</span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>outputListsSizes</span><span class='hs-layout'>)</span>
<a name="line-226"></a> <span class='hs-varid'>outputListsSizes</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span> <span class='hs-varid'>a</span>
<a name="line-227"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ParsedArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedArgCase</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ListArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>argLength</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>a</span> <span class='hs-layout'>}</span> <span class='hs-layout'>}</span>
<a name="line-228"></a> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>parsedOutputs</span> <span class='hs-varid'>pOp</span><span class='hs-keyglyph'>]</span>
<a name="line-229"></a> <span class='hs-varid'>buildOpParts</span> <span class='hs-keyglyph'>=</span>
<a name="line-230"></a> <span class='hs-str'>"opDef"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedOpName</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span> <span class='hs-conop'>:</span>
<a name="line-231"></a> <span class='hs-comment'>-- Renders tensor arguments.</span>
<a name="line-232"></a> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"& opAttr"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-varid'>n</span> <span class='hs-varop'><+></span>
<a name="line-233"></a> <span class='hs-str'>".~ tensorType (undefined ::"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>n</span> <span class='hs-varop'><></span> <span class='hs-str'>")"</span>
<a name="line-234"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>,</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span>
<a name="line-235"></a> <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span>
<a name="line-236"></a> <span class='hs-comment'>-- Renders mandatory attributes as function parameters.</span>
<a name="line-237"></a> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"& opAttr"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-varid'>n</span> <span class='hs-varop'><+></span> <span class='hs-str'>".~"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>n</span>
<a name="line-238"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>explicitInputAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>,</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span>
<a name="line-239"></a> <span class='hs-keyglyph'>]</span> <span class='hs-varop'>++</span>
<a name="line-240"></a> <span class='hs-comment'>-- Renders sizes of tensor list types having number_attr.</span>
<a name="line-241"></a> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"& opAttr"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderQuotedTFName</span> <span class='hs-varid'>n</span> <span class='hs-varop'><+></span> <span class='hs-str'>".~"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>n</span>
<a name="line-242"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>inferredListSizeAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>,</span> <span class='hs-keyword'>let</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span>
<a name="line-243"></a> <span class='hs-keyglyph'>]</span>
<a name="line-244"></a>
<a name="line-245"></a> <span class='hs-varid'>tensorArgs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varop'>.</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-varop'><$></span> <span class='hs-varid'>parsedInputs</span> <span class='hs-varid'>pOp</span>
<a name="line-246"></a>
<a name="line-247"></a><a name="extras"></a><span class='hs-comment'>-- | Write a comment with the inputs/outputs/attributes in proto format, for</span>
<a name="line-248"></a><span class='hs-comment'>-- debugging.</span>
<a name="line-249"></a><span class='hs-definition'>extras</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-250"></a><span class='hs-definition'>extras</span> <span class='hs-varid'>d</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-251"></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-252"></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-253"></a> <span class='hs-varop'>&</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-254"></a> <span class='hs-varop'>&</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-255"></a> <span class='hs-varop'>&</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-256"></a>
<a name="line-257"></a><a name="typeSig"></a><span class='hs-comment'>-- | The type signature for an op.</span>
<a name="line-258"></a><span class='hs-comment'>-- Of the form:</span>
<a name="line-259"></a><span class='hs-comment'>-- forall t1 t2 v1 v2 . (TensorType t1, TensorType t2)</span>
<a name="line-260"></a><span class='hs-comment'>-- => Float -> Tensor t1 v1 -> Tensor t2 v2</span>
<a name="line-261"></a><span class='hs-comment'>-- where "Float" is an explicit input attribute, "Tensor t1 v1" is an input, and</span>
<a name="line-262"></a><span class='hs-comment'>-- "Tensor t2 v2" is an output.</span>
<a name="line-263"></a><span class='hs-definition'>typeSig</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedOp</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-264"></a><span class='hs-definition'>typeSig</span> <span class='hs-varid'>pOp</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>constraints</span>
<a name="line-265"></a> <span class='hs-varop'><+/></span> <span class='hs-varid'>signatureFold</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>attrInput</span> <span class='hs-layout'>(</span><span class='hs-varid'>explicitInputAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-266"></a> <span class='hs-varop'>++</span> <span class='hs-varid'>map</span> <span class='hs-varid'>tensorArgAndComment</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedInputs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span>
<a name="line-267"></a> <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-268"></a> <span class='hs-keyword'>where</span>
<a name="line-269"></a> <span class='hs-varid'>constraints</span>
<a name="line-270"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>null</span> <span class='hs-layout'>(</span><span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>empty</span>
<a name="line-271"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"forall"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>sep</span> <span class='hs-varid'>typeParams</span> <span class='hs-varop'><+></span> <span class='hs-str'>"."</span> <span class='hs-varop'><+></span> <span class='hs-varid'>classConstraints</span> <span class='hs-varop'><+></span> <span class='hs-str'>"=>"</span>
<a name="line-272"></a> <span class='hs-varid'>typeParams</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>strictText</span> <span class='hs-varid'>v</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>k</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>parsedInputs</span> <span class='hs-varid'>pOp</span> <span class='hs-varop'>++</span> <span class='hs-varid'>parsedOutputs</span> <span class='hs-varid'>pOp</span><span class='hs-layout'>,</span>
<a name="line-273"></a> <span class='hs-conid'>ArgTensorEither</span> <span class='hs-varid'>v</span> <span class='hs-keyglyph'><-</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>parsedArgKind</span> <span class='hs-varid'>k</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span>
<a name="line-274"></a> <span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>renderHaskellName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-varid'>pOp</span><span class='hs-keyglyph'>]</span>
<a name="line-275"></a> <span class='hs-varid'>classConstraints</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tuple</span> <span class='hs-varop'>$</span> <span class='hs-varid'>concatMap</span> <span class='hs-varid'>tensorArgConstraint</span>
<a name="line-276"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-varid'>pOp</span>
<a name="line-277"></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'>-></span> <span class='hs-varid'>x</span> <span class='hs-varop'></></span> <span class='hs-str'>"->"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>y</span><span class='hs-layout'>)</span>
<a name="line-278"></a> <span class='hs-varid'>attrInput</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderAttrType</span> <span class='hs-layout'>(</span><span class='hs-varid'>attrInfo</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-varop'><+></span> <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'><+></span> <span class='hs-varid'>attrComment</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-279"></a> <span class='hs-varid'>renderAttrType</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrSingle</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderAttrBaseType</span> <span class='hs-varid'>a</span>
<a name="line-280"></a> <span class='hs-varid'>renderAttrType</span> <span class='hs-layout'>(</span><span class='hs-conid'>AttrList</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>brackets</span> <span class='hs-varop'>$</span> <span class='hs-varid'>renderAttrBaseType</span> <span class='hs-varid'>a</span>
<a name="line-281"></a> <span class='hs-varid'>renderAttrBaseType</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>\</span><span class='hs-keyword'>case</span>
<a name="line-282"></a> <span class='hs-conid'>AttrBytes</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"ByteString"</span>
<a name="line-283"></a> <span class='hs-conid'>AttrInt64</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Data.Int.Int64"</span>
<a name="line-284"></a> <span class='hs-conid'>AttrFloat</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Float"</span>
<a name="line-285"></a> <span class='hs-conid'>AttrBool</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Bool"</span>
<a name="line-286"></a> <span class='hs-conid'>AttrType</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"DataType"</span>
<a name="line-287"></a> <span class='hs-conid'>AttrShape</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Shape"</span>
<a name="line-288"></a> <span class='hs-conid'>AttrTensor</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"TensorProto"</span>
<a name="line-289"></a>
<a name="line-290"></a> <span class='hs-varid'>tensorArgAndComment</span> <span class='hs-varid'>t</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tensorArg</span> <span class='hs-varid'>t</span> <span class='hs-varop'><+></span> <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'><+></span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>t</span><span class='hs-layout'>)</span>
<a name="line-291"></a> <span class='hs-varid'>outputs</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>parsedOutputs</span> <span class='hs-varid'>pOp</span> <span class='hs-keyword'>of</span>
<a name="line-292"></a> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>wrapOutput</span> <span class='hs-str'>"ControlNode"</span>
<a name="line-293"></a> <span class='hs-comment'>-- TODO(judahjacobson): To improve indentation: `tensorArgAndComment a`</span>
<a name="line-294"></a> <span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>wrapOutput</span> <span class='hs-layout'>(</span><span class='hs-varid'>tensorArg</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-varop'><+></span> <span class='hs-str'>"-- ^"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>a</span>
<a name="line-295"></a> <span class='hs-keyword'>as</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>wrapOutput</span> <span class='hs-layout'>(</span><span class='hs-varid'>tuple</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-varid'>tensorArg</span> <span class='hs-keyword'>as</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varop'><+/></span> <span class='hs-varid'>resultComment</span> <span class='hs-keyword'>as</span>
<a name="line-296"></a> <span class='hs-varid'>wrapOutput</span> <span class='hs-varid'>o</span>
<a name="line-297"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>parsedOpIsMonadic</span> <span class='hs-varid'>pOp</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"Build"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>parens</span> <span class='hs-varid'>o</span>
<a name="line-298"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>o</span>
<a name="line-299"></a>
<a name="line-300"></a><a name="tensorArg"></a><span class='hs-comment'>-- | Render an op input or output.</span>
<a name="line-301"></a><span class='hs-comment'>-- For example: "Tensor Ref Int64", "Tensor v t", "ResourceHandle dtype"</span>
<a name="line-302"></a><span class='hs-definition'>tensorArg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedArg</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-303"></a><span class='hs-definition'>tensorArg</span> <span class='hs-varid'>p</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>parsedArgCase</span> <span class='hs-varid'>p</span> <span class='hs-keyword'>of</span>
<a name="line-304"></a> <span class='hs-conid'>SimpleArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>argType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>t</span> <span class='hs-layout'>}</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>tensorType</span> <span class='hs-varid'>t</span>
<a name="line-305"></a> <span class='hs-conid'>ListArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>argType</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>t</span> <span class='hs-layout'>}</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>brackets</span> <span class='hs-varop'>$</span> <span class='hs-varid'>tensorType</span> <span class='hs-varid'>t</span>
<a name="line-306"></a> <span class='hs-conid'>MixedListArg</span> <span class='hs-layout'>{</span><span class='hs-layout'>}</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"{{{tensorArg: can't handle heterogeneous lists}}}"</span>
<a name="line-307"></a> <span class='hs-keyword'>where</span>
<a name="line-308"></a> <span class='hs-varid'>tensorType</span> <span class='hs-varid'>t</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>let</span>
<a name="line-309"></a> <span class='hs-varid'>v</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>parsedArgKind</span> <span class='hs-varid'>p</span> <span class='hs-keyword'>of</span>
<a name="line-310"></a> <span class='hs-conid'>ArgTensorRef</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Tensor Ref"</span>
<a name="line-311"></a> <span class='hs-conid'>ArgTensorValue</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Tensor Value"</span>
<a name="line-312"></a> <span class='hs-conid'>ArgTensorEither</span> <span class='hs-varid'>v'</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"Tensor"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>strictText</span> <span class='hs-varid'>v'</span>
<a name="line-313"></a> <span class='hs-conid'>ArgResource</span> <span class='hs-keyglyph'>-></span> <span class='hs-str'>"ResourceHandle"</span>
<a name="line-314"></a> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyword'>case</span> <span class='hs-varid'>t</span> <span class='hs-keyword'>of</span>
<a name="line-315"></a> <span class='hs-conid'>ArgTypeFixed</span> <span class='hs-varid'>dt</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span> <span class='hs-varid'>dtTypeToHaskell</span> <span class='hs-varid'>dt</span>
<a name="line-316"></a> <span class='hs-conid'>ArgTypeAttr</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varid'>n</span>
<a name="line-317"></a> <span class='hs-keyword'>in</span> <span class='hs-varid'>v</span> <span class='hs-varop'><+></span> <span class='hs-varid'>a</span>
<a name="line-318"></a>
<a name="line-319"></a><a name="attrComment"></a><span class='hs-definition'>attrComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Attr</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-320"></a><span class='hs-definition'>attrComment</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>argComment'</span> <span class='hs-layout'>(</span><span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>attrDescription</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-321"></a>
<a name="line-322"></a><a name="argComment"></a><span class='hs-definition'>argComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedArg</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-323"></a><span class='hs-definition'>argComment</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>argComment'</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedArgName</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>parsedArgDescription</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span>
<a name="line-324"></a>
<a name="line-325"></a><a name="argComment'"></a><span class='hs-definition'>argComment'</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</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'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-326"></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-327"></a> <span class='hs-varid'>bold</span> <span class='hs-layout'>(</span><span class='hs-varid'>renderTFName</span> <span class='hs-varid'>argName</span><span class='hs-layout'>)</span> <span class='hs-varop'><></span> <span class='hs-varid'>splitMultilineText</span> <span class='hs-layout'>(</span><span class='hs-str'>":"</span> <span class='hs-varop'><+></span><span class='hs-layout'>)</span> <span class='hs-varid'>argDesc</span>
<a name="line-328"></a>
<a name="line-329"></a><a name="bold"></a><span class='hs-definition'>bold</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Doc</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-330"></a><span class='hs-definition'>bold</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"__"</span> <span class='hs-varop'><></span> <span class='hs-varid'>n</span> <span class='hs-varop'><></span> <span class='hs-str'>"__"</span>
<a name="line-331"></a>
<a name="line-332"></a><a name="resultComment"></a><span class='hs-comment'>-- | Comment for the outputs of an op.</span>
<a name="line-333"></a><span class='hs-comment'>-- For example:</span>
<a name="line-334"></a><span class='hs-comment'>-- -- ^ (__output1__, __output2__)</span>
<a name="line-335"></a><span class='hs-comment'>-- -- </span>
<a name="line-336"></a><span class='hs-comment'>-- -- * __output1__: description1</span>
<a name="line-337"></a><span class='hs-comment'>-- --</span>
<a name="line-338"></a><span class='hs-comment'>-- -- * __output2__: description2</span>
<a name="line-339"></a><span class='hs-definition'>resultComment</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>ParsedArg</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-340"></a><span class='hs-definition'>resultComment</span> <span class='hs-varid'>os</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>stack</span> <span class='hs-varop'>$</span> <span class='hs-varid'>flatten</span> <span class='hs-varid'>commentSummary</span> <span class='hs-conop'>:</span> <span class='hs-varid'>map</span> <span class='hs-varid'>commentDetails</span> <span class='hs-varid'>os</span>
<a name="line-341"></a> <span class='hs-keyword'>where</span>
<a name="line-342"></a> <span class='hs-varid'>commentSummary</span> <span class='hs-keyglyph'>=</span> <span class='hs-str'>"-- ^"</span> <span class='hs-varop'><+></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'>renderTFName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-varid'>o</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>os</span><span class='hs-keyglyph'>]</span>
<a name="line-343"></a> <span class='hs-varid'>commentDetails</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>=</span>
<a name="line-344"></a> <span class='hs-varid'>stack</span> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"--"</span>
<a name="line-345"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"-- *"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>argComment</span> <span class='hs-varid'>o</span>
<a name="line-346"></a> <span class='hs-keyglyph'>]</span>
<a name="line-347"></a>
<a name="line-348"></a><a name="tensorArgConstraint"></a><span class='hs-comment'>-- | Constraints for a given type parameter.</span>
<a name="line-349"></a><span class='hs-comment'>-- E.g.: ["TensorType t"] or ["TensorType t", "OneOf [Int64, Float] t"]</span>
<a name="line-350"></a><span class='hs-definition'>tensorArgConstraint</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Attr</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>DataType</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Doc</span><span class='hs-keyglyph'>]</span>
<a name="line-351"></a><span class='hs-definition'>tensorArgConstraint</span> <span class='hs-varid'>a</span>
<a name="line-352"></a> <span class='hs-keyglyph'>=</span> <span class='hs-layout'>(</span><span class='hs-str'>"TensorType"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>n</span>
<a name="line-353"></a> <span class='hs-conop'>:</span> <span class='hs-keyword'>if</span> <span class='hs-varid'>null</span> <span class='hs-varid'>typeList</span>
<a name="line-354"></a> <span class='hs-keyword'>then</span> <span class='hs-conid'>[]</span>
<a name="line-355"></a> <span class='hs-keyword'>else</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"OneOf"</span> <span class='hs-varop'><+></span> <span class='hs-str'>"'"</span> <span class='hs-varop'><></span> <span class='hs-varid'>brackets</span> <span class='hs-layout'>(</span><span class='hs-varid'>commasep</span> <span class='hs-varid'>typeList</span><span class='hs-layout'>)</span> <span class='hs-varop'><+></span> <span class='hs-varid'>n</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span>
<a name="line-356"></a> <span class='hs-keyword'>where</span>
<a name="line-357"></a> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>renderHaskellName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>a</span>
<a name="line-358"></a> <span class='hs-varid'>typeList</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>map</span> <span class='hs-varid'>strictText</span> <span class='hs-varop'>$</span>
<a name="line-359"></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-360"></a> <span class='hs-varid'>map</span> <span class='hs-varid'>dtTypeToHaskell</span> <span class='hs-varop'>$</span> <span class='hs-varid'>attrInfo</span> <span class='hs-varid'>a</span>
<a name="line-361"></a>
<a name="line-362"></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-363"></a><span class='hs-comment'>-- TensorFlow.Types.AllTensorTypes.</span>
<a name="line-364"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>DataType</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span>
<a name="line-365"></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-366"></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-367"></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-368"></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-369"></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-370"></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-371"></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-372"></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-373"></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-374"></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-375"></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-376"></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-377"></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-378"></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-379"></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-380"></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-381"></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-382"></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-383"></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-384"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-conid'>DT_RESOURCE</span> <span class='hs-keyglyph'>=</span>
<a name="line-385"></a> <span class='hs-varid'>error</span> <span class='hs-str'>"ResourceHandle must be prevented from getting here."</span>
<a name="line-386"></a><span class='hs-definition'>dtTypeToHaskell</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'>=</span>
<a name="line-387"></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-388"></a>
<a name="line-389"></a><a name="haddockComment"></a><span class='hs-comment'>-- | haddockComment escapes TensorFlow doc strings into haddock.</span>
<a name="line-390"></a><span class='hs-comment'>-- TODO(gnezdo): deal with the markup.</span>
<a name="line-391"></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'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-392"></a><span class='hs-definition'>haddockComment</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>strictText</span>
<a name="line-393"></a>
<a name="line-394"></a><a name="multilineComment"></a><span class='hs-comment'>-- | Generate a multiline comment. For example:</span>
<a name="line-395"></a><span class='hs-comment'>-- summary'</span>
<a name="line-396"></a><span class='hs-comment'>-- --</span>
<a name="line-397"></a><span class='hs-comment'>-- -- detail_line1</span>
<a name="line-398"></a><span class='hs-comment'>-- -- detail_line2</span>
<a name="line-399"></a><span class='hs-comment'>-- -- ...</span>
<a name="line-400"></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'>-></span> <span class='hs-conid'>Text</span><span class='hs-varop'>.</span><span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-401"></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-402"></a> <span class='hs-varid'>haddockComment</span> <span class='hs-varid'>summary'</span> <span class='hs-varop'></></span>
<a name="line-403"></a> <span class='hs-varid'>splitMultilineText</span> <span class='hs-varid'>insertParagraphAndComment</span> <span class='hs-varid'>detail</span>
<a name="line-404"></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'></></span> <span class='hs-str'>"--"</span> <span class='hs-varop'><+></span> <span class='hs-varid'>x</span>
<a name="line-405"></a>
<a name="line-406"></a><a name="splitMultilineText"></a><span class='hs-comment'>-- | Converts the given multi-line detail string into</span>
<a name="line-407"></a><span class='hs-comment'>-- a multi-line haddock. Applies the given lead to the</span>
<a name="line-408"></a><span class='hs-comment'>-- first line. Returns an empty document for empty detail.</span>
<a name="line-409"></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'>-></span> <span class='hs-conid'>Doc</span><span class='hs-layout'>)</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'>-></span> <span class='hs-conid'>Doc</span>
<a name="line-410"></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-411"></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-412"></a> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>empty</span>
<a name="line-413"></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'>-></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-414"></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'><+></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-415"></a>
<a name="line-416"></a><a name="indentation"></a><span class='hs-definition'>indentation</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span>
<a name="line-417"></a><span class='hs-definition'>indentation</span> <span class='hs-keyglyph'>=</span> <span class='hs-num'>4</span>
</pre></body>
</html>