mirror of
https://github.com/tensorflow/haskell.git
synced 2024-06-01 18:43:36 +02:00
6b19e54722
* Update README to refer to 2.3.0-gpu. * Remove old package documentation from haddock directory.
109 lines
15 KiB
HTML
109 lines
15 KiB
HTML
<!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><span id="line-2"></span><span class="hs-comment">--</span><span>
|
|
</span><span id="line-3"></span><span class="hs-comment">-- Licensed under the Apache License, Version 2.0 (the "License");</span><span>
|
|
</span><span id="line-4"></span><span class="hs-comment">-- you may not use this file except in compliance with the License.</span><span>
|
|
</span><span id="line-5"></span><span class="hs-comment">-- You may obtain a copy of the License at</span><span>
|
|
</span><span id="line-6"></span><span class="hs-comment">--</span><span>
|
|
</span><span id="line-7"></span><span class="hs-comment">-- http://www.apache.org/licenses/LICENSE-2.0</span><span>
|
|
</span><span id="line-8"></span><span class="hs-comment">--</span><span>
|
|
</span><span id="line-9"></span><span class="hs-comment">-- Unless required by applicable law or agreed to in writing, software</span><span>
|
|
</span><span id="line-10"></span><span class="hs-comment">-- distributed under the License is distributed on an "AS IS" BASIS,</span><span>
|
|
</span><span id="line-11"></span><span class="hs-comment">-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.</span><span>
|
|
</span><span id="line-12"></span><span class="hs-comment">-- See the License for the specific language governing permissions and</span><span>
|
|
</span><span id="line-13"></span><span class="hs-comment">-- limitations under the License.</span><span>
|
|
</span><span id="line-14"></span><span>
|
|
</span><span id="line-15"></span><span class="hs-pragma">{-# LANGUAGE BangPatterns #-}</span><span>
|
|
</span><span id="line-16"></span><span>
|
|
</span><span id="line-17"></span><span class="hs-comment">{-|
|
|
Module : TensorFlow.Internal.VarInt
|
|
Description : Encoders and decoders for varint types.
|
|
|
|
Originally taken from internal proto-lens code.
|
|
-}</span><span>
|
|
</span><span id="line-23"></span><span class="hs-keyword">module</span><span> </span><span class="hs-identifier">TensorFlow.Internal.VarInt</span><span>
|
|
</span><span id="line-24"></span><span> </span><span class="hs-special">(</span><span> </span><span class="annot"><a href="TensorFlow.Internal.VarInt.html#getVarInt"><span class="hs-identifier">getVarInt</span></a></span><span>
|
|
</span><span id="line-25"></span><span> </span><span class="hs-special">,</span><span> </span><span class="annot"><a href="TensorFlow.Internal.VarInt.html#putVarInt"><span class="hs-identifier">putVarInt</span></a></span><span>
|
|
</span><span id="line-26"></span><span> </span><span class="hs-special">)</span><span> </span><span class="hs-keyword">where</span><span>
|
|
</span><span id="line-27"></span><span>
|
|
</span><span id="line-28"></span><span class="hs-keyword">import</span><span> </span><span class="annot"><span class="hs-identifier">Data.Attoparsec.ByteString</span></span><span> </span><span class="hs-keyword">as</span><span> </span><span class="annot"><span class="hs-identifier">Parse</span></span><span>
|
|
</span><span id="line-29"></span><span class="hs-keyword">import</span><span> </span><span class="annot"><span class="hs-identifier">Data.Bits</span></span><span>
|
|
</span><span id="line-30"></span><span class="hs-keyword">import</span><span> </span><span class="annot"><span class="hs-identifier">Data.ByteString.Lazy.Builder</span></span><span> </span><span class="hs-keyword">as</span><span> </span><span class="annot"><span class="hs-identifier">Builder</span></span><span>
|
|
</span><span id="line-31"></span><span class="hs-keyword">import</span><span> </span><span class="annot"><span class="hs-identifier">Data.Word</span></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="hs-identifier">Word64</span></span><span class="hs-special">)</span><span>
|
|
</span><span id="line-32"></span><span>
|
|
</span><span id="line-33"></span><span class="hs-comment">-- | Decode an unsigned varint.</span><span>
|
|
</span><span id="line-34"></span><span class="annot"><a href="TensorFlow.Internal.VarInt.html#getVarInt"><span class="hs-identifier hs-type">getVarInt</span></a></span><span> </span><span class="hs-glyph">::</span><span> </span><span class="annot"><span class="hs-identifier hs-type">Parser</span></span><span> </span><span class="annot"><span class="hs-identifier hs-type">Word64</span></span><span>
|
|
</span><span id="line-35"></span><span id="getVarInt"><span class="annot"><span class="annottext">getVarInt :: Parser Word64
|
|
</span><a href="TensorFlow.Internal.VarInt.html#getVarInt"><span class="hs-identifier hs-var hs-var">getVarInt</span></a></span></span><span> </span><span class="hs-glyph">=</span><span> </span><span class="annot"><span class="annottext">Word64 -> Word64 -> Parser Word64
|
|
forall t. Num t => t -> t -> Parser ByteString t
|
|
</span><a href="#local-6989586621679163980"><span class="hs-identifier hs-var">loop</span></a></span><span> </span><span class="annot"><span class="hs-number">1</span></span><span> </span><span class="annot"><span class="hs-number">0</span></span><span>
|
|
</span><span id="line-36"></span><span> </span><span class="hs-keyword">where</span><span>
|
|
</span><span id="line-37"></span><span> </span><span id="local-6989586621679163980"><span class="annot"><span class="annottext">loop :: t -> t -> Parser ByteString t
|
|
</span><a href="#local-6989586621679163980"><span class="hs-identifier hs-var hs-var">loop</span></a></span></span><span> </span><span class="hs-glyph">!</span><span id="local-6989586621679163979"><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163979"><span class="hs-identifier hs-var">s</span></a></span></span><span> </span><span class="hs-glyph">!</span><span id="local-6989586621679163978"><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163978"><span class="hs-identifier hs-var">n</span></a></span></span><span> </span><span class="hs-glyph">=</span><span> </span><span class="hs-keyword">do</span><span>
|
|
</span><span id="line-38"></span><span> </span><span id="local-6989586621679163977"><span class="annot"><span class="annottext">Word8
|
|
</span><a href="#local-6989586621679163977"><span class="hs-identifier hs-var">b</span></a></span></span><span> </span><span class="hs-glyph"><-</span><span> </span><span class="annot"><span class="annottext">Parser Word8
|
|
</span><span class="hs-identifier hs-var">anyWord8</span></span><span>
|
|
</span><span id="line-39"></span><span> </span><span class="hs-keyword">let</span><span> </span><span id="local-6989586621679163975"><span class="annot"><span class="annottext">n' :: t
|
|
</span><a href="#local-6989586621679163975"><span class="hs-identifier hs-var hs-var">n'</span></a></span></span><span> </span><span class="hs-glyph">=</span><span> </span><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163978"><span class="hs-identifier hs-var">n</span></a></span><span> </span><span class="annot"><span class="annottext">t -> t -> t
|
|
forall a. Num a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">+</span></span><span> </span><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163979"><span class="hs-identifier hs-var">s</span></a></span><span> </span><span class="annot"><span class="annottext">t -> t -> t
|
|
forall a. Num a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">*</span></span><span> </span><span class="annot"><span class="annottext">Word8 -> t
|
|
forall a b. (Integral a, Num b) => a -> b
|
|
</span><span class="hs-identifier hs-var">fromIntegral</span></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="annottext">Word8
|
|
</span><a href="#local-6989586621679163977"><span class="hs-identifier hs-var">b</span></a></span><span> </span><span class="annot"><span class="annottext">Word8 -> Word8 -> Word8
|
|
forall a. Bits a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">.&.</span></span><span> </span><span class="annot"><span class="hs-number">127</span></span><span class="hs-special">)</span><span>
|
|
</span><span id="line-40"></span><span> </span><span class="hs-keyword">if</span><span> </span><span class="hs-special">(</span><span class="annot"><span class="annottext">Word8
|
|
</span><a href="#local-6989586621679163977"><span class="hs-identifier hs-var">b</span></a></span><span> </span><span class="annot"><span class="annottext">Word8 -> Word8 -> Word8
|
|
forall a. Bits a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">.&.</span></span><span> </span><span class="annot"><span class="hs-number">128</span></span><span class="hs-special">)</span><span> </span><span class="annot"><span class="annottext">Word8 -> Word8 -> Bool
|
|
forall a. Eq a => a -> a -> Bool
|
|
</span><span class="hs-operator hs-var">==</span></span><span> </span><span class="annot"><span class="hs-number">0</span></span><span>
|
|
</span><span id="line-41"></span><span> </span><span class="hs-keyword">then</span><span> </span><span class="annot"><span class="annottext">t -> Parser ByteString t
|
|
forall (m :: * -> *) a. Monad m => a -> m a
|
|
</span><span class="hs-identifier hs-var">return</span></span><span> </span><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163975"><span class="hs-identifier hs-var">n'</span></a></span><span>
|
|
</span><span id="line-42"></span><span> </span><span class="hs-keyword">else</span><span> </span><span class="annot"><span class="annottext">t -> t -> Parser ByteString t
|
|
</span><a href="#local-6989586621679163980"><span class="hs-identifier hs-var">loop</span></a></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="hs-number">128</span></span><span class="annot"><span class="annottext">t -> t -> t
|
|
forall a. Num a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">*</span></span><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163979"><span class="hs-identifier hs-var">s</span></a></span><span class="hs-special">)</span><span> </span><span class="annot"><span class="annottext">t
|
|
</span><a href="#local-6989586621679163975"><span class="hs-identifier hs-var">n'</span></a></span><span>
|
|
</span><span id="line-43"></span><span>
|
|
</span><span id="line-44"></span><span class="hs-comment">-- | Encode a Word64.</span><span>
|
|
</span><span id="line-45"></span><span class="annot"><a href="TensorFlow.Internal.VarInt.html#putVarInt"><span class="hs-identifier hs-type">putVarInt</span></a></span><span> </span><span class="hs-glyph">::</span><span> </span><span class="annot"><span class="hs-identifier hs-type">Word64</span></span><span> </span><span class="hs-glyph">-></span><span> </span><span class="annot"><span class="hs-identifier hs-type">Builder</span></span><span>
|
|
</span><span id="line-46"></span><span id="putVarInt"><span class="annot"><span class="annottext">putVarInt :: Word64 -> Builder
|
|
</span><a href="TensorFlow.Internal.VarInt.html#putVarInt"><span class="hs-identifier hs-var hs-var">putVarInt</span></a></span></span><span> </span><span id="local-6989586621679163971"><span class="annot"><span class="annottext">n :: Word64
|
|
</span><a href="#local-6989586621679163971"><span class="hs-identifier hs-var">n</span></a></span></span><span>
|
|
</span><span id="line-47"></span><span> </span><span class="hs-glyph">|</span><span> </span><span class="annot"><span class="annottext">Word64
|
|
</span><a href="#local-6989586621679163971"><span class="hs-identifier hs-var">n</span></a></span><span> </span><span class="annot"><span class="annottext">Word64 -> Word64 -> Bool
|
|
forall a. Ord a => a -> a -> Bool
|
|
</span><span class="hs-operator hs-var"><</span></span><span> </span><span class="annot"><span class="hs-number">128</span></span><span> </span><span class="hs-glyph">=</span><span> </span><span class="annot"><span class="annottext">Word8 -> Builder
|
|
</span><span class="hs-identifier hs-var">Builder.word8</span></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="annottext">Word64 -> Word8
|
|
forall a b. (Integral a, Num b) => a -> b
|
|
</span><span class="hs-identifier hs-var">fromIntegral</span></span><span> </span><span class="annot"><span class="annottext">Word64
|
|
</span><a href="#local-6989586621679163971"><span class="hs-identifier hs-var">n</span></a></span><span class="hs-special">)</span><span>
|
|
</span><span id="line-48"></span><span> </span><span class="hs-glyph">|</span><span> </span><span class="annot"><span class="annottext">Bool
|
|
</span><span class="hs-identifier hs-var">otherwise</span></span><span> </span><span class="hs-glyph">=</span><span> </span><span class="annot"><span class="annottext">Word8 -> Builder
|
|
</span><span class="hs-identifier hs-var">Builder.word8</span></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="annottext">Word64 -> Word8
|
|
forall a b. (Integral a, Num b) => a -> b
|
|
</span><span class="hs-identifier hs-var">fromIntegral</span></span><span> </span><span class="annot"><span class="annottext">(Word64 -> Word8) -> Word64 -> Word8
|
|
forall a b. (a -> b) -> a -> b
|
|
</span><span class="hs-operator hs-var">$</span></span><span> </span><span class="annot"><span class="annottext">Word64
|
|
</span><a href="#local-6989586621679163971"><span class="hs-identifier hs-var">n</span></a></span><span> </span><span class="annot"><span class="annottext">Word64 -> Word64 -> Word64
|
|
forall a. Bits a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">.&.</span></span><span> </span><span class="annot"><span class="hs-number">127</span></span><span> </span><span class="annot"><span class="annottext">Word64 -> Word64 -> Word64
|
|
forall a. Bits a => a -> a -> a
|
|
</span><span class="hs-operator hs-var">.|.</span></span><span> </span><span class="annot"><span class="hs-number">128</span></span><span class="hs-special">)</span><span>
|
|
</span><span id="line-49"></span><span> </span><span class="annot"><span class="annottext">Builder -> Builder -> Builder
|
|
forall a. Semigroup a => a -> a -> a
|
|
</span><span class="hs-operator hs-var"><></span></span><span> </span><span class="annot"><span class="annottext">Word64 -> Builder
|
|
</span><a href="TensorFlow.Internal.VarInt.html#putVarInt"><span class="hs-identifier hs-var">putVarInt</span></a></span><span> </span><span class="hs-special">(</span><span class="annot"><span class="annottext">Word64
|
|
</span><a href="#local-6989586621679163971"><span class="hs-identifier hs-var">n</span></a></span><span> </span><span class="annot"><span class="annottext">Word64 -> Int -> Word64
|
|
forall a. Bits a => a -> Int -> a
|
|
</span><span class="hs-operator hs-var">`shiftR`</span></span><span> </span><span class="annot"><span class="hs-number">7</span></span><span class="hs-special">)</span><span>
|
|
</span><span id="line-50"></span></pre></body></html> |