<?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/ParsedOp.hs</title>
<link type='text/css' rel='stylesheet' href='hscolour.css' />
</head>
<body>
<pre><a name="line-1"></a><span class='hs-comment'>-- | This module helps parse the proto OpDef into a Haskell type which is more</span>
<a name="line-2"></a><span class='hs-comment'>-- descriptive of how the attributes and arguments will be used in the</span>
<a name="line-3"></a><span class='hs-comment'>-- generated code.</span>
<a name="line-4"></a><span class='hs-comment'>{-# LANGUAGE LambdaCase #-}</span>
<a name="line-5"></a><span class='hs-comment'>{-# LANGUAGE OverloadedStrings #-}</span>
<a name="line-6"></a><span class='hs-comment'>{-# LANGUAGE RecordWildCards #-}</span>
<a name="line-7"></a><span class='hs-keyword'>module</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-8"></a> <span class='hs-layout'>(</span> <span class='hs-conid'>ParsedOp</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-9"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>Name</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-10"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>HaskellName</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-11"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>TFName</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-12"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>Attr</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-13"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>AttrType</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-14"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>AttrBaseType</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-15"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>ParsedArg</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-16"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>ParsedArgCase</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-17"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>ArgType</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-18"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>ArgKind</span><span class='hs-layout'>(</span><span class='hs-keyglyph'>..</span><span class='hs-layout'>)</span>
<a name="line-19"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parseOp</span>
<a name="line-20"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>camelCase</span>
<a name="line-21"></a> <span class='hs-layout'>)</span> <span class='hs-keyword'>where</span>
<a name="line-22"></a>
<a name="line-23"></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'>toUpper</span><span class='hs-layout'>,</span> <span class='hs-varid'>toLower</span><span class='hs-layout'>)</span>
<a name="line-24"></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-layout'>(</span><span class='hs-varid'>sortBy</span><span class='hs-layout'>)</span>
<a name="line-25"></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-varid'>nonEmpty</span><span class='hs-layout'>)</span>
<a name="line-26"></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'>mapMaybe</span><span class='hs-layout'>)</span>
<a name="line-27"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Monoid</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>
<a name="line-28"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Ord</span> <span class='hs-layout'>(</span><span class='hs-varid'>comparing</span><span class='hs-layout'>)</span>
<a name="line-29"></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-30"></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-31"></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-32"></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>
<a name="line-33"></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'>AttrValue</span> <span class='hs-layout'>(</span><span class='hs-varid'>list</span><span class='hs-layout'>)</span>
<a name="line-34"></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-35"></a> <span class='hs-layout'>(</span> <span class='hs-conid'>OpDef</span>
<a name="line-36"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>OpDef'ArgDef</span>
<a name="line-37"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>OpDef'AttrDef</span>
<a name="line-38"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>allowedValues</span>
<a name="line-39"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attr</span>
<a name="line-40"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>maybe'defaultValue</span>
<a name="line-41"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>description</span>
<a name="line-42"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>name</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'>isRef</span>
<a name="line-45"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>isStateful</span>
<a name="line-46"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>outputArg</span>
<a name="line-47"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>summary</span>
<a name="line-48"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>typeListAttr</span>
<a name="line-49"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>numberAttr</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> <span class='hs-varid'>type'</span>
<a name="line-52"></a> <span class='hs-layout'>)</span>
<a name="line-53"></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-conid'>DT_RESOURCE</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-54"></a>
<a name="line-55"></a><a name="ParsedOp"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>ParsedOp</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ParsedOp</span>
<a name="line-56"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedOpName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span>
<a name="line-57"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpSummary</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span>
<a name="line-58"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpDescription</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span>
<a name="line-59"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedInputs</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>ParsedArg</span><span class='hs-keyglyph'>]</span>
<a name="line-60"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOutputs</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>ParsedArg</span><span class='hs-keyglyph'>]</span>
<a name="line-61"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>explicitInputAttrs</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Attr</span> <span class='hs-conid'>AttrType</span><span class='hs-keyglyph'>]</span>
<a name="line-62"></a> <span class='hs-comment'>-- ^ Attributes that must be set explicitly when creating the op.</span>
<a name="line-63"></a> <span class='hs-comment'>-- Associated with the type of the attribute.</span>
<a name="line-64"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-keyglyph'>::</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>
<a name="line-65"></a> <span class='hs-comment'>-- ^ Attributes that are type parameters.</span>
<a name="line-66"></a> <span class='hs-comment'>-- Associated with the list of allowed types (see: TensorFlow.Types.OneOf).</span>
<a name="line-67"></a> <span class='hs-comment'>-- If this list is empty, then any type is acceptable.</span>
<a name="line-68"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>inferredListSizeAttrs</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>
<a name="line-69"></a> <span class='hs-comment'>-- Attributes which are list sizes (ints) that are inferred automatically</span>
<a name="line-70"></a> <span class='hs-comment'>-- from one or more of the input tensors.</span>
<a name="line-71"></a> <span class='hs-comment'>-- Associated with the list of tensors whose size it describes.</span>
<a name="line-72"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpIsMonadic</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Bool</span>
<a name="line-73"></a> <span class='hs-comment'>-- ^ Whether this op is stateful or takes a stateful input. Such ops</span>
<a name="line-74"></a> <span class='hs-comment'>-- should not be CSE'd and must be monadic in our API (i.e., return a</span>
<a name="line-75"></a> <span class='hs-comment'>-- Build action).</span>
<a name="line-76"></a> <span class='hs-layout'>}</span>
<a name="line-77"></a>
<a name="line-78"></a><a name="Name"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>Name</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Name</span>
<a name="line-79"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>haskellName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>HaskellName</span>
<a name="line-80"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>tfName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>TFName</span>
<a name="line-81"></a> <span class='hs-layout'>}</span>
<a name="line-82"></a>
<a name="line-83"></a><a name="TFName"></a><span class='hs-comment'>-- | A raw name as specified in the OpDef proto.</span>
<a name="line-84"></a><a name="TFName"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>TFName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>TFName</span> <span class='hs-layout'>{</span> <span class='hs-varid'>unTFName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-layout'>}</span>
<a name="line-85"></a> <span class='hs-keyword'>deriving</span> <span class='hs-layout'>(</span><span class='hs-conid'>Eq</span><span class='hs-layout'>,</span> <span class='hs-conid'>Ord</span><span class='hs-layout'>)</span>
<a name="line-86"></a>
<a name="line-87"></a><a name="HaskellName"></a><span class='hs-comment'>-- | A name that's appropriate for a variable in a Haskell source file.</span>
<a name="line-88"></a><a name="HaskellName"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>HaskellName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>HaskellName</span> <span class='hs-layout'>{</span> <span class='hs-varid'>unHaskellName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-layout'>}</span>
<a name="line-89"></a>
<a name="line-90"></a><a name="Attr"></a><span class='hs-comment'>-- | A named attribute, associated with some information about it.</span>
<a name="line-91"></a><a name="Attr"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>Attr</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Attr</span>
<a name="line-92"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>attrName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span>
<a name="line-93"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attrDescription</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span>
<a name="line-94"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attrInfo</span> <span class='hs-keyglyph'>::</span> <span class='hs-varid'>a</span>
<a name="line-95"></a> <span class='hs-layout'>}</span>
<a name="line-96"></a>
<a name="line-97"></a><a name="AttrType"></a><span class='hs-comment'>-- | The type of an attribute.</span>
<a name="line-98"></a><a name="AttrType"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>AttrType</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrBaseType</span>
<a name="line-99"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrBaseType</span>
<a name="line-100"></a> <span class='hs-keyword'>deriving</span> <span class='hs-conid'>Eq</span>
<a name="line-101"></a>
<a name="line-102"></a><a name="AttrBaseType"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>AttrBaseType</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>AttrBytes</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrInt64</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrFloat</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrBool</span>
<a name="line-103"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrType</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrShape</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>AttrTensor</span>
<a name="line-104"></a> <span class='hs-keyword'>deriving</span> <span class='hs-conid'>Eq</span>
<a name="line-105"></a>
<a name="line-106"></a><a name="ParsedArg"></a><span class='hs-comment'>-- | An input or output argument (Tensor) for an op.</span>
<a name="line-107"></a><a name="ParsedArg"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>ParsedArg</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ParsedArg</span>
<a name="line-108"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span>
<a name="line-109"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgDescription</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span>
<a name="line-110"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ParsedArgCase</span>
<a name="line-111"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgKind</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ArgKind</span>
<a name="line-112"></a> <span class='hs-layout'>}</span>
<a name="line-113"></a>
<a name="line-114"></a><a name="ParsedArgCase"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>ParsedArgCase</span>
<a name="line-115"></a> <span class='hs-keyglyph'>=</span> <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-conid'>ArgType</span> <span class='hs-layout'>}</span>
<a name="line-116"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ListArg</span>
<a name="line-117"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>argLength</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span> <span class='hs-comment'>-- ^ The attribute that specifies this list's length.</span>
<a name="line-118"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>argType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ArgType</span>
<a name="line-119"></a> <span class='hs-layout'>}</span>
<a name="line-120"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>MixedListArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>argTypeAttr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Name</span> <span class='hs-layout'>}</span>
<a name="line-121"></a> <span class='hs-comment'>-- ^ A heterogeneous list.</span>
<a name="line-122"></a> <span class='hs-comment'>-- TODO(judahjacobson): Implement this.</span>
<a name="line-123"></a>
<a name="line-124"></a><a name="ArgType"></a><span class='hs-comment'>-- | The type of an argument.</span>
<a name="line-125"></a><a name="ArgType"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>ArgType</span>
<a name="line-126"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTypeFixed</span> <span class='hs-conid'>DataType</span> <span class='hs-comment'>-- ^ A fixed type.</span>
<a name="line-127"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ArgTypeAttr</span> <span class='hs-conid'>Name</span> <span class='hs-comment'>-- ^ A type that depends on an attribute.</span>
<a name="line-128"></a>
<a name="line-129"></a><a name="ArgKind"></a><span class='hs-comment'>-- The kind of an op input or output (not including the argument type `a`).</span>
<a name="line-130"></a><a name="ArgKind"></a><span class='hs-keyword'>data</span> <span class='hs-conid'>ArgKind</span>
<a name="line-131"></a> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTensorRef</span> <span class='hs-comment'>-- Tensor Ref a</span>
<a name="line-132"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ArgTensorValue</span> <span class='hs-comment'>-- Tensor Value a</span>
<a name="line-133"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ArgTensorEither</span> <span class='hs-conid'>Text</span> <span class='hs-comment'>-- Tensor v a; the Text is the variable `v`</span>
<a name="line-134"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ArgResource</span> <span class='hs-comment'>-- Resource a</span>
<a name="line-135"></a>
<a name="line-136"></a><a name="isRefKind"></a><span class='hs-definition'>isRefKind</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>ArgKind</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span>
<a name="line-137"></a><span class='hs-definition'>isRefKind</span> <span class='hs-conid'>ArgTensorRef</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>True</span>
<a name="line-138"></a><span class='hs-definition'>isRefKind</span> <span class='hs-conid'>ArgResource</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>True</span>
<a name="line-139"></a><span class='hs-definition'>isRefKind</span> <span class='hs-keyword'>_</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>False</span>
<a name="line-140"></a>
<a name="line-141"></a><a name="makeName"></a><span class='hs-definition'>makeName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Name</span>
<a name="line-142"></a><span class='hs-definition'>makeName</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Name</span>
<a name="line-143"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>haskellName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>HaskellName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>fixReservedName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>lowCase</span> <span class='hs-varid'>n</span>
<a name="line-144"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>tfName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>TFName</span> <span class='hs-varid'>n</span>
<a name="line-145"></a> <span class='hs-layout'>}</span>
<a name="line-146"></a>
<a name="line-147"></a><a name="fixReservedName"></a><span class='hs-comment'>-- | Change a name so it doesn't conflict with any Haskell keywords.</span>
<a name="line-148"></a><span class='hs-definition'>fixReservedName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span>
<a name="line-149"></a><span class='hs-definition'>fixReservedName</span> <span class='hs-varid'>n</span>
<a name="line-150"></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'><></span> <span class='hs-str'>"'"</span>
<a name="line-151"></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-152"></a>
<a name="line-153"></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-154"></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-155"></a> <span class='hs-comment'>-- Haskell2010 keywords:</span>
<a name="line-156"></a> <span class='hs-comment'>-- https://www.haskell.org/onlinereport/haskell2010/haskellch2.html#x7-180002.4</span>
<a name="line-157"></a> <span class='hs-comment'>-- We don't include keywords that are allowed to be variable names,</span>
<a name="line-158"></a> <span class='hs-comment'>-- in particular: "as", "forall", and "hiding".</span>
<a name="line-159"></a> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"case"</span>
<a name="line-160"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"class"</span>
<a name="line-161"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"data"</span>
<a name="line-162"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"default"</span>
<a name="line-163"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"deriving"</span>
<a name="line-164"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"do"</span>
<a name="line-165"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"else"</span>
<a name="line-166"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"foreign"</span>
<a name="line-167"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"if"</span>
<a name="line-168"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"import"</span>
<a name="line-169"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"in"</span>
<a name="line-170"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"infix"</span>
<a name="line-171"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"infixl"</span>
<a name="line-172"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"infixr"</span>
<a name="line-173"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"instance"</span>
<a name="line-174"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"let"</span>
<a name="line-175"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"module"</span>
<a name="line-176"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"newtype"</span>
<a name="line-177"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"of"</span>
<a name="line-178"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"then"</span>
<a name="line-179"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"type"</span>
<a name="line-180"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"where"</span>
<a name="line-181"></a> <span class='hs-keyglyph'>]</span>
<a name="line-182"></a> <span class='hs-varop'>++</span> <span class='hs-comment'>-- Nonstandard extensions</span>
<a name="line-183"></a> <span class='hs-keyglyph'>[</span> <span class='hs-str'>"mdo"</span> <span class='hs-comment'>-- RecursiveDo</span>
<a name="line-184"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"rec"</span> <span class='hs-comment'>-- Arrows, RecursiveDo</span>
<a name="line-185"></a> <span class='hs-layout'>,</span> <span class='hs-str'>"proc"</span> <span class='hs-comment'>-- Arrows</span>
<a name="line-186"></a> <span class='hs-keyglyph'>]</span>
<a name="line-187"></a>
<a name="line-188"></a><a name="lowCase"></a><span class='hs-comment'>-- | Lower-case the given text.</span>
<a name="line-189"></a><span class='hs-definition'>lowCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span>
<a name="line-190"></a><span class='hs-definition'>lowCase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>forceCase</span> <span class='hs-varid'>toLower</span>
<a name="line-191"></a>
<a name="line-192"></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'>-></span> <span class='hs-conid'>Char</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span>
<a name="line-193"></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'>-></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-194"></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-195"></a>
<a name="line-196"></a><a name="camelCase"></a><span class='hs-definition'>camelCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span>
<a name="line-197"></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-198"></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-199"></a>
<a name="line-200"></a><a name="upCase"></a><span class='hs-comment'>-- | Upper-case the given text.</span>
<a name="line-201"></a><span class='hs-definition'>upCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span>
<a name="line-202"></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-203"></a>
<a name="line-204"></a>
<a name="line-205"></a><a name="parseOp"></a><span class='hs-definition'>parseOp</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ParsedOp</span>
<a name="line-206"></a><span class='hs-definition'>parseOp</span> <span class='hs-varid'>o</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ParsedOp</span>
<a name="line-207"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedOpName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>makeName</span> <span class='hs-varop'>$</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>name</span>
<a name="line-208"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpSummary</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>summary</span>
<a name="line-209"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpDescription</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>description</span>
<a name="line-210"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedOpIsMonadic</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>isStateful</span>
<a name="line-211"></a> <span class='hs-varop'>||</span> <span class='hs-varid'>any</span> <span class='hs-layout'>(</span><span class='hs-varid'>isRefKind</span> <span class='hs-varop'>.</span> <span class='hs-varid'>parsedArgKind</span><span class='hs-layout'>)</span> <span class='hs-varid'>parsedInputs</span>
<a name="line-212"></a> <span class='hs-layout'>,</span> <span class='hs-keyglyph'>..</span>
<a name="line-213"></a> <span class='hs-layout'>}</span>
<a name="line-214"></a> <span class='hs-keyword'>where</span>
<a name="line-215"></a> <span class='hs-varid'>parsedInputs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>zipWith</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>a</span> <span class='hs-varid'>v</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>parseArg</span> <span class='hs-varid'>a</span> <span class='hs-layout'>(</span><span class='hs-varid'>inputTensorKind</span> <span class='hs-varid'>a</span> <span class='hs-varid'>v</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-216"></a> <span class='hs-layout'>(</span><span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>inputArg</span><span class='hs-layout'>)</span> <span class='hs-varid'>tensorKindParams</span>
<a name="line-217"></a> <span class='hs-varid'>tensorKindParams</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"v"</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-layout'>(</span><span class='hs-varid'>show</span> <span class='hs-varid'>x</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-keyglyph'>[</span><span class='hs-num'>1</span><span class='hs-keyglyph'>::</span><span class='hs-conid'>Integer</span> <span class='hs-keyglyph'>..</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span>
<a name="line-218"></a> <span class='hs-varid'>parsedOutputs</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'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>parseArg</span> <span class='hs-varid'>a</span> <span class='hs-layout'>(</span><span class='hs-varid'>outputTensorKind</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>outputArg</span><span class='hs-layout'>)</span>
<a name="line-219"></a> <span class='hs-varid'>explicitInputAttrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>sortBy</span> <span class='hs-layout'>(</span><span class='hs-varid'>comparing</span> <span class='hs-layout'>(</span><span class='hs-varid'>tfName</span> <span class='hs-varop'>.</span> <span class='hs-varid'>attrName</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span>
<a name="line-220"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>mapMaybeAttrs</span> <span class='hs-layout'>(</span><span class='hs-varid'>getExplicitInputAttr</span> <span class='hs-varid'>implicitAttrs</span><span class='hs-layout'>)</span>
<a name="line-221"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attr</span>
<a name="line-222"></a> <span class='hs-varid'>inferredTypeAttrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mapMaybeAttrs</span> <span class='hs-varid'>getInferredTypeAttr</span> <span class='hs-varop'>$</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attr</span>
<a name="line-223"></a> <span class='hs-varid'>inferredListSizeAttrs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mapMaybeAttrs</span> <span class='hs-layout'>(</span><span class='hs-varid'>getInferredListSizeAttr</span> <span class='hs-varid'>parsedInputs</span><span class='hs-layout'>)</span>
<a name="line-224"></a> <span class='hs-varop'>$</span> <span class='hs-varid'>o</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>attr</span>
<a name="line-225"></a> <span class='hs-varid'>implicitAttrs</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> <span class='hs-varid'>map</span> <span class='hs-varid'>tfName</span> <span class='hs-varop'>$</span>
<a name="line-226"></a> <span class='hs-varid'>map</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>inferredTypeAttrs</span>
<a name="line-227"></a> <span class='hs-varop'>++</span> <span class='hs-varid'>map</span> <span class='hs-varid'>attrName</span> <span class='hs-varid'>inferredListSizeAttrs</span>
<a name="line-228"></a>
<a name="line-229"></a><a name="inputTensorKind"></a><span class='hs-comment'>-- TODO(judahjacobson): Some arguments should be refs.</span>
<a name="line-230"></a><span class='hs-definition'>inputTensorKind</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'ArgDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ArgKind</span>
<a name="line-231"></a><span class='hs-definition'>inputTensorKind</span> <span class='hs-varid'>a</span> <span class='hs-varid'>v</span>
<a name="line-232"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span> <span class='hs-varop'>==</span> <span class='hs-conid'>DT_RESOURCE</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgResource</span>
<a name="line-233"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>isRef</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTensorRef</span>
<a name="line-234"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTensorEither</span> <span class='hs-varid'>v</span>
<a name="line-235"></a>
<a name="line-236"></a><a name="outputTensorKind"></a><span class='hs-definition'>outputTensorKind</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'ArgDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ArgKind</span>
<a name="line-237"></a><span class='hs-definition'>outputTensorKind</span> <span class='hs-varid'>a</span>
<a name="line-238"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span> <span class='hs-varop'>==</span> <span class='hs-conid'>DT_RESOURCE</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgResource</span>
<a name="line-239"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>isRef</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTensorRef</span>
<a name="line-240"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTensorValue</span>
<a name="line-241"></a>
<a name="line-242"></a><a name="getExplicitInputAttr"></a><span class='hs-definition'>getExplicitInputAttr</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'>TFName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>OpDef'AttrDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-conid'>AttrType</span>
<a name="line-243"></a><span class='hs-definition'>getExplicitInputAttr</span> <span class='hs-varid'>implicitAttrs</span> <span class='hs-varid'>a</span>
<a name="line-244"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>TFName</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-varop'>`</span><span class='hs-conid'>Set</span><span class='hs-varop'>.</span><span class='hs-varid'>notMember</span><span class='hs-varop'>`</span> <span class='hs-varid'>implicitAttrs</span>
<a name="line-245"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>maybe'defaultValue</span> <span class='hs-varop'>==</span> <span class='hs-conid'>Nothing</span>
<a name="line-246"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>t</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>parseAttrType</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span><span class='hs-layout'>)</span>
<a name="line-247"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>t</span> <span class='hs-varop'>`elem`</span> <span class='hs-varid'>map</span> <span class='hs-conid'>AttrSingle</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AttrBool</span><span class='hs-layout'>,</span> <span class='hs-conid'>AttrInt64</span><span class='hs-layout'>,</span> <span class='hs-conid'>AttrFloat</span><span class='hs-layout'>,</span> <span class='hs-conid'>AttrShape</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>t</span>
<a name="line-248"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span>
<a name="line-249"></a>
<a name="line-250"></a><a name="getInferredTypeAttr"></a><span class='hs-definition'>getInferredTypeAttr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'AttrDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>DataType</span><span class='hs-keyglyph'>]</span>
<a name="line-251"></a><span class='hs-definition'>getInferredTypeAttr</span> <span class='hs-varid'>a</span>
<a name="line-252"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span> <span class='hs-varop'>==</span> <span class='hs-str'>"type"</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Just</span> <span class='hs-varop'>$</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>allowedValues</span> <span class='hs-varop'>.</span> <span class='hs-varid'>list</span> <span class='hs-varop'>.</span> <span class='hs-varid'>type'</span>
<a name="line-253"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span>
<a name="line-254"></a>
<a name="line-255"></a><a name="getInferredListSizeAttr"></a><span class='hs-definition'>getInferredListSizeAttr</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'>OpDef'AttrDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-layout'>(</span><span class='hs-conid'>NonEmpty</span> <span class='hs-conid'>Name</span><span class='hs-layout'>)</span>
<a name="line-256"></a><span class='hs-definition'>getInferredListSizeAttr</span> <span class='hs-varid'>inputs</span> <span class='hs-varid'>a</span>
<a name="line-257"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span> <span class='hs-varop'>==</span> <span class='hs-str'>"int"</span>
<a name="line-258"></a> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>nonEmpty</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>t</span> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>ParsedArg</span> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>t</span>
<a name="line-259"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgCase</span>
<a name="line-260"></a> <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'>n</span> <span class='hs-layout'>}</span>
<a name="line-261"></a> <span class='hs-layout'>}</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>inputs</span>
<a name="line-262"></a> <span class='hs-layout'>,</span> <span class='hs-conid'>TFName</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-varop'>==</span> <span class='hs-varid'>tfName</span> <span class='hs-varid'>n</span><span class='hs-keyglyph'>]</span>
<a name="line-263"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span>
<a name="line-264"></a>
<a name="line-265"></a><a name="mapMaybeAttrs"></a><span class='hs-comment'>-- | Like mapMaybe, but associates the attribute name/description with the given info.</span>
<a name="line-266"></a><span class='hs-definition'>mapMaybeAttrs</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>OpDef'AttrDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>OpDef'AttrDef</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Attr</span> <span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span>
<a name="line-267"></a><span class='hs-definition'>mapMaybeAttrs</span> <span class='hs-varid'>f</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>mapMaybe</span> <span class='hs-varop'>$</span> <span class='hs-keyglyph'>\</span><span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyword'>do</span>
<a name="line-268"></a> <span class='hs-varid'>x</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>f</span> <span class='hs-varid'>a</span>
<a name="line-269"></a> <span class='hs-conid'>Just</span> <span class='hs-conid'>Attr</span>
<a name="line-270"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>attrName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>makeName</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>
<a name="line-271"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attrDescription</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>description</span>
<a name="line-272"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>attrInfo</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>x</span>
<a name="line-273"></a> <span class='hs-layout'>}</span>
<a name="line-274"></a>
<a name="line-275"></a><a name="parseArg"></a><span class='hs-definition'>parseArg</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'ArgDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ArgKind</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ParsedArg</span>
<a name="line-276"></a><span class='hs-definition'>parseArg</span> <span class='hs-varid'>a</span> <span class='hs-varid'>tKind</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ParsedArg</span>
<a name="line-277"></a> <span class='hs-layout'>{</span> <span class='hs-varid'>parsedArgName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>makeName</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>
<a name="line-278"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgDescription</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>description</span>
<a name="line-279"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgCase</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>parseArgCase</span> <span class='hs-varid'>a</span>
<a name="line-280"></a> <span class='hs-layout'>,</span> <span class='hs-varid'>parsedArgKind</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>tKind</span>
<a name="line-281"></a> <span class='hs-layout'>}</span>
<a name="line-282"></a>
<a name="line-283"></a><a name="parseArgCase"></a><span class='hs-definition'>parseArgCase</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>OpDef'ArgDef</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>ParsedArgCase</span>
<a name="line-284"></a><span class='hs-definition'>parseArgCase</span> <span class='hs-varid'>a</span>
<a name="line-285"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeAttr</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>typeListAttr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>MixedListArg</span> <span class='hs-varid'>n</span>
<a name="line-286"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeAttr</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>numberAttr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ListArg</span> <span class='hs-varid'>n</span> <span class='hs-varid'>thisArgType</span>
<a name="line-287"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>SimpleArg</span> <span class='hs-varid'>thisArgType</span>
<a name="line-288"></a> <span class='hs-keyword'>where</span>
<a name="line-289"></a> <span class='hs-varid'>thisArgType</span>
<a name="line-290"></a> <span class='hs-keyglyph'>|</span> <span class='hs-conid'>Just</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>maybeAttr</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>typeAttr</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTypeAttr</span> <span class='hs-varid'>n</span>
<a name="line-291"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span> <span class='hs-varop'>==</span> <span class='hs-conid'>DT_RESOURCE</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTypeAttr</span> <span class='hs-layout'>(</span><span class='hs-varid'>makeName</span> <span class='hs-str'>"dtype"</span><span class='hs-layout'>)</span>
<a name="line-292"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>ArgTypeFixed</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span> <span class='hs-varop'>^.</span> <span class='hs-varid'>type'</span><span class='hs-layout'>)</span>
<a name="line-293"></a> <span class='hs-varid'>maybeAttr</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Maybe</span> <span class='hs-conid'>Name</span>
<a name="line-294"></a> <span class='hs-varid'>maybeAttr</span> <span class='hs-str'>""</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Nothing</span>
<a name="line-295"></a> <span class='hs-varid'>maybeAttr</span> <span class='hs-varid'>t</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Just</span> <span class='hs-varop'>$</span> <span class='hs-varid'>makeName</span> <span class='hs-varid'>t</span>
<a name="line-296"></a>
<a name="line-297"></a><a name="parseAttrType"></a><span class='hs-definition'>parseAttrType</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Text</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrType</span>
<a name="line-298"></a><span class='hs-definition'>parseAttrType</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>\</span><span class='hs-keyword'>case</span>
<a name="line-299"></a> <span class='hs-str'>"string"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrBytes</span>
<a name="line-300"></a> <span class='hs-str'>"int"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrInt64</span>
<a name="line-301"></a> <span class='hs-str'>"float"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrFloat</span>
<a name="line-302"></a> <span class='hs-str'>"bool"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrBool</span>
<a name="line-303"></a> <span class='hs-str'>"type"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrType</span>
<a name="line-304"></a> <span class='hs-str'>"shape"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrShape</span>
<a name="line-305"></a> <span class='hs-str'>"tensor"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrSingle</span> <span class='hs-conid'>AttrTensor</span>
<a name="line-306"></a> <span class='hs-str'>"list(string)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrBytes</span>
<a name="line-307"></a> <span class='hs-str'>"list(int)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrInt64</span>
<a name="line-308"></a> <span class='hs-str'>"list(float)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrFloat</span>
<a name="line-309"></a> <span class='hs-str'>"list(bool)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrBool</span>
<a name="line-310"></a> <span class='hs-str'>"list(type)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrType</span>
<a name="line-311"></a> <span class='hs-str'>"list(shape)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrShape</span>
<a name="line-312"></a> <span class='hs-str'>"list(tensor)"</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AttrList</span> <span class='hs-conid'>AttrTensor</span>
<a name="line-313"></a> <span class='hs-varid'>t</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>error</span> <span class='hs-varop'>$</span> <span class='hs-str'>"parseAttrType: unrecognized type "</span> <span class='hs-varop'>++</span> <span class='hs-varid'>show</span> <span class='hs-varid'>t</span>
</pre></body>
</html>