<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"><html xmlns="http://www.w3.org/1999/xhtml"><head><link rel="stylesheet" type="text/css" href="style.css" /><script type="text/javascript" src="highlight.js"></script></head><body><pre><span class="hs-comment">-- Copyright 2016 TensorFlow authors.</span><span>
</span><a name="line-2"></a><span class="hs-comment">--</span><span>
</span><a name="line-3"></a><span class="hs-comment">-- Licensed under the Apache License, Version 2.0 (the &quot;License&quot;);</span><span>
</span><a name="line-4"></a><span class="hs-comment">-- you may not use this file except in compliance with the License.</span><span>
</span><a name="line-5"></a><span class="hs-comment">-- You may obtain a copy of the License at</span><span>
</span><a name="line-6"></a><span class="hs-comment">--</span><span>
</span><a name="line-7"></a><span class="hs-comment">--     http://www.apache.org/licenses/LICENSE-2.0</span><span>
</span><a name="line-8"></a><span class="hs-comment">--</span><span>
</span><a name="line-9"></a><span class="hs-comment">-- Unless required by applicable law or agreed to in writing, software</span><span>
</span><a name="line-10"></a><span class="hs-comment">-- distributed under the License is distributed on an &quot;AS IS&quot; BASIS,</span><span>
</span><a name="line-11"></a><span class="hs-comment">-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span><span>
</span><a name="line-12"></a><span class="hs-comment">-- See the License for the specific language governing permissions and</span><span>
</span><a name="line-13"></a><span class="hs-comment">-- limitations under the License.</span><span>
</span><a name="line-14"></a><span>
</span><a name="line-15"></a><span class="hs-comment">-- | This module contains definitions for some built-in TensorFlow operations.</span><span>
</span><a name="line-16"></a><span class="hs-comment">--</span><span>
</span><a name="line-17"></a><span class="hs-comment">-- Note that certain, &quot;stateful&quot; ops like 'variable' and 'assign' return a</span><span>
</span><a name="line-18"></a><span class="hs-comment">-- 'Build' action (e.g., @Build (Tensor Ref a)@ instead of a pure value; the</span><span>
</span><a name="line-19"></a><span class="hs-comment">-- returned 'Tensor's are always rendered in the current 'Build' context.  This</span><span>
</span><a name="line-20"></a><span class="hs-comment">-- approach helps us avoid problems with inlining or common subexpression</span><span>
</span><a name="line-21"></a><span class="hs-comment">-- elimination, by writing</span><span>
</span><a name="line-22"></a><span class="hs-comment">--</span><span>
</span><a name="line-23"></a><span class="hs-comment">-- &gt; do</span><span>
</span><a name="line-24"></a><span class="hs-comment">-- &gt;     v &lt;- variable []</span><span>
</span><a name="line-25"></a><span class="hs-comment">-- &gt;     w &lt;- assign v 3</span><span>
</span><a name="line-26"></a><span class="hs-comment">-- &gt;     render $ w * w</span><span>
</span><a name="line-27"></a><span class="hs-comment">--</span><span>
</span><a name="line-28"></a><span class="hs-comment">-- instead of</span><span>
</span><a name="line-29"></a><span class="hs-comment">--</span><span>
</span><a name="line-30"></a><span class="hs-comment">-- &gt; let</span><span>
</span><a name="line-31"></a><span class="hs-comment">-- &gt;    v = variable []</span><span>
</span><a name="line-32"></a><span class="hs-comment">-- &gt;    w = assign v 3</span><span>
</span><a name="line-33"></a><span class="hs-comment">-- &gt; in w * w</span><span>
</span><a name="line-34"></a><span class="hs-comment">--</span><span>
</span><a name="line-35"></a><span class="hs-comment">-- since the latter could be reasonably transformed by the compiler into (or</span><span>
</span><a name="line-36"></a><span class="hs-comment">-- vice versa)</span><span>
</span><a name="line-37"></a><span class="hs-comment">--</span><span>
</span><a name="line-38"></a><span class="hs-comment">-- &gt; let</span><span>
</span><a name="line-39"></a><span class="hs-comment">-- &gt;    v = variable []</span><span>
</span><a name="line-40"></a><span class="hs-comment">-- &gt;    w = assign v 3</span><span>
</span><a name="line-41"></a><span class="hs-comment">-- &gt;    w' = assign v 3</span><span>
</span><a name="line-42"></a><span class="hs-comment">-- &gt; in w * w'</span><span>
</span><a name="line-43"></a><span class="hs-comment">--</span><span>
</span><a name="line-44"></a><span class="hs-comment">-- Ops should return a 'Build' action if their original 'OpDef' marks them as</span><span>
</span><a name="line-45"></a><span class="hs-comment">-- stateful, or if they take any Refs as input.  (This mirrors the rules that</span><span>
</span><a name="line-46"></a><span class="hs-comment">-- TensorFlow uses to avoid common subexpression elimination.)</span><span>
</span><a name="line-47"></a><span class="hs-pragma">{-# LANGUAGE ConstraintKinds #-}</span><span>
</span><a name="line-48"></a><span class="hs-pragma">{-# LANGUAGE DataKinds #-}</span><span>
</span><a name="line-49"></a><span class="hs-pragma">{-# LANGUAGE FlexibleContexts #-}</span><span>
</span><a name="line-50"></a><span class="hs-pragma">{-# LANGUAGE FlexibleInstances #-}</span><span>
</span><a name="line-51"></a><span class="hs-pragma">{-# LANGUAGE OverloadedLists #-}</span><span>
</span><a name="line-52"></a><span class="hs-pragma">{-# LANGUAGE OverloadedStrings #-}</span><span>
</span><a name="line-53"></a><span class="hs-pragma">{-# LANGUAGE RankNTypes #-}</span><span>
</span><a name="line-54"></a><span class="hs-pragma">{-# LANGUAGE ScopedTypeVariables #-}</span><span>
</span><a name="line-55"></a><span class="hs-pragma">{-# LANGUAGE TypeFamilies #-}</span><span>
</span><a name="line-56"></a><span class="hs-pragma">{-# LANGUAGE UndecidableInstances #-}</span><span>
</span><a name="line-57"></a><span class="hs-pragma">{-# OPTIONS_GHC -fno-warn-orphans #-}</span><span>
</span><a name="line-58"></a><span>
</span><a name="line-59"></a><span class="hs-keyword">module</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">Ops</span><span>
</span><a name="line-60"></a><span>    </span><span class="hs-special">(</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">add</span><span>
</span><a name="line-61"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">add'</span><span>
</span><a name="line-62"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">abs</span><span>
</span><a name="line-63"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">abs'</span><span>
</span><a name="line-64"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">addN</span><span>
</span><a name="line-65"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">addN'</span><span>
</span><a name="line-66"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">argMax</span><span>
</span><a name="line-67"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">argMax'</span><span>
</span><a name="line-68"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">assign</span><span>
</span><a name="line-69"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">assign'</span><span>
</span><a name="line-70"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">broadcastGradientArgs</span><span>
</span><a name="line-71"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">broadcastGradientArgs'</span><span>
</span><a name="line-72"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">cast</span><span>
</span><a name="line-73"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">cast'</span><span>
</span><a name="line-74"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">concat</span><span>
</span><a name="line-75"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">concat'</span><span>
</span><a name="line-76"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#constant"><span class="hs-identifier hs-var">constant</span></a><span>
</span><a name="line-77"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#constant%27"><span class="hs-identifier hs-var">constant'</span></a><span>
</span><a name="line-78"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">equal</span><span>
</span><a name="line-79"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">equal'</span><span>
</span><a name="line-80"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#expandDims"><span class="hs-identifier hs-var">expandDims</span></a><span>
</span><a name="line-81"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#expandDims%27"><span class="hs-identifier hs-var">expandDims'</span></a><span>
</span><a name="line-82"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#initializedVariable"><span class="hs-identifier hs-var">initializedVariable</span></a><span>
</span><a name="line-83"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#initializedVariable%27"><span class="hs-identifier hs-var">initializedVariable'</span></a><span>
</span><a name="line-84"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#zeroInitializedVariable"><span class="hs-identifier hs-var">zeroInitializedVariable</span></a><span>
</span><a name="line-85"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#zeroInitializedVariable%27"><span class="hs-identifier hs-var">zeroInitializedVariable'</span></a><span>
</span><a name="line-86"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">fill</span><span>
</span><a name="line-87"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">fill'</span><span>
</span><a name="line-88"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">identity</span><span>
</span><a name="line-89"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">identity'</span><span>
</span><a name="line-90"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">matMul</span><span>
</span><a name="line-91"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">matMul'</span><span>
</span><a name="line-92"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#matTranspose"><span class="hs-identifier hs-var">matTranspose</span></a><span>
</span><a name="line-93"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#matTranspose%27"><span class="hs-identifier hs-var">matTranspose'</span></a><span>
</span><a name="line-94"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mean</span><span>
</span><a name="line-95"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mean'</span><span>
</span><a name="line-96"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mul</span><span>
</span><a name="line-97"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mul'</span><span>
</span><a name="line-98"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">neg</span><span>
</span><a name="line-99"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">neg'</span><span>
</span><a name="line-100"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">oneHot</span><span>
</span><a name="line-101"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">oneHot'</span><span>
</span><a name="line-102"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">pack</span><span>
</span><a name="line-103"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">pack'</span><span>
</span><a name="line-104"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#placeholder"><span class="hs-identifier hs-var">placeholder</span></a><span>
</span><a name="line-105"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#placeholder%27"><span class="hs-identifier hs-var">placeholder'</span></a><span>
</span><a name="line-106"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range</span><span>
</span><a name="line-107"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range'</span><span>
</span><a name="line-108"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#reducedShape"><span class="hs-identifier hs-var">reducedShape</span></a><span>
</span><a name="line-109"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#reduceMean"><span class="hs-identifier hs-var">reduceMean</span></a><span>
</span><a name="line-110"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#reduceMean%27"><span class="hs-identifier hs-var">reduceMean'</span></a><span>
</span><a name="line-111"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">relu</span><span>
</span><a name="line-112"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">relu'</span><span>
</span><a name="line-113"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">reluGrad</span><span>
</span><a name="line-114"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">reluGrad'</span><span>
</span><a name="line-115"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">reshape</span><span>
</span><a name="line-116"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">reshape'</span><span>
</span><a name="line-117"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#restore"><span class="hs-identifier hs-var">restore</span></a><span>
</span><a name="line-118"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#restoreFromName"><span class="hs-identifier hs-var">restoreFromName</span></a><span>
</span><a name="line-119"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#save"><span class="hs-identifier hs-var">save</span></a><span>
</span><a name="line-120"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span>
</span><a name="line-121"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#scalar%27"><span class="hs-identifier hs-var">scalar'</span></a><span>
</span><a name="line-122"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#shape"><span class="hs-identifier hs-var">shape</span></a><span>
</span><a name="line-123"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#shape%27"><span class="hs-identifier hs-var">shape'</span></a><span>
</span><a name="line-124"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sign</span><span>
</span><a name="line-125"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sign'</span><span>
</span><a name="line-126"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">size</span><span>
</span><a name="line-127"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">size'</span><span>
</span><a name="line-128"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">softmax</span><span>
</span><a name="line-129"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">softmax'</span><span>
</span><a name="line-130"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">softmaxCrossEntropyWithLogits</span><span>
</span><a name="line-131"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">softmaxCrossEntropyWithLogits'</span><span>
</span><a name="line-132"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sparseToDense</span><span>
</span><a name="line-133"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sparseToDense'</span><span>
</span><a name="line-134"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sub</span><span>
</span><a name="line-135"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sub'</span><span>
</span><a name="line-136"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sum</span><span>
</span><a name="line-137"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sum'</span><span>
</span><a name="line-138"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#reduceSum"><span class="hs-identifier hs-var">reduceSum</span></a><span>
</span><a name="line-139"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#reduceSum%27"><span class="hs-identifier hs-var">reduceSum'</span></a><span>
</span><a name="line-140"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">transpose</span><span>
</span><a name="line-141"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">transpose'</span><span>
</span><a name="line-142"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#truncatedNormal"><span class="hs-identifier hs-var">truncatedNormal</span></a><span>
</span><a name="line-143"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#truncatedNormal%27"><span class="hs-identifier hs-var">truncatedNormal'</span></a><span>
</span><a name="line-144"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">variable</span><span>
</span><a name="line-145"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">variable'</span><span>
</span><a name="line-146"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#vector"><span class="hs-identifier hs-var">vector</span></a><span>
</span><a name="line-147"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#vector%27"><span class="hs-identifier hs-var">vector'</span></a><span>
</span><a name="line-148"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#zeros"><span class="hs-identifier hs-var">zeros</span></a><span>
</span><a name="line-149"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">zerosLike</span><span>
</span><a name="line-150"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">zerosLike'</span><span>
</span><a name="line-151"></a><span>    </span><span class="hs-special">,</span><span> </span><a href="TensorFlow.Ops.html#scalarize"><span class="hs-identifier hs-var">scalarize</span></a><span>
</span><a name="line-152"></a><span>    </span><span class="hs-special">)</span><span> </span><span class="hs-keyword">where</span><span>
</span><a name="line-153"></a><span>
</span><a name="line-154"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">ByteString</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">ByteString</span><span class="hs-special">)</span><span>
</span><a name="line-155"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">Complex</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Complex</span><span class="hs-special">)</span><span>
</span><a name="line-156"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">Int</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span class="hs-special">)</span><span>
</span><a name="line-157"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">Word</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Word16</span><span class="hs-special">)</span><span>
</span><a name="line-158"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Prelude</span><span> </span><span class="hs-keyword">hiding</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">abs</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">sum</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">concat</span><span class="hs-special">)</span><span>
</span><a name="line-159"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">ProtoLens</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">def</span><span class="hs-special">)</span><span>
</span><a name="line-160"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Data</span><span class="hs-operator">.</span><span class="hs-identifier">Text</span><span class="hs-operator">.</span><span class="hs-identifier">Encoding</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">encodeUtf8</span><span class="hs-special">)</span><span>
</span><a name="line-161"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Lens</span><span class="hs-operator">.</span><span class="hs-identifier">Family2</span><span> </span><span class="hs-special">(</span><span class="hs-special">(</span><span class="hs-operator hs-var">.~</span><span class="hs-special">)</span><span class="hs-special">,</span><span> </span><span class="hs-special">(</span><span class="hs-operator hs-var">&amp;</span><span class="hs-special">)</span><span class="hs-special">)</span><span>
</span><a name="line-162"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Text</span><span class="hs-operator">.</span><span class="hs-identifier">Printf</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">printf</span><span class="hs-special">)</span><span>
</span><a name="line-163"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">Proto</span><span class="hs-operator">.</span><span class="hs-identifier">Tensorflow</span><span class="hs-operator">.</span><span class="hs-identifier">Core</span><span class="hs-operator">.</span><span class="hs-identifier">Framework</span><span class="hs-operator">.</span><span class="hs-identifier">Tensor</span><span>
</span><a name="line-164"></a><span>    </span><span class="hs-special">(</span><span> </span><span class="hs-identifier hs-type">TensorProto</span><span>
</span><a name="line-165"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">dtype</span><span>
</span><a name="line-166"></a><span>    </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-var">tensorShape</span><span>
</span><a name="line-167"></a><span>    </span><span class="hs-special">)</span><span>
</span><a name="line-168"></a><span class="hs-keyword">import</span><span> </span><span class="hs-keyword">qualified</span><span> </span><span class="hs-identifier">Proto</span><span class="hs-operator">.</span><span class="hs-identifier">Tensorflow</span><span class="hs-operator">.</span><span class="hs-identifier">Core</span><span class="hs-operator">.</span><span class="hs-identifier">Framework</span><span class="hs-operator">.</span><span class="hs-identifier">TensorShape</span><span>
</span><a name="line-169"></a><span>  </span><span class="hs-keyword">as</span><span> </span><span class="hs-identifier">TensorShape</span><span>
</span><a name="line-170"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">Build</span><span>
</span><a name="line-171"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">BuildOp</span><span>
</span><a name="line-172"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">ControlFlow</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">group</span><span class="hs-special">)</span><span>
</span><a name="line-173"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">Tensor</span><span>
</span><a name="line-174"></a><span class="hs-keyword">import</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">Types</span><span>
</span><a name="line-175"></a><span>
</span><a name="line-176"></a><span class="hs-keyword">import</span><span> </span><span class="hs-keyword">qualified</span><span> </span><span class="hs-identifier">TensorFlow</span><span class="hs-operator">.</span><span class="hs-identifier">GenOps</span><span class="hs-operator">.</span><span class="hs-identifier">Core</span><span> </span><span class="hs-keyword">as</span><span> </span><span class="hs-identifier">CoreOps</span><span>
</span><a name="line-177"></a><span>
</span><a name="line-178"></a><span class="hs-keyword">import</span><span> </span><span class="hs-keyword">qualified</span><span> </span><span class="hs-identifier">Prelude</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">abs</span><span class="hs-special">)</span><span>
</span><a name="line-179"></a><span>
</span><a name="line-180"></a><span class="hs-comment">-- TODO: Look into hs-boot refactoring to allow mutually recursive imports.</span><span>
</span><a name="line-181"></a><span class="hs-comment">-- | Must be defined as an orphan because of the dependency order between Ops</span><span>
</span><a name="line-182"></a><span class="hs-comment">-- and Tensor.</span><span>
</span><a name="line-183"></a><span class="hs-comment">--</span><span>
</span><a name="line-184"></a><span class="hs-comment">-- The indirect constraint &quot;v ~ Value&quot; helps disambiguate types, for example in</span><span>
</span><a name="line-185"></a><span class="hs-comment">-- &quot;neg 1 :: Tensor Value Float&quot;, it helps find the type of the subexpression</span><span>
</span><a name="line-186"></a><span class="hs-comment">-- &quot;1&quot;.</span><span>
</span><a name="line-187"></a><span class="hs-keyword">instance</span><span> </span><span class="hs-special">(</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679041872"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-188"></a><span>         </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Num</span><span> </span><a href="#local-6989586621679041872"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-189"></a><span>         </span><span class="hs-special">,</span><span> </span><a href="#local-6989586621679041873"><span class="hs-identifier hs-type">v</span></a><span> </span><span class="hs-glyph">~</span><span> </span><span class="hs-identifier hs-type">Build</span><span>
</span><a name="line-190"></a><span>         </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span>
</span><a name="line-191"></a><span>                  </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679041872"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Num</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679041873"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679041872"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-keyword">where</span><span>
</span><a name="line-192"></a><span>    </span><span class="hs-special">(</span><a name="local-8214565720323790309"><span class="hs-operator">+</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">add</span><span>
</span><a name="line-193"></a><span>    </span><span class="hs-special">(</span><a name="local-8214565720323790308"><span class="hs-operator">*</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mul</span><span>
</span><a name="line-194"></a><span>    </span><span class="hs-special">(</span><span class="hs-glyph">-</span><span class="hs-special">)</span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sub</span><span>
</span><a name="line-195"></a><span>    </span><a name="local-8214565720323790307"><span class="hs-identifier">abs</span></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">abs</span><span>
</span><a name="line-196"></a><span>    </span><a name="local-3458764513820541088"><span class="hs-identifier">fromInteger</span></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><span class="hs-operator hs-var">.</span><span> </span><span class="hs-identifier hs-var">fromInteger</span><span>
</span><a name="line-197"></a><span>    </span><a name="local-8214565720323790306"><span class="hs-identifier">signum</span></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sign</span><span>
</span><a name="line-198"></a><span>    </span><a name="local-3458764513820541097"><span class="hs-identifier">negate</span></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">neg</span><span>
</span><a name="line-199"></a><span>
</span><a name="line-200"></a><span class="hs-identifier">matTranspose</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046514"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046515"><span class="hs-identifier hs-type">e</span></a><span> </span><a href="#local-6989586621679046514"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679046514"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-201"></a><a name="matTranspose"><a href="TensorFlow.Ops.html#matTranspose"><span class="hs-identifier">matTranspose</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#matTranspose%27"><span class="hs-identifier hs-var">matTranspose'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-202"></a><span>
</span><a name="line-203"></a><span class="hs-identifier">matTranspose'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046512"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046513"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679046512"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679046512"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-204"></a><a name="matTranspose%27"><a href="TensorFlow.Ops.html#matTranspose%27"><span class="hs-identifier">matTranspose'</span></a></a><span> </span><a name="local-6989586621679046516"><a href="#local-6989586621679046516"><span class="hs-identifier">params</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">flip</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">transpose'</span><span> </span><a href="#local-6989586621679046516"><span class="hs-identifier hs-var">params</span></a><span class="hs-special">)</span><span> </span><span class="hs-special">(</span><a href="TensorFlow.Ops.html#vector"><span class="hs-identifier hs-var">vector</span></a><span> </span><span class="hs-special">[</span><span class="hs-number">1</span><span class="hs-special">,</span><span> </span><span class="hs-number">0</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">]</span><span class="hs-special">)</span><span>
</span><a name="line-205"></a><span>
</span><a name="line-206"></a><span class="hs-identifier">placeholder</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046510"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046511"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046510"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Value</span><span> </span><a href="#local-6989586621679046511"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-207"></a><a name="placeholder"><a href="TensorFlow.Ops.html#placeholder"><span class="hs-identifier">placeholder</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#placeholder%27"><span class="hs-identifier hs-var">placeholder'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-208"></a><span>
</span><a name="line-209"></a><span class="hs-identifier">placeholder'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679046508"><a href="#local-6989586621679046508"><span class="hs-identifier">m</span></a></a><span> </span><a name="local-6989586621679046509"><a href="#local-6989586621679046509"><span class="hs-identifier">a</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046508"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046509"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-210"></a><span>             </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046508"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Value</span><span> </span><a href="#local-6989586621679046509"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-211"></a><a name="placeholder%27"><a href="TensorFlow.Ops.html#placeholder%27"><span class="hs-identifier">placeholder'</span></a></a><span> </span><a name="local-6989586621679046517"><a href="#local-6989586621679046517"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679046518"><a href="#local-6989586621679046518"><span class="hs-identifier">pShape</span></a></a><span>
</span><a name="line-212"></a><span>    </span><span class="hs-comment">-- Note: we don't use CoreOps.placeholder' since that op isn't stateful,</span><span>
</span><a name="line-213"></a><span>    </span><span class="hs-comment">-- and thus would be CSE'd.</span><span>
</span><a name="line-214"></a><span>    </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">build</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">buildOp</span><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">opDef</span><span> </span><span class="hs-string">&quot;Placeholder&quot;</span><span>
</span><a name="line-215"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;dtype&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-identifier hs-var">tensorType</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">undefined</span><span> </span><span class="hs-glyph">::</span><span> </span><a href="#local-6989586621679046509"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-216"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;shape&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><a href="#local-6989586621679046518"><span class="hs-identifier hs-var">pShape</span></a><span>
</span><a name="line-217"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><a href="#local-6989586621679046517"><span class="hs-identifier hs-var">params</span></a><span>
</span><a name="line-218"></a><span>
</span><a name="line-219"></a><span class="hs-comment">-- | Creates a variable initialized to the given value.</span><span>
</span><a name="line-220"></a><span class="hs-comment">-- Initialization happens next time session runs.</span><span>
</span><a name="line-221"></a><span class="hs-identifier">initializedVariable</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046505"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046506"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-222"></a><span>                    </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046507"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679046506"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046505"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679046506"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-223"></a><a name="initializedVariable"><a href="TensorFlow.Ops.html#initializedVariable"><span class="hs-identifier">initializedVariable</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#initializedVariable%27"><span class="hs-identifier hs-var">initializedVariable'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-224"></a><span>
</span><a name="line-225"></a><span class="hs-identifier">initializedVariable'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046502"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046503"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-226"></a><span>                    </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046504"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679046503"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046502"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679046503"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-227"></a><a name="initializedVariable%27"><a href="TensorFlow.Ops.html#initializedVariable%27"><span class="hs-identifier">initializedVariable'</span></a></a><span> </span><a name="local-6989586621679046613"><a href="#local-6989586621679046613"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679046614"><a href="#local-6989586621679046614"><span class="hs-identifier">initializer</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-keyword">do</span><span>
</span><a name="line-228"></a><span>    </span><a name="local-6989586621679046615"><a href="#local-6989586621679046615"><span class="hs-identifier">v</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">variable'</span><span> </span><a href="#local-6989586621679046613"><span class="hs-identifier hs-var">params</span></a><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span>  </span><span class="hs-comment">-- The shape is not known initially.</span><span>
</span><a name="line-229"></a><span>    </span><a name="local-6989586621679046616"><a href="#local-6989586621679046616"><span class="hs-identifier">i</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">assign'</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;validate_shape&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-identifier hs-var">False</span><span class="hs-special">)</span><span> </span><a href="#local-6989586621679046615"><span class="hs-identifier hs-var">v</span></a><span>
</span><a name="line-230"></a><span>                            </span><a href="#local-6989586621679046614"><span class="hs-identifier hs-var">initializer</span></a><span>
</span><a name="line-231"></a><span>    </span><span class="hs-identifier hs-var">addInitializer</span><span> </span><span class="hs-operator hs-var">=&lt;&lt;</span><span> </span><span class="hs-identifier hs-var">group</span><span> </span><a href="#local-6989586621679046616"><span class="hs-identifier hs-var">i</span></a><span>
</span><a name="line-232"></a><span>    </span><span class="hs-identifier hs-var">return</span><span> </span><a href="#local-6989586621679046615"><span class="hs-identifier hs-var">v</span></a><span>
</span><a name="line-233"></a><span>
</span><a name="line-234"></a><span class="hs-comment">-- | Creates a zero-initialized variable with the given shape.</span><span>
</span><a name="line-235"></a><span class="hs-identifier">zeroInitializedVariable</span><span>
</span><a name="line-236"></a><span>  </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046500"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046501"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Num</span><span> </span><a href="#local-6989586621679046501"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span>
</span><a name="line-237"></a><span>     </span><span class="hs-identifier hs-type">TensorFlow</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Types</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046500"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">TensorFlow</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Tensor</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679046501"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-238"></a><a name="zeroInitializedVariable"><a href="TensorFlow.Ops.html#zeroInitializedVariable"><span class="hs-identifier">zeroInitializedVariable</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#zeroInitializedVariable%27"><span class="hs-identifier hs-var">zeroInitializedVariable'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-239"></a><span>
</span><a name="line-240"></a><span class="hs-identifier">zeroInitializedVariable'</span><span>
</span><a name="line-241"></a><span>  </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046498"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046499"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Num</span><span> </span><a href="#local-6989586621679046499"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span>
</span><a name="line-242"></a><span>     </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">TensorFlow</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Types</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046498"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">TensorFlow</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Tensor</span><span class="hs-operator hs-type">.</span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679046499"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-243"></a><a name="zeroInitializedVariable%27"><a href="TensorFlow.Ops.html#zeroInitializedVariable%27"><span class="hs-identifier">zeroInitializedVariable'</span></a></a><span> </span><a name="local-6989586621679046617"><a href="#local-6989586621679046617"><span class="hs-identifier">params</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#initializedVariable%27"><span class="hs-identifier hs-var">initializedVariable'</span></a><span> </span><a href="#local-6989586621679046617"><span class="hs-identifier hs-var">params</span></a><span> </span><span class="hs-operator hs-var">.</span><span> </span><a href="TensorFlow.Ops.html#zeros"><span class="hs-identifier hs-var">zeros</span></a><span>
</span><a name="line-244"></a><span>
</span><a name="line-245"></a><span class="hs-comment">-- TODO: Support heterogeneous list of tensors.</span><span>
</span><a name="line-246"></a><span class="hs-identifier">save</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679046495"><a href="#local-6989586621679046495"><span class="hs-identifier">a</span></a></a><span> </span><a name="local-6989586621679046496"><a href="#local-6989586621679046496"><span class="hs-identifier">m</span></a></a><span> </span><a name="local-6989586621679046497"><a href="#local-6989586621679046497"><span class="hs-identifier">v</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Rendered</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046497"><span class="hs-identifier hs-type">v</span></a><span class="hs-special">)</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046496"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046495"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-247"></a><span>        </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">ByteString</span><span>    </span><span class="hs-comment">-- ^ File path.</span><span>
</span><a name="line-248"></a><span>        </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-special">[</span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679046497"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679046495"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">]</span><span>  </span><span class="hs-comment">-- ^ Tensors to save.</span><span>
</span><a name="line-249"></a><span>        </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046496"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-identifier hs-type">ControlNode</span><span>
</span><a name="line-250"></a><a name="save"><a href="TensorFlow.Ops.html#save"><span class="hs-identifier">save</span></a></a><span> </span><a name="local-6989586621679046618"><a href="#local-6989586621679046618"><span class="hs-identifier">path</span></a></a><span> </span><a name="local-6989586621679046619"><a href="#local-6989586621679046619"><span class="hs-identifier">xs</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">build</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-keyword">do</span><span>
</span><a name="line-251"></a><span>    </span><span class="hs-keyword">let</span><span> </span><a name="local-6989586621679046620"><a href="#local-6989586621679046620"><span class="hs-identifier">toByteStringTensor</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><span class="hs-operator hs-var">.</span><span> </span><span class="hs-identifier hs-var">encodeUtf8</span><span> </span><span class="hs-operator hs-var">.</span><span> </span><span class="hs-identifier hs-var">encodeOutput</span><span> </span><span class="hs-operator hs-var">.</span><span> </span><span class="hs-identifier hs-var">renderedOutput</span><span>
</span><a name="line-252"></a><span>    </span><span class="hs-keyword">let</span><span> </span><a name="local-6989586621679046621"><a href="#local-6989586621679046621"><span class="hs-identifier">names</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">fmap</span><span> </span><a href="#local-6989586621679046620"><span class="hs-identifier hs-var">toByteStringTensor</span></a><span> </span><a href="#local-6989586621679046619"><span class="hs-identifier hs-var">xs</span></a><span>
</span><a name="line-253"></a><span>    </span><span class="hs-keyword">let</span><span> </span><a name="local-6989586621679046622"><a href="#local-6989586621679046622"><span class="hs-identifier">types</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">replicate</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">length</span><span> </span><a href="#local-6989586621679046619"><span class="hs-identifier hs-var">xs</span></a><span class="hs-special">)</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">tensorType</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">undefined</span><span> </span><span class="hs-glyph">::</span><span> </span><a href="#local-6989586621679046495"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span class="hs-special">)</span><span>
</span><a name="line-254"></a><span>    </span><a name="local-6989586621679047751"><a href="#local-6989586621679047751"><span class="hs-identifier">names'</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildInputs</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">pack</span><span> </span><a href="#local-6989586621679046621"><span class="hs-identifier hs-var">names</span></a><span>
</span><a name="line-255"></a><span>    </span><a name="local-6989586621679047752"><a href="#local-6989586621679047752"><span class="hs-identifier">xs'</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildInputs</span><span> </span><a href="#local-6989586621679046619"><span class="hs-identifier hs-var">xs</span></a><span>
</span><a name="line-256"></a><span>    </span><a name="local-6989586621679047753"><a href="#local-6989586621679047753"><span class="hs-identifier">path'</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildInputs</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><a href="#local-6989586621679046618"><span class="hs-identifier hs-var">path</span></a><span>
</span><a name="line-257"></a><span>    </span><span class="hs-identifier hs-var">buildOp</span><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">opDef</span><span> </span><span class="hs-string">&quot;Save&quot;</span><span>
</span><a name="line-258"></a><span>                    </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;T&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><a href="#local-6989586621679046622"><span class="hs-identifier hs-var">types</span></a><span>
</span><a name="line-259"></a><span>                    </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opInputs</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-special">(</span><a href="#local-6989586621679047753"><span class="hs-identifier hs-var">path'</span></a><span> </span><span class="hs-operator hs-var">++</span><span> </span><a href="#local-6989586621679047751"><span class="hs-identifier hs-var">names'</span></a><span> </span><span class="hs-operator hs-var">++</span><span> </span><a href="#local-6989586621679047752"><span class="hs-identifier hs-var">xs'</span></a><span class="hs-special">)</span><span>
</span><a name="line-260"></a><span>
</span><a name="line-261"></a><span class="hs-comment">-- | Restore a tensor's value from a checkpoint file.</span><span>
</span><a name="line-262"></a><span class="hs-comment">--</span><span>
</span><a name="line-263"></a><span class="hs-comment">-- This version allows restoring from a checkpoint file that uses a different</span><span>
</span><a name="line-264"></a><span class="hs-comment">-- tensor name than the variable.</span><span>
</span><a name="line-265"></a><span class="hs-identifier">restoreFromName</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679046493"><a href="#local-6989586621679046493"><span class="hs-identifier">a</span></a></a><span> </span><a name="local-6989586621679046494"><a href="#local-6989586621679046494"><span class="hs-identifier">m</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679046494"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679046493"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-266"></a><span>                </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">ByteString</span><span>    </span><span class="hs-comment">-- ^ File path.</span><span>
</span><a name="line-267"></a><span>                </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">ByteString</span><span>    </span><span class="hs-comment">-- ^ Tensor name override.</span><span>
</span><a name="line-268"></a><span>                </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679046493"><span class="hs-identifier hs-type">a</span></a><span>  </span><span class="hs-comment">-- ^ Tensor to restore.</span><span>
</span><a name="line-269"></a><span>                </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679046494"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-identifier hs-type">ControlNode</span><span>
</span><a name="line-270"></a><a name="restoreFromName"><a href="TensorFlow.Ops.html#restoreFromName"><span class="hs-identifier">restoreFromName</span></a></a><span> </span><a name="local-6989586621679047754"><a href="#local-6989586621679047754"><span class="hs-identifier">path</span></a></a><span> </span><a name="local-6989586621679047755"><a href="#local-6989586621679047755"><span class="hs-identifier">name</span></a></a><span> </span><a name="local-6989586621679047756"><a href="#local-6989586621679047756"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">build</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-keyword">do</span><span>
</span><a name="line-271"></a><span>    </span><a name="local-6989586621679047757"><a href="#local-6989586621679047757"><span class="hs-identifier">path'</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildInputs</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><a href="#local-6989586621679047754"><span class="hs-identifier hs-var">path</span></a><span>
</span><a name="line-272"></a><span>    </span><a name="local-6989586621679047758"><a href="#local-6989586621679047758"><span class="hs-identifier">name'</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildInputs</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><a href="#local-6989586621679047755"><span class="hs-identifier hs-var">name</span></a><span>
</span><a name="line-273"></a><span>    </span><a name="local-6989586621679047759"><a href="#local-6989586621679047759"><span class="hs-identifier">restoreOp</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><span class="hs-identifier hs-var">buildOp</span><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">opDef</span><span> </span><span class="hs-string">&quot;Restore&quot;</span><span>
</span><a name="line-274"></a><span>                               </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;dt&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-identifier hs-var">tensorType</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">undefined</span><span> </span><span class="hs-glyph">::</span><span> </span><a href="#local-6989586621679046493"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-275"></a><span>                               </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">opInputs</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-special">(</span><a href="#local-6989586621679047757"><span class="hs-identifier hs-var">path'</span></a><span> </span><span class="hs-operator hs-var">++</span><span> </span><a href="#local-6989586621679047758"><span class="hs-identifier hs-var">name'</span></a><span class="hs-special">)</span><span>
</span><a name="line-276"></a><span>    </span><span class="hs-identifier hs-var">group</span><span> </span><span class="hs-operator hs-var">=&lt;&lt;</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">assign</span><span> </span><a href="#local-6989586621679047756"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-special">(</span><a href="#local-6989586621679047759"><span class="hs-identifier hs-var">restoreOp</span></a><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Value</span><span> </span><a href="#local-6989586621679046493"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-277"></a><span>
</span><a name="line-278"></a><span class="hs-comment">-- | Restore a tensor's value from a checkpoint file.</span><span>
</span><a name="line-279"></a><span class="hs-identifier">restore</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679045789"><a href="#local-6989586621679045789"><span class="hs-identifier">a</span></a></a><span> </span><a name="local-6989586621679045790"><a href="#local-6989586621679045790"><span class="hs-identifier">m</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679045790"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045789"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-280"></a><span>        </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">ByteString</span><span>    </span><span class="hs-comment">-- ^ File path.</span><span>
</span><a name="line-281"></a><span>        </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Ref</span><span> </span><a href="#local-6989586621679045789"><span class="hs-identifier hs-type">a</span></a><span>  </span><span class="hs-comment">-- ^ Tensor to restore.</span><span>
</span><a name="line-282"></a><span>        </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679045790"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-identifier hs-type">ControlNode</span><span>
</span><a name="line-283"></a><a name="restore"><a href="TensorFlow.Ops.html#restore"><span class="hs-identifier">restore</span></a></a><span> </span><a name="local-6989586621679047760"><a href="#local-6989586621679047760"><span class="hs-identifier">path</span></a></a><span> </span><a name="local-6989586621679047761"><a href="#local-6989586621679047761"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#restoreFromName"><span class="hs-identifier hs-var">restoreFromName</span></a><span> </span><a href="#local-6989586621679047760"><span class="hs-identifier hs-var">path</span></a><span> </span><a href="#local-6989586621679047762"><span class="hs-identifier hs-var">name</span></a><span> </span><a href="#local-6989586621679047761"><span class="hs-identifier hs-var">x</span></a><span>
</span><a name="line-284"></a><span>  </span><span class="hs-keyword">where</span><span>
</span><a name="line-285"></a><span>    </span><a name="local-6989586621679047762"><a href="#local-6989586621679047762"><span class="hs-identifier">name</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">encodeUtf8</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">encodeOutput</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">renderedOutput</span><span> </span><a href="#local-6989586621679047761"><span class="hs-identifier hs-var">x</span></a><span>
</span><a name="line-286"></a><span>
</span><a name="line-287"></a><span class="hs-comment">-- | Create a constant tensor.</span><span>
</span><a name="line-288"></a><span class="hs-comment">--</span><span>
</span><a name="line-289"></a><span class="hs-comment">-- The values should be in row major order, e.g.,</span><span>
</span><a name="line-290"></a><span class="hs-comment">--</span><span>
</span><a name="line-291"></a><span class="hs-comment">--   element 0:   index (0, ..., 0)</span><span>
</span><a name="line-292"></a><span class="hs-comment">--   element 1:   index (0, ..., 1)</span><span>
</span><a name="line-293"></a><span class="hs-comment">--   ...</span><span>
</span><a name="line-294"></a><span class="hs-identifier">constant</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045788"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-special">[</span><a href="#local-6989586621679045788"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">]</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045788"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-295"></a><a name="constant"><a href="TensorFlow.Ops.html#constant"><span class="hs-identifier">constant</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#constant%27"><span class="hs-identifier hs-var">constant'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-296"></a><span>
</span><a name="line-297"></a><span class="hs-identifier">constant'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679045787"><a href="#local-6989586621679045787"><span class="hs-identifier">a</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045787"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-special">[</span><a href="#local-6989586621679045787"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">]</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045787"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-298"></a><a name="constant%27"><a href="TensorFlow.Ops.html#constant%27"><span class="hs-identifier">constant'</span></a></a><span> </span><a name="local-6989586621679047763"><a href="#local-6989586621679047763"><span class="hs-identifier">params</span></a></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">Shape</span><span> </span><a name="local-6989586621679047764"><a href="#local-6989586621679047764"><span class="hs-identifier">cShape</span></a></a><span class="hs-special">)</span><span> </span><a name="local-6989586621679047765"><a href="#local-6989586621679047765"><span class="hs-identifier">values</span></a></a><span>
</span><a name="line-299"></a><span>    </span><span class="hs-glyph">|</span><span> </span><a href="#local-6989586621679047766"><span class="hs-identifier hs-var">invalidLength</span></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">error</span><span> </span><a href="#local-6989586621679047767"><span class="hs-identifier hs-var">invalidLengthMsg</span></a><span>
</span><a name="line-300"></a><span>    </span><span class="hs-glyph">|</span><span> </span><span class="hs-identifier hs-var">otherwise</span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">const'</span><span> </span><span class="hs-special">(</span><a href="#local-6989586621679047763"><span class="hs-identifier hs-var">params</span></a><span> </span><span class="hs-operator hs-var">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">opAttr</span><span> </span><span class="hs-string">&quot;value&quot;</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><a href="#local-6989586621679047768"><span class="hs-identifier hs-var">typedNode</span></a><span class="hs-special">)</span><span class="hs-special">)</span><span>
</span><a name="line-301"></a><span>  </span><span class="hs-keyword">where</span><span>
</span><a name="line-302"></a><span>    </span><a name="local-6989586621679047766"><a href="#local-6989586621679047766"><span class="hs-identifier">invalidLength</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">product</span><span> </span><a href="#local-6989586621679047764"><span class="hs-identifier hs-var">cShape</span></a><span> </span><span class="hs-operator hs-var">/=</span><span> </span><span class="hs-identifier hs-var">fromIntegral</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">length</span><span> </span><a href="#local-6989586621679047765"><span class="hs-identifier hs-var">values</span></a><span class="hs-special">)</span><span>
</span><a name="line-303"></a><span>    </span><a name="local-6989586621679047767"><a href="#local-6989586621679047767"><span class="hs-identifier">invalidLengthMsg</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">printf</span><span> </span><span class="hs-string">&quot;invalid tensor length: expected %d got %d&quot;</span><span>
</span><a name="line-304"></a><span>                              </span><span class="hs-special">(</span><span class="hs-identifier hs-var">product</span><span> </span><a href="#local-6989586621679047764"><span class="hs-identifier hs-var">cShape</span></a><span class="hs-special">)</span><span>
</span><a name="line-305"></a><span>                              </span><span class="hs-special">(</span><span class="hs-identifier hs-var">length</span><span> </span><a href="#local-6989586621679047765"><span class="hs-identifier hs-var">values</span></a><span class="hs-special">)</span><span>
</span><a name="line-306"></a><span>    </span><span class="hs-identifier">typedNode</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorProto</span><span>
</span><a name="line-307"></a><span>    </span><a name="local-6989586621679047768"><a href="#local-6989586621679047768"><span class="hs-identifier">typedNode</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">def</span><span>
</span><a name="line-308"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">dtype</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><span class="hs-identifier hs-var">tensorType</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">undefined</span><span> </span><span class="hs-glyph">::</span><span> </span><a href="#local-6989586621679045787"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-309"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">tensorShape</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">TensorShape</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">dim</span><span> </span><span class="hs-operator hs-var">.~</span><span>
</span><a name="line-310"></a><span>                      </span><span class="hs-special">[</span><span class="hs-identifier hs-var">def</span><span> </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">TensorShape</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">size</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><a href="#local-6989586621679052408"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-glyph">|</span><span> </span><a name="local-6989586621679052408"><a href="#local-6989586621679052408"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">&lt;-</span><span> </span><a href="#local-6989586621679047764"><span class="hs-identifier hs-var">cShape</span></a><span class="hs-special">]</span><span>
</span><a name="line-311"></a><span>                </span><span class="hs-operator hs-var">&amp;</span><span> </span><span class="hs-identifier hs-var">tensorVal</span><span> </span><span class="hs-operator hs-var">.~</span><span> </span><a href="#local-6989586621679047765"><span class="hs-identifier hs-var">values</span></a><span>
</span><a name="line-312"></a><span>
</span><a name="line-313"></a><span class="hs-comment">-- | Reshape a N-D tensor down to a scalar.</span><span>
</span><a name="line-314"></a><span class="hs-comment">--</span><span>
</span><a name="line-315"></a><span class="hs-comment">-- See `TensorFlow.GenOps.Core.reshape`.</span><span>
</span><a name="line-316"></a><span class="hs-identifier">scalarize</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045785"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045786"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679045785"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045785"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-317"></a><a name="scalarize"><a href="TensorFlow.Ops.html#scalarize"><span class="hs-identifier">scalarize</span></a></a><span> </span><a name="local-6989586621679052409"><a href="#local-6989586621679052409"><span class="hs-identifier">t</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">reshape</span><span> </span><a href="#local-6989586621679052409"><span class="hs-identifier hs-var">t</span></a><span> </span><span class="hs-special">(</span><a href="TensorFlow.Ops.html#vector"><span class="hs-identifier hs-var">vector</span></a><span> </span><a href="#local-6989586621679052410"><span class="hs-identifier hs-var">scalarShape</span></a><span class="hs-special">)</span><span>
</span><a name="line-318"></a><span>    </span><span class="hs-keyword">where</span><span>
</span><a name="line-319"></a><span>        </span><a name="local-6989586621679052410"><a href="#local-6989586621679052410"><span class="hs-identifier">scalarShape</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">[</span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">]</span><span>
</span><a name="line-320"></a><span>
</span><a name="line-321"></a><span class="hs-comment">-- | Sum a tensor down to a scalar</span><span>
</span><a name="line-322"></a><span class="hs-comment">-- Seee `TensorFlow.GenOps.Core.sum`</span><span>
</span><a name="line-323"></a><span class="hs-identifier">reduceSum</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span>
</span><a name="line-324"></a><span>                     </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679045783"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span>
</span><a name="line-325"></a><span>             </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045784"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679045783"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045783"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-326"></a><a name="reduceSum"><a href="TensorFlow.Ops.html#reduceSum"><span class="hs-identifier">reduceSum</span></a></a><span> </span><a name="local-6989586621679052411"><a href="#local-6989586621679052411"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sum</span><span> </span><a href="#local-6989586621679052411"><span class="hs-identifier hs-var">x</span></a><span> </span><a href="#local-6989586621679052412"><span class="hs-identifier hs-var">allAxes</span></a><span>
</span><a name="line-327"></a><span>  </span><span class="hs-keyword">where</span><span> </span><a name="local-6989586621679052412"><a href="#local-6989586621679052412"><span class="hs-identifier">allAxes</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range</span><span> </span><span class="hs-number">0</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">rank</span><span> </span><a href="#local-6989586621679052411"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">)</span><span> </span><span class="hs-number">1</span><span>
</span><a name="line-328"></a><span>
</span><a name="line-329"></a><span class="hs-identifier">reduceSum'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span>
</span><a name="line-330"></a><span>                      </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679045781"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span>
</span><a name="line-331"></a><span>              </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045782"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679045781"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045781"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-332"></a><a name="reduceSum%27"><a href="TensorFlow.Ops.html#reduceSum%27"><span class="hs-identifier">reduceSum'</span></a></a><span> </span><a name="local-6989586621679052413"><a href="#local-6989586621679052413"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679052414"><a href="#local-6989586621679052414"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">sum'</span><span> </span><a href="#local-6989586621679052413"><span class="hs-identifier hs-var">params</span></a><span> </span><a href="#local-6989586621679052414"><span class="hs-identifier hs-var">x</span></a><span> </span><a href="#local-6989586621679052415"><span class="hs-identifier hs-var">allAxes</span></a><span>
</span><a name="line-333"></a><span>  </span><span class="hs-keyword">where</span><span> </span><a name="local-6989586621679052415"><a href="#local-6989586621679052415"><span class="hs-identifier">allAxes</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range</span><span> </span><span class="hs-number">0</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">rank</span><span> </span><a href="#local-6989586621679052414"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">)</span><span> </span><span class="hs-number">1</span><span>
</span><a name="line-334"></a><span>
</span><a name="line-335"></a><span class="hs-comment">-- | Computes the mean of elements across dimensions of a tensor.</span><span>
</span><a name="line-336"></a><span class="hs-comment">-- See `TensorFlow.GenOps.Core.mean`</span><span>
</span><a name="line-337"></a><span class="hs-identifier">reduceMean</span><span>
</span><a name="line-338"></a><span>  </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045779"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-339"></a><span>     </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679045779"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-340"></a><span>     </span><span class="hs-special">)</span><span>
</span><a name="line-341"></a><span>  </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045780"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679045779"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045779"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-342"></a><a name="reduceMean"><a href="TensorFlow.Ops.html#reduceMean"><span class="hs-identifier">reduceMean</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#reduceMean%27"><span class="hs-identifier hs-var">reduceMean'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-343"></a><span>
</span><a name="line-344"></a><span class="hs-identifier">reduceMean'</span><span>
</span><a name="line-345"></a><span>  </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045777"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-346"></a><span>     </span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Complex</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679045777"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-347"></a><span>     </span><span class="hs-special">)</span><span>
</span><a name="line-348"></a><span>  </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045778"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679045777"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045777"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-349"></a><a name="reduceMean%27"><a href="TensorFlow.Ops.html#reduceMean%27"><span class="hs-identifier">reduceMean'</span></a></a><span> </span><a name="local-6989586621679052416"><a href="#local-6989586621679052416"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679052417"><a href="#local-6989586621679052417"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mean'</span><span> </span><a href="#local-6989586621679052416"><span class="hs-identifier hs-var">params</span></a><span> </span><a href="#local-6989586621679052417"><span class="hs-identifier hs-var">x</span></a><span> </span><a href="#local-6989586621679052418"><span class="hs-identifier hs-var">allAxes</span></a><span>
</span><a name="line-350"></a><span>  </span><span class="hs-keyword">where</span><span> </span><a name="local-6989586621679052418"><a href="#local-6989586621679052418"><span class="hs-identifier">allAxes</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range</span><span> </span><span class="hs-number">0</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">rank</span><span> </span><a href="#local-6989586621679052417"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">)</span><span> </span><span class="hs-number">1</span><span>
</span><a name="line-351"></a><span>
</span><a name="line-352"></a><span class="hs-comment">-- | Create a constant vector.</span><span>
</span><a name="line-353"></a><span class="hs-identifier">vector</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045776"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-special">[</span><a href="#local-6989586621679045776"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">]</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045776"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-354"></a><a name="vector"><a href="TensorFlow.Ops.html#vector"><span class="hs-identifier">vector</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#vector%27"><span class="hs-identifier hs-var">vector'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-355"></a><span>
</span><a name="line-356"></a><span class="hs-identifier">vector'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045775"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-special">[</span><a href="#local-6989586621679045775"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">]</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045775"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-357"></a><a name="vector%27"><a href="TensorFlow.Ops.html#vector%27"><span class="hs-identifier">vector'</span></a></a><span> </span><a name="local-6989586621679052419"><a href="#local-6989586621679052419"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679052420"><a href="#local-6989586621679052420"><span class="hs-identifier">xs</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#constant%27"><span class="hs-identifier hs-var">constant'</span></a><span> </span><a href="#local-6989586621679052419"><span class="hs-identifier hs-var">params</span></a><span> </span><span class="hs-special">[</span><span class="hs-identifier hs-var">fromIntegral</span><span> </span><span class="hs-operator hs-var">$</span><span> </span><span class="hs-identifier hs-var">length</span><span> </span><a href="#local-6989586621679052420"><span class="hs-identifier hs-var">xs</span></a><span class="hs-special">]</span><span> </span><a href="#local-6989586621679052420"><span class="hs-identifier hs-var">xs</span></a><span>
</span><a name="line-358"></a><span>
</span><a name="line-359"></a><span class="hs-comment">-- | Create a constant scalar.</span><span>
</span><a name="line-360"></a><span class="hs-identifier">scalar</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045774"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><a href="#local-6989586621679045774"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045774"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-361"></a><a name="scalar"><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier">scalar</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#scalar%27"><span class="hs-identifier hs-var">scalar'</span></a><span> </span><span class="hs-identifier hs-var">id</span><span>
</span><a name="line-362"></a><span>
</span><a name="line-363"></a><span class="hs-identifier">scalar'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679045773"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679045773"><span class="hs-identifier hs-type">a</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679045773"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-364"></a><a name="scalar%27"><a href="TensorFlow.Ops.html#scalar%27"><span class="hs-identifier">scalar'</span></a></a><span> </span><a name="local-6989586621679052421"><a href="#local-6989586621679052421"><span class="hs-identifier">params</span></a></a><span> </span><a name="local-6989586621679052422"><a href="#local-6989586621679052422"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#constant%27"><span class="hs-identifier hs-var">constant'</span></a><span> </span><a href="#local-6989586621679052421"><span class="hs-identifier hs-var">params</span></a><span> </span><span class="hs-special">[</span><span class="hs-special">]</span><span> </span><span class="hs-special">[</span><a href="#local-6989586621679052422"><span class="hs-identifier hs-var">x</span></a><span class="hs-special">]</span><span>
</span><a name="line-365"></a><span>
</span><a name="line-366"></a><span class="hs-comment">-- | Random tensor from the unit normal distribution with bounded values.</span><span>
</span><a name="line-367"></a><span class="hs-comment">--</span><span>
</span><a name="line-368"></a><span class="hs-comment">-- This is a type-restricted version of 'TensorFlow.GenOps.Core.truncatedNormal'.</span><span>
</span><a name="line-369"></a><span class="hs-identifier">truncatedNormal</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679045770"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span class="hs-identifier hs-type">Word16</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679045771"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-370"></a><span>                </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679045772"><span class="hs-identifier hs-type">v</span></a><span> </span><span class="hs-identifier hs-type">Int64</span><span>  </span><span class="hs-comment">-- ^ Shape.</span><span>
</span><a name="line-371"></a><span>                </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679045770"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Value</span><span> </span><a href="#local-6989586621679045771"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-372"></a><a name="truncatedNormal"><a href="TensorFlow.Ops.html#truncatedNormal"><span class="hs-identifier">truncatedNormal</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">truncatedNormal</span><span>
</span><a name="line-373"></a><span>
</span><a name="line-374"></a><span class="hs-identifier">truncatedNormal'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">MonadBuild</span><span> </span><a href="#local-6989586621679044972"><span class="hs-identifier hs-type">m</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span class="hs-identifier hs-type">Word16</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Double</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Float</span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679044973"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-375"></a><span>                </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044974"><span class="hs-identifier hs-type">v</span></a><span> </span><span class="hs-identifier hs-type">Int64</span><span>  </span><span class="hs-comment">-- ^ Shape.</span><span>
</span><a name="line-376"></a><span>                </span><span class="hs-glyph">-&gt;</span><span> </span><a href="#local-6989586621679044972"><span class="hs-identifier hs-type">m</span></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Value</span><span> </span><a href="#local-6989586621679044973"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span>
</span><a name="line-377"></a><a name="truncatedNormal%27"><a href="TensorFlow.Ops.html#truncatedNormal%27"><span class="hs-identifier">truncatedNormal'</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">truncatedNormal'</span><span>
</span><a name="line-378"></a><span>
</span><a name="line-379"></a><span class="hs-identifier">zeros</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-keyword">forall</span><span> </span><a name="local-6989586621679044971"><a href="#local-6989586621679044971"><span class="hs-identifier">a</span></a></a><span> </span><span class="hs-operator">.</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">Num</span><span> </span><a href="#local-6989586621679044971"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679044971"><span class="hs-identifier hs-type">a</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Shape</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679044971"><span class="hs-identifier hs-type">a</span></a><span>
</span><a name="line-380"></a><a name="zeros"><a href="TensorFlow.Ops.html#zeros"><span class="hs-identifier">zeros</span></a></a><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-var">Shape</span><span> </span><a name="local-6989586621679052423"><a href="#local-6989586621679052423"><span class="hs-identifier">s</span></a></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">fill</span><span> </span><span class="hs-special">(</span><a href="TensorFlow.Ops.html#vector"><span class="hs-identifier hs-var">vector</span></a><span> </span><a href="#local-6989586621679052423"><span class="hs-identifier hs-var">s</span></a><span class="hs-special">)</span><span> </span><span class="hs-special">(</span><a href="TensorFlow.Ops.html#scalar"><span class="hs-identifier hs-var">scalar</span></a><span> </span><span class="hs-number">0</span><span class="hs-special">)</span><span>
</span><a name="line-381"></a><span>
</span><a name="line-382"></a><span class="hs-identifier">shape</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679044969"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044970"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679044969"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span>
</span><a name="line-383"></a><a name="shape"><a href="TensorFlow.Ops.html#shape"><span class="hs-identifier">shape</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">shape</span><span>
</span><a name="line-384"></a><span>
</span><a name="line-385"></a><span class="hs-identifier">shape'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679044967"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044968"><span class="hs-identifier hs-type">v</span></a><span> </span><a href="#local-6989586621679044967"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span>
</span><a name="line-386"></a><a name="shape%27"><a href="TensorFlow.Ops.html#shape%27"><span class="hs-identifier">shape'</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">shape'</span><span>
</span><a name="line-387"></a><span>
</span><a name="line-388"></a><span class="hs-identifier">expandDims</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679044964"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044965"><span class="hs-identifier hs-type">v1</span></a><span> </span><a href="#local-6989586621679044964"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044966"><span class="hs-identifier hs-type">v2</span></a><span> </span><span class="hs-identifier hs-type">Int32</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679044964"><span class="hs-identifier hs-type">t</span></a><span>
</span><a name="line-389"></a><a name="expandDims"><a href="TensorFlow.Ops.html#expandDims"><span class="hs-identifier">expandDims</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">expandDims</span><span>
</span><a name="line-390"></a><span>
</span><a name="line-391"></a><span class="hs-identifier">expandDims'</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">TensorType</span><span> </span><a href="#local-6989586621679044961"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">=&gt;</span><span> </span><span class="hs-identifier hs-type">OpParams</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044962"><span class="hs-identifier hs-type">v1</span></a><span> </span><a href="#local-6989586621679044961"><span class="hs-identifier hs-type">t</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044963"><span class="hs-identifier hs-type">v2</span></a><span> </span><span class="hs-identifier hs-type">Int32</span><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><a href="#local-6989586621679044961"><span class="hs-identifier hs-type">t</span></a><span>
</span><a name="line-392"></a><a name="expandDims%27"><a href="TensorFlow.Ops.html#expandDims%27"><span class="hs-identifier">expandDims'</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">expandDims'</span><span>
</span><a name="line-393"></a><span>
</span><a name="line-394"></a><span class="hs-comment">-- | Helper function for reduction ops (translation of math_ops.reduced_shape).</span><span>
</span><a name="line-395"></a><span class="hs-identifier">reducedShape</span><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-special">(</span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span> </span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679044957"><span class="hs-identifier hs-type">t1</span></a><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">OneOf</span><span> </span><span class="hs-special">'</span><span class="hs-special">[</span><span> </span><span class="hs-identifier hs-type">Int32</span><span class="hs-special">,</span><span> </span><span class="hs-identifier hs-type">Int64</span><span> </span><span class="hs-special">]</span><span> </span><a href="#local-6989586621679044958"><span class="hs-identifier hs-type">t2</span></a><span class="hs-special">)</span><span> </span><span class="hs-glyph">=&gt;</span><span>
</span><a name="line-396"></a><span>                </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044959"><span class="hs-identifier hs-type">v1</span></a><span> </span><a href="#local-6989586621679044957"><span class="hs-identifier hs-type">t1</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><a href="#local-6989586621679044960"><span class="hs-identifier hs-type">v2</span></a><span> </span><a href="#local-6989586621679044958"><span class="hs-identifier hs-type">t2</span></a><span> </span><span class="hs-glyph">-&gt;</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span>
</span><a name="line-397"></a><a name="reducedShape"><a href="TensorFlow.Ops.html#reducedShape"><span class="hs-identifier">reducedShape</span></a></a><span> </span><a name="local-6989586621679052424"><a href="#local-6989586621679052424"><span class="hs-identifier">inputShape</span></a></a><span> </span><a name="local-6989586621679052425"><a href="#local-6989586621679052425"><span class="hs-identifier">axes</span></a></a><span> </span><span class="hs-glyph">=</span><span>
</span><a name="line-398"></a><span>    </span><span class="hs-keyword">let</span><span> </span><a name="local-6989586621679052426"><a href="#local-6989586621679052426"><span class="hs-identifier">inputShape32</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="#local-6989586621679052428"><span class="hs-identifier hs-var">toInt32</span></a><span> </span><a href="#local-6989586621679052424"><span class="hs-identifier hs-var">inputShape</span></a><span>         </span><span class="hs-comment">-- [2, 3, 5, 7]</span><span>
</span><a name="line-399"></a><span>        </span><a name="local-6989586621679052427"><a href="#local-6989586621679052427"><span class="hs-identifier">axes32</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="#local-6989586621679052428"><span class="hs-identifier hs-var">toInt32</span></a><span> </span><a href="#local-6989586621679052425"><span class="hs-identifier hs-var">axes</span></a><span>                     </span><span class="hs-comment">-- [1, 2]</span><span>
</span><a name="line-400"></a><span>        </span><a name="local-6989586621679052428"><a href="#local-6989586621679052428"><span class="hs-identifier">toInt32</span></a></a><span> </span><a name="local-6989586621679052432"><a href="#local-6989586621679052432"><span class="hs-identifier">x</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">cast</span><span> </span><a href="#local-6989586621679052432"><span class="hs-identifier hs-var">x</span></a><span> </span><span class="hs-glyph">::</span><span> </span><span class="hs-identifier hs-type">Tensor</span><span> </span><span class="hs-identifier hs-type">Build</span><span> </span><span class="hs-identifier hs-type">Int32</span><span>
</span><a name="line-401"></a><span>        </span><a name="local-6989586621679052429"><a href="#local-6989586621679052429"><span class="hs-identifier">inputRank</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">size</span><span> </span><a href="#local-6989586621679052426"><span class="hs-identifier hs-var">inputShape32</span></a><span>     </span><span class="hs-comment">-- 4</span><span>
</span><a name="line-402"></a><span>        </span><a name="local-6989586621679052430"><a href="#local-6989586621679052430"><span class="hs-identifier">axesMod</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-special">(</span><a href="#local-6989586621679052427"><span class="hs-identifier hs-var">axes32</span></a><span> </span><span class="hs-operator hs-var">+</span><span> </span><a href="#local-6989586621679052429"><span class="hs-identifier hs-var">inputRank</span></a><span class="hs-special">)</span><span> </span><span class="hs-special">`</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">mod</span><span class="hs-special">`</span><span> </span><a href="#local-6989586621679052429"><span class="hs-identifier hs-var">inputRank</span></a><span>
</span><a name="line-403"></a><span>        </span><a name="local-6989586621679052431"><a href="#local-6989586621679052431"><span class="hs-identifier">axesShape</span></a></a><span> </span><span class="hs-glyph">=</span><span> </span><a href="TensorFlow.Ops.html#shape"><span class="hs-identifier hs-var">shape</span></a><span> </span><a href="#local-6989586621679052430"><span class="hs-identifier hs-var">axesMod</span></a><span>                 </span><span class="hs-comment">-- [2]</span><span>
</span><a name="line-404"></a><span>    </span><span class="hs-keyword">in</span><span> </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">dynamicStitch</span><span>                      </span><span class="hs-comment">-- [2, 1, 1, 7]</span><span>
</span><a name="line-405"></a><span>         </span><span class="hs-special">[</span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">range</span><span> </span><span class="hs-number">0</span><span> </span><a href="#local-6989586621679052429"><span class="hs-identifier hs-var">inputRank</span></a><span> </span><span class="hs-number">1</span><span class="hs-special">,</span><span>            </span><span class="hs-comment">-- [0, 1, 2, 3]</span><span>
</span><a name="line-406"></a><span>           </span><a href="#local-6989586621679052430"><span class="hs-identifier hs-var">axesMod</span></a><span class="hs-special">]</span><span>                               </span><span class="hs-comment">-- [1, 2]</span><span>
</span><a name="line-407"></a><span>         </span><span class="hs-special">[</span><a href="#local-6989586621679052426"><span class="hs-identifier hs-var">inputShape32</span></a><span class="hs-special">,</span><span>                           </span><span class="hs-comment">-- [2, 3, 5, 7]</span><span>
</span><a name="line-408"></a><span>           </span><span class="hs-identifier hs-var">CoreOps</span><span class="hs-operator hs-var">.</span><span class="hs-identifier hs-var">fill</span><span> </span><a href="#local-6989586621679052431"><span class="hs-identifier hs-var">axesShape</span></a><span> </span><span class="hs-number">1</span><span class="hs-special">]</span><span>              </span><span class="hs-comment">-- [1, 1]</span><span>
</span><a name="line-409"></a></pre></body></html>