{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# OPTIONS_GHC -fno-warn-name-shadowing #-}
{-# OPTIONS_GHC -fno-warn-incomplete-patterns #-}
module TensorFlow.GenOps.Core where
import Data.ByteString (ByteString)
import Data.Complex (Complex)
import Data.Int (Int8, Int16, Int32, Int64)
import Data.Proxy (Proxy(Proxy))
import Data.Word (Word8, Word16, Word32, Word64)
import Lens.Family2 ((.~), (&))
import TensorFlow.Build
import TensorFlow.BuildOp
import TensorFlow.Tensor
import TensorFlow.Types
-- | 
abort :: forall m' . (MonadBuild m') => m' (ControlNode)
abort :: m' ControlNode
abort = OpParams -> m' ControlNode
forall (m' :: * -> *). MonadBuild m' => OpParams -> m' ControlNode
abort' OpParams
forall a. a -> a
id
abort' :: forall m' . (MonadBuild m') => OpParams ->
          m' (ControlNode)
abort' :: OpParams -> m' ControlNode
abort' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Abort"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr { name: "error_msg" type: "string" default_value { s: "" } }
attr {
  name: "exit_without_error" type: "bool" default_value { b: false }
}
-}
-- | 
abs :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                               Data.Int.Int8, Data.Word.Word16, Double,
                               Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
abs :: Tensor v'1 t -> Tensor Build t
abs = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
abs' OpParams
forall a. a -> a
id
abs' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                Data.Int.Int8, Data.Word.Word16, Double,
                                Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
abs' :: OpParams -> Tensor v'1 t -> Tensor Build t
abs' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Abs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
accumulateNV2 :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) =>
                 Shape -- ^ __shape__
                 -> [Tensor v'1 t] -- ^ __inputs__
                 -> Tensor Build t -- ^ __sum__
accumulateNV2 :: Shape -> [Tensor v'1 t] -> Tensor Build t
accumulateNV2 = OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
accumulateNV2' OpParams
forall a. a -> a
id
accumulateNV2' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) => OpParams ->
                  Shape -- ^ __shape__
                  -> [Tensor v'1 t] -- ^ __inputs__
                  -> Tensor Build t -- ^ __sum__
accumulateNV2' :: OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
accumulateNV2' op'options :: OpParams
op'options shape :: Shape
shape
               inputs :: [Tensor v'1 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AccumulateNV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs) :: Int64
{-
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "sum" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "shape" type: "shape" }
-}
-- | 
accumulatorApplyGradient :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                                       OneOf '[(Data.Complex.Complex Double),
                                                               (Data.Complex.Complex Float),
                                                               Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word32,
                                                               Data.Word.Word64,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] dtype) =>
                            Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                            -> Tensor v'3 dtype -- ^ __gradient__
                            -> m' (ControlNode)
accumulatorApplyGradient :: Tensor Ref ByteString
-> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
accumulatorApplyGradient = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
accumulatorApplyGradient' OpParams
forall a. a -> a
id
accumulatorApplyGradient' :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                                        OneOf '[(Data.Complex.Complex Double),
                                                                (Data.Complex.Complex Float),
                                                                Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Int.Int8,
                                                                Data.Word.Word16,
                                                                Data.Word.Word32,
                                                                Data.Word.Word64,
                                                                Data.Word.Word8,
                                                                Double,
                                                                Float] dtype) =>
                             OpParams ->
                             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                             -> Tensor v'3 dtype -- ^ __gradient__
                             -> m' (ControlNode)
accumulatorApplyGradient' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
accumulatorApplyGradient' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle local_step :: Tensor v'2 Int64
local_step
                          gradient :: Tensor v'3 dtype
gradient | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
local_step,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
gradient]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AccumulatorApplyGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "local_step" type: DT_INT64 }
input_arg { name: "gradient" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
accumulatorNumAccumulated :: forall m' . (MonadBuild m') =>
                             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                             -> m' (Tensor Value Data.Int.Int32) -- ^ __num_accumulated__
accumulatorNumAccumulated :: Tensor Ref ByteString -> m' (Tensor Value Int32)
accumulatorNumAccumulated = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
accumulatorNumAccumulated' OpParams
forall a. a -> a
id
accumulatorNumAccumulated' :: forall m' . (MonadBuild m') => OpParams ->
                              Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                              -> m' (Tensor Value Data.Int.Int32) -- ^ __num_accumulated__
accumulatorNumAccumulated' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
accumulatorNumAccumulated' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AccumulatorNumAccumulated"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "num_accumulated" type: DT_INT32 }
-}
-- | 
accumulatorSetGlobalStep :: forall v'2 m' . (MonadBuild m') =>
                            Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __new_global_step__
                            -> m' (ControlNode)
accumulatorSetGlobalStep :: Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
accumulatorSetGlobalStep = OpParams
-> Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
forall (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
accumulatorSetGlobalStep' OpParams
forall a. a -> a
id
accumulatorSetGlobalStep' :: forall v'2 m' . (MonadBuild m') => OpParams ->
                             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __new_global_step__
                             -> m' (ControlNode)
accumulatorSetGlobalStep' :: OpParams
-> Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
accumulatorSetGlobalStep' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle new_global_step :: Tensor v'2 Int64
new_global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
new_global_step]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AccumulatorSetGlobalStep"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "new_global_step" type: DT_INT64 }
-}
-- | 
accumulatorTakeGradient :: forall v'2 dtype m' . (MonadBuild m',
                                                  OneOf '[(Data.Complex.Complex Double),
                                                          (Data.Complex.Complex Float),
                                                          Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Int.Int8,
                                                          Data.Word.Word16,
                                                          Data.Word.Word32,
                                                          Data.Word.Word64,
                                                          Data.Word.Word8,
                                                          Double,
                                                          Float] dtype) =>
                           Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                           -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                           -> m' (Tensor Value dtype) -- ^ __average__
accumulatorTakeGradient :: Tensor Ref ByteString
-> Tensor v'2 Int32 -> m' (Tensor Value dtype)
accumulatorTakeGradient = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
forall (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
accumulatorTakeGradient' OpParams
forall a. a -> a
id
accumulatorTakeGradient' :: forall v'2 dtype m' . (MonadBuild m',
                                                   OneOf '[(Data.Complex.Complex Double),
                                                           (Data.Complex.Complex Float),
                                                           Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double,
                                                           Float] dtype) =>
                            OpParams ->
                            Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                            -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                            -> m' (Tensor Value dtype) -- ^ __average__
accumulatorTakeGradient' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
accumulatorTakeGradient' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle num_required :: Tensor v'2 Int32
num_required | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_required]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AccumulatorTakeGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "num_required" type: DT_INT32 }
output_arg { name: "average" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
acos :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
acos :: Tensor v'1 t -> Tensor Build t
acos = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
acos' OpParams
forall a. a -> a
id
acos' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
acos' :: OpParams -> Tensor v'1 t -> Tensor Build t
acos' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Acos"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
acosh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
acosh :: Tensor v'1 t -> Tensor Build t
acosh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
acosh' OpParams
forall a. a -> a
id
acosh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
acosh' :: OpParams -> Tensor v'1 t -> Tensor Build t
acosh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Acosh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
add :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float),
                                   Data.ByteString.ByteString, Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word8, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
add :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
add = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64,
    Int8, Word16, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
add' OpParams
forall a. a -> a
id
add' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.ByteString.ByteString, Data.Int.Int16,
                                    Data.Int.Int32, Data.Int.Int64,
                                    Data.Int.Int8, Data.Word.Word16,
                                    Data.Word.Word8, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
add' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
add' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Add"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_STRING
    }
  }
}
-}
-- | 
addManySparseToTensorsMap :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                        TensorType t) =>
                             Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                             -> Tensor v'2 t -- ^ __sparse_values__
                             -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                             -> m' (Tensor Value Data.Int.Int64) -- ^ __sparse_handles__
addManySparseToTensorsMap :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
addManySparseToTensorsMap = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
addManySparseToTensorsMap' OpParams
forall a. a -> a
id
addManySparseToTensorsMap' :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                         TensorType t) =>
                              OpParams ->
                              Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                              -> Tensor v'2 t -- ^ __sparse_values__
                              -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                              -> m' (Tensor Value Data.Int.Int64) -- ^ __sparse_handles__
addManySparseToTensorsMap' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
addManySparseToTensorsMap' op'options :: OpParams
op'options sparse_indices :: Tensor v'1 Int64
sparse_indices sparse_values :: Tensor v'2 t
sparse_values
                           sparse_shape :: Tensor v'3 Int64
sparse_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sparse_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sparse_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sparse_shape]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AddManySparseToTensorsMap"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_indices" type: DT_INT64 }
input_arg { name: "sparse_values" type_attr: "T" }
input_arg { name: "sparse_shape" type: DT_INT64 }
output_arg { name: "sparse_handles" type: DT_INT64 }
attr { name: "T" type: "type" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
addN :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Data.Word.Word32,
                                Data.Word.Word64, Data.Word.Word8, Double,
                                Float, Variant] t) =>
        [Tensor v'1 t] -- ^ __inputs__
        -> Tensor Build t -- ^ __sum__
addN :: [Tensor v'1 t] -> Tensor Build t
addN = OpParams -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float, Variant]
  t =>
OpParams -> [Tensor v'1 t] -> Tensor Build t
addN' OpParams
forall a. a -> a
id
addN' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Data.Word.Word32,
                                 Data.Word.Word64, Data.Word.Word8, Double,
                                 Float, Variant] t) => OpParams ->
         [Tensor v'1 t] -- ^ __inputs__
         -> Tensor Build t -- ^ __sum__
addN' :: OpParams -> [Tensor v'1 t] -> Tensor Build t
addN' op'options :: OpParams
op'options inputs :: [Tensor v'1 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AddN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs) :: Int64
{-
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "sum" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_VARIANT
    }
  }
}
-}
-- | 
addSparseToTensorsMap :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                    TensorType t) =>
                         Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                         -> Tensor v'2 t -- ^ __sparse_values__
                         -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                         -> m' (Tensor Value Data.Int.Int64) -- ^ __sparse_handle__
addSparseToTensorsMap :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
addSparseToTensorsMap = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
addSparseToTensorsMap' OpParams
forall a. a -> a
id
addSparseToTensorsMap' :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                     TensorType t) =>
                          OpParams ->
                          Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                          -> Tensor v'2 t -- ^ __sparse_values__
                          -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                          -> m' (Tensor Value Data.Int.Int64) -- ^ __sparse_handle__
addSparseToTensorsMap' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
addSparseToTensorsMap' op'options :: OpParams
op'options sparse_indices :: Tensor v'1 Int64
sparse_indices sparse_values :: Tensor v'2 t
sparse_values
                       sparse_shape :: Tensor v'3 Int64
sparse_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sparse_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sparse_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sparse_shape]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AddSparseToTensorsMap"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_indices" type: DT_INT64 }
input_arg { name: "sparse_values" type_attr: "T" }
input_arg { name: "sparse_shape" type: DT_INT64 }
output_arg { name: "sparse_handle" type: DT_INT64 }
attr { name: "T" type: "type" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
addV2 :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word8, Double,
                                     Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __y__
         -> Tensor Build t -- ^ __z__
addV2 :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
addV2 = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
addV2' OpParams
forall a. a -> a
id
addV2' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word8, Double,
                                      Float] t) => OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor v'2 t -- ^ __y__
          -> Tensor Build t -- ^ __z__
addV2' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
addV2' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AddV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
adjustContrast :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word8, Double,
                                                      Float] t) =>
                  Tensor v'1 t -- ^ __images__
                  -> Tensor v'2 Float -- ^ __contrast_factor__
                  -> Tensor v'3 Float -- ^ __min_value__
                  -> Tensor v'4 Float -- ^ __max_value__
                  -> Tensor Build Float -- ^ __output__
adjustContrast :: Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
adjustContrast = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
adjustContrast' OpParams
forall a. a -> a
id
adjustContrast' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word8, Double,
                                                       Float] t) => OpParams ->
                   Tensor v'1 t -- ^ __images__
                   -> Tensor v'2 Float -- ^ __contrast_factor__
                   -> Tensor v'3 Float -- ^ __min_value__
                   -> Tensor v'4 Float -- ^ __max_value__
                   -> Tensor Build Float -- ^ __output__
adjustContrast' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
adjustContrast' op'options :: OpParams
op'options images :: Tensor v'1 t
images contrast_factor :: Tensor v'2 Float
contrast_factor min_value :: Tensor v'3 Float
min_value
                max_value :: Tensor v'4 Float
max_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
contrast_factor,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AdjustContrast"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "contrast_factor" type: DT_FLOAT }
input_arg { name: "min_value" type: DT_FLOAT }
input_arg { name: "max_value" type: DT_FLOAT }
output_arg { name: "output" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
-}
-- | 
adjustContrastv2 :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                    Tensor v'1 t -- ^ __images__
                    -> Tensor v'2 Float -- ^ __contrast_factor__
                    -> Tensor Build t -- ^ __output__
adjustContrastv2 :: Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustContrastv2 = OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustContrastv2' OpParams
forall a. a -> a
id
adjustContrastv2' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __images__
                     -> Tensor v'2 Float -- ^ __contrast_factor__
                     -> Tensor Build t -- ^ __output__
adjustContrastv2' :: OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustContrastv2' op'options :: OpParams
op'options images :: Tensor v'1 t
images contrast_factor :: Tensor v'2 Float
contrast_factor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
contrast_factor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AdjustContrastv2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "contrast_factor" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
adjustHue :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
             Tensor v'1 t -- ^ __images__
             -> Tensor v'2 Float -- ^ __delta__
             -> Tensor Build t -- ^ __output__
adjustHue :: Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustHue = OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustHue' OpParams
forall a. a -> a
id
adjustHue' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __images__
              -> Tensor v'2 Float -- ^ __delta__
              -> Tensor Build t -- ^ __output__
adjustHue' :: OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustHue' op'options :: OpParams
op'options images :: Tensor v'1 t
images delta :: Tensor v'2 Float
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
delta]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AdjustHue"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "delta" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
adjustSaturation :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                    Tensor v'1 t -- ^ __images__
                    -> Tensor v'2 Float -- ^ __scale__
                    -> Tensor Build t -- ^ __output__
adjustSaturation :: Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustSaturation = OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustSaturation' OpParams
forall a. a -> a
id
adjustSaturation' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __images__
                     -> Tensor v'2 Float -- ^ __scale__
                     -> Tensor Build t -- ^ __output__
adjustSaturation' :: OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
adjustSaturation' op'options :: OpParams
op'options images :: Tensor v'1 t
images scale :: Tensor v'2 Float
scale | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
scale]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AdjustSaturation"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "scale" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
all :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
       Tensor v'1 Bool -- ^ __input__
       -> Tensor v'2 tidx -- ^ __reduction_indices__
       -> Tensor Build Bool -- ^ __output__
all :: Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
all = OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) tidx.
OneOf '[Int32, Int64] tidx =>
OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
all' OpParams
forall a. a -> a
id
all' :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
        OpParams ->
        Tensor v'1 Bool -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build Bool -- ^ __output__
all' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
all' op'options :: OpParams
op'options input :: Tensor v'1 Bool
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "All"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_BOOL }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type: DT_BOOL }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
allCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                       Data.Int.Int64 -- ^ __num_sampled__
                       -> Data.Int.Int64 -- ^ __num_true__
                       -> Bool -- ^ __unique__
                       -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                       -> m' ((Tensor Value Data.Int.Int64, Tensor Value Float,
                               Tensor Value Float))
                       -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                       --
                       -- * __sampled_candidates__
                       --
                       -- * __true_expected_count__
                       --
                       -- * __sampled_expected_count__
allCandidateSampler :: Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
allCandidateSampler = OpParams
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
allCandidateSampler' OpParams
forall a. a -> a
id
allCandidateSampler' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                        Data.Int.Int64 -- ^ __num_sampled__
                        -> Data.Int.Int64 -- ^ __num_true__
                        -> Bool -- ^ __unique__
                        -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                        -> m' ((Tensor Value Data.Int.Int64, Tensor Value Float,
                                Tensor Value Float))
                        -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                        --
                        -- * __sampled_candidates__
                        --
                        -- * __true_expected_count__
                        --
                        -- * __sampled_expected_count__
allCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
allCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true unique :: Bool
unique
                     true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AllCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
allToAll :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float), Bool,
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
            Data.Int.Int64 -- ^ __concat_dimension__
            -> Data.Int.Int64 -- ^ __split_count__
            -> Data.Int.Int64 -- ^ __split_dimension__
            -> Tensor v'1 t -- ^ __input__
            -> Tensor v'2 Data.Int.Int32 -- ^ __group_assignment__
            -> Tensor Build t -- ^ __output__
allToAll :: Int64
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor Build t
allToAll = OpParams
-> Int64
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Word16, Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Int64
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor Build t
allToAll' OpParams
forall a. a -> a
id
allToAll' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float), Bool,
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
             Data.Int.Int64 -- ^ __concat_dimension__
             -> Data.Int.Int64 -- ^ __split_count__
             -> Data.Int.Int64 -- ^ __split_dimension__
             -> Tensor v'1 t -- ^ __input__
             -> Tensor v'2 Data.Int.Int32 -- ^ __group_assignment__
             -> Tensor Build t -- ^ __output__
allToAll' :: OpParams
-> Int64
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor Build t
allToAll' op'options :: OpParams
op'options concat_dimension :: Int64
concat_dimension split_count :: Int64
split_count split_dimension :: Int64
split_dimension input :: Tensor v'1 t
input
          group_assignment :: Tensor v'2 Int32
group_assignment | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
group_assignment]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AllToAll"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "concat_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
concat_dimension
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "split_count" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
split_count
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "split_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
split_dimension
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "group_assignment" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_BOOL
    }
  }
}
attr { name: "concat_dimension" type: "int" }
attr { name: "split_dimension" type: "int" }
attr { name: "split_count" type: "int" }
-}
-- | 
angle :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float)] t,
                              OneOf '[Double, Float] tout) =>
         Tensor v'1 t -- ^ __input__
         -> Tensor Build tout -- ^ __output__
angle :: Tensor v'1 t -> Tensor Build tout
angle = OpParams -> Tensor v'1 t -> Tensor Build tout
forall (v'1 :: * -> *) t tout.
(OneOf '[Complex Double, Complex Float] t,
 OneOf '[Double, Float] tout) =>
OpParams -> Tensor v'1 t -> Tensor Build tout
angle' OpParams
forall a. a -> a
id
angle' :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float)] t,
                               OneOf '[Double, Float] tout) => OpParams ->
          Tensor v'1 t -- ^ __input__
          -> Tensor Build tout -- ^ __output__
angle' :: OpParams -> Tensor v'1 t -> Tensor Build tout
angle' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tout)
-> Build OpDef -> Tensor Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Angle"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "Tout" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
attr {
  name: "Tout"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
anonymousIterator :: forall m' . (MonadBuild m') =>
                     [DataType] -- ^ __output_types__
                     -> m' (Tensor Value ResourceHandle) -- ^ __handle__
anonymousIterator :: [DataType] -> m' (Tensor Value ResourceHandle)
anonymousIterator = OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
anonymousIterator' OpParams
forall a. a -> a
id
anonymousIterator' :: forall m' . (MonadBuild m') => OpParams ->
                      [DataType] -- ^ __output_types__
                      -> m' (Tensor Value ResourceHandle) -- ^ __handle__
anonymousIterator' :: OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
anonymousIterator' op'options :: OpParams
op'options output_types :: [DataType]
output_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
anonymousIteratorV2 :: forall m' . (MonadBuild m') =>
                       [DataType] -- ^ __output_types__
                       -> m' ((Tensor Value ResourceHandle,
                               Tensor Value Variant))
                       -- ^ (__handle__, __deleter__)
                       --
                       -- * __handle__
                       --
                       -- * __deleter__
anonymousIteratorV2 :: [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousIteratorV2 = OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousIteratorV2' OpParams
forall a. a -> a
id
anonymousIteratorV2' :: forall m' . (MonadBuild m') => OpParams ->
                        [DataType] -- ^ __output_types__
                        -> m' ((Tensor Value ResourceHandle,
                                Tensor Value Variant))
                        -- ^ (__handle__, __deleter__)
                        --
                        -- * __handle__
                        --
                        -- * __deleter__
anonymousIteratorV2' :: OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousIteratorV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Variant)
 -> m' (Tensor Value ResourceHandle, Tensor Value Variant))
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousIteratorV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "deleter" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
anonymousMemoryCache :: forall m' . (MonadBuild m') =>
                        m' ((Tensor Value ResourceHandle, Tensor Value Variant))
                        -- ^ (__handle__, __deleter__)
                        --
                        -- * __handle__
                        --
                        -- * __deleter__
anonymousMemoryCache :: m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMemoryCache = OpParams -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMemoryCache' OpParams
forall a. a -> a
id
anonymousMemoryCache' :: forall m' . (MonadBuild m') => OpParams ->
                         m' ((Tensor Value ResourceHandle,
                              Tensor Value Variant))
                         -- ^ (__handle__, __deleter__)
                         --
                         -- * __handle__
                         --
                         -- * __deleter__
anonymousMemoryCache' :: OpParams -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMemoryCache' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Variant)
 -> m' (Tensor Value ResourceHandle, Tensor Value Variant))
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousMemoryCache"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
anonymousMultiDeviceIterator :: forall m' . (MonadBuild m') =>
                                [DataType] -- ^ __output_types__
                                -> m' ((Tensor Value ResourceHandle,
                                        Tensor Value Variant))
                                -- ^ (__handle__, __deleter__)
                                --
                                -- * __handle__
                                --
                                -- * __deleter__
anonymousMultiDeviceIterator :: [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMultiDeviceIterator = OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMultiDeviceIterator' OpParams
forall a. a -> a
id
anonymousMultiDeviceIterator' :: forall m' . (MonadBuild m') => OpParams ->
                                 [DataType] -- ^ __output_types__
                                 -> m' ((Tensor Value ResourceHandle,
                                         Tensor Value Variant))
                                 -- ^ (__handle__, __deleter__)
                                 --
                                 -- * __handle__
                                 --
                                 -- * __deleter__
anonymousMultiDeviceIterator' :: OpParams
-> [DataType]
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousMultiDeviceIterator' op'options :: OpParams
op'options output_types :: [DataType]
output_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Variant)
 -> m' (Tensor Value ResourceHandle, Tensor Value Variant))
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousMultiDeviceIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "deleter" type: DT_VARIANT }
attr {
  name: "devices" type: "list(string)" has_minimum: true minimum: 1
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
anonymousRandomSeedGenerator :: forall v'1 v'2 m' . (MonadBuild m') =>
                                Tensor v'1 Data.Int.Int64 -- ^ __seed__
                                -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                                -> m' ((Tensor Value ResourceHandle,
                                        Tensor Value Variant))
                                -- ^ (__handle__, __deleter__)
                                --
                                -- * __handle__
                                --
                                -- * __deleter__
anonymousRandomSeedGenerator :: Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousRandomSeedGenerator = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousRandomSeedGenerator' OpParams
forall a. a -> a
id
anonymousRandomSeedGenerator' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                 OpParams ->
                                 Tensor v'1 Data.Int.Int64 -- ^ __seed__
                                 -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                                 -> m' ((Tensor Value ResourceHandle,
                                         Tensor Value Variant))
                                 -- ^ (__handle__, __deleter__)
                                 --
                                 -- * __handle__
                                 --
                                 -- * __deleter__
anonymousRandomSeedGenerator' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousRandomSeedGenerator' op'options :: OpParams
op'options seed :: Tensor v'1 Int64
seed seed2 :: Tensor v'2 Int64
seed2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Variant)
 -> m' (Tensor Value ResourceHandle, Tensor Value Variant))
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seed,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
seed2]
        [Int64]
-> OpDef
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousRandomSeedGenerator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
anonymousSeedGenerator :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                          Tensor v'1 Data.Int.Int64 -- ^ __seed__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                          -> Tensor v'3 Bool -- ^ __reshuffle__
                          -> m' ((Tensor Value ResourceHandle,
                                  Tensor Value Variant))
                          -- ^ (__handle__, __deleter__)
                          --
                          -- * __handle__
                          --
                          -- * __deleter__
anonymousSeedGenerator :: Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousSeedGenerator = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousSeedGenerator' OpParams
forall a. a -> a
id
anonymousSeedGenerator' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                           OpParams ->
                           Tensor v'1 Data.Int.Int64 -- ^ __seed__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                           -> Tensor v'3 Bool -- ^ __reshuffle__
                           -> m' ((Tensor Value ResourceHandle,
                                   Tensor Value Variant))
                           -- ^ (__handle__, __deleter__)
                           --
                           -- * __handle__
                           --
                           -- * __deleter__
anonymousSeedGenerator' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
anonymousSeedGenerator' op'options :: OpParams
op'options seed :: Tensor v'1 Int64
seed seed2 :: Tensor v'2 Int64
seed2 reshuffle :: Tensor v'3 Bool
reshuffle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Variant)
 -> m' (Tensor Value ResourceHandle, Tensor Value Variant))
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
-> m' (Tensor Value ResourceHandle, Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seed,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
seed2,
                                                             Tensor v'3 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Bool
reshuffle]
        [Int64]
-> OpDef
-> Build (Tensor Value ResourceHandle, Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AnonymousSeedGenerator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
input_arg { name: "reshuffle" type: DT_BOOL }
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
any :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
       Tensor v'1 Bool -- ^ __input__
       -> Tensor v'2 tidx -- ^ __reduction_indices__
       -> Tensor Build Bool -- ^ __output__
any :: Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
any = OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) tidx.
OneOf '[Int32, Int64] tidx =>
OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
any' OpParams
forall a. a -> a
id
any' :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
        OpParams ->
        Tensor v'1 Bool -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build Bool -- ^ __output__
any' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
any' op'options :: OpParams
op'options input :: Tensor v'1 Bool
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Any"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_BOOL }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type: DT_BOOL }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
applyAdaMax :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                      OneOf '[(Data.Complex.Complex Double),
                                                              (Data.Complex.Complex Float),
                                                              Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
               Tensor Ref t -- ^ __var__
               -> Tensor Ref t -- ^ __m__
               -> Tensor Ref t -- ^ __v__
               -> Tensor v'4 t -- ^ __beta1_power__
               -> Tensor v'5 t -- ^ __lr__
               -> Tensor v'6 t -- ^ __beta1__
               -> Tensor v'7 t -- ^ __beta2__
               -> Tensor v'8 t -- ^ __epsilon__
               -> Tensor v'9 t -- ^ __grad__
               -> m' (Tensor Ref t) -- ^ __out__
applyAdaMax :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyAdaMax = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyAdaMax' OpParams
forall a. a -> a
id
applyAdaMax' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                       OneOf '[(Data.Complex.Complex Double),
                                                               (Data.Complex.Complex Float),
                                                               Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word32,
                                                               Data.Word.Word64,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] t) =>
                OpParams ->
                Tensor Ref t -- ^ __var__
                -> Tensor Ref t -- ^ __m__
                -> Tensor Ref t -- ^ __v__
                -> Tensor v'4 t -- ^ __beta1_power__
                -> Tensor v'5 t -- ^ __lr__
                -> Tensor v'6 t -- ^ __beta1__
                -> Tensor v'7 t -- ^ __beta2__
                -> Tensor v'8 t -- ^ __epsilon__
                -> Tensor v'9 t -- ^ __grad__
                -> m' (Tensor Ref t) -- ^ __out__
applyAdaMax' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyAdaMax' op'options :: OpParams
op'options var :: Tensor Ref t
var m :: Tensor Ref t
m v :: Tensor Ref t
v beta1_power :: Tensor v'4 t
beta1_power lr :: Tensor v'5 t
lr beta1 :: Tensor v'6 t
beta1 beta2 :: Tensor v'7 t
beta2 epsilon :: Tensor v'8 t
epsilon
             grad :: Tensor v'9 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
m,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta1_power,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
beta2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdaMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "m" type_attr: "T" is_ref: true }
input_arg { name: "v" type_attr: "T" is_ref: true }
input_arg { name: "beta1_power" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "beta1" type_attr: "T" }
input_arg { name: "beta2" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyAdadelta :: forall v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t) =>
                 Tensor Ref t -- ^ __var__
                 -> Tensor Ref t -- ^ __accum__
                 -> Tensor Ref t -- ^ __accum_update__
                 -> Tensor v'4 t -- ^ __lr__
                 -> Tensor v'5 t -- ^ __rho__
                 -> Tensor v'6 t -- ^ __epsilon__
                 -> Tensor v'7 t -- ^ __grad__
                 -> m' (Tensor Ref t) -- ^ __out__
applyAdadelta :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAdadelta = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAdadelta' OpParams
forall a. a -> a
id
applyAdadelta' :: forall v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t) =>
                  OpParams ->
                  Tensor Ref t -- ^ __var__
                  -> Tensor Ref t -- ^ __accum__
                  -> Tensor Ref t -- ^ __accum_update__
                  -> Tensor v'4 t -- ^ __lr__
                  -> Tensor v'5 t -- ^ __rho__
                  -> Tensor v'6 t -- ^ __epsilon__
                  -> Tensor v'7 t -- ^ __grad__
                  -> m' (Tensor Ref t) -- ^ __out__
applyAdadelta' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAdadelta' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum accum_update :: Tensor Ref t
accum_update lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho epsilon :: Tensor v'6 t
epsilon
               grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum_update,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
epsilon,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdadelta"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "accum_update" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyAdagrad :: forall v'3 v'4 t m' . (MonadBuild m',
                                       OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) =>
                Tensor Ref t -- ^ __var__
                -> Tensor Ref t -- ^ __accum__
                -> Tensor v'3 t -- ^ __lr__
                -> Tensor v'4 t -- ^ __grad__
                -> m' (Tensor Ref t) -- ^ __out__
applyAdagrad :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> m' (Tensor Ref t)
applyAdagrad = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> m' (Tensor Ref t)
applyAdagrad' OpParams
forall a. a -> a
id
applyAdagrad' :: forall v'3 v'4 t m' . (MonadBuild m',
                                        OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) => OpParams ->
                 Tensor Ref t -- ^ __var__
                 -> Tensor Ref t -- ^ __accum__
                 -> Tensor v'3 t -- ^ __lr__
                 -> Tensor v'4 t -- ^ __grad__
                 -> m' (Tensor Ref t) -- ^ __out__
applyAdagrad' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> m' (Tensor Ref t)
applyAdagrad' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
applyAdagradDA :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                     OneOf '[(Data.Complex.Complex Double),
                                                             (Data.Complex.Complex Float),
                                                             Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                  Tensor Ref t -- ^ __var__
                  -> Tensor Ref t -- ^ __gradient_accumulator__
                  -> Tensor Ref t -- ^ __gradient_squared_accumulator__
                  -> Tensor v'4 t -- ^ __grad__
                  -> Tensor v'5 t -- ^ __lr__
                  -> Tensor v'6 t -- ^ __l1__
                  -> Tensor v'7 t -- ^ __l2__
                  -> Tensor v'8 Data.Int.Int64 -- ^ __global_step__
                  -> m' (Tensor Ref t) -- ^ __out__
applyAdagradDA :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' (Tensor Ref t)
applyAdagradDA = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' (Tensor Ref t)
applyAdagradDA' OpParams
forall a. a -> a
id
applyAdagradDA' :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                      OneOf '[(Data.Complex.Complex Double),
                                                              (Data.Complex.Complex Float),
                                                              Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                   OpParams ->
                   Tensor Ref t -- ^ __var__
                   -> Tensor Ref t -- ^ __gradient_accumulator__
                   -> Tensor Ref t -- ^ __gradient_squared_accumulator__
                   -> Tensor v'4 t -- ^ __grad__
                   -> Tensor v'5 t -- ^ __lr__
                   -> Tensor v'6 t -- ^ __l1__
                   -> Tensor v'7 t -- ^ __l2__
                   -> Tensor v'8 Data.Int.Int64 -- ^ __global_step__
                   -> m' (Tensor Ref t) -- ^ __out__
applyAdagradDA' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' (Tensor Ref t)
applyAdagradDA' op'options :: OpParams
op'options var :: Tensor Ref t
var gradient_accumulator :: Tensor Ref t
gradient_accumulator gradient_squared_accumulator :: Tensor Ref t
gradient_squared_accumulator
                grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2 global_step :: Tensor v'8 Int64
global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
gradient_accumulator,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
gradient_squared_accumulator,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Int64
global_step]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdagradDA"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg {
  name: "gradient_accumulator" type_attr: "T" is_ref: true
}
input_arg {
  name: "gradient_squared_accumulator" type_attr: "T" is_ref: true
}
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "global_step" type: DT_INT64 }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyAdagradV2 :: forall v'3 v'4 v'5 t m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                  Tensor Ref t -- ^ __var__
                  -> Tensor Ref t -- ^ __accum__
                  -> Tensor v'3 t -- ^ __lr__
                  -> Tensor v'4 t -- ^ __epsilon__
                  -> Tensor v'5 t -- ^ __grad__
                  -> m' (Tensor Ref t) -- ^ __out__
applyAdagradV2 :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyAdagradV2 = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyAdagradV2' OpParams
forall a. a -> a
id
applyAdagradV2' :: forall v'3 v'4 v'5 t m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                   Tensor Ref t -- ^ __var__
                   -> Tensor Ref t -- ^ __accum__
                   -> Tensor v'3 t -- ^ __lr__
                   -> Tensor v'4 t -- ^ __epsilon__
                   -> Tensor v'5 t -- ^ __grad__
                   -> m' (Tensor Ref t) -- ^ __out__
applyAdagradV2' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyAdagradV2' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr epsilon :: Tensor v'4 t
epsilon grad :: Tensor v'5 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
epsilon,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdagradV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
applyAdam :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m' . (MonadBuild m',
                                                         OneOf '[(Data.Complex.Complex Double),
                                                                 (Data.Complex.Complex Float),
                                                                 Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Int.Int8,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word32,
                                                                 Data.Word.Word64,
                                                                 Data.Word.Word8,
                                                                 Double,
                                                                 Float] t) =>
             Tensor Ref t -- ^ __var__
             -> Tensor Ref t -- ^ __m__
             -> Tensor Ref t -- ^ __v__
             -> Tensor v'4 t -- ^ __beta1_power__
             -> Tensor v'5 t -- ^ __beta2_power__
             -> Tensor v'6 t -- ^ __lr__
             -> Tensor v'7 t -- ^ __beta1__
             -> Tensor v'8 t -- ^ __beta2__
             -> Tensor v'9 t -- ^ __epsilon__
             -> Tensor v'10 t -- ^ __grad__
             -> m' (Tensor Ref t) -- ^ __out__
applyAdam :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
applyAdam = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
applyAdam' OpParams
forall a. a -> a
id
applyAdam' :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t m' . (MonadBuild m',
                                                          OneOf '[(Data.Complex.Complex Double),
                                                                  (Data.Complex.Complex Float),
                                                                  Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Int.Int64,
                                                                  Data.Int.Int8,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word32,
                                                                  Data.Word.Word64,
                                                                  Data.Word.Word8,
                                                                  Double,
                                                                  Float] t) =>
              OpParams ->
              Tensor Ref t -- ^ __var__
              -> Tensor Ref t -- ^ __m__
              -> Tensor Ref t -- ^ __v__
              -> Tensor v'4 t -- ^ __beta1_power__
              -> Tensor v'5 t -- ^ __beta2_power__
              -> Tensor v'6 t -- ^ __lr__
              -> Tensor v'7 t -- ^ __beta1__
              -> Tensor v'8 t -- ^ __beta2__
              -> Tensor v'9 t -- ^ __epsilon__
              -> Tensor v'10 t -- ^ __grad__
              -> m' (Tensor Ref t) -- ^ __out__
applyAdam' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
applyAdam' op'options :: OpParams
op'options var :: Tensor Ref t
var m :: Tensor Ref t
m v :: Tensor Ref t
v beta1_power :: Tensor v'4 t
beta1_power beta2_power :: Tensor v'5 t
beta2_power lr :: Tensor v'6 t
lr beta1 :: Tensor v'7 t
beta1 beta2 :: Tensor v'8 t
beta2 epsilon :: Tensor v'9 t
epsilon
           grad :: Tensor v'10 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
m,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta1_power,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
beta2_power,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
beta1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
beta2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
epsilon,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAdam"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "m" type_attr: "T" is_ref: true }
input_arg { name: "v" type_attr: "T" is_ref: true }
input_arg { name: "beta1_power" type_attr: "T" }
input_arg { name: "beta2_power" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "beta1" type_attr: "T" }
input_arg { name: "beta2" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
applyAddSign :: forall v'3 v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                   OneOf '[(Data.Complex.Complex Double),
                                                           (Data.Complex.Complex Float),
                                                           Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double, Float] t) =>
                Tensor Ref t -- ^ __var__
                -> Tensor Ref t -- ^ __m__
                -> Tensor v'3 t -- ^ __lr__
                -> Tensor v'4 t -- ^ __alpha__
                -> Tensor v'5 t -- ^ __sign_decay__
                -> Tensor v'6 t -- ^ __beta__
                -> Tensor v'7 t -- ^ __grad__
                -> m' (Tensor Ref t) -- ^ __out__
applyAddSign :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAddSign = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) (v'7 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAddSign' OpParams
forall a. a -> a
id
applyAddSign' :: forall v'3 v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                    OneOf '[(Data.Complex.Complex Double),
                                                            (Data.Complex.Complex Float),
                                                            Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word16,
                                                            Data.Word.Word32,
                                                            Data.Word.Word64,
                                                            Data.Word.Word8,
                                                            Double, Float] t) =>
                 OpParams ->
                 Tensor Ref t -- ^ __var__
                 -> Tensor Ref t -- ^ __m__
                 -> Tensor v'3 t -- ^ __lr__
                 -> Tensor v'4 t -- ^ __alpha__
                 -> Tensor v'5 t -- ^ __sign_decay__
                 -> Tensor v'6 t -- ^ __beta__
                 -> Tensor v'7 t -- ^ __grad__
                 -> m' (Tensor Ref t) -- ^ __out__
applyAddSign' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyAddSign' op'options :: OpParams
op'options var :: Tensor Ref t
var m :: Tensor Ref t
m lr :: Tensor v'3 t
lr alpha :: Tensor v'4 t
alpha sign_decay :: Tensor v'5 t
sign_decay beta :: Tensor v'6 t
beta
              grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
alpha,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
sign_decay,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyAddSign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "m" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "sign_decay" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyCenteredRMSProp :: forall v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                           OneOf '[(Data.Complex.Complex Double),
                                                                   (Data.Complex.Complex Float),
                                                                   Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Int.Int64,
                                                                   Data.Int.Int8,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word32,
                                                                   Data.Word.Word64,
                                                                   Data.Word.Word8,
                                                                   Double,
                                                                   Float] t) =>
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __mg__
                        -> Tensor Ref t -- ^ __ms__
                        -> Tensor Ref t -- ^ __mom__
                        -> Tensor v'5 t -- ^ __lr__
                        -> Tensor v'6 t -- ^ __rho__
                        -> Tensor v'7 t -- ^ __momentum__
                        -> Tensor v'8 t -- ^ __epsilon__
                        -> Tensor v'9 t -- ^ __grad__
                        -> m' (Tensor Ref t) -- ^ __out__
applyCenteredRMSProp :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyCenteredRMSProp = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
forall (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyCenteredRMSProp' OpParams
forall a. a -> a
id
applyCenteredRMSProp' :: forall v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                            OneOf '[(Data.Complex.Complex Double),
                                                                    (Data.Complex.Complex Float),
                                                                    Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Int.Int64,
                                                                    Data.Int.Int8,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word32,
                                                                    Data.Word.Word64,
                                                                    Data.Word.Word8,
                                                                    Double,
                                                                    Float] t) =>
                         OpParams ->
                         Tensor Ref t -- ^ __var__
                         -> Tensor Ref t -- ^ __mg__
                         -> Tensor Ref t -- ^ __ms__
                         -> Tensor Ref t -- ^ __mom__
                         -> Tensor v'5 t -- ^ __lr__
                         -> Tensor v'6 t -- ^ __rho__
                         -> Tensor v'7 t -- ^ __momentum__
                         -> Tensor v'8 t -- ^ __epsilon__
                         -> Tensor v'9 t -- ^ __grad__
                         -> m' (Tensor Ref t) -- ^ __out__
applyCenteredRMSProp' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyCenteredRMSProp' op'options :: OpParams
op'options var :: Tensor Ref t
var mg :: Tensor Ref t
mg ms :: Tensor Ref t
ms mom :: Tensor Ref t
mom lr :: Tensor v'5 t
lr rho :: Tensor v'6 t
rho momentum :: Tensor v'7 t
momentum epsilon :: Tensor v'8 t
epsilon
                      grad :: Tensor v'9 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mg,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ms,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mom,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
rho,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
momentum,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyCenteredRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "mg" type_attr: "T" is_ref: true }
input_arg { name: "ms" type_attr: "T" is_ref: true }
input_arg { name: "mom" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyFtrl :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t) =>
             Tensor Ref t -- ^ __var__
             -> Tensor Ref t -- ^ __accum__
             -> Tensor Ref t -- ^ __linear__
             -> Tensor v'4 t -- ^ __grad__
             -> Tensor v'5 t -- ^ __lr__
             -> Tensor v'6 t -- ^ __l1__
             -> Tensor v'7 t -- ^ __l2__
             -> Tensor v'8 t -- ^ __lr_power__
             -> m' (Tensor Ref t) -- ^ __out__
applyFtrl :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyFtrl = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyFtrl' OpParams
forall a. a -> a
id
applyFtrl' :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t) =>
              OpParams ->
              Tensor Ref t -- ^ __var__
              -> Tensor Ref t -- ^ __accum__
              -> Tensor Ref t -- ^ __linear__
              -> Tensor v'4 t -- ^ __grad__
              -> Tensor v'5 t -- ^ __lr__
              -> Tensor v'6 t -- ^ __l1__
              -> Tensor v'7 t -- ^ __l2__
              -> Tensor v'8 t -- ^ __lr_power__
              -> m' (Tensor Ref t) -- ^ __out__
applyFtrl' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyFtrl' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum linear :: Tensor Ref t
linear grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2
           lr_power :: Tensor v'8 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
lr_power]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyFtrl"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "linear" type_attr: "T" is_ref: true }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
applyFtrlV2 :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                      OneOf '[(Data.Complex.Complex Double),
                                                              (Data.Complex.Complex Float),
                                                              Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
               Tensor Ref t -- ^ __var__
               -> Tensor Ref t -- ^ __accum__
               -> Tensor Ref t -- ^ __linear__
               -> Tensor v'4 t -- ^ __grad__
               -> Tensor v'5 t -- ^ __lr__
               -> Tensor v'6 t -- ^ __l1__
               -> Tensor v'7 t -- ^ __l2__
               -> Tensor v'8 t -- ^ __l2_shrinkage__
               -> Tensor v'9 t -- ^ __lr_power__
               -> m' (Tensor Ref t) -- ^ __out__
applyFtrlV2 :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyFtrlV2 = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyFtrlV2' OpParams
forall a. a -> a
id
applyFtrlV2' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t m' . (MonadBuild m',
                                                       OneOf '[(Data.Complex.Complex Double),
                                                               (Data.Complex.Complex Float),
                                                               Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word32,
                                                               Data.Word.Word64,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] t) =>
                OpParams ->
                Tensor Ref t -- ^ __var__
                -> Tensor Ref t -- ^ __accum__
                -> Tensor Ref t -- ^ __linear__
                -> Tensor v'4 t -- ^ __grad__
                -> Tensor v'5 t -- ^ __lr__
                -> Tensor v'6 t -- ^ __l1__
                -> Tensor v'7 t -- ^ __l2__
                -> Tensor v'8 t -- ^ __l2_shrinkage__
                -> Tensor v'9 t -- ^ __lr_power__
                -> m' (Tensor Ref t) -- ^ __out__
applyFtrlV2' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
applyFtrlV2' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum linear :: Tensor Ref t
linear grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2 l2_shrinkage :: Tensor v'8 t
l2_shrinkage
             lr_power :: Tensor v'9 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2_shrinkage,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
lr_power]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyFtrlV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "linear" type_attr: "T" is_ref: true }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "l2_shrinkage" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
applyGradientDescent :: forall v'2 v'3 t m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t) =>
                        Tensor Ref t -- ^ __var__
                        -> Tensor v'2 t -- ^ __alpha__
                        -> Tensor v'3 t -- ^ __delta__
                        -> m' (Tensor Ref t) -- ^ __out__
applyGradientDescent :: Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> m' (Tensor Ref t)
applyGradientDescent = OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> m' (Tensor Ref t)
applyGradientDescent' OpParams
forall a. a -> a
id
applyGradientDescent' :: forall v'2 v'3 t m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t) => OpParams ->
                         Tensor Ref t -- ^ __var__
                         -> Tensor v'2 t -- ^ __alpha__
                         -> Tensor v'3 t -- ^ __delta__
                         -> m' (Tensor Ref t) -- ^ __out__
applyGradientDescent' :: OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> m' (Tensor Ref t)
applyGradientDescent' op'options :: OpParams
op'options var :: Tensor Ref t
var alpha :: Tensor v'2 t
alpha delta :: Tensor v'3 t
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
delta]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "delta" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyMomentum :: forall v'3 v'4 v'5 t m' . (MonadBuild m',
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float),
                                                    Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t) =>
                 Tensor Ref t -- ^ __var__
                 -> Tensor Ref t -- ^ __accum__
                 -> Tensor v'3 t -- ^ __lr__
                 -> Tensor v'4 t -- ^ __grad__
                 -> Tensor v'5 t -- ^ __momentum__
                 -> m' (Tensor Ref t) -- ^ __out__
applyMomentum :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyMomentum = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyMomentum' OpParams
forall a. a -> a
id
applyMomentum' :: forall v'3 v'4 v'5 t m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) => OpParams ->
                  Tensor Ref t -- ^ __var__
                  -> Tensor Ref t -- ^ __accum__
                  -> Tensor v'3 t -- ^ __lr__
                  -> Tensor v'4 t -- ^ __grad__
                  -> Tensor v'5 t -- ^ __momentum__
                  -> m' (Tensor Ref t) -- ^ __out__
applyMomentum' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyMomentum' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad momentum :: Tensor v'5 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
momentum]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
applyPowerSign :: forall v'3 v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                     OneOf '[(Data.Complex.Complex Double),
                                                             (Data.Complex.Complex Float),
                                                             Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                  Tensor Ref t -- ^ __var__
                  -> Tensor Ref t -- ^ __m__
                  -> Tensor v'3 t -- ^ __lr__
                  -> Tensor v'4 t -- ^ __logbase__
                  -> Tensor v'5 t -- ^ __sign_decay__
                  -> Tensor v'6 t -- ^ __beta__
                  -> Tensor v'7 t -- ^ __grad__
                  -> m' (Tensor Ref t) -- ^ __out__
applyPowerSign :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyPowerSign = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) (v'7 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyPowerSign' OpParams
forall a. a -> a
id
applyPowerSign' :: forall v'3 v'4 v'5 v'6 v'7 t m' . (MonadBuild m',
                                                      OneOf '[(Data.Complex.Complex Double),
                                                              (Data.Complex.Complex Float),
                                                              Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                   OpParams ->
                   Tensor Ref t -- ^ __var__
                   -> Tensor Ref t -- ^ __m__
                   -> Tensor v'3 t -- ^ __lr__
                   -> Tensor v'4 t -- ^ __logbase__
                   -> Tensor v'5 t -- ^ __sign_decay__
                   -> Tensor v'6 t -- ^ __beta__
                   -> Tensor v'7 t -- ^ __grad__
                   -> m' (Tensor Ref t) -- ^ __out__
applyPowerSign' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' (Tensor Ref t)
applyPowerSign' op'options :: OpParams
op'options var :: Tensor Ref t
var m :: Tensor Ref t
m lr :: Tensor v'3 t
lr logbase :: Tensor v'4 t
logbase sign_decay :: Tensor v'5 t
sign_decay beta :: Tensor v'6 t
beta
                grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
logbase,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
sign_decay,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyPowerSign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "m" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "logbase" type_attr: "T" }
input_arg { name: "sign_decay" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyProximalAdagrad :: forall v'3 v'4 v'5 v'6 t m' . (MonadBuild m',
                                                       OneOf '[(Data.Complex.Complex Double),
                                                               (Data.Complex.Complex Float),
                                                               Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word32,
                                                               Data.Word.Word64,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] t) =>
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __accum__
                        -> Tensor v'3 t -- ^ __lr__
                        -> Tensor v'4 t -- ^ __l1__
                        -> Tensor v'5 t -- ^ __l2__
                        -> Tensor v'6 t -- ^ __grad__
                        -> m' (Tensor Ref t) -- ^ __out__
applyProximalAdagrad :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' (Tensor Ref t)
applyProximalAdagrad = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' (Tensor Ref t)
applyProximalAdagrad' OpParams
forall a. a -> a
id
applyProximalAdagrad' :: forall v'3 v'4 v'5 v'6 t m' . (MonadBuild m',
                                                        OneOf '[(Data.Complex.Complex Double),
                                                                (Data.Complex.Complex Float),
                                                                Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Int.Int8,
                                                                Data.Word.Word16,
                                                                Data.Word.Word32,
                                                                Data.Word.Word64,
                                                                Data.Word.Word8,
                                                                Double,
                                                                Float] t) =>
                         OpParams ->
                         Tensor Ref t -- ^ __var__
                         -> Tensor Ref t -- ^ __accum__
                         -> Tensor v'3 t -- ^ __lr__
                         -> Tensor v'4 t -- ^ __l1__
                         -> Tensor v'5 t -- ^ __l2__
                         -> Tensor v'6 t -- ^ __grad__
                         -> m' (Tensor Ref t) -- ^ __out__
applyProximalAdagrad' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' (Tensor Ref t)
applyProximalAdagrad' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr l1 :: Tensor v'4 t
l1 l2 :: Tensor v'5 t
l2 grad :: Tensor v'6 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l1,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
l2,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyProximalAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyProximalGradientDescent :: forall v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] t) =>
                                Tensor Ref t -- ^ __var__
                                -> Tensor v'2 t -- ^ __alpha__
                                -> Tensor v'3 t -- ^ __l1__
                                -> Tensor v'4 t -- ^ __l2__
                                -> Tensor v'5 t -- ^ __delta__
                                -> m' (Tensor Ref t) -- ^ __out__
applyProximalGradientDescent :: Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyProximalGradientDescent = OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *)
       (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyProximalGradientDescent' OpParams
forall a. a -> a
id
applyProximalGradientDescent' :: forall v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                                OneOf '[(Data.Complex.Complex Double),
                                                                        (Data.Complex.Complex Float),
                                                                        Data.Int.Int16,
                                                                        Data.Int.Int32,
                                                                        Data.Int.Int64,
                                                                        Data.Int.Int8,
                                                                        Data.Word.Word16,
                                                                        Data.Word.Word32,
                                                                        Data.Word.Word64,
                                                                        Data.Word.Word8,
                                                                        Double,
                                                                        Float] t) =>
                                 OpParams ->
                                 Tensor Ref t -- ^ __var__
                                 -> Tensor v'2 t -- ^ __alpha__
                                 -> Tensor v'3 t -- ^ __l1__
                                 -> Tensor v'4 t -- ^ __l2__
                                 -> Tensor v'5 t -- ^ __delta__
                                 -> m' (Tensor Ref t) -- ^ __out__
applyProximalGradientDescent' :: OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Ref t)
applyProximalGradientDescent' op'options :: OpParams
op'options var :: Tensor Ref t
var alpha :: Tensor v'2 t
alpha l1 :: Tensor v'3 t
l1 l2 :: Tensor v'4 t
l2
                              delta :: Tensor v'5 t
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
l1,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l2,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
delta]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyProximalGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "delta" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
applyRMSProp :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                   OneOf '[(Data.Complex.Complex Double),
                                                           (Data.Complex.Complex Float),
                                                           Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double, Float] t) =>
                Tensor Ref t -- ^ __var__
                -> Tensor Ref t -- ^ __ms__
                -> Tensor Ref t -- ^ __mom__
                -> Tensor v'4 t -- ^ __lr__
                -> Tensor v'5 t -- ^ __rho__
                -> Tensor v'6 t -- ^ __momentum__
                -> Tensor v'7 t -- ^ __epsilon__
                -> Tensor v'8 t -- ^ __grad__
                -> m' (Tensor Ref t) -- ^ __out__
applyRMSProp :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyRMSProp = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyRMSProp' OpParams
forall a. a -> a
id
applyRMSProp' :: forall v'4 v'5 v'6 v'7 v'8 t m' . (MonadBuild m',
                                                    OneOf '[(Data.Complex.Complex Double),
                                                            (Data.Complex.Complex Float),
                                                            Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word16,
                                                            Data.Word.Word32,
                                                            Data.Word.Word64,
                                                            Data.Word.Word8,
                                                            Double, Float] t) =>
                 OpParams ->
                 Tensor Ref t -- ^ __var__
                 -> Tensor Ref t -- ^ __ms__
                 -> Tensor Ref t -- ^ __mom__
                 -> Tensor v'4 t -- ^ __lr__
                 -> Tensor v'5 t -- ^ __rho__
                 -> Tensor v'6 t -- ^ __momentum__
                 -> Tensor v'7 t -- ^ __epsilon__
                 -> Tensor v'8 t -- ^ __grad__
                 -> m' (Tensor Ref t) -- ^ __out__
applyRMSProp' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' (Tensor Ref t)
applyRMSProp' op'options :: OpParams
op'options var :: Tensor Ref t
var ms :: Tensor Ref t
ms mom :: Tensor Ref t
mom lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho momentum :: Tensor v'6 t
momentum epsilon :: Tensor v'7 t
epsilon
              grad :: Tensor v'8 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ms,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mom,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
epsilon,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
grad]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ApplyRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "ms" type_attr: "T" is_ref: true }
input_arg { name: "mom" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
approximateEqual :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) =>
                    Tensor v'1 t -- ^ __x__
                    -> Tensor v'2 t -- ^ __y__
                    -> Tensor Build Bool -- ^ __z__
approximateEqual :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
approximateEqual = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
approximateEqual' OpParams
forall a. a -> a
id
approximateEqual' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t) => OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 t -- ^ __y__
                     -> Tensor Build Bool -- ^ __z__
approximateEqual' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
approximateEqual' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ApproximateEqual"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "tolerance" type: "float" default_value { f: 1.0e-5 }
}
-}
-- | 
argMax :: forall v'1 v'2 t tidx
          output_type . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Bool,
                                 Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                 Data.Int.Int8, Data.Word.Word16,
                                 Data.Word.Word32, Data.Word.Word64,
                                 Data.Word.Word8, Double, Float] t,
                         OneOf '[Data.Int.Int32, Data.Int.Int64] tidx,
                         OneOf '[Data.Int.Int32, Data.Int.Int64] output_type) =>
          Tensor v'1 t -- ^ __input__
          -> Tensor v'2 tidx -- ^ __dimension__
          -> Tensor Build output_type -- ^ __output__
argMax :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMax = OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx output_type.
(OneOf
   '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
     Word16, Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) =>
OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMax' OpParams
forall a. a -> a
id
argMax' :: forall v'1 v'2 t tidx
           output_type . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Bool,
                                  Data.Int.Int16, Data.Int.Int32,
                                  Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t, OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tidx,
                          OneOf '[Data.Int.Int32,
                                  Data.Int.Int64] output_type) => OpParams ->
           Tensor v'1 t -- ^ __input__
           -> Tensor v'2 tidx -- ^ __dimension__
           -> Tensor Build output_type -- ^ __output__
argMax' :: OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMax' op'options :: OpParams
op'options input :: Tensor v'1 t
input dimension :: Tensor v'2 tidx
dimension | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build output_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build output_type)
-> Build OpDef -> Tensor Build output_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
dimension]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ArgMax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "dimension" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "output_type" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_BOOL
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "output_type"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
argMin :: forall v'1 v'2 t tidx
          output_type . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Bool,
                                 Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                 Data.Int.Int8, Data.Word.Word16,
                                 Data.Word.Word32, Data.Word.Word64,
                                 Data.Word.Word8, Double, Float] t,
                         OneOf '[Data.Int.Int32, Data.Int.Int64] tidx,
                         OneOf '[Data.Int.Int32, Data.Int.Int64] output_type) =>
          Tensor v'1 t -- ^ __input__
          -> Tensor v'2 tidx -- ^ __dimension__
          -> Tensor Build output_type -- ^ __output__
argMin :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMin = OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx output_type.
(OneOf
   '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
     Word16, Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] output_type) =>
OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMin' OpParams
forall a. a -> a
id
argMin' :: forall v'1 v'2 t tidx
           output_type . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Bool,
                                  Data.Int.Int16, Data.Int.Int32,
                                  Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t, OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tidx,
                          OneOf '[Data.Int.Int32,
                                  Data.Int.Int64] output_type) => OpParams ->
           Tensor v'1 t -- ^ __input__
           -> Tensor v'2 tidx -- ^ __dimension__
           -> Tensor Build output_type -- ^ __output__
argMin' :: OpParams
-> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
argMin' op'options :: OpParams
op'options input :: Tensor v'1 t
input dimension :: Tensor v'2 tidx
dimension | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build output_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build output_type)
-> Build OpDef -> Tensor Build output_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
dimension]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ArgMin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "dimension" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "output_type" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_BOOL
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "output_type"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
asString :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float), Bool,
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8, Double,
                                    Float] t) => Tensor v'1 t -- ^ __input__
            -> Tensor Build Data.ByteString.ByteString -- ^ __output__
asString :: Tensor v'1 t -> Tensor Build ByteString
asString = OpParams -> Tensor v'1 t -> Tensor Build ByteString
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build ByteString
asString' OpParams
forall a. a -> a
id
asString' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float), Bool,
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8, Double,
                                     Float] t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build Data.ByteString.ByteString -- ^ __output__
asString' :: OpParams -> Tensor v'1 t -> Tensor Build ByteString
asString' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AsString"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_STRING }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_BOOL
    }
  }
}
attr { name: "precision" type: "int" default_value { i: -1 } }
attr { name: "scientific" type: "bool" default_value { b: false } }
attr { name: "shortest" type: "bool" default_value { b: false } }
attr { name: "width" type: "int" default_value { i: -1 } }
attr { name: "fill" type: "string" default_value { s: "" } }
-}
-- | 
asin :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
asin :: Tensor v'1 t -> Tensor Build t
asin = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
asin' OpParams
forall a. a -> a
id
asin' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
asin' :: OpParams -> Tensor v'1 t -> Tensor Build t
asin' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Asin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
asinh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
asinh :: Tensor v'1 t -> Tensor Build t
asinh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
asinh' OpParams
forall a. a -> a
id
asinh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
asinh' :: OpParams -> Tensor v'1 t -> Tensor Build t
asinh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Asinh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
assert :: forall v'1 v'2 t m' . (MonadBuild m', TensorTypes t) =>
          Tensor v'1 Bool -- ^ __condition__
          -> TensorList (v'2) t -- ^ __data__
          -> m' (ControlNode)
assert :: Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
assert = OpParams -> Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (t :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes t) =>
OpParams -> Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
assert' OpParams
forall a. a -> a
id
assert' :: forall v'1 v'2 t m' . (MonadBuild m', TensorTypes t) => OpParams ->
           Tensor v'1 Bool -- ^ __condition__
           -> TensorList (v'2) t -- ^ __data__
           -> m' (ControlNode)
assert' :: OpParams -> Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
assert' op'options :: OpParams
op'options condition :: Tensor v'1 Bool
condition data' :: TensorList v'2 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
condition,
                                                             TensorList v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 t
data']
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Assert"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "condition" type: DT_BOOL }
input_arg { name: "data" type_list_attr: "T" }
attr { name: "T" type: "list(type)" has_minimum: true minimum: 1 }
attr { name: "summarize" type: "int" default_value { i: 3 } }
-}
-- | 
assertCardinalityDataset :: [DataType] -- ^ __output_types__
                            -> Tensor v'1 Variant -- ^ __input_dataset__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __cardinality__
                            -> Tensor Build Variant -- ^ __handle__
assertCardinalityDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
assertCardinalityDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
assertCardinalityDataset' OpParams
forall a. a -> a
id
assertCardinalityDataset' :: OpParams ->
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __cardinality__
                             -> Tensor Build Variant -- ^ __handle__
assertCardinalityDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
assertCardinalityDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                          cardinality :: Tensor v'2 Int64
cardinality | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
cardinality]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AssertCardinalityDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "cardinality" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
assertNextDataset :: [DataType] -- ^ __output_types__
                     -> Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 Data.ByteString.ByteString -- ^ __transformations__
                     -> Tensor Build Variant -- ^ __handle__
assertNextDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
assertNextDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
assertNextDataset' OpParams
forall a. a -> a
id
assertNextDataset' :: OpParams ->
                      [DataType] -- ^ __output_types__
                      -> Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __transformations__
                      -> Tensor Build Variant -- ^ __handle__
assertNextDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
assertNextDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                   transformations :: Tensor v'2 ByteString
transformations | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
transformations]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AssertNextDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "transformations" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
assign :: forall v'2 t m' . (MonadBuild m', TensorType t) =>
          Tensor Ref t -- ^ __ref__
          -> Tensor v'2 t -- ^ __value__
          -> m' (Tensor Ref t) -- ^ __output_ref__
assign :: Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assign = OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
forall (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assign' OpParams
forall a. a -> a
id
assign' :: forall v'2 t m' . (MonadBuild m', TensorType t) => OpParams ->
           Tensor Ref t -- ^ __ref__
           -> Tensor v'2 t -- ^ __value__
           -> m' (Tensor Ref t) -- ^ __output_ref__
assign' :: OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assign' op'options :: OpParams
op'options ref :: Tensor Ref t
ref value :: Tensor v'2 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Assign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr {
  name: "validate_shape" type: "bool" default_value { b: true }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
assignAdd :: forall v'2 t m' . (MonadBuild m',
                                OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
             Tensor Ref t -- ^ __ref__
             -> Tensor v'2 t -- ^ __value__
             -> m' (Tensor Ref t) -- ^ __output_ref__
assignAdd :: Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignAdd = OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
forall (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignAdd' OpParams
forall a. a -> a
id
assignAdd' :: forall v'2 t m' . (MonadBuild m',
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 t -- ^ __value__
              -> m' (Tensor Ref t) -- ^ __output_ref__
assignAdd' :: OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignAdd' op'options :: OpParams
op'options ref :: Tensor Ref t
ref value :: Tensor v'2 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AssignAdd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
assignAddVariableOp :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                  TensorType dtype) =>
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 dtype -- ^ __value__
                       -> m' (ControlNode)
assignAddVariableOp :: Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignAddVariableOp = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignAddVariableOp' OpParams
forall a. a -> a
id
assignAddVariableOp' :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                   TensorType dtype) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __resource__
                        -> Tensor v'2 dtype -- ^ __value__
                        -> m' (ControlNode)
assignAddVariableOp' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignAddVariableOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource value :: Tensor v'2 dtype
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 dtype
value]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AssignAddVariableOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
assignSub :: forall v'2 t m' . (MonadBuild m',
                                OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
             Tensor Ref t -- ^ __ref__
             -> Tensor v'2 t -- ^ __value__
             -> m' (Tensor Ref t) -- ^ __output_ref__
assignSub :: Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignSub = OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
forall (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignSub' OpParams
forall a. a -> a
id
assignSub' :: forall v'2 t m' . (MonadBuild m',
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 t -- ^ __value__
              -> m' (Tensor Ref t) -- ^ __output_ref__
assignSub' :: OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
assignSub' op'options :: OpParams
op'options ref :: Tensor Ref t
ref value :: Tensor v'2 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AssignSub"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
assignSubVariableOp :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                  TensorType dtype) =>
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 dtype -- ^ __value__
                       -> m' (ControlNode)
assignSubVariableOp :: Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignSubVariableOp = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignSubVariableOp' OpParams
forall a. a -> a
id
assignSubVariableOp' :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                   TensorType dtype) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __resource__
                        -> Tensor v'2 dtype -- ^ __value__
                        -> m' (ControlNode)
assignSubVariableOp' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignSubVariableOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource value :: Tensor v'2 dtype
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 dtype
value]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AssignSubVariableOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
assignVariableOp :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                               TensorType dtype) =>
                    Tensor v'1 ResourceHandle -- ^ __resource__
                    -> Tensor v'2 dtype -- ^ __value__
                    -> m' (ControlNode)
assignVariableOp :: Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignVariableOp = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignVariableOp' OpParams
forall a. a -> a
id
assignVariableOp' :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                TensorType dtype) => OpParams ->
                     Tensor v'1 ResourceHandle -- ^ __resource__
                     -> Tensor v'2 dtype -- ^ __value__
                     -> m' (ControlNode)
assignVariableOp' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
assignVariableOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource value :: Tensor v'2 dtype
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 dtype
value]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "AssignVariableOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
atan :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
atan :: Tensor v'1 t -> Tensor Build t
atan = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
atan' OpParams
forall a. a -> a
id
atan' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
atan' :: OpParams -> Tensor v'1 t -> Tensor Build t
atan' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Atan"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
atan2 :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __y__
         -> Tensor v'2 t -- ^ __x__
         -> Tensor Build t -- ^ __z__
atan2 :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
atan2 = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
atan2' OpParams
forall a. a -> a
id
atan2' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __y__
          -> Tensor v'2 t -- ^ __x__
          -> Tensor Build t -- ^ __z__
atan2' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
atan2' op'options :: OpParams
op'options y :: Tensor v'1 t
y x :: Tensor v'2 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Atan2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
atanh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
atanh :: Tensor v'1 t -> Tensor Build t
atanh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
atanh' OpParams
forall a. a -> a
id
atanh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
atanh' :: OpParams -> Tensor v'1 t -> Tensor Build t
atanh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Atanh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
audioSpectrogram :: Data.Int.Int64 -- ^ __stride__
                    -> Data.Int.Int64 -- ^ __window_size__
                    -> Tensor v'1 Float -- ^ __input__
                    -> Tensor Build Float -- ^ __spectrogram__
audioSpectrogram :: Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
audioSpectrogram = OpParams
-> Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
forall (v'1 :: * -> *).
OpParams
-> Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
audioSpectrogram' OpParams
forall a. a -> a
id
audioSpectrogram' :: OpParams ->
                     Data.Int.Int64 -- ^ __stride__
                     -> Data.Int.Int64 -- ^ __window_size__
                     -> Tensor v'1 Float -- ^ __input__
                     -> Tensor Build Float -- ^ __spectrogram__
audioSpectrogram' :: OpParams
-> Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
audioSpectrogram' op'options :: OpParams
op'options stride :: Int64
stride window_size :: Int64
window_size input :: Tensor v'1 Float
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AudioSpectrogram"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "stride" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
stride
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "window_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
window_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_FLOAT }
output_arg { name: "spectrogram" type: DT_FLOAT }
attr { name: "window_size" type: "int" }
attr { name: "stride" type: "int" }
attr {
  name: "magnitude_squared" type: "bool" default_value { b: false }
}
-}
-- | 
audioSummary :: Float -- ^ __sample_rate__
                -> Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                -> Tensor v'2 Float -- ^ __tensor__
                -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
audioSummary :: Float
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor Build ByteString
audioSummary = OpParams
-> Float
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Float
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor Build ByteString
audioSummary' OpParams
forall a. a -> a
id
audioSummary' :: OpParams ->
                 Float -- ^ __sample_rate__
                 -> Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                 -> Tensor v'2 Float -- ^ __tensor__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
audioSummary' :: OpParams
-> Float
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor Build ByteString
audioSummary' op'options :: OpParams
op'options sample_rate :: Float
sample_rate tag :: Tensor v'1 ByteString
tag tensor :: Tensor v'2 Float
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tag,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AudioSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sample_rate" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
sample_rate
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type: DT_FLOAT }
output_arg { name: "summary" type: DT_STRING }
attr { name: "sample_rate" type: "float" }
attr {
  name: "max_outputs"
  type: "int"
  default_value { i: 3 }
  has_minimum: true
  minimum: 1
}
-}
-- | 
audioSummaryV2 :: Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                  -> Tensor v'2 Float -- ^ __tensor__
                  -> Tensor v'3 Float -- ^ __sample_rate__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
audioSummaryV2 :: Tensor v'1 ByteString
-> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build ByteString
audioSummaryV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build ByteString
audioSummaryV2' OpParams
forall a. a -> a
id
audioSummaryV2' :: OpParams ->
                   Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                   -> Tensor v'2 Float -- ^ __tensor__
                   -> Tensor v'3 Float -- ^ __sample_rate__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
audioSummaryV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build ByteString
audioSummaryV2' op'options :: OpParams
op'options tag :: Tensor v'1 ByteString
tag tensor :: Tensor v'2 Float
tensor sample_rate :: Tensor v'3 Float
sample_rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tag,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
tensor,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
sample_rate]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AudioSummaryV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type: DT_FLOAT }
input_arg { name: "sample_rate" type: DT_FLOAT }
output_arg { name: "summary" type: DT_STRING }
attr {
  name: "max_outputs"
  type: "int"
  default_value { i: 3 }
  has_minimum: true
  minimum: 1
}
-}
-- | 
autoShardDataset :: [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __num_workers__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                    -> Tensor Build Variant -- ^ __handle__
autoShardDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
autoShardDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
autoShardDataset' OpParams
forall a. a -> a
id
autoShardDataset' :: OpParams ->
                     [DataType] -- ^ __output_types__
                     -> Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __num_workers__
                     -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                     -> Tensor Build Variant -- ^ __handle__
autoShardDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
autoShardDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset num_workers :: Tensor v'2 Int64
num_workers
                  index :: Tensor v'3 Int64
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_workers,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
index]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AutoShardDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_workers" type: DT_INT64 }
input_arg { name: "index" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "auto_shard_policy" type: "int" default_value { i: 0 }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
avgPool :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           ByteString -- ^ __padding__
           -> Tensor v'1 t -- ^ __value__
           -> Tensor Build t -- ^ __output__
avgPool :: ByteString -> Tensor v'1 t -> Tensor Build t
avgPool = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool' OpParams
forall a. a -> a
id
avgPool' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            ByteString -- ^ __padding__
            -> Tensor v'1 t -- ^ __value__
            -> Tensor Build t -- ^ __output__
avgPool' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool' op'options :: OpParams
op'options padding :: ByteString
padding value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AvgPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
avgPool3D :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             ByteString -- ^ __padding__
             -> Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
avgPool3D :: ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3D = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3D' OpParams
forall a. a -> a
id
avgPool3D' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              ByteString -- ^ __padding__
              -> Tensor v'1 t -- ^ __input__
              -> Tensor Build t -- ^ __output__
avgPool3D' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
avgPool3D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AvgPool3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
avgPool3DGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) =>
                 ByteString -- ^ __padding__
                 -> Tensor v'1 Data.Int.Int32 -- ^ __orig_input_shape__
                 -> Tensor v'2 t -- ^ __grad__
                 -> Tensor Build t -- ^ __output__
avgPool3DGrad :: ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPool3DGrad = OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPool3DGrad' OpParams
forall a. a -> a
id
avgPool3DGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                              Float] t) => OpParams ->
                  ByteString -- ^ __padding__
                  -> Tensor v'1 Data.Int.Int32 -- ^ __orig_input_shape__
                  -> Tensor v'2 t -- ^ __grad__
                  -> Tensor Build t -- ^ __output__
avgPool3DGrad' :: OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPool3DGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input_shape :: Tensor v'1 Int32
orig_input_shape grad :: Tensor v'2 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
orig_input_shape,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AvgPool3DGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input_shape" type: DT_INT32 }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
avgPoolGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                           Float] t) =>
               ByteString -- ^ __padding__
               -> Tensor v'1 Data.Int.Int32 -- ^ __orig_input_shape__
               -> Tensor v'2 t -- ^ __grad__
               -> Tensor Build t -- ^ __output__
avgPoolGrad :: ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPoolGrad = OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPoolGrad' OpParams
forall a. a -> a
id
avgPoolGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) => OpParams ->
                ByteString -- ^ __padding__
                -> Tensor v'1 Data.Int.Int32 -- ^ __orig_input_shape__
                -> Tensor v'2 t -- ^ __grad__
                -> Tensor Build t -- ^ __output__
avgPoolGrad' :: OpParams
-> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
avgPoolGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input_shape :: Tensor v'1 Int32
orig_input_shape grad :: Tensor v'2 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
orig_input_shape,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "AvgPoolGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input_shape" type: DT_INT32 }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
bandedTriangularSolve :: forall v'1 v'2
                         t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Word.Word16, Double, Float] t) =>
                         Tensor v'1 t -- ^ __matrix__
                         -> Tensor v'2 t -- ^ __rhs__
                         -> Tensor Build t -- ^ __output__
bandedTriangularSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bandedTriangularSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bandedTriangularSolve' OpParams
forall a. a -> a
id
bandedTriangularSolve' :: forall v'1 v'2
                          t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Word.Word16, Double, Float] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __matrix__
                          -> Tensor v'2 t -- ^ __rhs__
                          -> Tensor Build t -- ^ __output__
bandedTriangularSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bandedTriangularSolve' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BandedTriangularSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "lower" type: "bool" default_value { b: true } }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
barrier :: forall m' . (MonadBuild m') => [DataType] -- ^ __component_types__
           -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
barrier :: [DataType] -> m' (Tensor Ref ByteString)
barrier = OpParams -> [DataType] -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Ref ByteString)
barrier' OpParams
forall a. a -> a
id
barrier' :: forall m' . (MonadBuild m') => OpParams ->
            [DataType] -- ^ __component_types__
            -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
barrier' :: OpParams -> [DataType] -> m' (Tensor Ref ByteString)
barrier' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Barrier"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
barrierClose :: forall m' . (MonadBuild m') =>
                Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                -> m' (ControlNode)
barrierClose :: Tensor Ref ByteString -> m' ControlNode
barrierClose = OpParams -> Tensor Ref ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' ControlNode
barrierClose' OpParams
forall a. a -> a
id
barrierClose' :: forall m' . (MonadBuild m') => OpParams ->
                 Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                 -> m' (ControlNode)
barrierClose' :: OpParams -> Tensor Ref ByteString -> m' ControlNode
barrierClose' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BarrierClose"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "cancel_pending_enqueues"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
barrierIncompleteSize :: forall m' . (MonadBuild m') =>
                         Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                         -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
barrierIncompleteSize :: Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierIncompleteSize = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierIncompleteSize' OpParams
forall a. a -> a
id
barrierIncompleteSize' :: forall m' . (MonadBuild m') => OpParams ->
                          Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                          -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
barrierIncompleteSize' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierIncompleteSize' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BarrierIncompleteSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
barrierInsertMany :: forall v'2 v'3 t m' . (MonadBuild m', TensorType t) =>
                     Data.Int.Int64 -- ^ __component_index__
                     -> Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.ByteString.ByteString -- ^ __keys__
                     -> Tensor v'3 t -- ^ __values__
                     -> m' (ControlNode)
barrierInsertMany :: Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 t
-> m' ControlNode
barrierInsertMany = OpParams
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 t
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 t
-> m' ControlNode
barrierInsertMany' OpParams
forall a. a -> a
id
barrierInsertMany' :: forall v'2 v'3 t m' . (MonadBuild m', TensorType t) =>
                      OpParams ->
                      Data.Int.Int64 -- ^ __component_index__
                      -> Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __keys__
                      -> Tensor v'3 t -- ^ __values__
                      -> m' (ControlNode)
barrierInsertMany' :: OpParams
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 t
-> m' ControlNode
barrierInsertMany' op'options :: OpParams
op'options component_index :: Int64
component_index handle :: Tensor Ref ByteString
handle keys :: Tensor v'2 ByteString
keys
                   values :: Tensor v'3 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
keys,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BarrierInsertMany"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
component_index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "keys" type: DT_STRING }
input_arg { name: "values" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "component_index" type: "int" }
-}
-- | 
barrierReadySize :: forall m' . (MonadBuild m') =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
barrierReadySize :: Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierReadySize = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierReadySize' OpParams
forall a. a -> a
id
barrierReadySize' :: forall m' . (MonadBuild m') => OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
barrierReadySize' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
barrierReadySize' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BarrierReadySize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
barrierTakeMany :: forall v'2 component_types m' . (MonadBuild m',
                                                    TensorTypes component_types) =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __num_elements__
                   -> m' ((Tensor Value Data.Int.Int64,
                           Tensor Value Data.ByteString.ByteString,
                           TensorList (Value) component_types))
                   -- ^ (__indices__, __keys__, __values__)
                   --
                   -- * __indices__
                   --
                   -- * __keys__
                   --
                   -- * __values__
barrierTakeMany :: Tensor Ref ByteString
-> Tensor v'2 Int32
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
barrierTakeMany = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
forall (v'2 :: * -> *) (component_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
barrierTakeMany' OpParams
forall a. a -> a
id
barrierTakeMany' :: forall v'2 component_types m' . (MonadBuild m',
                                                     TensorTypes component_types) =>
                    OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __num_elements__
                    -> m' ((Tensor Value Data.Int.Int64,
                            Tensor Value Data.ByteString.ByteString,
                            TensorList (Value) component_types))
                    -- ^ (__indices__, __keys__, __values__)
                    --
                    -- * __indices__
                    --
                    -- * __keys__
                    --
                    -- * __values__
barrierTakeMany' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
barrierTakeMany' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle num_elements :: Tensor v'2 Int32
num_elements | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Int64, Tensor Value ByteString,
   TensorList Value component_types)
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Int64, Tensor Value ByteString,
    TensorList Value component_types)
 -> m'
      (Tensor Value Int64, Tensor Value ByteString,
       TensorList Value component_types))
-> Build
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
-> m'
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_elements]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value ByteString,
      TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BarrierTakeMany"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "num_elements" type: DT_INT32 }
output_arg { name: "indices" type: DT_INT64 }
output_arg { name: "keys" type: DT_STRING }
output_arg { name: "values" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "allow_small_batch" type: "bool" default_value { b: false }
}
attr {
  name: "wait_for_incomplete" type: "bool" default_value { b: false }
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
batch :: forall v'1 t . (TensorTypes t) =>
         Data.Int.Int64 -- ^ __batch_timeout_micros__
         -> Data.Int.Int64 -- ^ __grad_timeout_micros__
         -> Data.Int.Int64 -- ^ __max_batch_size__
         -> Data.Int.Int64 -- ^ __num_batch_threads__
         -> TensorList (v'1) t -- ^ __in_tensors__
         -> (TensorList (Build) t, Tensor Build Data.Int.Int64,
             Tensor Build Data.Int.Int64)
         -- ^ (__batched_tensors__, __batch_index__, __id__)
         --
         -- * __batched_tensors__
         --
         -- * __batch_index__
         --
         -- * __id__
batch :: Int64
-> Int64
-> Int64
-> Int64
-> TensorList v'1 t
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
batch = OpParams
-> Int64
-> Int64
-> Int64
-> Int64
-> TensorList v'1 t
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
forall (v'1 :: * -> *) (t :: [*]).
TensorTypes t =>
OpParams
-> Int64
-> Int64
-> Int64
-> Int64
-> TensorList v'1 t
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
batch' OpParams
forall a. a -> a
id
batch' :: forall v'1 t . (TensorTypes t) => OpParams ->
          Data.Int.Int64 -- ^ __batch_timeout_micros__
          -> Data.Int.Int64 -- ^ __grad_timeout_micros__
          -> Data.Int.Int64 -- ^ __max_batch_size__
          -> Data.Int.Int64 -- ^ __num_batch_threads__
          -> TensorList (v'1) t -- ^ __in_tensors__
          -> (TensorList (Build) t, Tensor Build Data.Int.Int64,
              Tensor Build Data.Int.Int64)
          -- ^ (__batched_tensors__, __batch_index__, __id__)
          --
          -- * __batched_tensors__
          --
          -- * __batch_index__
          --
          -- * __id__
batch' :: OpParams
-> Int64
-> Int64
-> Int64
-> Int64
-> TensorList v'1 t
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
batch' op'options :: OpParams
op'options batch_timeout_micros :: Int64
batch_timeout_micros grad_timeout_micros :: Int64
grad_timeout_micros max_batch_size :: Int64
max_batch_size
       num_batch_threads :: Int64
num_batch_threads in_tensors :: TensorList v'1 t
in_tensors | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (TensorList Build t, Tensor Build Int64, Tensor Build Int64))
-> Build OpDef
-> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 t
in_tensors]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Batch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "batch_timeout_micros" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
batch_timeout_micros
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "grad_timeout_micros" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
grad_timeout_micros
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_batch_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_batch_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_batch_threads" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_batch_threads
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "in_tensors" type_list_attr: "T" }
output_arg { name: "batched_tensors" type_list_attr: "T" }
output_arg { name: "batch_index" type: DT_INT64 }
output_arg { name: "id" type: DT_INT64 }
attr { name: "num_batch_threads" type: "int" }
attr { name: "max_batch_size" type: "int" }
attr {
  name: "max_enqueued_batches" type: "int" default_value { i: 10 }
}
attr { name: "batch_timeout_micros" type: "int" }
attr {
  name: "allowed_batch_sizes"
  type: "list(int)"
  default_value { list { } }
}
attr { name: "grad_timeout_micros" type: "int" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "batching_queue" type: "string" default_value { s: "" }
}
attr { name: "T" type: "list(type)" has_minimum: true minimum: 1 }
-}
-- | 
batchCholesky :: forall v'1 t . (OneOf '[Double, Float] t) =>
                 Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
batchCholesky :: Tensor v'1 t -> Tensor Build t
batchCholesky = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchCholesky' OpParams
forall a. a -> a
id
batchCholesky' :: forall v'1 t . (OneOf '[Double, Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor Build t -- ^ __output__
batchCholesky' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchCholesky' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchCholesky"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchCholeskyGrad :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                     Tensor v'1 t -- ^ __l__
                     -> Tensor v'2 t -- ^ __grad__
                     -> Tensor Build t -- ^ __output__
batchCholeskyGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchCholeskyGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchCholeskyGrad' OpParams
forall a. a -> a
id
batchCholeskyGrad' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                      OpParams ->
                      Tensor v'1 t -- ^ __l__
                      -> Tensor v'2 t -- ^ __grad__
                      -> Tensor Build t -- ^ __output__
batchCholeskyGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchCholeskyGrad' op'options :: OpParams
op'options l :: Tensor v'1 t
l grad :: Tensor v'2 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
l,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchCholeskyGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "l" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
batchDataset :: [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                -> Tensor Build Variant -- ^ __handle__
batchDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
batchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
batchDataset' OpParams
forall a. a -> a
id
batchDataset' :: OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                 -> Tensor Build Variant -- ^ __handle__
batchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
batchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
              batch_size :: Tensor v'2 Int64
batch_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
batchDatasetV2 :: [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                  -> Tensor v'3 Bool -- ^ __drop_remainder__
                  -> Tensor Build Variant -- ^ __handle__
batchDatasetV2 :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> Tensor Build Variant
batchDatasetV2 = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> Tensor Build Variant
batchDatasetV2' OpParams
forall a. a -> a
id
batchDatasetV2' :: OpParams ->
                   [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                   -> Tensor v'3 Bool -- ^ __drop_remainder__
                   -> Tensor Build Variant -- ^ __handle__
batchDatasetV2' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Bool
-> Tensor Build Variant
batchDatasetV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset batch_size :: Tensor v'2 Int64
batch_size
                drop_remainder :: Tensor v'3 Bool
drop_remainder | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size,
                                                             Tensor v'3 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Bool
drop_remainder]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchDatasetV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
input_arg { name: "drop_remainder" type: DT_BOOL }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "parallel_copy" type: "bool" default_value { b: false }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
batchFFT :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
            -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT' OpParams
forall a. a -> a
id
batchFFT' :: OpParams ->
             Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
             -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchFFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchFFT2D :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
              -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT2D = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT2D' OpParams
forall a. a -> a
id
batchFFT2D' :: OpParams ->
               Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
               -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT2D' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT2D' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchFFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchFFT3D :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
              -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT3D = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT3D' OpParams
forall a. a -> a
id
batchFFT3D' :: OpParams ->
               Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
               -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchFFT3D' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchFFT3D' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchFFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchIFFT :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
             -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT' OpParams
forall a. a -> a
id
batchIFFT' :: OpParams ->
              Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
              -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchIFFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchIFFT2D :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
               -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT2D = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT2D' OpParams
forall a. a -> a
id
batchIFFT2D' :: OpParams ->
                Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
                -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT2D' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT2D' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchIFFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchIFFT3D :: Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
               -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT3D = OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT3D' OpParams
forall a. a -> a
id
batchIFFT3D' :: OpParams ->
                Tensor v'1 (Data.Complex.Complex Float) -- ^ __input__
                -> Tensor Build (Data.Complex.Complex Float) -- ^ __output__
batchIFFT3D' :: OpParams
-> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
batchIFFT3D' op'options :: OpParams
op'options input :: Tensor v'1 (Complex Float)
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build (Complex Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build (Complex Float))
-> Build OpDef -> Tensor Build (Complex Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 (Complex Float) -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 (Complex Float)
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchIFFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_COMPLEX64 }
output_arg { name: "output" type: DT_COMPLEX64 }
-}
-- | 
batchMatMul :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Int.Int32, Data.Int.Int64,
                                           Data.Word.Word16, Double,
                                           Float] t) => Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __output__
batchMatMul :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMul = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMul' OpParams
forall a. a -> a
id
batchMatMul' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] t) => OpParams ->
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __y__
                -> Tensor Build t -- ^ __output__
batchMatMul' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMul' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr { name: "adj_x" type: "bool" default_value { b: false } }
attr { name: "adj_y" type: "bool" default_value { b: false } }
-}
-- | 
batchMatMulV2 :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int32, Data.Int.Int64,
                                             Data.Word.Word16, Double,
                                             Float] t) =>
                 Tensor v'1 t -- ^ __x__
                 -> Tensor v'2 t -- ^ __y__
                 -> Tensor Build t -- ^ __output__
batchMatMulV2 :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMulV2 = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMulV2' OpParams
forall a. a -> a
id
batchMatMulV2' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int32, Data.Int.Int64,
                                              Data.Word.Word16, Double,
                                              Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __x__
                  -> Tensor v'2 t -- ^ __y__
                  -> Tensor Build t -- ^ __output__
batchMatMulV2' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatMulV2' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatMulV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr { name: "adj_x" type: "bool" default_value { b: false } }
attr { name: "adj_y" type: "bool" default_value { b: false } }
-}
-- | 
batchMatrixBandPart :: forall v'1 v'2 v'3 t . (TensorType t) =>
                       Tensor v'1 t -- ^ __input__
                       -> Tensor v'2 Data.Int.Int64 -- ^ __num_lower__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __num_upper__
                       -> Tensor Build t -- ^ __band__
batchMatrixBandPart :: Tensor v'1 t
-> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
batchMatrixBandPart = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
batchMatrixBandPart' OpParams
forall a. a -> a
id
batchMatrixBandPart' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                        Tensor v'1 t -- ^ __input__
                        -> Tensor v'2 Data.Int.Int64 -- ^ __num_lower__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __num_upper__
                        -> Tensor Build t -- ^ __band__
batchMatrixBandPart' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
batchMatrixBandPart' op'options :: OpParams
op'options input :: Tensor v'1 t
input num_lower :: Tensor v'2 Int64
num_lower num_upper :: Tensor v'3 Int64
num_upper | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_lower,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
num_upper]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixBandPart"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "num_lower" type: DT_INT64 }
input_arg { name: "num_upper" type: DT_INT64 }
output_arg { name: "band" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
batchMatrixDeterminant :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Double, Float] t) =>
                          Tensor v'1 t -- ^ __input__
                          -> Tensor Build t -- ^ __output__
batchMatrixDeterminant :: Tensor v'1 t -> Tensor Build t
batchMatrixDeterminant = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDeterminant' OpParams
forall a. a -> a
id
batchMatrixDeterminant' :: forall v'1
                           t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float), Double,
                                        Float] t) => OpParams ->
                           Tensor v'1 t -- ^ __input__
                           -> Tensor Build t -- ^ __output__
batchMatrixDeterminant' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDeterminant' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixDeterminant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
batchMatrixDiag :: forall v'1 t . (TensorType t) =>
                   Tensor v'1 t -- ^ __diagonal__
                   -> Tensor Build t -- ^ __output__
batchMatrixDiag :: Tensor v'1 t -> Tensor Build t
batchMatrixDiag = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDiag' OpParams
forall a. a -> a
id
batchMatrixDiag' :: forall v'1 t . (TensorType t) => OpParams ->
                    Tensor v'1 t -- ^ __diagonal__
                    -> Tensor Build t -- ^ __output__
batchMatrixDiag' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDiag' op'options :: OpParams
op'options diagonal :: Tensor v'1 t
diagonal | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonal]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixDiag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonal" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
batchMatrixDiagPart :: forall v'1 t . (TensorType t) =>
                       Tensor v'1 t -- ^ __input__
                       -> Tensor Build t -- ^ __diagonal__
batchMatrixDiagPart :: Tensor v'1 t -> Tensor Build t
batchMatrixDiagPart = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDiagPart' OpParams
forall a. a -> a
id
batchMatrixDiagPart' :: forall v'1 t . (TensorType t) => OpParams ->
                        Tensor v'1 t -- ^ __input__
                        -> Tensor Build t -- ^ __diagonal__
batchMatrixDiagPart' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixDiagPart' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixDiagPart"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "diagonal" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
batchMatrixInverse :: forall v'1 t . (OneOf '[Double, Float] t) =>
                      Tensor v'1 t -- ^ __input__
                      -> Tensor Build t -- ^ __output__
batchMatrixInverse :: Tensor v'1 t -> Tensor Build t
batchMatrixInverse = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixInverse' OpParams
forall a. a -> a
id
batchMatrixInverse' :: forall v'1 t . (OneOf '[Double, Float] t) => OpParams ->
                       Tensor v'1 t -- ^ __input__
                       -> Tensor Build t -- ^ __output__
batchMatrixInverse' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchMatrixInverse' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixInverse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchMatrixSetDiag :: forall v'1 v'2 t . (TensorType t) =>
                      Tensor v'1 t -- ^ __input__
                      -> Tensor v'2 t -- ^ __diagonal__
                      -> Tensor Build t -- ^ __output__
batchMatrixSetDiag :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSetDiag = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSetDiag' OpParams
forall a. a -> a
id
batchMatrixSetDiag' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                       Tensor v'1 t -- ^ __input__
                       -> Tensor v'2 t -- ^ __diagonal__
                       -> Tensor Build t -- ^ __output__
batchMatrixSetDiag' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSetDiag' op'options :: OpParams
op'options input :: Tensor v'1 t
input diagonal :: Tensor v'2 t
diagonal | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
diagonal]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixSetDiag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "diagonal" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
batchMatrixSolve :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                    Tensor v'1 t -- ^ __matrix__
                    -> Tensor v'2 t -- ^ __rhs__
                    -> Tensor Build t -- ^ __output__
batchMatrixSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSolve' OpParams
forall a. a -> a
id
batchMatrixSolve' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __matrix__
                     -> Tensor v'2 t -- ^ __rhs__
                     -> Tensor Build t -- ^ __output__
batchMatrixSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixSolve' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchMatrixSolveLs :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                      Tensor v'1 t -- ^ __matrix__
                      -> Tensor v'2 t -- ^ __rhs__
                      -> Tensor v'3 Double -- ^ __l2_regularizer__
                      -> Tensor Build t -- ^ __output__
batchMatrixSolveLs :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
batchMatrixSolveLs = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
batchMatrixSolveLs' OpParams
forall a. a -> a
id
batchMatrixSolveLs' :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __matrix__
                       -> Tensor v'2 t -- ^ __rhs__
                       -> Tensor v'3 Double -- ^ __l2_regularizer__
                       -> Tensor Build t -- ^ __output__
batchMatrixSolveLs' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
batchMatrixSolveLs' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs l2_regularizer :: Tensor v'3 Double
l2_regularizer | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs,
                                                             Tensor v'3 Double -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Double
l2_regularizer]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixSolveLs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
input_arg { name: "l2_regularizer" type: DT_DOUBLE }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
attr { name: "fast" type: "bool" default_value { b: true } }
-}
-- | 
batchMatrixTriangularSolve :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                              Tensor v'1 t -- ^ __matrix__
                              -> Tensor v'2 t -- ^ __rhs__
                              -> Tensor Build t -- ^ __output__
batchMatrixTriangularSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixTriangularSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixTriangularSolve' OpParams
forall a. a -> a
id
batchMatrixTriangularSolve' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                               OpParams ->
                               Tensor v'1 t -- ^ __matrix__
                               -> Tensor v'2 t -- ^ __rhs__
                               -> Tensor Build t -- ^ __output__
batchMatrixTriangularSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
batchMatrixTriangularSolve' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchMatrixTriangularSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "lower" type: "bool" default_value { b: true } }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchNormWithGlobalNormalization :: forall v'1 v'2 v'3 v'4 v'5
                                    t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t) =>
                                    Bool -- ^ __scale_after_normalization__
                                    -> Float -- ^ __variance_epsilon__
                                    -> Tensor v'1 t -- ^ __t__
                                    -> Tensor v'2 t -- ^ __m__
                                    -> Tensor v'3 t -- ^ __v__
                                    -> Tensor v'4 t -- ^ __beta__
                                    -> Tensor v'5 t -- ^ __gamma__
                                    -> Tensor Build t -- ^ __result__
batchNormWithGlobalNormalization :: Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor Build t
batchNormWithGlobalNormalization = OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor Build t
batchNormWithGlobalNormalization' OpParams
forall a. a -> a
id
batchNormWithGlobalNormalization' :: forall v'1 v'2 v'3 v'4 v'5
                                     t . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t) => OpParams ->
                                     Bool -- ^ __scale_after_normalization__
                                     -> Float -- ^ __variance_epsilon__
                                     -> Tensor v'1 t -- ^ __t__
                                     -> Tensor v'2 t -- ^ __m__
                                     -> Tensor v'3 t -- ^ __v__
                                     -> Tensor v'4 t -- ^ __beta__
                                     -> Tensor v'5 t -- ^ __gamma__
                                     -> Tensor Build t -- ^ __result__
batchNormWithGlobalNormalization' :: OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor Build t
batchNormWithGlobalNormalization' op'options :: OpParams
op'options scale_after_normalization :: Bool
scale_after_normalization
                                  variance_epsilon :: Float
variance_epsilon t :: Tensor v'1 t
t m :: Tensor v'2 t
m v :: Tensor v'3 t
v beta :: Tensor v'4 t
beta
                                  gamma :: Tensor v'5 t
gamma | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
t,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
gamma]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchNormWithGlobalNormalization"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "scale_after_normalization" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
scale_after_normalization
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "variance_epsilon" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
variance_epsilon
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "t" type_attr: "T" }
input_arg { name: "m" type_attr: "T" }
input_arg { name: "v" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
input_arg { name: "gamma" type_attr: "T" }
output_arg { name: "result" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "variance_epsilon" type: "float" }
attr { name: "scale_after_normalization" type: "bool" }
-}
-- | 
batchNormWithGlobalNormalizationGrad :: forall v'1 v'2 v'3 v'4 v'5
                                        t . (OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                                        Bool -- ^ __scale_after_normalization__
                                        -> Float -- ^ __variance_epsilon__
                                        -> Tensor v'1 t -- ^ __t__
                                        -> Tensor v'2 t -- ^ __m__
                                        -> Tensor v'3 t -- ^ __v__
                                        -> Tensor v'4 t -- ^ __gamma__
                                        -> Tensor v'5 t -- ^ __backprop__
                                        -> (Tensor Build t, Tensor Build t,
                                            Tensor Build t, Tensor Build t,
                                            Tensor Build t)
                                        -- ^ (__dx__, __dm__, __dv__, __db__, __dg__)
                                        --
                                        -- * __dx__
                                        --
                                        -- * __dm__
                                        --
                                        -- * __dv__
                                        --
                                        -- * __db__
                                        --
                                        -- * __dg__
batchNormWithGlobalNormalizationGrad :: Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
batchNormWithGlobalNormalizationGrad = OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
batchNormWithGlobalNormalizationGrad' OpParams
forall a. a -> a
id
batchNormWithGlobalNormalizationGrad' :: forall v'1 v'2 v'3 v'4 v'5
                                         t . (OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                                         Bool -- ^ __scale_after_normalization__
                                         -> Float -- ^ __variance_epsilon__
                                         -> Tensor v'1 t -- ^ __t__
                                         -> Tensor v'2 t -- ^ __m__
                                         -> Tensor v'3 t -- ^ __v__
                                         -> Tensor v'4 t -- ^ __gamma__
                                         -> Tensor v'5 t -- ^ __backprop__
                                         -> (Tensor Build t, Tensor Build t,
                                             Tensor Build t, Tensor Build t,
                                             Tensor Build t)
                                         -- ^ (__dx__, __dm__, __dv__, __db__, __dg__)
                                         --
                                         -- * __dx__
                                         --
                                         -- * __dm__
                                         --
                                         -- * __dv__
                                         --
                                         -- * __db__
                                         --
                                         -- * __dg__
batchNormWithGlobalNormalizationGrad' :: OpParams
-> Bool
-> Float
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
batchNormWithGlobalNormalizationGrad' op'options :: OpParams
op'options scale_after_normalization :: Bool
scale_after_normalization
                                      variance_epsilon :: Float
variance_epsilon t :: Tensor v'1 t
t m :: Tensor v'2 t
m v :: Tensor v'3 t
v gamma :: Tensor v'4 t
gamma
                                      backprop :: Tensor v'5 t
backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
t,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
gamma,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchNormWithGlobalNormalizationGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "scale_after_normalization" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
scale_after_normalization
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "variance_epsilon" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
variance_epsilon
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "t" type_attr: "T" }
input_arg { name: "m" type_attr: "T" }
input_arg { name: "v" type_attr: "T" }
input_arg { name: "gamma" type_attr: "T" }
input_arg { name: "backprop" type_attr: "T" }
output_arg { name: "dx" type_attr: "T" }
output_arg { name: "dm" type_attr: "T" }
output_arg { name: "dv" type_attr: "T" }
output_arg { name: "db" type_attr: "T" }
output_arg { name: "dg" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "variance_epsilon" type: "float" }
attr { name: "scale_after_normalization" type: "bool" }
-}
-- | 
batchSelfAdjointEig :: forall v'1 t . (OneOf '[Double, Float] t) =>
                       Tensor v'1 t -- ^ __input__
                       -> Tensor Build t -- ^ __output__
batchSelfAdjointEig :: Tensor v'1 t -> Tensor Build t
batchSelfAdjointEig = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
batchSelfAdjointEig' OpParams
forall a. a -> a
id
batchSelfAdjointEig' :: forall v'1 t . (OneOf '[Double, Float] t) => OpParams ->
                        Tensor v'1 t -- ^ __input__
                        -> Tensor Build t -- ^ __output__
batchSelfAdjointEig' :: OpParams -> Tensor v'1 t -> Tensor Build t
batchSelfAdjointEig' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchSelfAdjointEig"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchSelfAdjointEigV2 :: forall v'1 t . (OneOf '[Double, Float] t) =>
                         Tensor v'1 t -- ^ __input__
                         -> (Tensor Build t, Tensor Build t) -- ^ (__e__, __v__)
                         --
                         -- * __e__
                         --
                         -- * __v__
batchSelfAdjointEigV2 :: Tensor v'1 t -> (Tensor Build t, Tensor Build t)
batchSelfAdjointEigV2 = OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
batchSelfAdjointEigV2' OpParams
forall a. a -> a
id
batchSelfAdjointEigV2' :: forall v'1 t . (OneOf '[Double, Float] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __input__
                          -> (Tensor Build t, Tensor Build t)
                          -- ^ (__e__, __v__)
                          --
                          -- * __e__
                          --
                          -- * __v__
batchSelfAdjointEigV2' :: OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
batchSelfAdjointEigV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchSelfAdjointEigV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "e" type_attr: "T" }
output_arg { name: "v" type_attr: "T" }
attr { name: "compute_v" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
batchSvd :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float), Double,
                                    Float] t) => Tensor v'1 t -- ^ __input__
            -> (Tensor Build t, Tensor Build t, Tensor Build t)
            -- ^ (__s__, __u__, __v__)
            --
            -- * __s__
            --
            -- * __u__
            --
            -- * __v__
batchSvd :: Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
batchSvd = OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
batchSvd' OpParams
forall a. a -> a
id
batchSvd' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float), Double,
                                     Float] t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> (Tensor Build t, Tensor Build t, Tensor Build t)
             -- ^ (__s__, __u__, __v__)
             --
             -- * __s__
             --
             -- * __u__
             --
             -- * __v__
batchSvd' :: OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
batchSvd' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchSvd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "s" type_attr: "T" }
output_arg { name: "u" type_attr: "T" }
output_arg { name: "v" type_attr: "T" }
attr { name: "compute_uv" type: "bool" default_value { b: true } }
attr {
  name: "full_matrices" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
batchToSpace :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tidx) =>
                Data.Int.Int64 -- ^ __block_size__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor v'2 tidx -- ^ __crops__
                -> Tensor Build t -- ^ __output__
batchToSpace :: Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
batchToSpace = OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
batchToSpace' OpParams
forall a. a -> a
id
batchToSpace' :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tidx) =>
                 OpParams ->
                 Data.Int.Int64 -- ^ __block_size__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 tidx -- ^ __crops__
                 -> Tensor Build t -- ^ __output__
batchToSpace' :: OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
batchToSpace' op'options :: OpParams
op'options block_size :: Int64
block_size input :: Tensor v'1 t
input crops :: Tensor v'2 tidx
crops | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
crops]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchToSpace"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "block_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
block_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "crops" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "block_size" type: "int" has_minimum: true minimum: 2
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
batchToSpaceND :: forall v'1 v'2 v'3 t tblock_shape tcrops . (TensorType t,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tblock_shape,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tcrops) =>
                  Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 tblock_shape -- ^ __block_shape__
                  -> Tensor v'3 tcrops -- ^ __crops__
                  -> Tensor Build t -- ^ __output__
batchToSpaceND :: Tensor v'1 t
-> Tensor v'2 tblock_shape -> Tensor v'3 tcrops -> Tensor Build t
batchToSpaceND = OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       tblock_shape tcrops.
(TensorType t, OneOf '[Int32, Int64] tblock_shape,
 OneOf '[Int32, Int64] tcrops) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
batchToSpaceND' OpParams
forall a. a -> a
id
batchToSpaceND' :: forall v'1 v'2 v'3 t tblock_shape tcrops . (TensorType t,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tblock_shape,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tcrops) =>
                   OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 tblock_shape -- ^ __block_shape__
                   -> Tensor v'3 tcrops -- ^ __crops__
                   -> Tensor Build t -- ^ __output__
batchToSpaceND' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
batchToSpaceND' op'options :: OpParams
op'options input :: Tensor v'1 t
input block_shape :: Tensor v'2 tblock_shape
block_shape crops :: Tensor v'3 tcrops
crops | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tblock_shape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tblock_shape
block_shape,
                                                             Tensor v'3 tcrops -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tcrops
crops]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BatchToSpaceND"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tblock_shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tblock_shape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tblock_shape
forall a. HasCallStack => a
undefined :: tblock_shape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcrops" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcrops -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcrops
forall a. HasCallStack => a
undefined :: tcrops)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "block_shape" type_attr: "Tblock_shape" }
input_arg { name: "crops" type_attr: "Tcrops" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tblock_shape"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tcrops"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
besselI0 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselI0 :: Tensor v'1 t -> Tensor Build t
besselI0 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselI0' OpParams
forall a. a -> a
id
besselI0' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselI0' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselI0' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselI0"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselI0e :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselI0e :: Tensor v'1 t -> Tensor Build t
besselI0e = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselI0e' OpParams
forall a. a -> a
id
besselI0e' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
besselI0e' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselI0e' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselI0e"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselI1 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselI1 :: Tensor v'1 t -> Tensor Build t
besselI1 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselI1' OpParams
forall a. a -> a
id
besselI1' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselI1' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselI1' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselI1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselI1e :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselI1e :: Tensor v'1 t -> Tensor Build t
besselI1e = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselI1e' OpParams
forall a. a -> a
id
besselI1e' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
besselI1e' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselI1e' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselI1e"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselJ0 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselJ0 :: Tensor v'1 t -> Tensor Build t
besselJ0 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselJ0' OpParams
forall a. a -> a
id
besselJ0' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselJ0' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselJ0' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselJ0"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselJ1 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselJ1 :: Tensor v'1 t -> Tensor Build t
besselJ1 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselJ1' OpParams
forall a. a -> a
id
besselJ1' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselJ1' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselJ1' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselJ1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselK0 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselK0 :: Tensor v'1 t -> Tensor Build t
besselK0 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselK0' OpParams
forall a. a -> a
id
besselK0' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselK0' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselK0' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselK0"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselK0e :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselK0e :: Tensor v'1 t -> Tensor Build t
besselK0e = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselK0e' OpParams
forall a. a -> a
id
besselK0e' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
besselK0e' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselK0e' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselK0e"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselK1 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselK1 :: Tensor v'1 t -> Tensor Build t
besselK1 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselK1' OpParams
forall a. a -> a
id
besselK1' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselK1' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselK1' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselK1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselK1e :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselK1e :: Tensor v'1 t -> Tensor Build t
besselK1e = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselK1e' OpParams
forall a. a -> a
id
besselK1e' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
besselK1e' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselK1e' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselK1e"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselY0 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselY0 :: Tensor v'1 t -> Tensor Build t
besselY0 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselY0' OpParams
forall a. a -> a
id
besselY0' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselY0' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselY0' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselY0"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
besselY1 :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
besselY1 :: Tensor v'1 t -> Tensor Build t
besselY1 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
besselY1' OpParams
forall a. a -> a
id
besselY1' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
besselY1' :: OpParams -> Tensor v'1 t -> Tensor Build t
besselY1' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BesselY1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
betainc :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
           Tensor v'1 t -- ^ __a__
           -> Tensor v'2 t -- ^ __b__
           -> Tensor v'3 t -- ^ __x__
           -> Tensor Build t -- ^ __z__
betainc :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
betainc = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
betainc' OpParams
forall a. a -> a
id
betainc' :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __a__
            -> Tensor v'2 t -- ^ __b__
            -> Tensor v'3 t -- ^ __x__
            -> Tensor Build t -- ^ __z__
betainc' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
betainc' op'options :: OpParams
op'options a :: Tensor v'1 t
a b :: Tensor v'2 t
b x :: Tensor v'3 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Betainc"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
biasAdd :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) =>
           Tensor v'1 t -- ^ __value__
           -> Tensor v'2 t -- ^ __bias__
           -> Tensor Build t -- ^ __output__
biasAdd :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAdd = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAdd' OpParams
forall a. a -> a
id
biasAdd' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __value__
            -> Tensor v'2 t -- ^ __bias__
            -> Tensor Build t -- ^ __output__
biasAdd' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAdd' op'options :: OpParams
op'options value :: Tensor v'1 t
value bias :: Tensor v'2 t
bias | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
bias]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BiasAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
input_arg { name: "bias" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
-}
-- | 
biasAddGrad :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) =>
               Tensor v'1 t -- ^ __out_backprop__
               -> Tensor Build t -- ^ __output__
biasAddGrad :: Tensor v'1 t -> Tensor Build t
biasAddGrad = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
biasAddGrad' OpParams
forall a. a -> a
id
biasAddGrad' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
                Tensor v'1 t -- ^ __out_backprop__
                -> Tensor Build t -- ^ __output__
biasAddGrad' :: OpParams -> Tensor v'1 t -> Tensor Build t
biasAddGrad' op'options :: OpParams
op'options out_backprop :: Tensor v'1 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BiasAddGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
-}
-- | 
biasAddV1 :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) =>
             Tensor v'1 t -- ^ __value__
             -> Tensor v'2 t -- ^ __bias__
             -> Tensor Build t -- ^ __output__
biasAddV1 :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAddV1 = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAddV1' OpParams
forall a. a -> a
id
biasAddV1' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) => OpParams ->
              Tensor v'1 t -- ^ __value__
              -> Tensor v'2 t -- ^ __bias__
              -> Tensor Build t -- ^ __output__
biasAddV1' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
biasAddV1' op'options :: OpParams
op'options value :: Tensor v'1 t
value bias :: Tensor v'2 t
bias | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
bias]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BiasAddV1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
input_arg { name: "bias" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
bincount :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Double, Float] t) =>
            Tensor v'1 Data.Int.Int32 -- ^ __arr__
            -> Tensor v'2 Data.Int.Int32 -- ^ __size__
            -> Tensor v'3 t -- ^ __weights__
            -> Tensor Build t -- ^ __bins__
bincount :: Tensor v'1 Int32
-> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
bincount = OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
bincount' OpParams
forall a. a -> a
id
bincount' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Double, Float] t) => OpParams ->
             Tensor v'1 Data.Int.Int32 -- ^ __arr__
             -> Tensor v'2 Data.Int.Int32 -- ^ __size__
             -> Tensor v'3 t -- ^ __weights__
             -> Tensor Build t -- ^ __bins__
bincount' :: OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
bincount' op'options :: OpParams
op'options arr :: Tensor v'1 Int32
arr size :: Tensor v'2 Int32
size weights :: Tensor v'3 t
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
arr,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Bincount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "arr" type: DT_INT32 }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "weights" type_attr: "T" }
output_arg { name: "bins" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
bitcast :: forall v'1 t type' . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] type') =>
           Tensor v'1 t -- ^ __input__
           -> Tensor Build type' -- ^ __output__
bitcast :: Tensor v'1 t -> Tensor Build type'
bitcast = OpParams -> Tensor v'1 t -> Tensor Build type'
forall (v'1 :: * -> *) t type'.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   type') =>
OpParams -> Tensor v'1 t -> Tensor Build type'
bitcast' OpParams
forall a. a -> a
id
bitcast' :: forall v'1 t type' . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t,
                                  OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] type') => OpParams ->
            Tensor v'1 t -- ^ __input__
            -> Tensor Build type' -- ^ __output__
bitcast' :: OpParams -> Tensor v'1 t -> Tensor Build type'
bitcast' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build type'
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build type')
-> Build OpDef -> Tensor Build type'
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Bitcast"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ type' -> DataType
forall a. TensorType a => a -> DataType
tensorType (type'
forall a. HasCallStack => a
undefined :: type')
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "type" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT64
      type: DT_INT32
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
      type: DT_INT8
      type: DT_INT16
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT16
      type: DT_QUINT16
      type: DT_QINT32
    }
  }
}
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT64
      type: DT_INT32
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
      type: DT_INT8
      type: DT_INT16
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT16
      type: DT_QUINT16
      type: DT_QINT32
    }
  }
}
-}
-- | 
bitwiseAnd :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64,
                                          Data.Word.Word8] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build t -- ^ __z__
bitwiseAnd :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseAnd = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseAnd' OpParams
forall a. a -> a
id
bitwiseAnd' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64,
                                           Data.Word.Word8] t) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __z__
bitwiseAnd' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseAnd' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BitwiseAnd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
bitwiseOr :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64,
                                         Data.Word.Word8] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
bitwiseOr :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseOr = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseOr' OpParams
forall a. a -> a
id
bitwiseOr' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64,
                                          Data.Word.Word8] t) => OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build t -- ^ __z__
bitwiseOr' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseOr' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BitwiseOr"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
bitwiseXor :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64,
                                          Data.Word.Word8] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build t -- ^ __z__
bitwiseXor :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseXor = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseXor' OpParams
forall a. a -> a
id
bitwiseXor' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64,
                                           Data.Word.Word8] t) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __z__
bitwiseXor' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
bitwiseXor' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BitwiseXor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
blockLSTM :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
             t . (OneOf '[Data.Word.Word16, Float] t) =>
             Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
             -> Tensor v'2 t -- ^ __x__
             -> Tensor v'3 t -- ^ __cs_prev__
             -> Tensor v'4 t -- ^ __h_prev__
             -> Tensor v'5 t -- ^ __w__
             -> Tensor v'6 t -- ^ __wci__
             -> Tensor v'7 t -- ^ __wcf__
             -> Tensor v'8 t -- ^ __wco__
             -> Tensor v'9 t -- ^ __b__
             -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
                 Tensor Build t, Tensor Build t, Tensor Build t)
             -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
             --
             -- * __i__
             --
             -- * __cs__
             --
             -- * __f__
             --
             -- * __o__
             --
             -- * __ci__
             --
             -- * __co__
             --
             -- * __h__
blockLSTM :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTM = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTM' OpParams
forall a. a -> a
id
blockLSTM' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
              t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
              Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
              -> Tensor v'2 t -- ^ __x__
              -> Tensor v'3 t -- ^ __cs_prev__
              -> Tensor v'4 t -- ^ __h_prev__
              -> Tensor v'5 t -- ^ __w__
              -> Tensor v'6 t -- ^ __wci__
              -> Tensor v'7 t -- ^ __wcf__
              -> Tensor v'8 t -- ^ __wco__
              -> Tensor v'9 t -- ^ __b__
              -> (Tensor Build t, Tensor Build t, Tensor Build t,
                  Tensor Build t, Tensor Build t, Tensor Build t,
                  Tensor Build t)
              -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
              --
              -- * __i__
              --
              -- * __cs__
              --
              -- * __f__
              --
              -- * __o__
              --
              -- * __ci__
              --
              -- * __co__
              --
              -- * __h__
blockLSTM' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTM' op'options :: OpParams
op'options seq_len_max :: Tensor v'1 Int64
seq_len_max x :: Tensor v'2 t
x cs_prev :: Tensor v'3 t
cs_prev h_prev :: Tensor v'4 t
h_prev w :: Tensor v'5 t
w wci :: Tensor v'6 t
wci wcf :: Tensor v'7 t
wcf wco :: Tensor v'8 t
wco
           b :: Tensor v'9 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seq_len_max,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
cs_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
h_prev,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
w,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wci,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wcf,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
wco,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BlockLSTM"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seq_len_max" type: DT_INT64 }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "i" type_attr: "T" }
output_arg { name: "cs" type_attr: "T" }
output_arg { name: "f" type_attr: "T" }
output_arg { name: "o" type_attr: "T" }
output_arg { name: "ci" type_attr: "T" }
output_arg { name: "co" type_attr: "T" }
output_arg { name: "h" type_attr: "T" }
attr { name: "forget_bias" type: "float" default_value { f: 1.0 } }
attr { name: "cell_clip" type: "float" default_value { f: 3.0 } }
attr {
  name: "use_peephole" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
blockLSTMGrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 v'13
                 v'14 v'15 v'16 v'17 v'18 t . (OneOf '[Data.Word.Word16,
                                                       Float] t) =>
                 Bool -- ^ __use_peephole__
                 -> Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
                 -> Tensor v'2 t -- ^ __x__
                 -> Tensor v'3 t -- ^ __cs_prev__
                 -> Tensor v'4 t -- ^ __h_prev__
                 -> Tensor v'5 t -- ^ __w__
                 -> Tensor v'6 t -- ^ __wci__
                 -> Tensor v'7 t -- ^ __wcf__
                 -> Tensor v'8 t -- ^ __wco__
                 -> Tensor v'9 t -- ^ __b__
                 -> Tensor v'10 t -- ^ __i__
                 -> Tensor v'11 t -- ^ __cs__
                 -> Tensor v'12 t -- ^ __f__
                 -> Tensor v'13 t -- ^ __o__
                 -> Tensor v'14 t -- ^ __ci__
                 -> Tensor v'15 t -- ^ __co__
                 -> Tensor v'16 t -- ^ __h__
                 -> Tensor v'17 t -- ^ __cs_grad__
                 -> Tensor v'18 t -- ^ __h_grad__
                 -> (Tensor Build t, Tensor Build t, Tensor Build t,
                     Tensor Build t, Tensor Build t, Tensor Build t,
                     Tensor Build t, Tensor Build t)
                 -- ^ (__x_grad__, __cs_prev_grad__, __h_prev_grad__, __w_grad__, __wci_grad__, __wcf_grad__, __wco_grad__, __b_grad__)
                 --
                 -- * __x_grad__
                 --
                 -- * __cs_prev_grad__
                 --
                 -- * __h_prev_grad__
                 --
                 -- * __w_grad__
                 --
                 -- * __wci_grad__
                 --
                 -- * __wcf_grad__
                 --
                 -- * __wco_grad__
                 --
                 -- * __b_grad__
blockLSTMGrad :: Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGrad = OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (v'13 :: * -> *) (v'14 :: * -> *) (v'15 :: * -> *)
       (v'16 :: * -> *) (v'17 :: * -> *) (v'18 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGrad' OpParams
forall a. a -> a
id
blockLSTMGrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12 v'13
                  v'14 v'15 v'16 v'17 v'18 t . (OneOf '[Data.Word.Word16,
                                                        Float] t) => OpParams ->
                  Bool -- ^ __use_peephole__
                  -> Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
                  -> Tensor v'2 t -- ^ __x__
                  -> Tensor v'3 t -- ^ __cs_prev__
                  -> Tensor v'4 t -- ^ __h_prev__
                  -> Tensor v'5 t -- ^ __w__
                  -> Tensor v'6 t -- ^ __wci__
                  -> Tensor v'7 t -- ^ __wcf__
                  -> Tensor v'8 t -- ^ __wco__
                  -> Tensor v'9 t -- ^ __b__
                  -> Tensor v'10 t -- ^ __i__
                  -> Tensor v'11 t -- ^ __cs__
                  -> Tensor v'12 t -- ^ __f__
                  -> Tensor v'13 t -- ^ __o__
                  -> Tensor v'14 t -- ^ __ci__
                  -> Tensor v'15 t -- ^ __co__
                  -> Tensor v'16 t -- ^ __h__
                  -> Tensor v'17 t -- ^ __cs_grad__
                  -> Tensor v'18 t -- ^ __h_grad__
                  -> (Tensor Build t, Tensor Build t, Tensor Build t,
                      Tensor Build t, Tensor Build t, Tensor Build t,
                      Tensor Build t, Tensor Build t)
                  -- ^ (__x_grad__, __cs_prev_grad__, __h_prev_grad__, __w_grad__, __wci_grad__, __wcf_grad__, __wco_grad__, __b_grad__)
                  --
                  -- * __x_grad__
                  --
                  -- * __cs_prev_grad__
                  --
                  -- * __h_prev_grad__
                  --
                  -- * __w_grad__
                  --
                  -- * __wci_grad__
                  --
                  -- * __wcf_grad__
                  --
                  -- * __wco_grad__
                  --
                  -- * __b_grad__
blockLSTMGrad' :: OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGrad' op'options :: OpParams
op'options use_peephole :: Bool
use_peephole seq_len_max :: Tensor v'1 Int64
seq_len_max x :: Tensor v'2 t
x cs_prev :: Tensor v'3 t
cs_prev h_prev :: Tensor v'4 t
h_prev w :: Tensor v'5 t
w wci :: Tensor v'6 t
wci wcf :: Tensor v'7 t
wcf
               wco :: Tensor v'8 t
wco b :: Tensor v'9 t
b i :: Tensor v'10 t
i cs :: Tensor v'11 t
cs f :: Tensor v'12 t
f o :: Tensor v'13 t
o ci :: Tensor v'14 t
ci co :: Tensor v'15 t
co h :: Tensor v'16 t
h cs_grad :: Tensor v'17 t
cs_grad h_grad :: Tensor v'18 t
h_grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seq_len_max,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
cs_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
h_prev,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
w,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wci,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wcf,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
wco,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
b,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
i,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
cs,
                                                             Tensor v'12 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 t
f,
                                                             Tensor v'13 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'13 t
o,
                                                             Tensor v'14 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'14 t
ci,
                                                             Tensor v'15 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'15 t
co,
                                                             Tensor v'16 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'16 t
h,
                                                             Tensor v'17 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'17 t
cs_grad,
                                                             Tensor v'18 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'18 t
h_grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BlockLSTMGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "use_peephole" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
use_peephole
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seq_len_max" type: DT_INT64 }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
input_arg { name: "i" type_attr: "T" }
input_arg { name: "cs" type_attr: "T" }
input_arg { name: "f" type_attr: "T" }
input_arg { name: "o" type_attr: "T" }
input_arg { name: "ci" type_attr: "T" }
input_arg { name: "co" type_attr: "T" }
input_arg { name: "h" type_attr: "T" }
input_arg { name: "cs_grad" type_attr: "T" }
input_arg { name: "h_grad" type_attr: "T" }
output_arg { name: "x_grad" type_attr: "T" }
output_arg { name: "cs_prev_grad" type_attr: "T" }
output_arg { name: "h_prev_grad" type_attr: "T" }
output_arg { name: "w_grad" type_attr: "T" }
output_arg { name: "wci_grad" type_attr: "T" }
output_arg { name: "wcf_grad" type_attr: "T" }
output_arg { name: "wco_grad" type_attr: "T" }
output_arg { name: "b_grad" type_attr: "T" }
attr { name: "use_peephole" type: "bool" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
blockLSTMGradV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                   v'13 v'14 v'15 v'16 v'17 v'18 t . (OneOf '[Data.Word.Word16,
                                                              Float] t) =>
                   Bool -- ^ __use_peephole__
                   -> Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
                   -> Tensor v'2 t -- ^ __x__
                   -> Tensor v'3 t -- ^ __cs_prev__
                   -> Tensor v'4 t -- ^ __h_prev__
                   -> Tensor v'5 t -- ^ __w__
                   -> Tensor v'6 t -- ^ __wci__
                   -> Tensor v'7 t -- ^ __wcf__
                   -> Tensor v'8 t -- ^ __wco__
                   -> Tensor v'9 t -- ^ __b__
                   -> Tensor v'10 t -- ^ __i__
                   -> Tensor v'11 t -- ^ __cs__
                   -> Tensor v'12 t -- ^ __f__
                   -> Tensor v'13 t -- ^ __o__
                   -> Tensor v'14 t -- ^ __ci__
                   -> Tensor v'15 t -- ^ __co__
                   -> Tensor v'16 t -- ^ __h__
                   -> Tensor v'17 t -- ^ __cs_grad__
                   -> Tensor v'18 t -- ^ __h_grad__
                   -> (Tensor Build t, Tensor Build t, Tensor Build t,
                       Tensor Build t, Tensor Build t, Tensor Build t,
                       Tensor Build t, Tensor Build t)
                   -- ^ (__x_grad__, __cs_prev_grad__, __h_prev_grad__, __w_grad__, __wci_grad__, __wcf_grad__, __wco_grad__, __b_grad__)
                   --
                   -- * __x_grad__
                   --
                   -- * __cs_prev_grad__
                   --
                   -- * __h_prev_grad__
                   --
                   -- * __w_grad__
                   --
                   -- * __wci_grad__
                   --
                   -- * __wcf_grad__
                   --
                   -- * __wco_grad__
                   --
                   -- * __b_grad__
blockLSTMGradV2 :: Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGradV2 = OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (v'13 :: * -> *) (v'14 :: * -> *) (v'15 :: * -> *)
       (v'16 :: * -> *) (v'17 :: * -> *) (v'18 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGradV2' OpParams
forall a. a -> a
id
blockLSTMGradV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                    v'13 v'14 v'15 v'16 v'17 v'18 t . (OneOf '[Data.Word.Word16,
                                                               Float] t) =>
                    OpParams ->
                    Bool -- ^ __use_peephole__
                    -> Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
                    -> Tensor v'2 t -- ^ __x__
                    -> Tensor v'3 t -- ^ __cs_prev__
                    -> Tensor v'4 t -- ^ __h_prev__
                    -> Tensor v'5 t -- ^ __w__
                    -> Tensor v'6 t -- ^ __wci__
                    -> Tensor v'7 t -- ^ __wcf__
                    -> Tensor v'8 t -- ^ __wco__
                    -> Tensor v'9 t -- ^ __b__
                    -> Tensor v'10 t -- ^ __i__
                    -> Tensor v'11 t -- ^ __cs__
                    -> Tensor v'12 t -- ^ __f__
                    -> Tensor v'13 t -- ^ __o__
                    -> Tensor v'14 t -- ^ __ci__
                    -> Tensor v'15 t -- ^ __co__
                    -> Tensor v'16 t -- ^ __h__
                    -> Tensor v'17 t -- ^ __cs_grad__
                    -> Tensor v'18 t -- ^ __h_grad__
                    -> (Tensor Build t, Tensor Build t, Tensor Build t,
                        Tensor Build t, Tensor Build t, Tensor Build t,
                        Tensor Build t, Tensor Build t)
                    -- ^ (__x_grad__, __cs_prev_grad__, __h_prev_grad__, __w_grad__, __wci_grad__, __wcf_grad__, __wco_grad__, __b_grad__)
                    --
                    -- * __x_grad__
                    --
                    -- * __cs_prev_grad__
                    --
                    -- * __h_prev_grad__
                    --
                    -- * __w_grad__
                    --
                    -- * __wci_grad__
                    --
                    -- * __wcf_grad__
                    --
                    -- * __wco_grad__
                    --
                    -- * __b_grad__
blockLSTMGradV2' :: OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> Tensor v'17 t
-> Tensor v'18 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMGradV2' op'options :: OpParams
op'options use_peephole :: Bool
use_peephole seq_len_max :: Tensor v'1 Int64
seq_len_max x :: Tensor v'2 t
x cs_prev :: Tensor v'3 t
cs_prev h_prev :: Tensor v'4 t
h_prev w :: Tensor v'5 t
w wci :: Tensor v'6 t
wci wcf :: Tensor v'7 t
wcf
                 wco :: Tensor v'8 t
wco b :: Tensor v'9 t
b i :: Tensor v'10 t
i cs :: Tensor v'11 t
cs f :: Tensor v'12 t
f o :: Tensor v'13 t
o ci :: Tensor v'14 t
ci co :: Tensor v'15 t
co h :: Tensor v'16 t
h cs_grad :: Tensor v'17 t
cs_grad h_grad :: Tensor v'18 t
h_grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seq_len_max,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
cs_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
h_prev,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
w,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wci,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wcf,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
wco,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
b,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
i,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
cs,
                                                             Tensor v'12 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 t
f,
                                                             Tensor v'13 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'13 t
o,
                                                             Tensor v'14 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'14 t
ci,
                                                             Tensor v'15 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'15 t
co,
                                                             Tensor v'16 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'16 t
h,
                                                             Tensor v'17 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'17 t
cs_grad,
                                                             Tensor v'18 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'18 t
h_grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BlockLSTMGradV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "use_peephole" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
use_peephole
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seq_len_max" type: DT_INT64 }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
input_arg { name: "i" type_attr: "T" }
input_arg { name: "cs" type_attr: "T" }
input_arg { name: "f" type_attr: "T" }
input_arg { name: "o" type_attr: "T" }
input_arg { name: "ci" type_attr: "T" }
input_arg { name: "co" type_attr: "T" }
input_arg { name: "h" type_attr: "T" }
input_arg { name: "cs_grad" type_attr: "T" }
input_arg { name: "h_grad" type_attr: "T" }
output_arg { name: "x_grad" type_attr: "T" }
output_arg { name: "cs_prev_grad" type_attr: "T" }
output_arg { name: "h_prev_grad" type_attr: "T" }
output_arg { name: "w_grad" type_attr: "T" }
output_arg { name: "wci_grad" type_attr: "T" }
output_arg { name: "wcf_grad" type_attr: "T" }
output_arg { name: "wco_grad" type_attr: "T" }
output_arg { name: "b_grad" type_attr: "T" }
attr { name: "use_peephole" type: "bool" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
blockLSTMV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
               t . (OneOf '[Data.Word.Word16, Float] t) =>
               Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
               -> Tensor v'2 t -- ^ __x__
               -> Tensor v'3 t -- ^ __cs_prev__
               -> Tensor v'4 t -- ^ __h_prev__
               -> Tensor v'5 t -- ^ __w__
               -> Tensor v'6 t -- ^ __wci__
               -> Tensor v'7 t -- ^ __wcf__
               -> Tensor v'8 t -- ^ __wco__
               -> Tensor v'9 t -- ^ __b__
               -> (Tensor Build t, Tensor Build t, Tensor Build t,
                   Tensor Build t, Tensor Build t, Tensor Build t,
                   Tensor Build t)
               -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
               --
               -- * __i__
               --
               -- * __cs__
               --
               -- * __f__
               --
               -- * __o__
               --
               -- * __ci__
               --
               -- * __co__
               --
               -- * __h__
blockLSTMV2 :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMV2 = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMV2' OpParams
forall a. a -> a
id
blockLSTMV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
                Tensor v'1 Data.Int.Int64 -- ^ __seq_len_max__
                -> Tensor v'2 t -- ^ __x__
                -> Tensor v'3 t -- ^ __cs_prev__
                -> Tensor v'4 t -- ^ __h_prev__
                -> Tensor v'5 t -- ^ __w__
                -> Tensor v'6 t -- ^ __wci__
                -> Tensor v'7 t -- ^ __wcf__
                -> Tensor v'8 t -- ^ __wco__
                -> Tensor v'9 t -- ^ __b__
                -> (Tensor Build t, Tensor Build t, Tensor Build t,
                    Tensor Build t, Tensor Build t, Tensor Build t,
                    Tensor Build t)
                -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
                --
                -- * __i__
                --
                -- * __cs__
                --
                -- * __f__
                --
                -- * __o__
                --
                -- * __ci__
                --
                -- * __co__
                --
                -- * __h__
blockLSTMV2' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
blockLSTMV2' op'options :: OpParams
op'options seq_len_max :: Tensor v'1 Int64
seq_len_max x :: Tensor v'2 t
x cs_prev :: Tensor v'3 t
cs_prev h_prev :: Tensor v'4 t
h_prev w :: Tensor v'5 t
w wci :: Tensor v'6 t
wci wcf :: Tensor v'7 t
wcf wco :: Tensor v'8 t
wco
             b :: Tensor v'9 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seq_len_max,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
cs_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
h_prev,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
w,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wci,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wcf,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
wco,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BlockLSTMV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seq_len_max" type: DT_INT64 }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "i" type_attr: "T" }
output_arg { name: "cs" type_attr: "T" }
output_arg { name: "f" type_attr: "T" }
output_arg { name: "o" type_attr: "T" }
output_arg { name: "ci" type_attr: "T" }
output_arg { name: "co" type_attr: "T" }
output_arg { name: "h" type_attr: "T" }
attr { name: "cell_clip" type: "float" default_value { f: 0.0 } }
attr {
  name: "use_peephole" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
boostedTreesAggregateStats :: Data.Int.Int64 -- ^ __max_splits__
                              -> Data.Int.Int64 -- ^ __num_buckets__
                              -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                              -> Tensor v'2 Float -- ^ __gradients__
                              -> Tensor v'3 Float -- ^ __hessians__
                              -> Tensor v'4 Data.Int.Int32 -- ^ __feature__
                              -> Tensor Build Float -- ^ __stats_summary__
boostedTreesAggregateStats :: Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
boostedTreesAggregateStats = OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
boostedTreesAggregateStats' OpParams
forall a. a -> a
id
boostedTreesAggregateStats' :: OpParams ->
                               Data.Int.Int64 -- ^ __max_splits__
                               -> Data.Int.Int64 -- ^ __num_buckets__
                               -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                               -> Tensor v'2 Float -- ^ __gradients__
                               -> Tensor v'3 Float -- ^ __hessians__
                               -> Tensor v'4 Data.Int.Int32 -- ^ __feature__
                               -> Tensor Build Float -- ^ __stats_summary__
boostedTreesAggregateStats' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
boostedTreesAggregateStats' op'options :: OpParams
op'options max_splits :: Int64
max_splits num_buckets :: Int64
num_buckets node_ids :: Tensor v'1 Int32
node_ids gradients :: Tensor v'2 Float
gradients
                            hessians :: Tensor v'3 Float
hessians feature :: Tensor v'4 Int32
feature | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_ids,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
gradients,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
hessians,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
feature]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesAggregateStats"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_splits" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_splits
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "node_ids" type: DT_INT32 }
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "hessians" type: DT_FLOAT }
input_arg { name: "feature" type: DT_INT32 }
output_arg { name: "stats_summary" type: DT_FLOAT }
attr {
  name: "max_splits" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesBucketize :: [Tensor v'1 Float] -- ^ __float_values__
                         -> [Tensor v'2 Float] -- ^ __bucket_boundaries__
                         -> [Tensor Build Data.Int.Int32] -- ^ __buckets__
boostedTreesBucketize :: [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
boostedTreesBucketize = OpParams
-> [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
boostedTreesBucketize' OpParams
forall a. a -> a
id
boostedTreesBucketize' :: OpParams ->
                          [Tensor v'1 Float] -- ^ __float_values__
                          -> [Tensor v'2 Float] -- ^ __bucket_boundaries__
                          -> [Tensor Build Data.Int.Int32] -- ^ __buckets__
boostedTreesBucketize' :: OpParams
-> [Tensor v'1 Float] -> [Tensor v'2 Float] -> [Tensor Build Int32]
boostedTreesBucketize' op'options :: OpParams
op'options float_values :: [Tensor v'1 Float]
float_values
                       bucket_boundaries :: [Tensor v'2 Float]
bucket_boundaries | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("float_values", [Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
float_values),
                                                                            ("bucket_boundaries", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
bucket_boundaries)])] =
    [Int64] -> Build OpDef -> [Tensor Build Int32]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_features] (Build OpDef -> [Tensor Build Int32])
-> Build OpDef -> [Tensor Build Int32]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Float]
float_values,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
bucket_boundaries]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesBucketize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
float_values) :: Int64
{-
input_arg {
  name: "float_values" type: DT_FLOAT number_attr: "num_features"
}
input_arg {
  name: "bucket_boundaries"
  type: DT_FLOAT
  number_attr: "num_features"
}
output_arg {
  name: "buckets" type: DT_INT32 number_attr: "num_features"
}
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesCalculateBestFeatureSplit :: Data.Int.Int64 -- ^ __logits_dimension__
                                         -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                         -> Tensor v'2 Float -- ^ __stats_summary__
                                         -> Tensor v'3 Float -- ^ __l1__
                                         -> Tensor v'4 Float -- ^ __l2__
                                         -> Tensor v'5 Float -- ^ __tree_complexity__
                                         -> Tensor v'6 Float -- ^ __min_node_weight__
                                         -> (Tensor Build Data.Int.Int32,
                                             Tensor Build Float,
                                             Tensor Build Data.Int.Int32,
                                             Tensor Build Data.Int.Int32,
                                             Tensor Build Float,
                                             Tensor Build Float,
                                             Tensor Build Data.ByteString.ByteString)
                                         -- ^ (__node_ids__, __gains__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                         --
                                         -- * __node_ids__
                                         --
                                         -- * __gains__
                                         --
                                         -- * __feature_dimensions__
                                         --
                                         -- * __thresholds__
                                         --
                                         -- * __left_node_contribs__
                                         --
                                         -- * __right_node_contribs__
                                         --
                                         -- * __split_with_default_directions__
boostedTreesCalculateBestFeatureSplit :: Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplit = OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplit' OpParams
forall a. a -> a
id
boostedTreesCalculateBestFeatureSplit' :: OpParams ->
                                          Data.Int.Int64 -- ^ __logits_dimension__
                                          -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                          -> Tensor v'2 Float -- ^ __stats_summary__
                                          -> Tensor v'3 Float -- ^ __l1__
                                          -> Tensor v'4 Float -- ^ __l2__
                                          -> Tensor v'5 Float -- ^ __tree_complexity__
                                          -> Tensor v'6 Float -- ^ __min_node_weight__
                                          -> (Tensor Build Data.Int.Int32,
                                              Tensor Build Float,
                                              Tensor Build Data.Int.Int32,
                                              Tensor Build Data.Int.Int32,
                                              Tensor Build Float,
                                              Tensor Build Float,
                                              Tensor Build Data.ByteString.ByteString)
                                          -- ^ (__node_ids__, __gains__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                          --
                                          -- * __node_ids__
                                          --
                                          -- * __gains__
                                          --
                                          -- * __feature_dimensions__
                                          --
                                          -- * __thresholds__
                                          --
                                          -- * __left_node_contribs__
                                          --
                                          -- * __right_node_contribs__
                                          --
                                          -- * __split_with_default_directions__
boostedTreesCalculateBestFeatureSplit' :: OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplit' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension node_id_range :: Tensor v'1 Int32
node_id_range
                                       stats_summary :: Tensor v'2 Float
stats_summary l1 :: Tensor v'3 Float
l1 l2 :: Tensor v'4 Float
l2 tree_complexity :: Tensor v'5 Float
tree_complexity
                                       min_node_weight :: Tensor v'6 Float
min_node_weight | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
     Tensor Build Int32, Tensor Build Float, Tensor Build Float,
     Tensor Build ByteString))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_id_range,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
stats_summary,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
l1,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
l2,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
tree_complexity,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_node_weight]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesCalculateBestFeatureSplit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "node_id_range" type: DT_INT32 }
input_arg { name: "stats_summary" type: DT_FLOAT }
input_arg { name: "l1" type: DT_FLOAT }
input_arg { name: "l2" type: DT_FLOAT }
input_arg { name: "tree_complexity" type: DT_FLOAT }
input_arg { name: "min_node_weight" type: DT_FLOAT }
output_arg { name: "node_ids" type: DT_INT32 }
output_arg { name: "gains" type: DT_FLOAT }
output_arg { name: "feature_dimensions" type: DT_INT32 }
output_arg { name: "thresholds" type: DT_INT32 }
output_arg { name: "left_node_contribs" type: DT_FLOAT }
output_arg { name: "right_node_contribs" type: DT_FLOAT }
output_arg {
  name: "split_with_default_directions" type: DT_STRING
}
attr {
  name: "logits_dimension" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "split_type"
  type: "string"
  default_value { s: "inequality" }
  allowed_values { list { s: "inequality" s: "equality" } }
}
-}
-- | 
boostedTreesCalculateBestFeatureSplitV2 :: Data.Int.Int64 -- ^ __logits_dimension__
                                           -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                           -> [Tensor v'2 Float] -- ^ __stats_summaries_list__
                                           -> Tensor v'3 Data.ByteString.ByteString -- ^ __split_types__
                                           -> Tensor v'4 Data.Int.Int32 -- ^ __candidate_feature_ids__
                                           -> Tensor v'5 Float -- ^ __l1__
                                           -> Tensor v'6 Float -- ^ __l2__
                                           -> Tensor v'7 Float -- ^ __tree_complexity__
                                           -> Tensor v'8 Float -- ^ __min_node_weight__
                                           -> (Tensor Build Data.Int.Int32,
                                               Tensor Build Float,
                                               Tensor Build Data.Int.Int32,
                                               Tensor Build Data.Int.Int32,
                                               Tensor Build Data.Int.Int32,
                                               Tensor Build Float,
                                               Tensor Build Float,
                                               Tensor Build Data.ByteString.ByteString)
                                           -- ^ (__node_ids__, __gains__, __feature_ids__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                           --
                                           -- * __node_ids__
                                           --
                                           -- * __gains__
                                           --
                                           -- * __feature_ids__
                                           --
                                           -- * __feature_dimensions__
                                           --
                                           -- * __thresholds__
                                           --
                                           -- * __left_node_contribs__
                                           --
                                           -- * __right_node_contribs__
                                           --
                                           -- * __split_with_default_directions__
boostedTreesCalculateBestFeatureSplitV2 :: Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 ByteString
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplitV2 = OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 ByteString
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 ByteString
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplitV2' OpParams
forall a. a -> a
id
boostedTreesCalculateBestFeatureSplitV2' :: OpParams ->
                                            Data.Int.Int64 -- ^ __logits_dimension__
                                            -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                            -> [Tensor v'2 Float] -- ^ __stats_summaries_list__
                                            -> Tensor v'3 Data.ByteString.ByteString -- ^ __split_types__
                                            -> Tensor v'4 Data.Int.Int32 -- ^ __candidate_feature_ids__
                                            -> Tensor v'5 Float -- ^ __l1__
                                            -> Tensor v'6 Float -- ^ __l2__
                                            -> Tensor v'7 Float -- ^ __tree_complexity__
                                            -> Tensor v'8 Float -- ^ __min_node_weight__
                                            -> (Tensor Build Data.Int.Int32,
                                                Tensor Build Float,
                                                Tensor Build Data.Int.Int32,
                                                Tensor Build Data.Int.Int32,
                                                Tensor Build Data.Int.Int32,
                                                Tensor Build Float,
                                                Tensor Build Float,
                                                Tensor Build Data.ByteString.ByteString)
                                            -- ^ (__node_ids__, __gains__, __feature_ids__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                            --
                                            -- * __node_ids__
                                            --
                                            -- * __gains__
                                            --
                                            -- * __feature_ids__
                                            --
                                            -- * __feature_dimensions__
                                            --
                                            -- * __thresholds__
                                            --
                                            -- * __left_node_contribs__
                                            --
                                            -- * __right_node_contribs__
                                            --
                                            -- * __split_with_default_directions__
boostedTreesCalculateBestFeatureSplitV2' :: OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 ByteString
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
boostedTreesCalculateBestFeatureSplitV2' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension
                                         node_id_range :: Tensor v'1 Int32
node_id_range stats_summaries_list :: [Tensor v'2 Float]
stats_summaries_list
                                         split_types :: Tensor v'3 ByteString
split_types candidate_feature_ids :: Tensor v'4 Int32
candidate_feature_ids l1 :: Tensor v'5 Float
l1 l2 :: Tensor v'6 Float
l2
                                         tree_complexity :: Tensor v'7 Float
tree_complexity
                                         min_node_weight :: Tensor v'8 Float
min_node_weight | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("stats_summaries_list", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
stats_summaries_list)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
     Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
     Tensor Build Float, Tensor Build ByteString))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Int32, Tensor Build Float,
    Tensor Build Float, Tensor Build ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_id_range,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
stats_summaries_list,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
split_types,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
candidate_feature_ids,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
l1,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
l2,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
tree_complexity,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_node_weight]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesCalculateBestFeatureSplitV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
stats_summaries_list) :: Int64
{-
input_arg { name: "node_id_range" type: DT_INT32 }
input_arg {
  name: "stats_summaries_list"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg { name: "split_types" type: DT_STRING }
input_arg { name: "candidate_feature_ids" type: DT_INT32 }
input_arg { name: "l1" type: DT_FLOAT }
input_arg { name: "l2" type: DT_FLOAT }
input_arg { name: "tree_complexity" type: DT_FLOAT }
input_arg { name: "min_node_weight" type: DT_FLOAT }
output_arg { name: "node_ids" type: DT_INT32 }
output_arg { name: "gains" type: DT_FLOAT }
output_arg { name: "feature_ids" type: DT_INT32 }
output_arg { name: "feature_dimensions" type: DT_INT32 }
output_arg { name: "thresholds" type: DT_INT32 }
output_arg { name: "left_node_contribs" type: DT_FLOAT }
output_arg { name: "right_node_contribs" type: DT_FLOAT }
output_arg {
  name: "split_with_default_directions" type: DT_STRING
}
attr {
  name: "num_features" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "logits_dimension" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesCalculateBestGainsPerFeature :: Data.Int.Int64 -- ^ __max_splits__
                                            -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                            -> [Tensor v'2 Float] -- ^ __stats_summary_list__
                                            -> Tensor v'3 Float -- ^ __l1__
                                            -> Tensor v'4 Float -- ^ __l2__
                                            -> Tensor v'5 Float -- ^ __tree_complexity__
                                            -> Tensor v'6 Float -- ^ __min_node_weight__
                                            -> ([Tensor Build Data.Int.Int32],
                                                [Tensor Build Float],
                                                [Tensor Build Data.Int.Int32],
                                                [Tensor Build Float],
                                                [Tensor Build Float])
                                            -- ^ (__node_ids_list__, __gains_list__, __thresholds_list__, __left_node_contribs_list__, __right_node_contribs_list__)
                                            --
                                            -- * __node_ids_list__
                                            --
                                            -- * __gains_list__
                                            --
                                            -- * __thresholds_list__
                                            --
                                            -- * __left_node_contribs_list__
                                            --
                                            -- * __right_node_contribs_list__
boostedTreesCalculateBestGainsPerFeature :: Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
boostedTreesCalculateBestGainsPerFeature = OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
boostedTreesCalculateBestGainsPerFeature' OpParams
forall a. a -> a
id
boostedTreesCalculateBestGainsPerFeature' :: OpParams ->
                                             Data.Int.Int64 -- ^ __max_splits__
                                             -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                             -> [Tensor v'2 Float] -- ^ __stats_summary_list__
                                             -> Tensor v'3 Float -- ^ __l1__
                                             -> Tensor v'4 Float -- ^ __l2__
                                             -> Tensor v'5 Float -- ^ __tree_complexity__
                                             -> Tensor v'6 Float -- ^ __min_node_weight__
                                             -> ([Tensor Build Data.Int.Int32],
                                                 [Tensor Build Float],
                                                 [Tensor Build Data.Int.Int32],
                                                 [Tensor Build Float],
                                                 [Tensor Build Float])
                                             -- ^ (__node_ids_list__, __gains_list__, __thresholds_list__, __left_node_contribs_list__, __right_node_contribs_list__)
                                             --
                                             -- * __node_ids_list__
                                             --
                                             -- * __gains_list__
                                             --
                                             -- * __thresholds_list__
                                             --
                                             -- * __left_node_contribs_list__
                                             --
                                             -- * __right_node_contribs_list__
boostedTreesCalculateBestGainsPerFeature' :: OpParams
-> Int64
-> Tensor v'1 Int32
-> [Tensor v'2 Float]
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
boostedTreesCalculateBestGainsPerFeature' op'options :: OpParams
op'options max_splits :: Int64
max_splits node_id_range :: Tensor v'1 Int32
node_id_range
                                          stats_summary_list :: [Tensor v'2 Float]
stats_summary_list l1 :: Tensor v'3 Float
l1 l2 :: Tensor v'4 Float
l2
                                          tree_complexity :: Tensor v'5 Float
tree_complexity
                                          min_node_weight :: Tensor v'6 Float
min_node_weight | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("stats_summary_list", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
stats_summary_list)])] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_features, Int64
num_features, Int64
num_features, Int64
num_features,
            Int64
num_features] (Build OpDef
 -> ([Tensor Build Int32], [Tensor Build Float],
     [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float]))
-> Build OpDef
-> ([Tensor Build Int32], [Tensor Build Float],
    [Tensor Build Int32], [Tensor Build Float], [Tensor Build Float])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_id_range,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
stats_summary_list,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
l1,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
l2,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
tree_complexity,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_node_weight]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesCalculateBestGainsPerFeature"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_splits" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_splits
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
stats_summary_list) :: Int64
{-
input_arg { name: "node_id_range" type: DT_INT32 }
input_arg {
  name: "stats_summary_list"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg { name: "l1" type: DT_FLOAT }
input_arg { name: "l2" type: DT_FLOAT }
input_arg { name: "tree_complexity" type: DT_FLOAT }
input_arg { name: "min_node_weight" type: DT_FLOAT }
output_arg {
  name: "node_ids_list" type: DT_INT32 number_attr: "num_features"
}
output_arg {
  name: "gains_list" type: DT_FLOAT number_attr: "num_features"
}
output_arg {
  name: "thresholds_list" type: DT_INT32 number_attr: "num_features"
}
output_arg {
  name: "left_node_contribs_list"
  type: DT_FLOAT
  number_attr: "num_features"
}
output_arg {
  name: "right_node_contribs_list"
  type: DT_FLOAT
  number_attr: "num_features"
}
attr {
  name: "max_splits" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_features" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesCenterBias :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                          Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                          -> Tensor v'2 Float -- ^ __mean_gradients__
                          -> Tensor v'3 Float -- ^ __mean_hessians__
                          -> Tensor v'4 Float -- ^ __l1__
                          -> Tensor v'5 Float -- ^ __l2__
                          -> m' (Tensor Value Bool) -- ^ __continue_centering__
boostedTreesCenterBias :: Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' (Tensor Value Bool)
boostedTreesCenterBias = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' (Tensor Value Bool)
boostedTreesCenterBias' OpParams
forall a. a -> a
id
boostedTreesCenterBias' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                           OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                           -> Tensor v'2 Float -- ^ __mean_gradients__
                           -> Tensor v'3 Float -- ^ __mean_hessians__
                           -> Tensor v'4 Float -- ^ __l1__
                           -> Tensor v'5 Float -- ^ __l2__
                           -> m' (Tensor Value Bool) -- ^ __continue_centering__
boostedTreesCenterBias' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' (Tensor Value Bool)
boostedTreesCenterBias' op'options :: OpParams
op'options tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle mean_gradients :: Tensor v'2 Float
mean_gradients
                        mean_hessians :: Tensor v'3 Float
mean_hessians l1 :: Tensor v'4 Float
l1 l2 :: Tensor v'5 Float
l2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
mean_gradients,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
mean_hessians,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
l1,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
l2]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesCenterBias"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg { name: "mean_gradients" type: DT_FLOAT }
input_arg { name: "mean_hessians" type: DT_FLOAT }
input_arg { name: "l1" type: DT_FLOAT }
input_arg { name: "l2" type: DT_FLOAT }
output_arg { name: "continue_centering" type: DT_BOOL }
-}
-- | 
boostedTreesCreateEnsemble :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                              Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __stamp_token__
                              -> Tensor v'3 Data.ByteString.ByteString -- ^ __tree_ensemble_serialized__
                              -> m' (ControlNode)
boostedTreesCreateEnsemble :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
boostedTreesCreateEnsemble = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
boostedTreesCreateEnsemble' OpParams
forall a. a -> a
id
boostedTreesCreateEnsemble' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                               -> Tensor v'2 Data.Int.Int64 -- ^ __stamp_token__
                               -> Tensor v'3 Data.ByteString.ByteString -- ^ __tree_ensemble_serialized__
                               -> m' (ControlNode)
boostedTreesCreateEnsemble' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
boostedTreesCreateEnsemble' op'options :: OpParams
op'options tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle stamp_token :: Tensor v'2 Int64
stamp_token
                            tree_ensemble_serialized :: Tensor v'3 ByteString
tree_ensemble_serialized | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
stamp_token,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tree_ensemble_serialized]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesCreateEnsemble"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg { name: "stamp_token" type: DT_INT64 }
input_arg { name: "tree_ensemble_serialized" type: DT_STRING }
-}
-- | 
boostedTreesCreateQuantileStreamResource :: forall v'1 v'2 v'3
                                            m' . (MonadBuild m') =>
                                            Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                            -> Tensor v'2 Float -- ^ __epsilon__
                                            -> Tensor v'3 Data.Int.Int64 -- ^ __num_streams__
                                            -> m' (ControlNode)
boostedTreesCreateQuantileStreamResource :: Tensor v'1 ResourceHandle
-> Tensor v'2 Float -> Tensor v'3 Int64 -> m' ControlNode
boostedTreesCreateQuantileStreamResource = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> m' ControlNode
boostedTreesCreateQuantileStreamResource' OpParams
forall a. a -> a
id
boostedTreesCreateQuantileStreamResource' :: forall v'1 v'2 v'3
                                             m' . (MonadBuild m') => OpParams ->
                                             Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                             -> Tensor v'2 Float -- ^ __epsilon__
                                             -> Tensor v'3 Data.Int.Int64 -- ^ __num_streams__
                                             -> m' (ControlNode)
boostedTreesCreateQuantileStreamResource' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> m' ControlNode
boostedTreesCreateQuantileStreamResource' op'options :: OpParams
op'options
                                          quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle
                                          epsilon :: Tensor v'2 Float
epsilon
                                          num_streams :: Tensor v'3 Int64
num_streams | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
epsilon,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
num_streams]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesCreateQuantileStreamResource"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
input_arg { name: "epsilon" type: DT_FLOAT }
input_arg { name: "num_streams" type: DT_INT64 }
attr {
  name: "max_elements" type: "int" default_value { i: 1099511627776 }
}
-}
-- | 
boostedTreesDeserializeEnsemble :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                   Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                   -> Tensor v'2 Data.Int.Int64 -- ^ __stamp_token__
                                   -> Tensor v'3 Data.ByteString.ByteString -- ^ __tree_ensemble_serialized__
                                   -> m' (ControlNode)
boostedTreesDeserializeEnsemble :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
boostedTreesDeserializeEnsemble = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
boostedTreesDeserializeEnsemble' OpParams
forall a. a -> a
id
boostedTreesDeserializeEnsemble' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                    OpParams ->
                                    Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                    -> Tensor v'2 Data.Int.Int64 -- ^ __stamp_token__
                                    -> Tensor v'3 Data.ByteString.ByteString -- ^ __tree_ensemble_serialized__
                                    -> m' (ControlNode)
boostedTreesDeserializeEnsemble' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
boostedTreesDeserializeEnsemble' op'options :: OpParams
op'options tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle stamp_token :: Tensor v'2 Int64
stamp_token
                                 tree_ensemble_serialized :: Tensor v'3 ByteString
tree_ensemble_serialized | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
stamp_token,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tree_ensemble_serialized]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesDeserializeEnsemble"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg { name: "stamp_token" type: DT_INT64 }
input_arg { name: "tree_ensemble_serialized" type: DT_STRING }
-}
-- | 
boostedTreesEnsembleResourceHandleOp :: forall m' . (MonadBuild m') =>
                                        m' (Tensor Value ResourceHandle) -- ^ __resource__
boostedTreesEnsembleResourceHandleOp :: m' (Tensor Value ResourceHandle)
boostedTreesEnsembleResourceHandleOp = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
boostedTreesEnsembleResourceHandleOp' OpParams
forall a. a -> a
id
boostedTreesEnsembleResourceHandleOp' :: forall m' . (MonadBuild m') =>
                                         OpParams ->
                                         m' (Tensor Value ResourceHandle) -- ^ __resource__
boostedTreesEnsembleResourceHandleOp' :: OpParams -> m' (Tensor Value ResourceHandle)
boostedTreesEnsembleResourceHandleOp' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesEnsembleResourceHandleOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resource" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
boostedTreesExampleDebugOutputs :: forall v'1 v'2 m' . (MonadBuild m') =>
                                   Data.Int.Int64 -- ^ __logits_dimension__
                                   -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                   -> [Tensor v'2 Data.Int.Int32] -- ^ __bucketized_features__
                                   -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __examples_debug_outputs_serialized__
boostedTreesExampleDebugOutputs :: Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value ByteString)
boostedTreesExampleDebugOutputs = OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value ByteString)
boostedTreesExampleDebugOutputs' OpParams
forall a. a -> a
id
boostedTreesExampleDebugOutputs' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                    OpParams ->
                                    Data.Int.Int64 -- ^ __logits_dimension__
                                    -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                    -> [Tensor v'2 Data.Int.Int32] -- ^ __bucketized_features__
                                    -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __examples_debug_outputs_serialized__
boostedTreesExampleDebugOutputs' :: OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value ByteString)
boostedTreesExampleDebugOutputs' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension
                                 tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle
                                 bucketized_features :: [Tensor v'2 Int32]
bucketized_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_bucketized_features", [("bucketized_features", [Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
bucketized_features)])] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             [Tensor v'2 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int32]
bucketized_features]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesExampleDebugOutputs"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_bucketized_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_bucketized_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_bucketized_features :: Int64
num_bucketized_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
bucketized_features) :: Int64
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg {
  name: "bucketized_features"
  type: DT_INT32
  number_attr: "num_bucketized_features"
}
output_arg {
  name: "examples_debug_outputs_serialized" type: DT_STRING
}
attr {
  name: "num_bucketized_features"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr { name: "logits_dimension" type: "int" }
-}
-- | 
boostedTreesFlushQuantileSummaries :: forall v'1 m' . (MonadBuild m') =>
                                      Data.Int.Int64 -- ^ __num_features__
                                      -> Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                      -> m' ([Tensor Value Float]) -- ^ __summaries__
boostedTreesFlushQuantileSummaries :: Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesFlushQuantileSummaries = OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesFlushQuantileSummaries' OpParams
forall a. a -> a
id
boostedTreesFlushQuantileSummaries' :: forall v'1 m' . (MonadBuild m') =>
                                       OpParams ->
                                       Data.Int.Int64 -- ^ __num_features__
                                       -> Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                       -> m' ([Tensor Value Float]) -- ^ __summaries__
boostedTreesFlushQuantileSummaries' :: OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesFlushQuantileSummaries' op'options :: OpParams
op'options num_features :: Int64
num_features
                                    quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build [Tensor Value Float] -> m' [Tensor Value Float]
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build [Tensor Value Float] -> m' [Tensor Value Float])
-> Build [Tensor Value Float] -> m' [Tensor Value Float]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle]
        [Int64] -> OpDef -> Build [Tensor Value Float]
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [Int64
num_features] (OpType -> OpDef
opDef "BoostedTreesFlushQuantileSummaries"
                                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
output_arg {
  name: "summaries" type: DT_FLOAT number_attr: "num_features"
}
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesGetEnsembleStates :: forall v'1 m' . (MonadBuild m') =>
                                 Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                 -> m' ((Tensor Value Data.Int.Int64,
                                         Tensor Value Data.Int.Int32,
                                         Tensor Value Data.Int.Int32,
                                         Tensor Value Data.Int.Int32,
                                         Tensor Value Data.Int.Int32))
                                 -- ^ (__stamp_token__, __num_trees__, __num_finalized_trees__, __num_attempted_layers__, __last_layer_nodes_range__)
                                 --
                                 -- * __stamp_token__
                                 --
                                 -- * __num_trees__
                                 --
                                 -- * __num_finalized_trees__
                                 --
                                 -- * __num_attempted_layers__
                                 --
                                 -- * __last_layer_nodes_range__
boostedTreesGetEnsembleStates :: Tensor v'1 ResourceHandle
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
boostedTreesGetEnsembleStates = OpParams
-> Tensor v'1 ResourceHandle
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
boostedTreesGetEnsembleStates' OpParams
forall a. a -> a
id
boostedTreesGetEnsembleStates' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                                  Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                  -> m' ((Tensor Value Data.Int.Int64,
                                          Tensor Value Data.Int.Int32,
                                          Tensor Value Data.Int.Int32,
                                          Tensor Value Data.Int.Int32,
                                          Tensor Value Data.Int.Int32))
                                  -- ^ (__stamp_token__, __num_trees__, __num_finalized_trees__, __num_attempted_layers__, __last_layer_nodes_range__)
                                  --
                                  -- * __stamp_token__
                                  --
                                  -- * __num_trees__
                                  --
                                  -- * __num_finalized_trees__
                                  --
                                  -- * __num_attempted_layers__
                                  --
                                  -- * __last_layer_nodes_range__
boostedTreesGetEnsembleStates' :: OpParams
-> Tensor v'1 ResourceHandle
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
boostedTreesGetEnsembleStates' op'options :: OpParams
op'options
                               tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
   Tensor Value Int32, Tensor Value Int32)
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
    Tensor Value Int32, Tensor Value Int32)
 -> m'
      (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
       Tensor Value Int32, Tensor Value Int32))
-> Build
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
-> m'
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32,
      Tensor Value Int32, Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesGetEnsembleStates"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
output_arg { name: "stamp_token" type: DT_INT64 }
output_arg { name: "num_trees" type: DT_INT32 }
output_arg { name: "num_finalized_trees" type: DT_INT32 }
output_arg { name: "num_attempted_layers" type: DT_INT32 }
output_arg { name: "last_layer_nodes_range" type: DT_INT32 }
-}
-- | 
boostedTreesMakeQuantileSummaries :: [Tensor v'1 Float] -- ^ __float_values__
                                     -> Tensor v'2 Float -- ^ __example_weights__
                                     -> Tensor v'3 Float -- ^ __epsilon__
                                     -> [Tensor Build Float] -- ^ __summaries__
boostedTreesMakeQuantileSummaries :: [Tensor v'1 Float]
-> Tensor v'2 Float -> Tensor v'3 Float -> [Tensor Build Float]
boostedTreesMakeQuantileSummaries = OpParams
-> [Tensor v'1 Float]
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor Build Float]
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [Tensor v'1 Float]
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor Build Float]
boostedTreesMakeQuantileSummaries' OpParams
forall a. a -> a
id
boostedTreesMakeQuantileSummaries' :: OpParams ->
                                      [Tensor v'1 Float] -- ^ __float_values__
                                      -> Tensor v'2 Float -- ^ __example_weights__
                                      -> Tensor v'3 Float -- ^ __epsilon__
                                      -> [Tensor Build Float] -- ^ __summaries__
boostedTreesMakeQuantileSummaries' :: OpParams
-> [Tensor v'1 Float]
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor Build Float]
boostedTreesMakeQuantileSummaries' op'options :: OpParams
op'options float_values :: [Tensor v'1 Float]
float_values example_weights :: Tensor v'2 Float
example_weights
                                   epsilon :: Tensor v'3 Float
epsilon | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("float_values", [Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
float_values)])] =
    [Int64] -> Build OpDef -> [Tensor Build Float]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_features] (Build OpDef -> [Tensor Build Float])
-> Build OpDef -> [Tensor Build Float]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Float]
float_values,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
example_weights,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
epsilon]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesMakeQuantileSummaries"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
float_values) :: Int64
{-
input_arg {
  name: "float_values" type: DT_FLOAT number_attr: "num_features"
}
input_arg { name: "example_weights" type: DT_FLOAT }
input_arg { name: "epsilon" type: DT_FLOAT }
output_arg {
  name: "summaries" type: DT_FLOAT number_attr: "num_features"
}
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesMakeStatsSummary :: Data.Int.Int64 -- ^ __max_splits__
                                -> Data.Int.Int64 -- ^ __num_buckets__
                                -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                                -> Tensor v'2 Float -- ^ __gradients__
                                -> Tensor v'3 Float -- ^ __hessians__
                                -> [Tensor v'4 Data.Int.Int32] -- ^ __bucketized_features_list__
                                -> Tensor Build Float -- ^ __stats_summary__
boostedTreesMakeStatsSummary :: Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor v'4 Int32]
-> Tensor Build Float
boostedTreesMakeStatsSummary = OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor v'4 Int32]
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor v'4 Int32]
-> Tensor Build Float
boostedTreesMakeStatsSummary' OpParams
forall a. a -> a
id
boostedTreesMakeStatsSummary' :: OpParams ->
                                 Data.Int.Int64 -- ^ __max_splits__
                                 -> Data.Int.Int64 -- ^ __num_buckets__
                                 -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                                 -> Tensor v'2 Float -- ^ __gradients__
                                 -> Tensor v'3 Float -- ^ __hessians__
                                 -> [Tensor v'4 Data.Int.Int32] -- ^ __bucketized_features_list__
                                 -> Tensor Build Float -- ^ __stats_summary__
boostedTreesMakeStatsSummary' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> [Tensor v'4 Int32]
-> Tensor Build Float
boostedTreesMakeStatsSummary' op'options :: OpParams
op'options max_splits :: Int64
max_splits num_buckets :: Int64
num_buckets node_ids :: Tensor v'1 Int32
node_ids
                              gradients :: Tensor v'2 Float
gradients hessians :: Tensor v'3 Float
hessians
                              bucketized_features_list :: [Tensor v'4 Int32]
bucketized_features_list | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("bucketized_features_list", [Tensor v'4 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Int32]
bucketized_features_list)])] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_ids,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
gradients,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
hessians,
                                                             [Tensor v'4 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Int32]
bucketized_features_list]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesMakeStatsSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_splits" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_splits
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Int32]
bucketized_features_list) :: Int64
{-
input_arg { name: "node_ids" type: DT_INT32 }
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "hessians" type: DT_FLOAT }
input_arg {
  name: "bucketized_features_list"
  type: DT_INT32
  number_attr: "num_features"
}
output_arg { name: "stats_summary" type: DT_FLOAT }
attr {
  name: "max_splits" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_features" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesPredict :: forall v'1 v'2 m' . (MonadBuild m') =>
                       Data.Int.Int64 -- ^ __logits_dimension__
                       -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                       -> [Tensor v'2 Data.Int.Int32] -- ^ __bucketized_features__
                       -> m' (Tensor Value Float) -- ^ __logits__
boostedTreesPredict :: Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value Float)
boostedTreesPredict = OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value Float)
boostedTreesPredict' OpParams
forall a. a -> a
id
boostedTreesPredict' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                        Data.Int.Int64 -- ^ __logits_dimension__
                        -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                        -> [Tensor v'2 Data.Int.Int32] -- ^ __bucketized_features__
                        -> m' (Tensor Value Float) -- ^ __logits__
boostedTreesPredict' :: OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> m' (Tensor Value Float)
boostedTreesPredict' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle
                     bucketized_features :: [Tensor v'2 Int32]
bucketized_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_bucketized_features", [("bucketized_features", [Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
bucketized_features)])] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             [Tensor v'2 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int32]
bucketized_features]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesPredict"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_bucketized_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_bucketized_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_bucketized_features :: Int64
num_bucketized_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
bucketized_features) :: Int64
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg {
  name: "bucketized_features"
  type: DT_INT32
  number_attr: "num_bucketized_features"
}
output_arg { name: "logits" type: DT_FLOAT }
attr {
  name: "num_bucketized_features"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr { name: "logits_dimension" type: "int" }
-}
-- | 
boostedTreesQuantileStreamResourceAddSummaries :: forall v'1 v'2
                                                  m' . (MonadBuild m') =>
                                                  Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                  -> [Tensor v'2 Float] -- ^ __summaries__
                                                  -> m' (ControlNode)
boostedTreesQuantileStreamResourceAddSummaries :: Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceAddSummaries = OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
boostedTreesQuantileStreamResourceAddSummaries' OpParams
forall a. a -> a
id
boostedTreesQuantileStreamResourceAddSummaries' :: forall v'1 v'2
                                                   m' . (MonadBuild m') =>
                                                   OpParams ->
                                                   Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                   -> [Tensor v'2 Float] -- ^ __summaries__
                                                   -> m' (ControlNode)
boostedTreesQuantileStreamResourceAddSummaries' :: OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
boostedTreesQuantileStreamResourceAddSummaries' op'options :: OpParams
op'options
                                                quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle
                                                summaries :: [Tensor v'2 Float]
summaries | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("summaries", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
summaries)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
summaries]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesQuantileStreamResourceAddSummaries"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
summaries) :: Int64
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
input_arg {
  name: "summaries" type: DT_FLOAT number_attr: "num_features"
}
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesQuantileStreamResourceDeserialize :: forall v'1 v'2
                                                 m' . (MonadBuild m') =>
                                                 Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                 -> [Tensor v'2 Float] -- ^ __bucket_boundaries__
                                                 -> m' (ControlNode)
boostedTreesQuantileStreamResourceDeserialize :: Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
boostedTreesQuantileStreamResourceDeserialize = OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
boostedTreesQuantileStreamResourceDeserialize' OpParams
forall a. a -> a
id
boostedTreesQuantileStreamResourceDeserialize' :: forall v'1 v'2
                                                  m' . (MonadBuild m') =>
                                                  OpParams ->
                                                  Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                  -> [Tensor v'2 Float] -- ^ __bucket_boundaries__
                                                  -> m' (ControlNode)
boostedTreesQuantileStreamResourceDeserialize' :: OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Float]
-> m' ControlNode
boostedTreesQuantileStreamResourceDeserialize' op'options :: OpParams
op'options
                                               quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle
                                               bucket_boundaries :: [Tensor v'2 Float]
bucket_boundaries | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_streams", [("bucket_boundaries", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
bucket_boundaries)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
bucket_boundaries]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesQuantileStreamResourceDeserialize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_streams" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_streams
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_streams :: Int64
num_streams = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
bucket_boundaries) :: Int64
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
input_arg {
  name: "bucket_boundaries" type: DT_FLOAT number_attr: "num_streams"
}
attr {
  name: "num_streams" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesQuantileStreamResourceFlush :: forall v'1 v'2
                                           m' . (MonadBuild m') =>
                                           Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                           -> Tensor v'2 Data.Int.Int64 -- ^ __num_buckets__
                                           -> m' (ControlNode)
boostedTreesQuantileStreamResourceFlush :: Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
boostedTreesQuantileStreamResourceFlush = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
boostedTreesQuantileStreamResourceFlush' OpParams
forall a. a -> a
id
boostedTreesQuantileStreamResourceFlush' :: forall v'1 v'2
                                            m' . (MonadBuild m') => OpParams ->
                                            Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                            -> Tensor v'2 Data.Int.Int64 -- ^ __num_buckets__
                                            -> m' (ControlNode)
boostedTreesQuantileStreamResourceFlush' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
boostedTreesQuantileStreamResourceFlush' op'options :: OpParams
op'options
                                         quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle
                                         num_buckets :: Tensor v'2 Int64
num_buckets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_buckets]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesQuantileStreamResourceFlush"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
input_arg { name: "num_buckets" type: DT_INT64 }
attr {
  name: "generate_quantiles" type: "bool" default_value { b: false }
}
-}
-- | 
boostedTreesQuantileStreamResourceGetBucketBoundaries :: forall v'1
                                                         m' . (MonadBuild m') =>
                                                         Data.Int.Int64 -- ^ __num_features__
                                                         -> Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                         -> m' ([Tensor Value Float]) -- ^ __bucket_boundaries__
boostedTreesQuantileStreamResourceGetBucketBoundaries :: Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesQuantileStreamResourceGetBucketBoundaries = OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesQuantileStreamResourceGetBucketBoundaries' OpParams
forall a. a -> a
id
boostedTreesQuantileStreamResourceGetBucketBoundaries' :: forall v'1
                                                          m' . (MonadBuild m') =>
                                                          OpParams ->
                                                          Data.Int.Int64 -- ^ __num_features__
                                                          -> Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                          -> m' ([Tensor Value Float]) -- ^ __bucket_boundaries__
boostedTreesQuantileStreamResourceGetBucketBoundaries' :: OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
boostedTreesQuantileStreamResourceGetBucketBoundaries' op'options :: OpParams
op'options num_features :: Int64
num_features
                                                       quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build [Tensor Value Float] -> m' [Tensor Value Float]
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build [Tensor Value Float] -> m' [Tensor Value Float])
-> Build [Tensor Value Float] -> m' [Tensor Value Float]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle]
        [Int64] -> OpDef -> Build [Tensor Value Float]
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [Int64
num_features] (OpType -> OpDef
opDef "BoostedTreesQuantileStreamResourceGetBucketBoundaries"
                                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
output_arg {
  name: "bucket_boundaries"
  type: DT_FLOAT
  number_attr: "num_features"
}
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesQuantileStreamResourceHandleOp :: forall m' . (MonadBuild m') =>
                                              m' (Tensor Value ResourceHandle) -- ^ __resource__
boostedTreesQuantileStreamResourceHandleOp :: m' (Tensor Value ResourceHandle)
boostedTreesQuantileStreamResourceHandleOp = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
boostedTreesQuantileStreamResourceHandleOp' OpParams
forall a. a -> a
id
boostedTreesQuantileStreamResourceHandleOp' :: forall m' . (MonadBuild m') =>
                                               OpParams ->
                                               m' (Tensor Value ResourceHandle) -- ^ __resource__
boostedTreesQuantileStreamResourceHandleOp' :: OpParams -> m' (Tensor Value ResourceHandle)
boostedTreesQuantileStreamResourceHandleOp' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesQuantileStreamResourceHandleOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resource" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
boostedTreesSerializeEnsemble :: forall v'1 m' . (MonadBuild m') =>
                                 Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                 -> m' ((Tensor Value Data.Int.Int64,
                                         Tensor Value Data.ByteString.ByteString))
                                 -- ^ (__stamp_token__, __tree_ensemble_serialized__)
                                 --
                                 -- * __stamp_token__
                                 --
                                 -- * __tree_ensemble_serialized__
boostedTreesSerializeEnsemble :: Tensor v'1 ResourceHandle
-> m' (Tensor Value Int64, Tensor Value ByteString)
boostedTreesSerializeEnsemble = OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Int64, Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Int64, Tensor Value ByteString)
boostedTreesSerializeEnsemble' OpParams
forall a. a -> a
id
boostedTreesSerializeEnsemble' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                                  Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                  -> m' ((Tensor Value Data.Int.Int64,
                                          Tensor Value Data.ByteString.ByteString))
                                  -- ^ (__stamp_token__, __tree_ensemble_serialized__)
                                  --
                                  -- * __stamp_token__
                                  --
                                  -- * __tree_ensemble_serialized__
boostedTreesSerializeEnsemble' :: OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Int64, Tensor Value ByteString)
boostedTreesSerializeEnsemble' op'options :: OpParams
op'options
                               tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value ByteString)
-> m' (Tensor Value Int64, Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value ByteString)
 -> m' (Tensor Value Int64, Tensor Value ByteString))
-> Build (Tensor Value Int64, Tensor Value ByteString)
-> m' (Tensor Value Int64, Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle]
        [Int64]
-> OpDef -> Build (Tensor Value Int64, Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesSerializeEnsemble"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
output_arg { name: "stamp_token" type: DT_INT64 }
output_arg { name: "tree_ensemble_serialized" type: DT_STRING }
-}
-- | 
boostedTreesSparseAggregateStats :: Data.Int.Int64 -- ^ __max_splits__
                                    -> Data.Int.Int64 -- ^ __num_buckets__
                                    -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                                    -> Tensor v'2 Float -- ^ __gradients__
                                    -> Tensor v'3 Float -- ^ __hessians__
                                    -> Tensor v'4 Data.Int.Int32 -- ^ __feature_indices__
                                    -> Tensor v'5 Data.Int.Int32 -- ^ __feature_values__
                                    -> Tensor v'6 Data.Int.Int32 -- ^ __feature_shape__
                                    -> (Tensor Build Data.Int.Int32,
                                        Tensor Build Float,
                                        Tensor Build Data.Int.Int32)
                                    -- ^ (__stats_summary_indices__, __stats_summary_values__, __stats_summary_shape__)
                                    --
                                    -- * __stats_summary_indices__
                                    --
                                    -- * __stats_summary_values__
                                    --
                                    -- * __stats_summary_shape__
boostedTreesSparseAggregateStats :: Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor v'6 Int32
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
boostedTreesSparseAggregateStats = OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor v'6 Int32
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *).
OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor v'6 Int32
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
boostedTreesSparseAggregateStats' OpParams
forall a. a -> a
id
boostedTreesSparseAggregateStats' :: OpParams ->
                                     Data.Int.Int64 -- ^ __max_splits__
                                     -> Data.Int.Int64 -- ^ __num_buckets__
                                     -> Tensor v'1 Data.Int.Int32 -- ^ __node_ids__
                                     -> Tensor v'2 Float -- ^ __gradients__
                                     -> Tensor v'3 Float -- ^ __hessians__
                                     -> Tensor v'4 Data.Int.Int32 -- ^ __feature_indices__
                                     -> Tensor v'5 Data.Int.Int32 -- ^ __feature_values__
                                     -> Tensor v'6 Data.Int.Int32 -- ^ __feature_shape__
                                     -> (Tensor Build Data.Int.Int32,
                                         Tensor Build Float,
                                         Tensor Build Data.Int.Int32)
                                     -- ^ (__stats_summary_indices__, __stats_summary_values__, __stats_summary_shape__)
                                     --
                                     -- * __stats_summary_indices__
                                     --
                                     -- * __stats_summary_values__
                                     --
                                     -- * __stats_summary_shape__
boostedTreesSparseAggregateStats' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor v'6 Int32
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
boostedTreesSparseAggregateStats' op'options :: OpParams
op'options max_splits :: Int64
max_splits num_buckets :: Int64
num_buckets node_ids :: Tensor v'1 Int32
node_ids
                                  gradients :: Tensor v'2 Float
gradients hessians :: Tensor v'3 Float
hessians feature_indices :: Tensor v'4 Int32
feature_indices
                                  feature_values :: Tensor v'5 Int32
feature_values
                                  feature_shape :: Tensor v'6 Int32
feature_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_ids,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
gradients,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
hessians,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
feature_indices,
                                                             Tensor v'5 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int32
feature_values,
                                                             Tensor v'6 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int32
feature_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesSparseAggregateStats"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_splits" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_splits
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "node_ids" type: DT_INT32 }
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "hessians" type: DT_FLOAT }
input_arg { name: "feature_indices" type: DT_INT32 }
input_arg { name: "feature_values" type: DT_INT32 }
input_arg { name: "feature_shape" type: DT_INT32 }
output_arg { name: "stats_summary_indices" type: DT_INT32 }
output_arg { name: "stats_summary_values" type: DT_FLOAT }
output_arg { name: "stats_summary_shape" type: DT_INT32 }
attr {
  name: "max_splits" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
boostedTreesSparseCalculateBestFeatureSplit :: Data.Int.Int64 -- ^ __logits_dimension__
                                               -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                               -> Tensor v'2 Data.Int.Int32 -- ^ __stats_summary_indices__
                                               -> Tensor v'3 Float -- ^ __stats_summary_values__
                                               -> Tensor v'4 Data.Int.Int32 -- ^ __stats_summary_shape__
                                               -> Tensor v'5 Float -- ^ __l1__
                                               -> Tensor v'6 Float -- ^ __l2__
                                               -> Tensor v'7 Float -- ^ __tree_complexity__
                                               -> Tensor v'8 Float -- ^ __min_node_weight__
                                               -> (Tensor Build Data.Int.Int32,
                                                   Tensor Build Float,
                                                   Tensor Build Data.Int.Int32,
                                                   Tensor Build Data.Int.Int32,
                                                   Tensor Build Float,
                                                   Tensor Build Float,
                                                   Tensor Build Data.ByteString.ByteString)
                                               -- ^ (__node_ids__, __gains__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                               --
                                               -- * __node_ids__
                                               --
                                               -- * __gains__
                                               --
                                               -- * __feature_dimensions__
                                               --
                                               -- * __thresholds__
                                               --
                                               -- * __left_node_contribs__
                                               --
                                               -- * __right_node_contribs__
                                               --
                                               -- * __split_with_default_directions__
boostedTreesSparseCalculateBestFeatureSplit :: Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesSparseCalculateBestFeatureSplit = OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesSparseCalculateBestFeatureSplit' OpParams
forall a. a -> a
id
boostedTreesSparseCalculateBestFeatureSplit' :: OpParams ->
                                                Data.Int.Int64 -- ^ __logits_dimension__
                                                -> Tensor v'1 Data.Int.Int32 -- ^ __node_id_range__
                                                -> Tensor v'2 Data.Int.Int32 -- ^ __stats_summary_indices__
                                                -> Tensor v'3 Float -- ^ __stats_summary_values__
                                                -> Tensor v'4 Data.Int.Int32 -- ^ __stats_summary_shape__
                                                -> Tensor v'5 Float -- ^ __l1__
                                                -> Tensor v'6 Float -- ^ __l2__
                                                -> Tensor v'7 Float -- ^ __tree_complexity__
                                                -> Tensor v'8 Float -- ^ __min_node_weight__
                                                -> (Tensor Build Data.Int.Int32,
                                                    Tensor Build Float,
                                                    Tensor Build Data.Int.Int32,
                                                    Tensor Build Data.Int.Int32,
                                                    Tensor Build Float,
                                                    Tensor Build Float,
                                                    Tensor Build Data.ByteString.ByteString)
                                                -- ^ (__node_ids__, __gains__, __feature_dimensions__, __thresholds__, __left_node_contribs__, __right_node_contribs__, __split_with_default_directions__)
                                                --
                                                -- * __node_ids__
                                                --
                                                -- * __gains__
                                                --
                                                -- * __feature_dimensions__
                                                --
                                                -- * __thresholds__
                                                --
                                                -- * __left_node_contribs__
                                                --
                                                -- * __right_node_contribs__
                                                --
                                                -- * __split_with_default_directions__
boostedTreesSparseCalculateBestFeatureSplit' :: OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
boostedTreesSparseCalculateBestFeatureSplit' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension
                                             node_id_range :: Tensor v'1 Int32
node_id_range stats_summary_indices :: Tensor v'2 Int32
stats_summary_indices
                                             stats_summary_values :: Tensor v'3 Float
stats_summary_values
                                             stats_summary_shape :: Tensor v'4 Int32
stats_summary_shape l1 :: Tensor v'5 Float
l1 l2 :: Tensor v'6 Float
l2
                                             tree_complexity :: Tensor v'7 Float
tree_complexity
                                             min_node_weight :: Tensor v'8 Float
min_node_weight | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
     Tensor Build Int32, Tensor Build Float, Tensor Build Float,
     Tensor Build ByteString))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Float, Tensor Build Int32,
    Tensor Build Int32, Tensor Build Float, Tensor Build Float,
    Tensor Build ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
node_id_range,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
stats_summary_indices,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
stats_summary_values,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
stats_summary_shape,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
l1,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
l2,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
tree_complexity,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_node_weight]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BoostedTreesSparseCalculateBestFeatureSplit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "node_id_range" type: DT_INT32 }
input_arg { name: "stats_summary_indices" type: DT_INT32 }
input_arg { name: "stats_summary_values" type: DT_FLOAT }
input_arg { name: "stats_summary_shape" type: DT_INT32 }
input_arg { name: "l1" type: DT_FLOAT }
input_arg { name: "l2" type: DT_FLOAT }
input_arg { name: "tree_complexity" type: DT_FLOAT }
input_arg { name: "min_node_weight" type: DT_FLOAT }
output_arg { name: "node_ids" type: DT_INT32 }
output_arg { name: "gains" type: DT_FLOAT }
output_arg { name: "feature_dimensions" type: DT_INT32 }
output_arg { name: "thresholds" type: DT_INT32 }
output_arg { name: "left_node_contribs" type: DT_FLOAT }
output_arg { name: "right_node_contribs" type: DT_FLOAT }
output_arg {
  name: "split_with_default_directions" type: DT_STRING
}
attr {
  name: "logits_dimension" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "split_type"
  type: "string"
  default_value { s: "inequality" }
  allowed_values { list { s: "inequality" } }
}
-}
-- | 
boostedTreesTrainingPredict :: forall v'1 v'2 v'3 v'4 m' . (MonadBuild m') =>
                               Data.Int.Int64 -- ^ __logits_dimension__
                               -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                               -> Tensor v'2 Data.Int.Int32 -- ^ __cached_tree_ids__
                               -> Tensor v'3 Data.Int.Int32 -- ^ __cached_node_ids__
                               -> [Tensor v'4 Data.Int.Int32] -- ^ __bucketized_features__
                               -> m' ((Tensor Value Float,
                                       Tensor Value Data.Int.Int32,
                                       Tensor Value Data.Int.Int32))
                               -- ^ (__partial_logits__, __tree_ids__, __node_ids__)
                               --
                               -- * __partial_logits__
                               --
                               -- * __tree_ids__
                               --
                               -- * __node_ids__
boostedTreesTrainingPredict :: Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 Int32]
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
boostedTreesTrainingPredict = OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 Int32]
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 Int32]
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
boostedTreesTrainingPredict' OpParams
forall a. a -> a
id
boostedTreesTrainingPredict' :: forall v'1 v'2 v'3 v'4 m' . (MonadBuild m') =>
                                OpParams ->
                                Data.Int.Int64 -- ^ __logits_dimension__
                                -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                -> Tensor v'2 Data.Int.Int32 -- ^ __cached_tree_ids__
                                -> Tensor v'3 Data.Int.Int32 -- ^ __cached_node_ids__
                                -> [Tensor v'4 Data.Int.Int32] -- ^ __bucketized_features__
                                -> m' ((Tensor Value Float,
                                        Tensor Value Data.Int.Int32,
                                        Tensor Value Data.Int.Int32))
                                -- ^ (__partial_logits__, __tree_ids__, __node_ids__)
                                --
                                -- * __partial_logits__
                                --
                                -- * __tree_ids__
                                --
                                -- * __node_ids__
boostedTreesTrainingPredict' :: OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 Int32]
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
boostedTreesTrainingPredict' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle
                             cached_tree_ids :: Tensor v'2 Int32
cached_tree_ids cached_node_ids :: Tensor v'3 Int32
cached_node_ids
                             bucketized_features :: [Tensor v'4 Int32]
bucketized_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_bucketized_features", [("bucketized_features", [Tensor v'4 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Int32]
bucketized_features)])] =
    Build (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
 -> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32))
-> Build
     (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
-> m' (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
cached_tree_ids,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
cached_node_ids,
                                                             [Tensor v'4 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Int32]
bucketized_features]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Int32, Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesTrainingPredict"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_bucketized_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_bucketized_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_bucketized_features :: Int64
num_bucketized_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Int32]
bucketized_features) :: Int64
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg { name: "cached_tree_ids" type: DT_INT32 }
input_arg { name: "cached_node_ids" type: DT_INT32 }
input_arg {
  name: "bucketized_features"
  type: DT_INT32
  number_attr: "num_bucketized_features"
}
output_arg { name: "partial_logits" type: DT_FLOAT }
output_arg { name: "tree_ids" type: DT_INT32 }
output_arg { name: "node_ids" type: DT_INT32 }
attr {
  name: "num_bucketized_features"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr { name: "logits_dimension" type: "int" }
-}
-- | 
boostedTreesUpdateEnsemble :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                              m' . (MonadBuild m') =>
                              Data.Int.Int64 -- ^ __pruning_mode__
                              -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __feature_ids__
                              -> [Tensor v'3 Data.Int.Int32] -- ^ __node_ids__
                              -> [Tensor v'4 Float] -- ^ __gains__
                              -> [Tensor v'5 Data.Int.Int32] -- ^ __thresholds__
                              -> [Tensor v'6 Float] -- ^ __left_node_contribs__
                              -> [Tensor v'7 Float] -- ^ __right_node_contribs__
                              -> Tensor v'8 Data.Int.Int32 -- ^ __max_depth__
                              -> Tensor v'9 Float -- ^ __learning_rate__
                              -> m' (ControlNode)
boostedTreesUpdateEnsemble :: Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> [Tensor v'3 Int32]
-> [Tensor v'4 Float]
-> [Tensor v'5 Int32]
-> [Tensor v'6 Float]
-> [Tensor v'7 Float]
-> Tensor v'8 Int32
-> Tensor v'9 Float
-> m' ControlNode
boostedTreesUpdateEnsemble = OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> [Tensor v'3 Int32]
-> [Tensor v'4 Float]
-> [Tensor v'5 Int32]
-> [Tensor v'6 Float]
-> [Tensor v'7 Float]
-> Tensor v'8 Int32
-> Tensor v'9 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> [Tensor v'3 Int32]
-> [Tensor v'4 Float]
-> [Tensor v'5 Int32]
-> [Tensor v'6 Float]
-> [Tensor v'7 Float]
-> Tensor v'8 Int32
-> Tensor v'9 Float
-> m' ControlNode
boostedTreesUpdateEnsemble' OpParams
forall a. a -> a
id
boostedTreesUpdateEnsemble' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                               m' . (MonadBuild m') => OpParams ->
                               Data.Int.Int64 -- ^ __pruning_mode__
                               -> Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                               -> Tensor v'2 Data.Int.Int32 -- ^ __feature_ids__
                               -> [Tensor v'3 Data.Int.Int32] -- ^ __node_ids__
                               -> [Tensor v'4 Float] -- ^ __gains__
                               -> [Tensor v'5 Data.Int.Int32] -- ^ __thresholds__
                               -> [Tensor v'6 Float] -- ^ __left_node_contribs__
                               -> [Tensor v'7 Float] -- ^ __right_node_contribs__
                               -> Tensor v'8 Data.Int.Int32 -- ^ __max_depth__
                               -> Tensor v'9 Float -- ^ __learning_rate__
                               -> m' (ControlNode)
boostedTreesUpdateEnsemble' :: OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> [Tensor v'3 Int32]
-> [Tensor v'4 Float]
-> [Tensor v'5 Int32]
-> [Tensor v'6 Float]
-> [Tensor v'7 Float]
-> Tensor v'8 Int32
-> Tensor v'9 Float
-> m' ControlNode
boostedTreesUpdateEnsemble' op'options :: OpParams
op'options pruning_mode :: Int64
pruning_mode tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle
                            feature_ids :: Tensor v'2 Int32
feature_ids node_ids :: [Tensor v'3 Int32]
node_ids gains :: [Tensor v'4 Float]
gains thresholds :: [Tensor v'5 Int32]
thresholds
                            left_node_contribs :: [Tensor v'6 Float]
left_node_contribs right_node_contribs :: [Tensor v'7 Float]
right_node_contribs max_depth :: Tensor v'8 Int32
max_depth
                            learning_rate :: Tensor v'9 Float
learning_rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("node_ids", [Tensor v'3 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int32]
node_ids),
                                                                             ("gains", [Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
gains),
                                                                             ("thresholds", [Tensor v'5 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 Int32]
thresholds),
                                                                             ("left_node_contribs", [Tensor v'6 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'6 Float]
left_node_contribs),
                                                                             ("right_node_contribs", [Tensor v'7 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'7 Float]
right_node_contribs)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
feature_ids,
                                                             [Tensor v'3 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int32]
node_ids,
                                                             [Tensor v'4 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Float]
gains,
                                                             [Tensor v'5 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'5 Int32]
thresholds,
                                                             [Tensor v'6 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'6 Float]
left_node_contribs,
                                                             [Tensor v'7 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'7 Float]
right_node_contribs,
                                                             Tensor v'8 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Int32
max_depth,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
learning_rate]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesUpdateEnsemble"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "pruning_mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
pruning_mode
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int32]
node_ids) :: Int64
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg { name: "feature_ids" type: DT_INT32 }
input_arg {
  name: "node_ids" type: DT_INT32 number_attr: "num_features"
}
input_arg {
  name: "gains" type: DT_FLOAT number_attr: "num_features"
}
input_arg {
  name: "thresholds" type: DT_INT32 number_attr: "num_features"
}
input_arg {
  name: "left_node_contribs"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg {
  name: "right_node_contribs"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg { name: "max_depth" type: DT_INT32 }
input_arg { name: "learning_rate" type: DT_FLOAT }
attr { name: "pruning_mode" type: "int" has_minimum: true }
attr { name: "num_features" type: "int" has_minimum: true }
-}
-- | 
boostedTreesUpdateEnsembleV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                                v'11 v'12 m' . (MonadBuild m') =>
                                Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                -> [Tensor v'2 Data.Int.Int32] -- ^ __feature_ids__
                                -> [Tensor v'3 Data.Int.Int32] -- ^ __dimension_ids__
                                -> [Tensor v'4 Data.Int.Int32] -- ^ __node_ids__
                                -> [Tensor v'5 Float] -- ^ __gains__
                                -> [Tensor v'6 Data.Int.Int32] -- ^ __thresholds__
                                -> [Tensor v'7 Float] -- ^ __left_node_contribs__
                                -> [Tensor v'8 Float] -- ^ __right_node_contribs__
                                -> [Tensor v'9 Data.ByteString.ByteString] -- ^ __split_types__
                                -> Tensor v'10 Data.Int.Int32 -- ^ __max_depth__
                                -> Tensor v'11 Float -- ^ __learning_rate__
                                -> Tensor v'12 Data.Int.Int32 -- ^ __pruning_mode__
                                -> m' (ControlNode)
boostedTreesUpdateEnsembleV2 :: Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> [Tensor v'3 Int32]
-> [Tensor v'4 Int32]
-> [Tensor v'5 Float]
-> [Tensor v'6 Int32]
-> [Tensor v'7 Float]
-> [Tensor v'8 Float]
-> [Tensor v'9 ByteString]
-> Tensor v'10 Int32
-> Tensor v'11 Float
-> Tensor v'12 Int32
-> m' ControlNode
boostedTreesUpdateEnsembleV2 = OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> [Tensor v'3 Int32]
-> [Tensor v'4 Int32]
-> [Tensor v'5 Float]
-> [Tensor v'6 Int32]
-> [Tensor v'7 Float]
-> [Tensor v'8 Float]
-> [Tensor v'9 ByteString]
-> Tensor v'10 Int32
-> Tensor v'11 Float
-> Tensor v'12 Int32
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> [Tensor v'3 Int32]
-> [Tensor v'4 Int32]
-> [Tensor v'5 Float]
-> [Tensor v'6 Int32]
-> [Tensor v'7 Float]
-> [Tensor v'8 Float]
-> [Tensor v'9 ByteString]
-> Tensor v'10 Int32
-> Tensor v'11 Float
-> Tensor v'12 Int32
-> m' ControlNode
boostedTreesUpdateEnsembleV2' OpParams
forall a. a -> a
id
boostedTreesUpdateEnsembleV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                                 v'11 v'12 m' . (MonadBuild m') => OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                 -> [Tensor v'2 Data.Int.Int32] -- ^ __feature_ids__
                                 -> [Tensor v'3 Data.Int.Int32] -- ^ __dimension_ids__
                                 -> [Tensor v'4 Data.Int.Int32] -- ^ __node_ids__
                                 -> [Tensor v'5 Float] -- ^ __gains__
                                 -> [Tensor v'6 Data.Int.Int32] -- ^ __thresholds__
                                 -> [Tensor v'7 Float] -- ^ __left_node_contribs__
                                 -> [Tensor v'8 Float] -- ^ __right_node_contribs__
                                 -> [Tensor v'9 Data.ByteString.ByteString] -- ^ __split_types__
                                 -> Tensor v'10 Data.Int.Int32 -- ^ __max_depth__
                                 -> Tensor v'11 Float -- ^ __learning_rate__
                                 -> Tensor v'12 Data.Int.Int32 -- ^ __pruning_mode__
                                 -> m' (ControlNode)
boostedTreesUpdateEnsembleV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 Int32]
-> [Tensor v'3 Int32]
-> [Tensor v'4 Int32]
-> [Tensor v'5 Float]
-> [Tensor v'6 Int32]
-> [Tensor v'7 Float]
-> [Tensor v'8 Float]
-> [Tensor v'9 ByteString]
-> Tensor v'10 Int32
-> Tensor v'11 Float
-> Tensor v'12 Int32
-> m' ControlNode
boostedTreesUpdateEnsembleV2' op'options :: OpParams
op'options tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle feature_ids :: [Tensor v'2 Int32]
feature_ids
                              dimension_ids :: [Tensor v'3 Int32]
dimension_ids node_ids :: [Tensor v'4 Int32]
node_ids gains :: [Tensor v'5 Float]
gains thresholds :: [Tensor v'6 Int32]
thresholds
                              left_node_contribs :: [Tensor v'7 Float]
left_node_contribs right_node_contribs :: [Tensor v'8 Float]
right_node_contribs split_types :: [Tensor v'9 ByteString]
split_types
                              max_depth :: Tensor v'10 Int32
max_depth learning_rate :: Tensor v'11 Float
learning_rate
                              pruning_mode :: Tensor v'12 Int32
pruning_mode | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("dimension_ids", [Tensor v'3 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int32]
dimension_ids),
                                                                              ("node_ids", [Tensor v'4 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Int32]
node_ids),
                                                                              ("gains", [Tensor v'5 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 Float]
gains),
                                                                              ("thresholds", [Tensor v'6 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'6 Int32]
thresholds),
                                                                              ("left_node_contribs", [Tensor v'7 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'7 Float]
left_node_contribs),
                                                                              ("right_node_contribs", [Tensor v'8 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'8 Float]
right_node_contribs),
                                                                              ("split_types", [Tensor v'9 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'9 ByteString]
split_types)]),
                                                            ("num_groups", [("feature_ids", [Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
feature_ids)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle,
                                                             [Tensor v'2 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int32]
feature_ids,
                                                             [Tensor v'3 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int32]
dimension_ids,
                                                             [Tensor v'4 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Int32]
node_ids,
                                                             [Tensor v'5 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'5 Float]
gains,
                                                             [Tensor v'6 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'6 Int32]
thresholds,
                                                             [Tensor v'7 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'7 Float]
left_node_contribs,
                                                             [Tensor v'8 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'8 Float]
right_node_contribs,
                                                             [Tensor v'9 ByteString] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'9 ByteString]
split_types,
                                                             Tensor v'10 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 Int32
max_depth,
                                                             Tensor v'11 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 Float
learning_rate,
                                                             Tensor v'12 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 Int32
pruning_mode]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "BoostedTreesUpdateEnsembleV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_groups" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_groups
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int32]
dimension_ids) :: Int64
    num_groups :: Int64
num_groups = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
feature_ids) :: Int64
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
input_arg {
  name: "feature_ids" type: DT_INT32 number_attr: "num_groups"
}
input_arg {
  name: "dimension_ids" type: DT_INT32 number_attr: "num_features"
}
input_arg {
  name: "node_ids" type: DT_INT32 number_attr: "num_features"
}
input_arg {
  name: "gains" type: DT_FLOAT number_attr: "num_features"
}
input_arg {
  name: "thresholds" type: DT_INT32 number_attr: "num_features"
}
input_arg {
  name: "left_node_contribs"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg {
  name: "right_node_contribs"
  type: DT_FLOAT
  number_attr: "num_features"
}
input_arg {
  name: "split_types" type: DT_STRING number_attr: "num_features"
}
input_arg { name: "max_depth" type: DT_INT32 }
input_arg { name: "learning_rate" type: DT_FLOAT }
input_arg { name: "pruning_mode" type: DT_INT32 }
attr { name: "num_features" type: "int" has_minimum: true }
attr {
  name: "logits_dimension" type: "int" default_value { i: 1 }
}
attr {
  name: "num_groups"
  type: "int"
  default_value { i: 1 }
  has_minimum: true
  minimum: 1
}
-}
-- | 
broadcastArgs :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] t) =>
                 Tensor v'1 t -- ^ __s0__
                 -> Tensor v'2 t -- ^ __s1__
                 -> Tensor Build t -- ^ __r0__
broadcastArgs :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
broadcastArgs = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
broadcastArgs' OpParams
forall a. a -> a
id
broadcastArgs' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] t) => OpParams ->
                  Tensor v'1 t -- ^ __s0__
                  -> Tensor v'2 t -- ^ __s1__
                  -> Tensor Build t -- ^ __r0__
broadcastArgs' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
broadcastArgs' op'options :: OpParams
op'options s0 :: Tensor v'1 t
s0 s1 :: Tensor v'2 t
s1 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
s0,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
s1]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BroadcastArgs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "s0" type_attr: "T" }
input_arg { name: "s1" type_attr: "T" }
output_arg { name: "r0" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
broadcastGradientArgs :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] t) =>
                         Tensor v'1 t -- ^ __s0__
                         -> Tensor v'2 t -- ^ __s1__
                         -> (Tensor Build t, Tensor Build t)
                         -- ^ (__r0__, __r1__)
                         --
                         -- * __r0__
                         --
                         -- * __r1__
broadcastGradientArgs :: Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
broadcastGradientArgs = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
broadcastGradientArgs' OpParams
forall a. a -> a
id
broadcastGradientArgs' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __s0__
                          -> Tensor v'2 t -- ^ __s1__
                          -> (Tensor Build t, Tensor Build t)
                          -- ^ (__r0__, __r1__)
                          --
                          -- * __r0__
                          --
                          -- * __r1__
broadcastGradientArgs' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
broadcastGradientArgs' op'options :: OpParams
op'options s0 :: Tensor v'1 t
s0 s1 :: Tensor v'2 t
s1 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
s0,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
s1]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BroadcastGradientArgs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "s0" type_attr: "T" }
input_arg { name: "s1" type_attr: "T" }
output_arg { name: "r0" type_attr: "T" }
output_arg { name: "r1" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
broadcastTo :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tidx) =>
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 tidx -- ^ __shape__
               -> Tensor Build t -- ^ __output__
broadcastTo :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
broadcastTo = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
broadcastTo' OpParams
forall a. a -> a
id
broadcastTo' :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tidx) =>
                OpParams ->
                Tensor v'1 t -- ^ __input__
                -> Tensor v'2 tidx -- ^ __shape__
                -> Tensor Build t -- ^ __output__
broadcastTo' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
broadcastTo' op'options :: OpParams
op'options input :: Tensor v'1 t
input shape :: Tensor v'2 tidx
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BroadcastTo"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "shape" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
bucketize :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64, Double,
                                     Float] t) => Tensor v'1 t -- ^ __input__
             -> Tensor Build Data.Int.Int32 -- ^ __output__
bucketize :: Tensor v'1 t -> Tensor Build Int32
bucketize = OpParams -> Tensor v'1 t -> Tensor Build Int32
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build Int32
bucketize' OpParams
forall a. a -> a
id
bucketize' :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64, Double,
                                      Float] t) => OpParams ->
              Tensor v'1 t -- ^ __input__
              -> Tensor Build Data.Int.Int32 -- ^ __output__
bucketize' :: OpParams -> Tensor v'1 t -> Tensor Build Int32
bucketize' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Bucketize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_INT32 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "boundaries" type: "list(float)" }
-}
-- | 
bytesProducedStatsDataset :: [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                             -> Tensor Build Variant -- ^ __handle__
bytesProducedStatsDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
bytesProducedStatsDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
bytesProducedStatsDataset' OpParams
forall a. a -> a
id
bytesProducedStatsDataset' :: OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                              -> Tensor Build Variant -- ^ __handle__
bytesProducedStatsDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
bytesProducedStatsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                           tag :: Tensor v'2 ByteString
tag | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tag]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "BytesProducedStatsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "tag" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
cSRSparseMatrixComponents :: forall v'1 v'2
                             type' . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Double, Float] type') =>
                             Tensor v'1 Variant -- ^ __csr_sparse_matrix__
                             -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                             -> (Tensor Build Data.Int.Int32,
                                 Tensor Build Data.Int.Int32,
                                 Tensor Build type')
                             -- ^ (__row_ptrs__, __col_inds__, __values__)
                             --
                             -- * __row_ptrs__
                             --
                             -- * __col_inds__
                             --
                             -- * __values__
cSRSparseMatrixComponents :: Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
cSRSparseMatrixComponents = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
forall (v'1 :: * -> *) (v'2 :: * -> *) type'.
OneOf '[Complex Double, Complex Float, Double, Float] type' =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
cSRSparseMatrixComponents' OpParams
forall a. a -> a
id
cSRSparseMatrixComponents' :: forall v'1 v'2
                              type' . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Double, Float] type') =>
                              OpParams ->
                              Tensor v'1 Variant -- ^ __csr_sparse_matrix__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                              -> (Tensor Build Data.Int.Int32,
                                  Tensor Build Data.Int.Int32,
                                  Tensor Build type')
                              -- ^ (__row_ptrs__, __col_inds__, __values__)
                              --
                              -- * __row_ptrs__
                              --
                              -- * __col_inds__
                              --
                              -- * __values__
cSRSparseMatrixComponents' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
cSRSparseMatrixComponents' op'options :: OpParams
op'options csr_sparse_matrix :: Tensor v'1 Variant
csr_sparse_matrix
                           index :: Tensor v'2 Int32
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Int32, Tensor Build type'))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
csr_sparse_matrix,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CSRSparseMatrixComponents"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ type' -> DataType
forall a. TensorType a => a -> DataType
tensorType (type'
forall a. HasCallStack => a
undefined :: type')
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "csr_sparse_matrix" type: DT_VARIANT }
input_arg { name: "index" type: DT_INT32 }
output_arg { name: "row_ptrs" type: DT_INT32 }
output_arg { name: "col_inds" type: DT_INT32 }
output_arg { name: "values" type_attr: "type" }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
cSRSparseMatrixToDense :: forall v'1
                          type' . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float), Double,
                                           Float] type') =>
                          Tensor v'1 Variant -- ^ __sparse_input__
                          -> Tensor Build type' -- ^ __dense_output__
cSRSparseMatrixToDense :: Tensor v'1 Variant -> Tensor Build type'
cSRSparseMatrixToDense = OpParams -> Tensor v'1 Variant -> Tensor Build type'
forall (v'1 :: * -> *) type'.
OneOf '[Complex Double, Complex Float, Double, Float] type' =>
OpParams -> Tensor v'1 Variant -> Tensor Build type'
cSRSparseMatrixToDense' OpParams
forall a. a -> a
id
cSRSparseMatrixToDense' :: forall v'1
                           type' . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Double, Float] type') => OpParams ->
                           Tensor v'1 Variant -- ^ __sparse_input__
                           -> Tensor Build type' -- ^ __dense_output__
cSRSparseMatrixToDense' :: OpParams -> Tensor v'1 Variant -> Tensor Build type'
cSRSparseMatrixToDense' op'options :: OpParams
op'options sparse_input :: Tensor v'1 Variant
sparse_input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build type'
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build type')
-> Build OpDef -> Tensor Build type'
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
sparse_input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CSRSparseMatrixToDense"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ type' -> DataType
forall a. TensorType a => a -> DataType
tensorType (type'
forall a. HasCallStack => a
undefined :: type')
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_input" type: DT_VARIANT }
output_arg { name: "dense_output" type_attr: "type" }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
cSRSparseMatrixToSparseTensor :: forall v'1
                                 type' . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Double, Float] type') =>
                                 Tensor v'1 Variant -- ^ __sparse_matrix__
                                 -> (Tensor Build Data.Int.Int64,
                                     Tensor Build type',
                                     Tensor Build Data.Int.Int64)
                                 -- ^ (__indices__, __values__, __dense_shape__)
                                 --
                                 -- * __indices__
                                 --
                                 -- * __values__
                                 --
                                 -- * __dense_shape__
cSRSparseMatrixToSparseTensor :: Tensor v'1 Variant
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
cSRSparseMatrixToSparseTensor = OpParams
-> Tensor v'1 Variant
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
forall (v'1 :: * -> *) type'.
OneOf '[Complex Double, Complex Float, Double, Float] type' =>
OpParams
-> Tensor v'1 Variant
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
cSRSparseMatrixToSparseTensor' OpParams
forall a. a -> a
id
cSRSparseMatrixToSparseTensor' :: forall v'1
                                  type' . (OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float),
                                                   Double, Float] type') =>
                                  OpParams ->
                                  Tensor v'1 Variant -- ^ __sparse_matrix__
                                  -> (Tensor Build Data.Int.Int64,
                                      Tensor Build type',
                                      Tensor Build Data.Int.Int64)
                                  -- ^ (__indices__, __values__, __dense_shape__)
                                  --
                                  -- * __indices__
                                  --
                                  -- * __values__
                                  --
                                  -- * __dense_shape__
cSRSparseMatrixToSparseTensor' :: OpParams
-> Tensor v'1 Variant
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
cSRSparseMatrixToSparseTensor' op'options :: OpParams
op'options sparse_matrix :: Tensor v'1 Variant
sparse_matrix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build type', Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
sparse_matrix]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CSRSparseMatrixToSparseTensor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ type' -> DataType
forall a. TensorType a => a -> DataType
tensorType (type'
forall a. HasCallStack => a
undefined :: type')
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_matrix" type: DT_VARIANT }
output_arg { name: "indices" type: DT_INT64 }
output_arg { name: "values" type_attr: "type" }
output_arg { name: "dense_shape" type: DT_INT64 }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
cSVDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types
              m' . (MonadBuild m', OneOfs '[Data.ByteString.ByteString,
                                            Data.Int.Int32, Data.Int.Int64,
                                            Double, Float] output_types) =>
              Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
              -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
              -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
              -> Tensor v'4 Bool -- ^ __header__
              -> Tensor v'5 Data.ByteString.ByteString -- ^ __field_delim__
              -> Tensor v'6 Bool -- ^ __use_quote_delim__
              -> Tensor v'7 Data.ByteString.ByteString -- ^ __na_value__
              -> Tensor v'8 Data.Int.Int64 -- ^ __select_cols__
              -> TensorList (v'9) output_types -- ^ __record_defaults__
              -> m' (Tensor Value Variant) -- ^ __handle__
cSVDataset :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
cSVDataset = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (output_types :: [*])
       (m' :: * -> *).
(MonadBuild m',
 OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
cSVDataset' OpParams
forall a. a -> a
id
cSVDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 output_types
               m' . (MonadBuild m', OneOfs '[Data.ByteString.ByteString,
                                             Data.Int.Int32, Data.Int.Int64,
                                             Double, Float] output_types) =>
               OpParams ->
               Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
               -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
               -> Tensor v'4 Bool -- ^ __header__
               -> Tensor v'5 Data.ByteString.ByteString -- ^ __field_delim__
               -> Tensor v'6 Bool -- ^ __use_quote_delim__
               -> Tensor v'7 Data.ByteString.ByteString -- ^ __na_value__
               -> Tensor v'8 Data.Int.Int64 -- ^ __select_cols__
               -> TensorList (v'9) output_types -- ^ __record_defaults__
               -> m' (Tensor Value Variant) -- ^ __handle__
cSVDataset' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
cSVDataset' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames compression_type :: Tensor v'2 ByteString
compression_type buffer_size :: Tensor v'3 Int64
buffer_size header :: Tensor v'4 Bool
header field_delim :: Tensor v'5 ByteString
field_delim
            use_quote_delim :: Tensor v'6 Bool
use_quote_delim na_value :: Tensor v'7 ByteString
na_value select_cols :: Tensor v'8 Int64
select_cols
            record_defaults :: TensorList v'9 output_types
record_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
compression_type,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
buffer_size,
                                                             Tensor v'4 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Bool
header,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
field_delim,
                                                             Tensor v'6 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Bool
use_quote_delim,
                                                             Tensor v'7 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 ByteString
na_value,
                                                             Tensor v'8 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Int64
select_cols,
                                                             TensorList v'9 output_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'9 output_types
record_defaults]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CSVDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "header" type: DT_BOOL }
input_arg { name: "field_delim" type: DT_STRING }
input_arg { name: "use_quote_delim" type: DT_BOOL }
input_arg { name: "na_value" type: DT_STRING }
input_arg { name: "select_cols" type: DT_INT64 }
input_arg {
  name: "record_defaults" type_list_attr: "output_types"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_STRING
    }
  }
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
cTCBeamSearchDecoder :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                        Data.Int.Int64 -- ^ __beam_width__
                        -> Data.Int.Int64 -- ^ __top_paths__
                        -> Tensor v'1 t -- ^ __inputs__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __sequence_length__
                        -> ([Tensor Build Data.Int.Int64],
                            [Tensor Build Data.Int.Int64],
                            [Tensor Build Data.Int.Int64], Tensor Build t)
                        -- ^ (__decoded_indices__, __decoded_values__, __decoded_shape__, __log_probability__)
                        --
                        -- * __decoded_indices__
                        --
                        -- * __decoded_values__
                        --
                        -- * __decoded_shape__
                        --
                        -- * __log_probability__
cTCBeamSearchDecoder :: Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
cTCBeamSearchDecoder = OpParams
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
cTCBeamSearchDecoder' OpParams
forall a. a -> a
id
cTCBeamSearchDecoder' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                         OpParams ->
                         Data.Int.Int64 -- ^ __beam_width__
                         -> Data.Int.Int64 -- ^ __top_paths__
                         -> Tensor v'1 t -- ^ __inputs__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __sequence_length__
                         -> ([Tensor Build Data.Int.Int64],
                             [Tensor Build Data.Int.Int64],
                             [Tensor Build Data.Int.Int64], Tensor Build t)
                         -- ^ (__decoded_indices__, __decoded_values__, __decoded_shape__, __log_probability__)
                         --
                         -- * __decoded_indices__
                         --
                         -- * __decoded_values__
                         --
                         -- * __decoded_shape__
                         --
                         -- * __log_probability__
cTCBeamSearchDecoder' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int32
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
cTCBeamSearchDecoder' op'options :: OpParams
op'options beam_width :: Int64
beam_width top_paths :: Int64
top_paths inputs :: Tensor v'1 t
inputs
                      sequence_length :: Tensor v'2 Int32
sequence_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
top_paths, Int64
top_paths, Int64
top_paths] (Build OpDef
 -> ([Tensor Build Int64], [Tensor Build Int64],
     [Tensor Build Int64], Tensor Build t))
-> Build OpDef
-> ([Tensor Build Int64], [Tensor Build Int64],
    [Tensor Build Int64], Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
inputs,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
sequence_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CTCBeamSearchDecoder"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "beam_width" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
beam_width
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "top_paths" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
top_paths
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_attr: "T" }
input_arg { name: "sequence_length" type: DT_INT32 }
output_arg {
  name: "decoded_indices" type: DT_INT64 number_attr: "top_paths"
}
output_arg {
  name: "decoded_values" type: DT_INT64 number_attr: "top_paths"
}
output_arg {
  name: "decoded_shape" type: DT_INT64 number_attr: "top_paths"
}
output_arg { name: "log_probability" type_attr: "T" }
attr {
  name: "beam_width" type: "int" has_minimum: true minimum: 1
}
attr { name: "top_paths" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "merge_repeated" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
cTCGreedyDecoder :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                    Tensor v'1 t -- ^ __inputs__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __sequence_length__
                    -> (Tensor Build Data.Int.Int64,
                        Tensor Build Data.Int.Int64,
                        Tensor Build Data.Int.Int64, Tensor Build t)
                    -- ^ (__decoded_indices__, __decoded_values__, __decoded_shape__, __log_probability__)
                    --
                    -- * __decoded_indices__
                    --
                    -- * __decoded_values__
                    --
                    -- * __decoded_shape__
                    --
                    -- * __log_probability__
cTCGreedyDecoder :: Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
cTCGreedyDecoder = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
cTCGreedyDecoder' OpParams
forall a. a -> a
id
cTCGreedyDecoder' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __inputs__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __sequence_length__
                     -> (Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64, Tensor Build t)
                     -- ^ (__decoded_indices__, __decoded_values__, __decoded_shape__, __log_probability__)
                     --
                     -- * __decoded_indices__
                     --
                     -- * __decoded_values__
                     --
                     -- * __decoded_shape__
                     --
                     -- * __log_probability__
cTCGreedyDecoder' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
cTCGreedyDecoder' op'options :: OpParams
op'options inputs :: Tensor v'1 t
inputs sequence_length :: Tensor v'2 Int32
sequence_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64,
    Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
inputs,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
sequence_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CTCGreedyDecoder"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_attr: "T" }
input_arg { name: "sequence_length" type: DT_INT32 }
output_arg { name: "decoded_indices" type: DT_INT64 }
output_arg { name: "decoded_values" type: DT_INT64 }
output_arg { name: "decoded_shape" type: DT_INT64 }
output_arg { name: "log_probability" type_attr: "T" }
attr {
  name: "merge_repeated" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
cTCLoss :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Double, Float] t) =>
           Tensor v'1 t -- ^ __inputs__
           -> Tensor v'2 Data.Int.Int64 -- ^ __labels_indices__
           -> Tensor v'3 Data.Int.Int32 -- ^ __labels_values__
           -> Tensor v'4 Data.Int.Int32 -- ^ __sequence_length__
           -> (Tensor Build t, Tensor Build t) -- ^ (__loss__, __gradient__)
           --
           -- * __loss__
           --
           -- * __gradient__
cTCLoss :: Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build t, Tensor Build t)
cTCLoss = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build t, Tensor Build t)
cTCLoss' OpParams
forall a. a -> a
id
cTCLoss' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __inputs__
            -> Tensor v'2 Data.Int.Int64 -- ^ __labels_indices__
            -> Tensor v'3 Data.Int.Int32 -- ^ __labels_values__
            -> Tensor v'4 Data.Int.Int32 -- ^ __sequence_length__
            -> (Tensor Build t, Tensor Build t) -- ^ (__loss__, __gradient__)
            --
            -- * __loss__
            --
            -- * __gradient__
cTCLoss' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build t, Tensor Build t)
cTCLoss' op'options :: OpParams
op'options inputs :: Tensor v'1 t
inputs labels_indices :: Tensor v'2 Int64
labels_indices labels_values :: Tensor v'3 Int32
labels_values
         sequence_length :: Tensor v'4 Int32
sequence_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
inputs,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
labels_indices,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
labels_values,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
sequence_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CTCLoss"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_attr: "T" }
input_arg { name: "labels_indices" type: DT_INT64 }
input_arg { name: "labels_values" type: DT_INT32 }
input_arg { name: "sequence_length" type: DT_INT32 }
output_arg { name: "loss" type_attr: "T" }
output_arg { name: "gradient" type_attr: "T" }
attr {
  name: "preprocess_collapse_repeated"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "ctc_merge_repeated" type: "bool" default_value { b: true }
}
attr {
  name: "ignore_longer_outputs_than_inputs"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
cTCLossV2 :: Tensor v'1 Float -- ^ __inputs__
             -> Tensor v'2 Data.Int.Int64 -- ^ __labels_indices__
             -> Tensor v'3 Data.Int.Int32 -- ^ __labels_values__
             -> Tensor v'4 Data.Int.Int32 -- ^ __sequence_length__
             -> (Tensor Build Float, Tensor Build Float)
             -- ^ (__loss__, __gradient__)
             --
             -- * __loss__
             --
             -- * __gradient__
cTCLossV2 :: Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build Float, Tensor Build Float)
cTCLossV2 = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build Float, Tensor Build Float)
cTCLossV2' OpParams
forall a. a -> a
id
cTCLossV2' :: OpParams ->
              Tensor v'1 Float -- ^ __inputs__
              -> Tensor v'2 Data.Int.Int64 -- ^ __labels_indices__
              -> Tensor v'3 Data.Int.Int32 -- ^ __labels_values__
              -> Tensor v'4 Data.Int.Int32 -- ^ __sequence_length__
              -> (Tensor Build Float, Tensor Build Float)
              -- ^ (__loss__, __gradient__)
              --
              -- * __loss__
              --
              -- * __gradient__
cTCLossV2' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> (Tensor Build Float, Tensor Build Float)
cTCLossV2' op'options :: OpParams
op'options inputs :: Tensor v'1 Float
inputs labels_indices :: Tensor v'2 Int64
labels_indices labels_values :: Tensor v'3 Int32
labels_values
           sequence_length :: Tensor v'4 Int32
sequence_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Float, Tensor Build Float))
-> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
inputs,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
labels_indices,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
labels_values,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
sequence_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CTCLossV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_FLOAT }
input_arg { name: "labels_indices" type: DT_INT64 }
input_arg { name: "labels_values" type: DT_INT32 }
input_arg { name: "sequence_length" type: DT_INT32 }
output_arg { name: "loss" type: DT_FLOAT }
output_arg { name: "gradient" type: DT_FLOAT }
attr {
  name: "preprocess_collapse_repeated"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "ctc_merge_repeated" type: "bool" default_value { b: true }
}
attr {
  name: "ignore_longer_outputs_than_inputs"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
cacheDataset :: [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                -> Tensor Build Variant -- ^ __handle__
cacheDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
cacheDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
cacheDataset' OpParams
forall a. a -> a
id
cacheDataset' :: OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                 -> Tensor Build Variant -- ^ __handle__
cacheDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
cacheDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
              filename :: Tensor v'2 ByteString
filename | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CacheDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "filename" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
cacheDatasetV2 :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                  [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                  -> Tensor v'3 ResourceHandle -- ^ __cache__
                  -> m' (Tensor Value Variant) -- ^ __handle__
cacheDatasetV2 :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
cacheDatasetV2 = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
cacheDatasetV2' OpParams
forall a. a -> a
id
cacheDatasetV2' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                   [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                   -> Tensor v'3 ResourceHandle -- ^ __cache__
                   -> m' (Tensor Value Variant) -- ^ __handle__
cacheDatasetV2' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
cacheDatasetV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset filename :: Tensor v'2 ByteString
filename
                cache :: Tensor v'3 ResourceHandle
cache | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
cache]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CacheDatasetV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "cache" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
cast :: forall v'1 srcT dstT . (TensorType srcT, TensorType dstT) =>
        Tensor v'1 srcT -- ^ __x__
        -> Tensor Build dstT -- ^ __y__
cast :: Tensor v'1 srcT -> Tensor Build dstT
cast = OpParams -> Tensor v'1 srcT -> Tensor Build dstT
forall (v'1 :: * -> *) srcT dstT.
(TensorType srcT, TensorType dstT) =>
OpParams -> Tensor v'1 srcT -> Tensor Build dstT
cast' OpParams
forall a. a -> a
id
cast' :: forall v'1 srcT dstT . (TensorType srcT, TensorType dstT) =>
         OpParams ->
         Tensor v'1 srcT -- ^ __x__
         -> Tensor Build dstT -- ^ __y__
cast' :: OpParams -> Tensor v'1 srcT -> Tensor Build dstT
cast' op'options :: OpParams
op'options x :: Tensor v'1 srcT
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dstT
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dstT)
-> Build OpDef -> Tensor Build dstT
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 srcT -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 srcT
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cast"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "SrcT" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ srcT -> DataType
forall a. TensorType a => a -> DataType
tensorType (srcT
forall a. HasCallStack => a
undefined :: srcT)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "DstT" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dstT -> DataType
forall a. TensorType a => a -> DataType
tensorType (dstT
forall a. HasCallStack => a
undefined :: dstT)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "SrcT" }
output_arg { name: "y" type_attr: "DstT" }
attr { name: "SrcT" type: "type" }
attr { name: "DstT" type: "type" }
attr { name: "Truncate" type: "bool" default_value { b: false } }
-}
-- | 
ceil :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
ceil :: Tensor v'1 t -> Tensor Build t
ceil = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
ceil' OpParams
forall a. a -> a
id
ceil' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
ceil' :: OpParams -> Tensor v'1 t -> Tensor Build t
ceil' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Ceil"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
checkNumerics :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                           Double, Float] t) =>
                 ByteString -- ^ __message__
                 -> Tensor v'1 t -- ^ __tensor__
                 -> m' (Tensor Value t) -- ^ __output__
checkNumerics :: ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumerics = OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumerics' OpParams
forall a. a -> a
id
checkNumerics' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                            Double, Float] t) =>
                  OpParams ->
                  ByteString -- ^ __message__
                  -> Tensor v'1 t -- ^ __tensor__
                  -> m' (Tensor Value t) -- ^ __output__
checkNumerics' :: OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumerics' op'options :: OpParams
op'options message :: ByteString
message tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CheckNumerics"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "message" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
message
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "message" type: "string" }
-}
-- | 
checkNumericsV2 :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                             Double,
                                                             Float] t) =>
                   ByteString -- ^ __message__
                   -> Tensor v'1 t -- ^ __tensor__
                   -> m' (Tensor Value t) -- ^ __output__
checkNumericsV2 :: ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumericsV2 = OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumericsV2' OpParams
forall a. a -> a
id
checkNumericsV2' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                    OpParams ->
                    ByteString -- ^ __message__
                    -> Tensor v'1 t -- ^ __tensor__
                    -> m' (Tensor Value t) -- ^ __output__
checkNumericsV2' :: OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
checkNumericsV2' op'options :: OpParams
op'options message :: ByteString
message tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CheckNumericsV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "message" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
message
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "message" type: "string" }
-}
-- | 
cholesky :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
cholesky :: Tensor v'1 t -> Tensor Build t
cholesky = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
cholesky' OpParams
forall a. a -> a
id
cholesky' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
cholesky' :: OpParams -> Tensor v'1 t -> Tensor Build t
cholesky' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cholesky"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
choleskyGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) => Tensor v'1 t -- ^ __l__
                -> Tensor v'2 t -- ^ __grad__
                -> Tensor Build t -- ^ __output__
choleskyGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
choleskyGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
choleskyGrad' OpParams
forall a. a -> a
id
choleskyGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                 Tensor v'1 t -- ^ __l__
                 -> Tensor v'2 t -- ^ __grad__
                 -> Tensor Build t -- ^ __output__
choleskyGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
choleskyGrad' op'options :: OpParams
op'options l :: Tensor v'1 t
l grad :: Tensor v'2 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
l,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CholeskyGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "l" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
-}
-- | 
chooseFastestDataset :: Data.Int.Int64 -- ^ __num_experiments__
                        -> [DataType] -- ^ __output_types__
                        -> [Tensor v'1 Variant] -- ^ __input_datasets__
                        -> Tensor Build Variant -- ^ __handle__
chooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
chooseFastestDataset = OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
chooseFastestDataset' OpParams
forall a. a -> a
id
chooseFastestDataset' :: OpParams ->
                         Data.Int.Int64 -- ^ __num_experiments__
                         -> [DataType] -- ^ __output_types__
                         -> [Tensor v'1 Variant] -- ^ __input_datasets__
                         -> Tensor Build Variant -- ^ __handle__
chooseFastestDataset' :: OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
chooseFastestDataset' op'options :: OpParams
op'options num_experiments :: Int64
num_experiments output_types :: [DataType]
output_types
                      input_datasets :: [Tensor v'1 Variant]
input_datasets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("input_datasets", [Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Variant] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Variant]
input_datasets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ChooseFastestDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_experiments" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_experiments
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets) :: Int64
{-
input_arg {
  name: "input_datasets" type: DT_VARIANT number_attr: "N"
}
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "num_experiments" type: "int" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
clipByValue :: forall v'1 v'2 v'3 t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) =>
               Tensor v'1 t -- ^ __t__
               -> Tensor v'2 t -- ^ __clip_value_min__
               -> Tensor v'3 t -- ^ __clip_value_max__
               -> Tensor Build t -- ^ __output__
clipByValue :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
clipByValue = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
clipByValue' OpParams
forall a. a -> a
id
clipByValue' :: forall v'1 v'2 v'3 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) => OpParams ->
                Tensor v'1 t -- ^ __t__
                -> Tensor v'2 t -- ^ __clip_value_min__
                -> Tensor v'3 t -- ^ __clip_value_max__
                -> Tensor Build t -- ^ __output__
clipByValue' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
clipByValue' op'options :: OpParams
op'options t :: Tensor v'1 t
t clip_value_min :: Tensor v'2 t
clip_value_min clip_value_max :: Tensor v'3 t
clip_value_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
t,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
clip_value_min,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
clip_value_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ClipByValue"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "t" type_attr: "T" }
input_arg { name: "clip_value_min" type_attr: "T" }
input_arg { name: "clip_value_max" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
closeSummaryWriter :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> m' (ControlNode)
closeSummaryWriter :: Tensor v'1 ResourceHandle -> m' ControlNode
closeSummaryWriter = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
closeSummaryWriter' OpParams
forall a. a -> a
id
closeSummaryWriter' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __writer__
                       -> m' (ControlNode)
closeSummaryWriter' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
closeSummaryWriter' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CloseSummaryWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
-}
-- | 
collectiveBcastRecv :: forall t m' . (MonadBuild m', OneOf '[Bool,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Word.Word16,
                                                             Double,
                                                             Float] t) =>
                       Data.Int.Int64 -- ^ __group_key__
                       -> Data.Int.Int64 -- ^ __group_size__
                       -> Data.Int.Int64 -- ^ __instance_key__
                       -> Shape -- ^ __shape__
                       -> m' (Tensor Value t) -- ^ __data__
collectiveBcastRecv :: Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
collectiveBcastRecv = OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m',
 OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) =>
OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
collectiveBcastRecv' OpParams
forall a. a -> a
id
collectiveBcastRecv' :: forall t m' . (MonadBuild m', OneOf '[Bool,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                        OpParams ->
                        Data.Int.Int64 -- ^ __group_key__
                        -> Data.Int.Int64 -- ^ __group_size__
                        -> Data.Int.Int64 -- ^ __instance_key__
                        -> Shape -- ^ __shape__
                        -> m' (Tensor Value t) -- ^ __data__
collectiveBcastRecv' :: OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
collectiveBcastRecv' op'options :: OpParams
op'options group_key :: Int64
group_key group_size :: Int64
group_size instance_key :: Int64
instance_key
                     shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CollectiveBcastRecv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_size
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "instance_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
instance_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "data" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BOOL
      type: DT_FLOAT
      type: DT_HALF
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "group_size" type: "int" }
attr { name: "group_key" type: "int" }
attr { name: "instance_key" type: "int" }
attr { name: "shape" type: "shape" }
attr {
  name: "communication_hint"
  type: "string"
  default_value { s: "auto" }
}
attr {
  name: "timeout_seconds" type: "float" default_value { f: 0.0 }
}
-}
-- | 
collectiveBcastSend :: forall v'1 t m' . (MonadBuild m', OneOf '[Bool,
                                                                 Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Word.Word16,
                                                                 Double,
                                                                 Float] t) =>
                       Data.Int.Int64 -- ^ __group_key__
                       -> Data.Int.Int64 -- ^ __group_size__
                       -> Data.Int.Int64 -- ^ __instance_key__
                       -> Shape -- ^ __shape__
                       -> Tensor v'1 t -- ^ __input__
                       -> m' (Tensor Value t) -- ^ __data__
collectiveBcastSend :: Int64
-> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
collectiveBcastSend = OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveBcastSend' OpParams
forall a. a -> a
id
collectiveBcastSend' :: forall v'1 t m' . (MonadBuild m', OneOf '[Bool,
                                                                  Data.Int.Int32,
                                                                  Data.Int.Int64,
                                                                  Data.Word.Word16,
                                                                  Double,
                                                                  Float] t) =>
                        OpParams ->
                        Data.Int.Int64 -- ^ __group_key__
                        -> Data.Int.Int64 -- ^ __group_size__
                        -> Data.Int.Int64 -- ^ __instance_key__
                        -> Shape -- ^ __shape__
                        -> Tensor v'1 t -- ^ __input__
                        -> m' (Tensor Value t) -- ^ __data__
collectiveBcastSend' :: OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveBcastSend' op'options :: OpParams
op'options group_key :: Int64
group_key group_size :: Int64
group_size instance_key :: Int64
instance_key shape :: Shape
shape
                     input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CollectiveBcastSend"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_size
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "instance_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
instance_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "data" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BOOL
      type: DT_FLOAT
      type: DT_HALF
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "group_size" type: "int" }
attr { name: "group_key" type: "int" }
attr { name: "instance_key" type: "int" }
attr { name: "shape" type: "shape" }
attr {
  name: "communication_hint"
  type: "string"
  default_value { s: "auto" }
}
attr {
  name: "timeout_seconds" type: "float" default_value { f: 0.0 }
}
-}
-- | 
collectiveGather :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                    Data.Int.Int64 -- ^ __group_key__
                    -> Data.Int.Int64 -- ^ __group_size__
                    -> Data.Int.Int64 -- ^ __instance_key__
                    -> Shape -- ^ __shape__
                    -> Tensor v'1 t -- ^ __input__
                    -> m' (Tensor Value t) -- ^ __data__
collectiveGather :: Int64
-> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
collectiveGather = OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveGather' OpParams
forall a. a -> a
id
collectiveGather' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                     OpParams ->
                     Data.Int.Int64 -- ^ __group_key__
                     -> Data.Int.Int64 -- ^ __group_size__
                     -> Data.Int.Int64 -- ^ __instance_key__
                     -> Shape -- ^ __shape__
                     -> Tensor v'1 t -- ^ __input__
                     -> m' (Tensor Value t) -- ^ __data__
collectiveGather' :: OpParams
-> Int64
-> Int64
-> Int64
-> Shape
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveGather' op'options :: OpParams
op'options group_key :: Int64
group_key group_size :: Int64
group_size instance_key :: Int64
instance_key shape :: Shape
shape
                  input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CollectiveGather"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_size
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "instance_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
instance_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "data" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_HALF
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "group_size" type: "int" }
attr { name: "group_key" type: "int" }
attr { name: "instance_key" type: "int" }
attr { name: "shape" type: "shape" }
attr {
  name: "communication_hint"
  type: "string"
  default_value { s: "auto" }
}
attr {
  name: "timeout_seconds" type: "float" default_value { f: 0.0 }
}
-}
-- | 
collectivePermute :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t) =>
                     Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __source_target_pairs__
                     -> Tensor Build t -- ^ __output__
collectivePermute :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
collectivePermute = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
collectivePermute' OpParams
forall a. a -> a
id
collectivePermute' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __input__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __source_target_pairs__
                      -> Tensor Build t -- ^ __output__
collectivePermute' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
collectivePermute' op'options :: OpParams
op'options input :: Tensor v'1 t
input source_target_pairs :: Tensor v'2 Int32
source_target_pairs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
source_target_pairs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CollectivePermute"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "source_target_pairs" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
collectiveReduce :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                    ByteString -- ^ __final_op__
                    -> Data.Int.Int64 -- ^ __group_key__
                    -> Data.Int.Int64 -- ^ __group_size__
                    -> Data.Int.Int64 -- ^ __instance_key__
                    -> ByteString -- ^ __merge_op__
                    -> Tensor v'1 t -- ^ __input__
                    -> m' (Tensor Value t) -- ^ __data__
collectiveReduce :: ByteString
-> Int64
-> Int64
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveReduce = OpParams
-> ByteString
-> Int64
-> Int64
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> ByteString
-> Int64
-> Int64
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveReduce' OpParams
forall a. a -> a
id
collectiveReduce' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                     OpParams ->
                     ByteString -- ^ __final_op__
                     -> Data.Int.Int64 -- ^ __group_key__
                     -> Data.Int.Int64 -- ^ __group_size__
                     -> Data.Int.Int64 -- ^ __instance_key__
                     -> ByteString -- ^ __merge_op__
                     -> Tensor v'1 t -- ^ __input__
                     -> m' (Tensor Value t) -- ^ __data__
collectiveReduce' :: OpParams
-> ByteString
-> Int64
-> Int64
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
collectiveReduce' op'options :: OpParams
op'options final_op :: ByteString
final_op group_key :: Int64
group_key group_size :: Int64
group_size instance_key :: Int64
instance_key merge_op :: ByteString
merge_op
                  input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CollectiveReduce"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "final_op" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
final_op
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "group_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
group_size
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "instance_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
instance_key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "merge_op" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
merge_op
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "data" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_HALF
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "group_size" type: "int" }
attr { name: "group_key" type: "int" }
attr { name: "instance_key" type: "int" }
attr {
  name: "merge_op"
  type: "string"
  allowed_values { list { s: "Min" s: "Max" s: "Mul" s: "Add" } }
}
attr {
  name: "final_op"
  type: "string"
  allowed_values { list { s: "Id" s: "Div" } }
}
attr { name: "subdiv_offsets" type: "list(int)" }
attr {
  name: "wait_for" type: "list(int)" default_value { list { } }
}
attr {
  name: "communication_hint"
  type: "string"
  default_value { s: "auto" }
}
attr {
  name: "timeout_seconds" type: "float" default_value { f: 0.0 }
}
-}
-- | 
combinedNonMaxSuppression :: Tensor v'1 Float -- ^ __boxes__
                             -> Tensor v'2 Float -- ^ __scores__
                             -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size_per_class__
                             -> Tensor v'4 Data.Int.Int32 -- ^ __max_total_size__
                             -> Tensor v'5 Float -- ^ __iou_threshold__
                             -> Tensor v'6 Float -- ^ __score_threshold__
                             -> (Tensor Build Float, Tensor Build Float,
                                 Tensor Build Float,
                                 Tensor Build Data.Int.Int32)
                             -- ^ (__nmsed_boxes__, __nmsed_scores__, __nmsed_classes__, __valid_detections__)
                             --
                             -- * __nmsed_boxes__
                             --
                             -- * __nmsed_scores__
                             --
                             -- * __nmsed_classes__
                             --
                             -- * __valid_detections__
combinedNonMaxSuppression :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
combinedNonMaxSuppression = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
combinedNonMaxSuppression' OpParams
forall a. a -> a
id
combinedNonMaxSuppression' :: OpParams ->
                              Tensor v'1 Float -- ^ __boxes__
                              -> Tensor v'2 Float -- ^ __scores__
                              -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size_per_class__
                              -> Tensor v'4 Data.Int.Int32 -- ^ __max_total_size__
                              -> Tensor v'5 Float -- ^ __iou_threshold__
                              -> Tensor v'6 Float -- ^ __score_threshold__
                              -> (Tensor Build Float, Tensor Build Float,
                                  Tensor Build Float,
                                  Tensor Build Data.Int.Int32)
                              -- ^ (__nmsed_boxes__, __nmsed_scores__, __nmsed_classes__, __valid_detections__)
                              --
                              -- * __nmsed_boxes__
                              --
                              -- * __nmsed_scores__
                              --
                              -- * __nmsed_classes__
                              --
                              -- * __valid_detections__
combinedNonMaxSuppression' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
combinedNonMaxSuppression' op'options :: OpParams
op'options boxes :: Tensor v'1 Float
boxes scores :: Tensor v'2 Float
scores max_output_size_per_class :: Tensor v'3 Int32
max_output_size_per_class
                           max_total_size :: Tensor v'4 Int32
max_total_size iou_threshold :: Tensor v'5 Float
iou_threshold
                           score_threshold :: Tensor v'6 Float
score_threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
     Tensor Build Int32))
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float,
    Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
boxes,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size_per_class,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
max_total_size,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
iou_threshold,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
score_threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CombinedNonMaxSuppression"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "scores" type: DT_FLOAT }
input_arg { name: "max_output_size_per_class" type: DT_INT32 }
input_arg { name: "max_total_size" type: DT_INT32 }
input_arg { name: "iou_threshold" type: DT_FLOAT }
input_arg { name: "score_threshold" type: DT_FLOAT }
output_arg { name: "nmsed_boxes" type: DT_FLOAT }
output_arg { name: "nmsed_scores" type: DT_FLOAT }
output_arg { name: "nmsed_classes" type: DT_FLOAT }
output_arg { name: "valid_detections" type: DT_INT32 }
attr {
  name: "pad_per_class" type: "bool" default_value { b: false }
}
attr { name: "clip_boxes" type: "bool" default_value { b: true } }
-}
-- | 
compareAndBitpack :: forall v'1 v'2 t . (OneOf '[Bool, Data.Int.Int16,
                                                 Data.Int.Int32, Data.Int.Int64,
                                                 Data.Int.Int8,
                                                 Data.Word.Word16, Double,
                                                 Float] t) =>
                     Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 t -- ^ __threshold__
                     -> Tensor Build Data.Word.Word8 -- ^ __output__
compareAndBitpack :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
compareAndBitpack = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
compareAndBitpack' OpParams
forall a. a -> a
id
compareAndBitpack' :: forall v'1 v'2 t . (OneOf '[Bool, Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16, Double,
                                                  Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __input__
                      -> Tensor v'2 t -- ^ __threshold__
                      -> Tensor Build Data.Word.Word8 -- ^ __output__
compareAndBitpack' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
compareAndBitpack' op'options :: OpParams
op'options input :: Tensor v'1 t
input threshold :: Tensor v'2 t
threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CompareAndBitpack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "threshold" type_attr: "T" }
output_arg { name: "output" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BOOL
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
complex :: forall v'1 v'2 t tout . (OneOf '[Double, Float] t,
                                    OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float)] tout) =>
           Tensor v'1 t -- ^ __real__
           -> Tensor v'2 t -- ^ __imag__
           -> Tensor Build tout -- ^ __out__
complex :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
complex = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
forall (v'1 :: * -> *) (v'2 :: * -> *) t tout.
(OneOf '[Double, Float] t,
 OneOf '[Complex Double, Complex Float] tout) =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
complex' OpParams
forall a. a -> a
id
complex' :: forall v'1 v'2 t tout . (OneOf '[Double, Float] t,
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float)] tout) =>
            OpParams ->
            Tensor v'1 t -- ^ __real__
            -> Tensor v'2 t -- ^ __imag__
            -> Tensor Build tout -- ^ __out__
complex' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
complex' op'options :: OpParams
op'options real :: Tensor v'1 t
real imag :: Tensor v'2 t
imag | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tout)
-> Build OpDef -> Tensor Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
real,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
imag]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Complex"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "real" type_attr: "T" }
input_arg { name: "imag" type_attr: "T" }
output_arg { name: "out" type_attr: "Tout" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tout"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
complexAbs :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float)] t,
                                   OneOf '[Double, Float] tout) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor Build tout -- ^ __y__
complexAbs :: Tensor v'1 t -> Tensor Build tout
complexAbs = OpParams -> Tensor v'1 t -> Tensor Build tout
forall (v'1 :: * -> *) t tout.
(OneOf '[Complex Double, Complex Float] t,
 OneOf '[Double, Float] tout) =>
OpParams -> Tensor v'1 t -> Tensor Build tout
complexAbs' OpParams
forall a. a -> a
id
complexAbs' :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float)] t,
                                    OneOf '[Double, Float] tout) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor Build tout -- ^ __y__
complexAbs' :: OpParams -> Tensor v'1 t -> Tensor Build tout
complexAbs' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tout)
-> Build OpDef -> Tensor Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ComplexAbs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "Tout" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
attr {
  name: "Tout"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
compressElement :: forall v'1 input_types . (TensorTypes input_types) =>
                   TensorList (v'1) input_types -- ^ __components__
                   -> Tensor Build Variant -- ^ __compressed__
compressElement :: TensorList v'1 input_types -> Tensor Build Variant
compressElement = OpParams -> TensorList v'1 input_types -> Tensor Build Variant
forall (v'1 :: * -> *) (input_types :: [*]).
TensorTypes input_types =>
OpParams -> TensorList v'1 input_types -> Tensor Build Variant
compressElement' OpParams
forall a. a -> a
id
compressElement' :: forall v'1 input_types . (TensorTypes input_types) =>
                    OpParams ->
                    TensorList (v'1) input_types -- ^ __components__
                    -> Tensor Build Variant -- ^ __compressed__
compressElement' :: OpParams -> TensorList v'1 input_types -> Tensor Build Variant
compressElement' op'options :: OpParams
op'options components :: TensorList v'1 input_types
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 input_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 input_types
components]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CompressElement"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy input_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy input_types
forall k (t :: k). Proxy t
Proxy :: Proxy input_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "components" type_list_attr: "input_types" }
output_arg { name: "compressed" type: DT_VARIANT }
attr {
  name: "input_types" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | 
computeAccidentalHits :: Data.Int.Int64 -- ^ __num_true__
                         -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __sampled_candidates__
                         -> (Tensor Build Data.Int.Int32,
                             Tensor Build Data.Int.Int64, Tensor Build Float)
                         -- ^ (__indices__, __ids__, __weights__)
                         --
                         -- * __indices__
                         --
                         -- * __ids__
                         --
                         -- * __weights__
computeAccidentalHits :: Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
computeAccidentalHits = OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
computeAccidentalHits' OpParams
forall a. a -> a
id
computeAccidentalHits' :: OpParams ->
                          Data.Int.Int64 -- ^ __num_true__
                          -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __sampled_candidates__
                          -> (Tensor Build Data.Int.Int32,
                              Tensor Build Data.Int.Int64, Tensor Build Float)
                          -- ^ (__indices__, __ids__, __weights__)
                          --
                          -- * __indices__
                          --
                          -- * __ids__
                          --
                          -- * __weights__
computeAccidentalHits' :: OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
computeAccidentalHits' op'options :: OpParams
op'options num_true :: Int64
num_true true_classes :: Tensor v'1 Int64
true_classes
                       sampled_candidates :: Tensor v'2 Int64
sampled_candidates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build Int64, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
sampled_candidates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ComputeAccidentalHits"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
input_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "indices" type: DT_INT32 }
output_arg { name: "ids" type: DT_INT64 }
output_arg { name: "weights" type: DT_FLOAT }
attr { name: "num_true" type: "int" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
concat :: forall v'1 v'2 t . (TensorType t) =>
          Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
          -> [Tensor v'2 t] -- ^ __values__
          -> Tensor Build t -- ^ __output__
concat :: Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
concat = OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
concat' OpParams
forall a. a -> a
id
concat' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
           Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
           -> [Tensor v'2 t] -- ^ __values__
           -> Tensor Build t -- ^ __output__
concat' :: OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
concat' op'options :: OpParams
op'options concat_dim :: Tensor v'1 Int32
concat_dim
        values :: [Tensor v'2 t]
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("values", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
values)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
concat_dim,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Concat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
values) :: Int64
{-
input_arg { name: "concat_dim" type: DT_INT32 }
input_arg { name: "values" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "T" type: "type" }
-}
-- | 
concatOffset :: Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
                -> [Tensor v'2 Data.Int.Int32] -- ^ __shape__
                -> [Tensor Build Data.Int.Int32] -- ^ __offset__
concatOffset :: Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
concatOffset = OpParams
-> Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
concatOffset' OpParams
forall a. a -> a
id
concatOffset' :: OpParams ->
                 Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
                 -> [Tensor v'2 Data.Int.Int32] -- ^ __shape__
                 -> [Tensor Build Data.Int.Int32] -- ^ __offset__
concatOffset' :: OpParams
-> Tensor v'1 Int32 -> [Tensor v'2 Int32] -> [Tensor Build Int32]
concatOffset' op'options :: OpParams
op'options concat_dim :: Tensor v'1 Int32
concat_dim
              shape :: [Tensor v'2 Int32]
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("shape", [Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
shape)])] =
    [Int64] -> Build OpDef -> [Tensor Build Int32]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
n] (Build OpDef -> [Tensor Build Int32])
-> Build OpDef -> [Tensor Build Int32]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
concat_dim,
                                                             [Tensor v'2 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int32]
shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ConcatOffset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int32]
shape) :: Int64
{-
input_arg { name: "concat_dim" type: DT_INT32 }
input_arg { name: "shape" type: DT_INT32 number_attr: "N" }
output_arg { name: "offset" type: DT_INT32 number_attr: "N" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
-}
-- | 
concatV2 :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tidx) =>
            [Tensor v'1 t] -- ^ __values__
            -> Tensor v'2 tidx -- ^ __axis__
            -> Tensor Build t -- ^ __output__
concatV2 :: [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
concatV2 = OpParams -> [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
concatV2' OpParams
forall a. a -> a
id
concatV2' :: forall v'1 v'2 t tidx . (TensorType t, OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tidx) =>
             OpParams ->
             [Tensor v'1 t] -- ^ __values__
             -> Tensor v'2 tidx -- ^ __axis__
             -> Tensor Build t -- ^ __output__
concatV2' :: OpParams -> [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
concatV2' op'options :: OpParams
op'options values :: [Tensor v'1 t]
values
          axis :: Tensor v'2 tidx
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("values", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
values,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ConcatV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values) :: Int64
{-
input_arg { name: "values" type_attr: "T" number_attr: "N" }
input_arg { name: "axis" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "T" type: "type" }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
concatenateDataset :: [DataType] -- ^ __output_types__
                      -> Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor v'2 Variant -- ^ __another_dataset__
                      -> Tensor Build Variant -- ^ __handle__
concatenateDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
concatenateDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
concatenateDataset' OpParams
forall a. a -> a
id
concatenateDataset' :: OpParams ->
                       [DataType] -- ^ __output_types__
                       -> Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor v'2 Variant -- ^ __another_dataset__
                       -> Tensor Build Variant -- ^ __handle__
concatenateDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
concatenateDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                    another_dataset :: Tensor v'2 Variant
another_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
another_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ConcatenateDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "another_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
conditionalAccumulator :: forall m' . (MonadBuild m') => DataType -- ^ __dtype__
                          -> Shape -- ^ __shape__
                          -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
conditionalAccumulator :: DataType -> Shape -> m' (Tensor Ref ByteString)
conditionalAccumulator = OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
conditionalAccumulator' OpParams
forall a. a -> a
id
conditionalAccumulator' :: forall m' . (MonadBuild m') => OpParams ->
                           DataType -- ^ __dtype__
                           -> Shape -- ^ __shape__
                           -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
conditionalAccumulator' :: OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
conditionalAccumulator' op'options :: OpParams
op'options dtype :: DataType
dtype shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ConditionalAccumulator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "shape" type: "shape" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "reduction_type"
  type: "string"
  default_value { s: "MEAN" }
  allowed_values { list { s: "MEAN" s: "SUM" } }
}
-}
-- | 
configureDistributedTPU :: forall m' . (MonadBuild m') =>
                           m' (Tensor Value Data.ByteString.ByteString) -- ^ __topology__
configureDistributedTPU :: m' (Tensor Value ByteString)
configureDistributedTPU = OpParams -> m' (Tensor Value ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ByteString)
configureDistributedTPU' OpParams
forall a. a -> a
id
configureDistributedTPU' :: forall m' . (MonadBuild m') => OpParams ->
                            m' (Tensor Value Data.ByteString.ByteString) -- ^ __topology__
configureDistributedTPU' :: OpParams -> m' (Tensor Value ByteString)
configureDistributedTPU' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ConfigureDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "topology" type: DT_STRING }
attr {
  name: "embedding_config" type: "string" default_value { s: "" }
}
attr {
  name: "tpu_embedding_config" type: "string" default_value { s: "" }
}
attr {
  name: "is_global_init" type: "bool" default_value { b: false }
}
attr {
  name: "enable_whole_mesh_compilations"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "compilation_failure_closes_chips"
  type: "bool"
  default_value { b: true }
}
-}
-- | 
configureTPUEmbedding :: forall m' . (MonadBuild m') =>
                         ByteString -- ^ __config__
                         -> m' (ControlNode)
configureTPUEmbedding :: ByteString -> m' ControlNode
configureTPUEmbedding = OpParams -> ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> ByteString -> m' ControlNode
configureTPUEmbedding' OpParams
forall a. a -> a
id
configureTPUEmbedding' :: forall m' . (MonadBuild m') => OpParams ->
                          ByteString -- ^ __config__
                          -> m' (ControlNode)
configureTPUEmbedding' :: OpParams -> ByteString -> m' ControlNode
configureTPUEmbedding' op'options :: OpParams
op'options config :: ByteString
config | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ConfigureTPUEmbedding"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
config
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr { name: "config" type: "string" }
-}
-- | 
conj :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Variant] t) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor Build t -- ^ __output__
conj :: Tensor v'1 t -> Tensor Build t
conj = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Variant] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
conj' OpParams
forall a. a -> a
id
conj' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Variant] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build t -- ^ __output__
conj' :: OpParams -> Tensor v'1 t -> Tensor Build t
conj' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conj"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values {
    list { type: DT_COMPLEX64 type: DT_COMPLEX128 type: DT_VARIANT }
  }
}
-}
-- | 
conjugateTranspose :: forall v'1 v'2 t tperm . (TensorType t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tperm) =>
                      Tensor v'1 t -- ^ __x__
                      -> Tensor v'2 tperm -- ^ __perm__
                      -> Tensor Build t -- ^ __y__
conjugateTranspose :: Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
conjugateTranspose = OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
conjugateTranspose' OpParams
forall a. a -> a
id
conjugateTranspose' :: forall v'1 v'2 t tperm . (TensorType t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tperm) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __x__
                       -> Tensor v'2 tperm -- ^ __perm__
                       -> Tensor Build t -- ^ __y__
conjugateTranspose' :: OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
conjugateTranspose' op'options :: OpParams
op'options x :: Tensor v'1 t
x perm :: Tensor v'2 tperm
perm | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 tperm -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tperm
perm]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ConjugateTranspose"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tperm" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tperm -> DataType
forall a. TensorType a => a -> DataType
tensorType (tperm
forall a. HasCallStack => a
undefined :: tperm)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "perm" type_attr: "Tperm" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tperm"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
const :: forall dtype . (TensorType dtype) => Tensor Build dtype -- ^ __output__
const :: Tensor Build dtype
const = OpParams -> Tensor Build dtype
forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
const' OpParams
forall a. a -> a
id
const' :: forall dtype . (TensorType dtype) => OpParams ->
          Tensor Build dtype -- ^ __output__
const' :: OpParams -> Tensor Build dtype
const' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Const"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "value" type: "tensor" }
attr { name: "dtype" type: "type" }
-}
-- | 
consumeMutexLock :: forall v'1 m' . (MonadBuild m') =>
                    Tensor v'1 Variant -- ^ __mutex_lock__
                    -> m' (ControlNode)
consumeMutexLock :: Tensor v'1 Variant -> m' ControlNode
consumeMutexLock = OpParams -> Tensor v'1 Variant -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 Variant -> m' ControlNode
consumeMutexLock' OpParams
forall a. a -> a
id
consumeMutexLock' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                     Tensor v'1 Variant -- ^ __mutex_lock__
                     -> m' (ControlNode)
consumeMutexLock' :: OpParams -> Tensor v'1 Variant -> m' ControlNode
consumeMutexLock' op'options :: OpParams
op'options mutex_lock :: Tensor v'1 Variant
mutex_lock | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
mutex_lock]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ConsumeMutexLock"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "mutex_lock" type: DT_VARIANT }
-}
-- | 
controlTrigger :: forall m' . (MonadBuild m') => m' (ControlNode)
controlTrigger :: m' ControlNode
controlTrigger = OpParams -> m' ControlNode
forall (m' :: * -> *). MonadBuild m' => OpParams -> m' ControlNode
controlTrigger' OpParams
forall a. a -> a
id
controlTrigger' :: forall m' . (MonadBuild m') => OpParams ->
                   m' (ControlNode)
controlTrigger' :: OpParams -> m' ControlNode
controlTrigger' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ControlTrigger"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-

-}
-- | 
conv2D :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Word.Word16, Double,
                                      Float] t) => ByteString -- ^ __padding__
          -> Tensor v'1 t -- ^ __input__
          -> Tensor v'2 t -- ^ __filter__
          -> Tensor Build t -- ^ __output__
conv2D :: ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv2D = OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Word16, Double, Float] t =>
OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv2D' OpParams
forall a. a -> a
id
conv2D' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Word.Word16, Double,
                                       Float] t) => OpParams ->
           ByteString -- ^ __padding__
           -> Tensor v'1 t -- ^ __input__
           -> Tensor v'2 t -- ^ __filter__
           -> Tensor Build t -- ^ __output__
conv2D' :: OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv2D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "use_cudnn_on_gpu" type: "bool" default_value { b: true }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv2DBackpropFilter :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                        Double, Float] t) =>
                        ByteString -- ^ __padding__
                        -> Tensor v'1 t -- ^ __input__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                        -> Tensor v'3 t -- ^ __out_backprop__
                        -> Tensor Build t -- ^ __output__
conv2DBackpropFilter :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropFilter = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropFilter' OpParams
forall a. a -> a
id
conv2DBackpropFilter' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                         Double, Float] t) =>
                         OpParams ->
                         ByteString -- ^ __padding__
                         -> Tensor v'1 t -- ^ __input__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                         -> Tensor v'3 t -- ^ __out_backprop__
                         -> Tensor Build t -- ^ __output__
conv2DBackpropFilter' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropFilter' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter_sizes :: Tensor v'2 Int32
filter_sizes
                      out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
filter_sizes,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv2DBackpropFilter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter_sizes" type: DT_INT32 }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "use_cudnn_on_gpu" type: "bool" default_value { b: true }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv2DBackpropInput :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32,
                                                       Data.Word.Word16, Double,
                                                       Float] t) =>
                       ByteString -- ^ __padding__
                       -> Tensor v'1 Data.Int.Int32 -- ^ __input_sizes__
                       -> Tensor v'2 t -- ^ __filter__
                       -> Tensor v'3 t -- ^ __out_backprop__
                       -> Tensor Build t -- ^ __output__
conv2DBackpropInput :: ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropInput = OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int32, Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropInput' OpParams
forall a. a -> a
id
conv2DBackpropInput' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32,
                                                        Data.Word.Word16,
                                                        Double, Float] t) =>
                        OpParams ->
                        ByteString -- ^ __padding__
                        -> Tensor v'1 Data.Int.Int32 -- ^ __input_sizes__
                        -> Tensor v'2 t -- ^ __filter__
                        -> Tensor v'3 t -- ^ __out_backprop__
                        -> Tensor Build t -- ^ __output__
conv2DBackpropInput' :: OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv2DBackpropInput' op'options :: OpParams
op'options padding :: ByteString
padding input_sizes :: Tensor v'1 Int32
input_sizes filter :: Tensor v'2 t
filter
                     out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
input_sizes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv2DBackpropInput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_sizes" type: DT_INT32 }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "use_cudnn_on_gpu" type: "bool" default_value { b: true }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv3D :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          ByteString -- ^ __padding__
          -> Tensor v'1 t -- ^ __input__
          -> Tensor v'2 t -- ^ __filter__
          -> Tensor Build t -- ^ __output__
conv3D :: ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv3D = OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv3D' OpParams
forall a. a -> a
id
conv3D' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           ByteString -- ^ __padding__
           -> Tensor v'1 t -- ^ __input__
           -> Tensor v'2 t -- ^ __filter__
           -> Tensor Build t -- ^ __output__
conv3D' :: OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
conv3D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv3DBackpropFilter :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                        Double, Float] t) =>
                        ByteString -- ^ __padding__
                        -> Tensor v'1 t -- ^ __input__
                        -> Tensor v'2 t -- ^ __filter__
                        -> Tensor v'3 t -- ^ __out_backprop__
                        -> Tensor Build t -- ^ __output__
conv3DBackpropFilter :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
conv3DBackpropFilter = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropFilter' OpParams
forall a. a -> a
id
conv3DBackpropFilter' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                         Double, Float] t) =>
                         OpParams ->
                         ByteString -- ^ __padding__
                         -> Tensor v'1 t -- ^ __input__
                         -> Tensor v'2 t -- ^ __filter__
                         -> Tensor v'3 t -- ^ __out_backprop__
                         -> Tensor Build t -- ^ __output__
conv3DBackpropFilter' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropFilter' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
                      out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv3DBackpropFilter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv3DBackpropFilterV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                          Double, Float] t) =>
                          ByteString -- ^ __padding__
                          -> Tensor v'1 t -- ^ __input__
                          -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                          -> Tensor v'3 t -- ^ __out_backprop__
                          -> Tensor Build t -- ^ __output__
conv3DBackpropFilterV2 :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropFilterV2 = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropFilterV2' OpParams
forall a. a -> a
id
conv3DBackpropFilterV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                           Double, Float] t) =>
                           OpParams ->
                           ByteString -- ^ __padding__
                           -> Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                           -> Tensor v'3 t -- ^ __out_backprop__
                           -> Tensor Build t -- ^ __output__
conv3DBackpropFilterV2' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropFilterV2' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter_sizes :: Tensor v'2 Int32
filter_sizes
                        out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
filter_sizes,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv3DBackpropFilterV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter_sizes" type: DT_INT32 }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv3DBackpropInput :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16, Double,
                                                       Float] t) =>
                       ByteString -- ^ __padding__
                       -> Tensor v'1 t -- ^ __input__
                       -> Tensor v'2 t -- ^ __filter__
                       -> Tensor v'3 t -- ^ __out_backprop__
                       -> Tensor Build t -- ^ __output__
conv3DBackpropInput :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
conv3DBackpropInput = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropInput' OpParams
forall a. a -> a
id
conv3DBackpropInput' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                        Double, Float] t) =>
                        OpParams ->
                        ByteString -- ^ __padding__
                        -> Tensor v'1 t -- ^ __input__
                        -> Tensor v'2 t -- ^ __filter__
                        -> Tensor v'3 t -- ^ __out_backprop__
                        -> Tensor Build t -- ^ __output__
conv3DBackpropInput' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropInput' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
                     out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv3DBackpropInput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
conv3DBackpropInputV2 :: forall v'1 v'2 v'3 t
                         tshape . (OneOf '[Data.Word.Word16, Double, Float] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tshape) =>
                         ByteString -- ^ __padding__
                         -> Tensor v'1 tshape -- ^ __input_sizes__
                         -> Tensor v'2 t -- ^ __filter__
                         -> Tensor v'3 t -- ^ __out_backprop__
                         -> Tensor Build t -- ^ __output__
conv3DBackpropInputV2 :: ByteString
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropInputV2 = OpParams
-> ByteString
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tshape.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape) =>
OpParams
-> ByteString
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropInputV2' OpParams
forall a. a -> a
id
conv3DBackpropInputV2' :: forall v'1 v'2 v'3 t
                          tshape . (OneOf '[Data.Word.Word16, Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tshape) =>
                          OpParams ->
                          ByteString -- ^ __padding__
                          -> Tensor v'1 tshape -- ^ __input_sizes__
                          -> Tensor v'2 t -- ^ __filter__
                          -> Tensor v'3 t -- ^ __out_backprop__
                          -> Tensor Build t -- ^ __output__
conv3DBackpropInputV2' :: OpParams
-> ByteString
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
conv3DBackpropInputV2' op'options :: OpParams
op'options padding :: ByteString
padding input_sizes :: Tensor v'1 tshape
input_sizes filter :: Tensor v'2 t
filter
                       out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tshape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tshape
input_sizes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Conv3DBackpropInputV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tshape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tshape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tshape
forall a. HasCallStack => a
undefined :: tshape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_sizes" type_attr: "Tshape" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "Tshape"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
copy :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
        -> Tensor Build t -- ^ __output__
copy :: Tensor v'1 t -> Tensor Build t
copy = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
copy' OpParams
forall a. a -> a
id
copy' :: forall v'1 t . (TensorType t) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build t -- ^ __output__
copy' :: OpParams -> Tensor v'1 t -> Tensor Build t
copy' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Copy"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "tensor_name" type: "string" default_value { s: "" } }
attr {
  name: "debug_ops_spec"
  type: "list(string)"
  default_value { list { } }
}
-}
-- | 
copyHost :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
copyHost :: Tensor v'1 t -> Tensor Build t
copyHost = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
copyHost' OpParams
forall a. a -> a
id
copyHost' :: forall v'1 t . (TensorType t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
copyHost' :: OpParams -> Tensor v'1 t -> Tensor Build t
copyHost' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CopyHost"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "tensor_name" type: "string" default_value { s: "" } }
attr {
  name: "debug_ops_spec"
  type: "list(string)"
  default_value { list { } }
}
-}
-- | 
cos :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
cos :: Tensor v'1 t -> Tensor Build t
cos = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
cos' OpParams
forall a. a -> a
id
cos' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
cos' :: OpParams -> Tensor v'1 t -> Tensor Build t
cos' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cos"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
cosh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
cosh :: Tensor v'1 t -> Tensor Build t
cosh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
cosh' OpParams
forall a. a -> a
id
cosh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
cosh' :: OpParams -> Tensor v'1 t -> Tensor Build t
cosh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cosh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
countUpTo :: forall t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] t) =>
             Data.Int.Int64 -- ^ __limit__
             -> Tensor Ref t -- ^ __ref__
             -> m' (Tensor Value t) -- ^ __output__
countUpTo :: Int64 -> Tensor Ref t -> m' (Tensor Value t)
countUpTo = OpParams -> Int64 -> Tensor Ref t -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] t) =>
OpParams -> Int64 -> Tensor Ref t -> m' (Tensor Value t)
countUpTo' OpParams
forall a. a -> a
id
countUpTo' :: forall t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] t) =>
              OpParams ->
              Data.Int.Int64 -- ^ __limit__
              -> Tensor Ref t -- ^ __ref__
              -> m' (Tensor Value t) -- ^ __output__
countUpTo' :: OpParams -> Int64 -> Tensor Ref t -> m' (Tensor Value t)
countUpTo' op'options :: OpParams
op'options limit :: Int64
limit ref :: Tensor Ref t
ref | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CountUpTo"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "limit" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
limit
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" }
attr { name: "limit" type: "int" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
createSummaryDbWriter :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                         Tensor v'1 ResourceHandle -- ^ __writer__
                         -> Tensor v'2 Data.ByteString.ByteString -- ^ __db_uri__
                         -> Tensor v'3 Data.ByteString.ByteString -- ^ __experiment_name__
                         -> Tensor v'4 Data.ByteString.ByteString -- ^ __run_name__
                         -> Tensor v'5 Data.ByteString.ByteString -- ^ __user_name__
                         -> m' (ControlNode)
createSummaryDbWriter :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryDbWriter = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryDbWriter' OpParams
forall a. a -> a
id
createSummaryDbWriter' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                          OpParams ->
                          Tensor v'1 ResourceHandle -- ^ __writer__
                          -> Tensor v'2 Data.ByteString.ByteString -- ^ __db_uri__
                          -> Tensor v'3 Data.ByteString.ByteString -- ^ __experiment_name__
                          -> Tensor v'4 Data.ByteString.ByteString -- ^ __run_name__
                          -> Tensor v'5 Data.ByteString.ByteString -- ^ __user_name__
                          -> m' (ControlNode)
createSummaryDbWriter' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryDbWriter' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer db_uri :: Tensor v'2 ByteString
db_uri experiment_name :: Tensor v'3 ByteString
experiment_name run_name :: Tensor v'4 ByteString
run_name
                       user_name :: Tensor v'5 ByteString
user_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
db_uri,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
experiment_name,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
run_name,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
user_name]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CreateSummaryDbWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "db_uri" type: DT_STRING }
input_arg { name: "experiment_name" type: DT_STRING }
input_arg { name: "run_name" type: DT_STRING }
input_arg { name: "user_name" type: DT_STRING }
-}
-- | 
createSummaryFileWriter :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                           Tensor v'1 ResourceHandle -- ^ __writer__
                           -> Tensor v'2 Data.ByteString.ByteString -- ^ __logdir__
                           -> Tensor v'3 Data.Int.Int32 -- ^ __max_queue__
                           -> Tensor v'4 Data.Int.Int32 -- ^ __flush_millis__
                           -> Tensor v'5 Data.ByteString.ByteString -- ^ __filename_suffix__
                           -> m' (ControlNode)
createSummaryFileWriter :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryFileWriter = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryFileWriter' OpParams
forall a. a -> a
id
createSummaryFileWriter' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                            OpParams ->
                            Tensor v'1 ResourceHandle -- ^ __writer__
                            -> Tensor v'2 Data.ByteString.ByteString -- ^ __logdir__
                            -> Tensor v'3 Data.Int.Int32 -- ^ __max_queue__
                            -> Tensor v'4 Data.Int.Int32 -- ^ __flush_millis__
                            -> Tensor v'5 Data.ByteString.ByteString -- ^ __filename_suffix__
                            -> m' (ControlNode)
createSummaryFileWriter' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 ByteString
-> m' ControlNode
createSummaryFileWriter' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer logdir :: Tensor v'2 ByteString
logdir max_queue :: Tensor v'3 Int32
max_queue flush_millis :: Tensor v'4 Int32
flush_millis
                         filename_suffix :: Tensor v'5 ByteString
filename_suffix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
logdir,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_queue,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
flush_millis,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
filename_suffix]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CreateSummaryFileWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "logdir" type: DT_STRING }
input_arg { name: "max_queue" type: DT_INT32 }
input_arg { name: "flush_millis" type: DT_INT32 }
input_arg { name: "filename_suffix" type: DT_STRING }
-}
-- | 
cropAndResize :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                 Tensor v'1 t -- ^ __image__
                 -> Tensor v'2 Float -- ^ __boxes__
                 -> Tensor v'3 Data.Int.Int32 -- ^ __box_ind__
                 -> Tensor v'4 Data.Int.Int32 -- ^ __crop_size__
                 -> Tensor Build Float -- ^ __crops__
cropAndResize :: Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResize = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResize' OpParams
forall a. a -> a
id
cropAndResize' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __image__
                  -> Tensor v'2 Float -- ^ __boxes__
                  -> Tensor v'3 Data.Int.Int32 -- ^ __box_ind__
                  -> Tensor v'4 Data.Int.Int32 -- ^ __crop_size__
                  -> Tensor Build Float -- ^ __crops__
cropAndResize' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResize' op'options :: OpParams
op'options image :: Tensor v'1 t
image boxes :: Tensor v'2 Float
boxes box_ind :: Tensor v'3 Int32
box_ind crop_size :: Tensor v'4 Int32
crop_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
image,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
boxes,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
box_ind,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
crop_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CropAndResize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image" type_attr: "T" }
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "box_ind" type: DT_INT32 }
input_arg { name: "crop_size" type: DT_INT32 }
output_arg { name: "crops" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_UINT16
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr {
  name: "method"
  type: "string"
  default_value { s: "bilinear" }
  allowed_values { list { s: "bilinear" s: "nearest" } }
}
attr {
  name: "extrapolation_value" type: "float" default_value { f: 0.0 }
}
-}
-- | 
cropAndResizeGradBoxes :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                          Tensor v'1 Float -- ^ __grads__
                          -> Tensor v'2 t -- ^ __image__
                          -> Tensor v'3 Float -- ^ __boxes__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __box_ind__
                          -> Tensor Build Float -- ^ __output__
cropAndResizeGradBoxes :: Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResizeGradBoxes = OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResizeGradBoxes' OpParams
forall a. a -> a
id
cropAndResizeGradBoxes' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] t) =>
                           OpParams ->
                           Tensor v'1 Float -- ^ __grads__
                           -> Tensor v'2 t -- ^ __image__
                           -> Tensor v'3 Float -- ^ __boxes__
                           -> Tensor v'4 Data.Int.Int32 -- ^ __box_ind__
                           -> Tensor Build Float -- ^ __output__
cropAndResizeGradBoxes' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Int32
-> Tensor Build Float
cropAndResizeGradBoxes' op'options :: OpParams
op'options grads :: Tensor v'1 Float
grads image :: Tensor v'2 t
image boxes :: Tensor v'3 Float
boxes
                        box_ind :: Tensor v'4 Int32
box_ind | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
grads,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
image,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
boxes,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
box_ind]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CropAndResizeGradBoxes"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type: DT_FLOAT }
input_arg { name: "image" type_attr: "T" }
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "box_ind" type: DT_INT32 }
output_arg { name: "output" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_UINT16
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr {
  name: "method"
  type: "string"
  default_value { s: "bilinear" }
  allowed_values { list { s: "bilinear" } }
}
-}
-- | 
cropAndResizeGradImage :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                          Tensor v'1 Float -- ^ __grads__
                          -> Tensor v'2 Float -- ^ __boxes__
                          -> Tensor v'3 Data.Int.Int32 -- ^ __box_ind__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __image_size__
                          -> Tensor Build t -- ^ __output__
cropAndResizeGradImage :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build t
cropAndResizeGradImage = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build t
cropAndResizeGradImage' OpParams
forall a. a -> a
id
cropAndResizeGradImage' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                           OpParams ->
                           Tensor v'1 Float -- ^ __grads__
                           -> Tensor v'2 Float -- ^ __boxes__
                           -> Tensor v'3 Data.Int.Int32 -- ^ __box_ind__
                           -> Tensor v'4 Data.Int.Int32 -- ^ __image_size__
                           -> Tensor Build t -- ^ __output__
cropAndResizeGradImage' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor Build t
cropAndResizeGradImage' op'options :: OpParams
op'options grads :: Tensor v'1 Float
grads boxes :: Tensor v'2 Float
boxes box_ind :: Tensor v'3 Int32
box_ind
                        image_size :: Tensor v'4 Int32
image_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
grads,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
boxes,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
box_ind,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
image_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CropAndResizeGradImage"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type: DT_FLOAT }
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "box_ind" type: DT_INT32 }
input_arg { name: "image_size" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_FLOAT type: DT_HALF type: DT_DOUBLE }
  }
}
attr {
  name: "method"
  type: "string"
  default_value { s: "bilinear" }
  allowed_values { list { s: "bilinear" s: "nearest" } }
}
-}
-- | 
cross :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => Tensor v'1 t -- ^ __a__
         -> Tensor v'2 t -- ^ __b__
         -> Tensor Build t -- ^ __product__
cross :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
cross = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
cross' OpParams
forall a. a -> a
id
cross' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) => OpParams ->
          Tensor v'1 t -- ^ __a__
          -> Tensor v'2 t -- ^ __b__
          -> Tensor Build t -- ^ __product__
cross' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
cross' op'options :: OpParams
op'options a :: Tensor v'1 t
a b :: Tensor v'2 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cross"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "product" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
crossReplicaSum :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Word.Word16,
                                               Data.Word.Word32, Float] t) =>
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __group_assignment__
                   -> Tensor Build t -- ^ __output__
crossReplicaSum :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
crossReplicaSum = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Word16, Word32, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
crossReplicaSum' OpParams
forall a. a -> a
id
crossReplicaSum' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word32, Float] t) =>
                    OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __group_assignment__
                    -> Tensor Build t -- ^ __output__
crossReplicaSum' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
crossReplicaSum' op'options :: OpParams
op'options input :: Tensor v'1 t
input group_assignment :: Tensor v'2 Int32
group_assignment | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
group_assignment]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CrossReplicaSum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "group_assignment" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_FLOAT type: DT_INT32 type: DT_UINT32
    }
  }
}
-}
-- | 
cudnnRNN :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                           OneOf '[Data.Word.Word16, Double,
                                                   Float] t) =>
            Tensor v'1 t -- ^ __input__
            -> Tensor v'2 t -- ^ __input_h__
            -> Tensor v'3 t -- ^ __input_c__
            -> Tensor v'4 t -- ^ __params__
            -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                    Tensor Value t))
            -- ^ (__output__, __output_h__, __output_c__, __reserve_space__)
            --
            -- * __output__
            --
            -- * __output_h__
            --
            -- * __output_c__
            --
            -- * __reserve_space__
cudnnRNN :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNN = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNN' OpParams
forall a. a -> a
id
cudnnRNN' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                            OneOf '[Data.Word.Word16, Double,
                                                    Float] t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor v'2 t -- ^ __input_h__
             -> Tensor v'3 t -- ^ __input_c__
             -> Tensor v'4 t -- ^ __params__
             -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                     Tensor Value t))
             -- ^ (__output__, __output_h__, __output_c__, __reserve_space__)
             --
             -- * __output__
             --
             -- * __output_h__
             --
             -- * __output_c__
             --
             -- * __reserve_space__
cudnnRNN' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNN' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNN"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_h" type_attr: "T" }
output_arg { name: "output_c" type_attr: "T" }
output_arg { name: "reserve_space" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
cudnnRNNBackprop :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 t
                    m' . (MonadBuild m', OneOf '[Data.Word.Word16, Double,
                                                 Float] t) =>
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 t -- ^ __input_h__
                    -> Tensor v'3 t -- ^ __input_c__
                    -> Tensor v'4 t -- ^ __params__
                    -> Tensor v'5 t -- ^ __output__
                    -> Tensor v'6 t -- ^ __output_h__
                    -> Tensor v'7 t -- ^ __output_c__
                    -> Tensor v'8 t -- ^ __output_backprop__
                    -> Tensor v'9 t -- ^ __output_h_backprop__
                    -> Tensor v'10 t -- ^ __output_c_backprop__
                    -> Tensor v'11 t -- ^ __reserve_space__
                    -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                            Tensor Value t))
                    -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                    --
                    -- * __input_backprop__
                    --
                    -- * __input_h_backprop__
                    --
                    -- * __input_c_backprop__
                    --
                    -- * __params_backprop__
cudnnRNNBackprop :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackprop = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackprop' OpParams
forall a. a -> a
id
cudnnRNNBackprop' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 t
                     m' . (MonadBuild m', OneOf '[Data.Word.Word16, Double,
                                                  Float] t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 t -- ^ __input_h__
                     -> Tensor v'3 t -- ^ __input_c__
                     -> Tensor v'4 t -- ^ __params__
                     -> Tensor v'5 t -- ^ __output__
                     -> Tensor v'6 t -- ^ __output_h__
                     -> Tensor v'7 t -- ^ __output_c__
                     -> Tensor v'8 t -- ^ __output_backprop__
                     -> Tensor v'9 t -- ^ __output_h_backprop__
                     -> Tensor v'10 t -- ^ __output_c_backprop__
                     -> Tensor v'11 t -- ^ __reserve_space__
                     -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                             Tensor Value t))
                     -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                     --
                     -- * __input_backprop__
                     --
                     -- * __input_h_backprop__
                     --
                     -- * __input_c_backprop__
                     --
                     -- * __params_backprop__
cudnnRNNBackprop' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackprop' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params output :: Tensor v'5 t
output output_h :: Tensor v'6 t
output_h
                  output_c :: Tensor v'7 t
output_c output_backprop :: Tensor v'8 t
output_backprop output_h_backprop :: Tensor v'9 t
output_h_backprop output_c_backprop :: Tensor v'10 t
output_c_backprop
                  reserve_space :: Tensor v'11 t
reserve_space | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
output,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
output_h,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
output_c,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
output_backprop,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
output_h_backprop,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
output_c_backprop,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
reserve_space]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNNBackprop"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
input_arg { name: "output" type_attr: "T" }
input_arg { name: "output_h" type_attr: "T" }
input_arg { name: "output_c" type_attr: "T" }
input_arg { name: "output_backprop" type_attr: "T" }
input_arg { name: "output_h_backprop" type_attr: "T" }
input_arg { name: "output_c_backprop" type_attr: "T" }
input_arg { name: "reserve_space" type_attr: "T" }
output_arg { name: "input_backprop" type_attr: "T" }
output_arg { name: "input_h_backprop" type_attr: "T" }
output_arg { name: "input_c_backprop" type_attr: "T" }
output_arg { name: "params_backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNBackpropV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                      t m' . (MonadBuild m', OneOf '[Data.Word.Word16, Double,
                                                     Float] t) =>
                      Tensor v'1 t -- ^ __input__
                      -> Tensor v'2 t -- ^ __input_h__
                      -> Tensor v'3 t -- ^ __input_c__
                      -> Tensor v'4 t -- ^ __params__
                      -> Tensor v'5 t -- ^ __output__
                      -> Tensor v'6 t -- ^ __output_h__
                      -> Tensor v'7 t -- ^ __output_c__
                      -> Tensor v'8 t -- ^ __output_backprop__
                      -> Tensor v'9 t -- ^ __output_h_backprop__
                      -> Tensor v'10 t -- ^ __output_c_backprop__
                      -> Tensor v'11 t -- ^ __reserve_space__
                      -> Tensor v'12 Data.Int.Int8 -- ^ __host_reserved__
                      -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                              Tensor Value t))
                      -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                      --
                      -- * __input_backprop__
                      --
                      -- * __input_h_backprop__
                      --
                      -- * __input_c_backprop__
                      --
                      -- * __params_backprop__
cudnnRNNBackpropV2 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV2' OpParams
forall a. a -> a
id
cudnnRNNBackpropV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                       t m' . (MonadBuild m', OneOf '[Data.Word.Word16, Double,
                                                      Float] t) => OpParams ->
                       Tensor v'1 t -- ^ __input__
                       -> Tensor v'2 t -- ^ __input_h__
                       -> Tensor v'3 t -- ^ __input_c__
                       -> Tensor v'4 t -- ^ __params__
                       -> Tensor v'5 t -- ^ __output__
                       -> Tensor v'6 t -- ^ __output_h__
                       -> Tensor v'7 t -- ^ __output_c__
                       -> Tensor v'8 t -- ^ __output_backprop__
                       -> Tensor v'9 t -- ^ __output_h_backprop__
                       -> Tensor v'10 t -- ^ __output_c_backprop__
                       -> Tensor v'11 t -- ^ __reserve_space__
                       -> Tensor v'12 Data.Int.Int8 -- ^ __host_reserved__
                       -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                               Tensor Value t))
                       -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                       --
                       -- * __input_backprop__
                       --
                       -- * __input_h_backprop__
                       --
                       -- * __input_c_backprop__
                       --
                       -- * __params_backprop__
cudnnRNNBackpropV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params output :: Tensor v'5 t
output output_h :: Tensor v'6 t
output_h
                    output_c :: Tensor v'7 t
output_c output_backprop :: Tensor v'8 t
output_backprop output_h_backprop :: Tensor v'9 t
output_h_backprop output_c_backprop :: Tensor v'10 t
output_c_backprop
                    reserve_space :: Tensor v'11 t
reserve_space host_reserved :: Tensor v'12 Int8
host_reserved | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
output,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
output_h,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
output_c,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
output_backprop,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
output_h_backprop,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
output_c_backprop,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
reserve_space,
                                                             Tensor v'12 Int8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 Int8
host_reserved]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNNBackpropV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
input_arg { name: "output" type_attr: "T" }
input_arg { name: "output_h" type_attr: "T" }
input_arg { name: "output_c" type_attr: "T" }
input_arg { name: "output_backprop" type_attr: "T" }
input_arg { name: "output_h_backprop" type_attr: "T" }
input_arg { name: "output_c_backprop" type_attr: "T" }
input_arg { name: "reserve_space" type_attr: "T" }
input_arg { name: "host_reserved" type: DT_INT8 }
output_arg { name: "input_backprop" type_attr: "T" }
output_arg { name: "input_h_backprop" type_attr: "T" }
output_arg { name: "input_c_backprop" type_attr: "T" }
output_arg { name: "params_backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNBackpropV3 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                      v'13 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                          Double, Float] t) =>
                      Tensor v'1 t -- ^ __input__
                      -> Tensor v'2 t -- ^ __input_h__
                      -> Tensor v'3 t -- ^ __input_c__
                      -> Tensor v'4 t -- ^ __params__
                      -> Tensor v'5 Data.Int.Int32 -- ^ __sequence_lengths__
                      -> Tensor v'6 t -- ^ __output__
                      -> Tensor v'7 t -- ^ __output_h__
                      -> Tensor v'8 t -- ^ __output_c__
                      -> Tensor v'9 t -- ^ __output_backprop__
                      -> Tensor v'10 t -- ^ __output_h_backprop__
                      -> Tensor v'11 t -- ^ __output_c_backprop__
                      -> Tensor v'12 t -- ^ __reserve_space__
                      -> Tensor v'13 Data.Int.Int8 -- ^ __host_reserved__
                      -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                              Tensor Value t))
                      -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                      --
                      -- * __input_backprop__
                      --
                      -- * __input_h_backprop__
                      --
                      -- * __input_c_backprop__
                      --
                      -- * __params_backprop__
cudnnRNNBackpropV3 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (v'13 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV3' OpParams
forall a. a -> a
id
cudnnRNNBackpropV3' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                       v'13 t m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                           Double, Float] t) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __input__
                       -> Tensor v'2 t -- ^ __input_h__
                       -> Tensor v'3 t -- ^ __input_c__
                       -> Tensor v'4 t -- ^ __params__
                       -> Tensor v'5 Data.Int.Int32 -- ^ __sequence_lengths__
                       -> Tensor v'6 t -- ^ __output__
                       -> Tensor v'7 t -- ^ __output_h__
                       -> Tensor v'8 t -- ^ __output_c__
                       -> Tensor v'9 t -- ^ __output_backprop__
                       -> Tensor v'10 t -- ^ __output_h_backprop__
                       -> Tensor v'11 t -- ^ __output_c_backprop__
                       -> Tensor v'12 t -- ^ __reserve_space__
                       -> Tensor v'13 Data.Int.Int8 -- ^ __host_reserved__
                       -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                               Tensor Value t))
                       -- ^ (__input_backprop__, __input_h_backprop__, __input_c_backprop__, __params_backprop__)
                       --
                       -- * __input_backprop__
                       --
                       -- * __input_h_backprop__
                       --
                       -- * __input_c_backprop__
                       --
                       -- * __params_backprop__
cudnnRNNBackpropV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 Int8
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
cudnnRNNBackpropV3' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params sequence_lengths :: Tensor v'5 Int32
sequence_lengths
                    output :: Tensor v'6 t
output output_h :: Tensor v'7 t
output_h output_c :: Tensor v'8 t
output_c output_backprop :: Tensor v'9 t
output_backprop output_h_backprop :: Tensor v'10 t
output_h_backprop
                    output_c_backprop :: Tensor v'11 t
output_c_backprop reserve_space :: Tensor v'12 t
reserve_space
                    host_reserved :: Tensor v'13 Int8
host_reserved | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params,
                                                             Tensor v'5 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int32
sequence_lengths,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
output,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
output_h,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
output_c,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
output_backprop,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
output_h_backprop,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
output_c_backprop,
                                                             Tensor v'12 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 t
reserve_space,
                                                             Tensor v'13 Int8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'13 Int8
host_reserved]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNNBackpropV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
input_arg { name: "sequence_lengths" type: DT_INT32 }
input_arg { name: "output" type_attr: "T" }
input_arg { name: "output_h" type_attr: "T" }
input_arg { name: "output_c" type_attr: "T" }
input_arg { name: "output_backprop" type_attr: "T" }
input_arg { name: "output_h_backprop" type_attr: "T" }
input_arg { name: "output_c_backprop" type_attr: "T" }
input_arg { name: "reserve_space" type_attr: "T" }
input_arg { name: "host_reserved" type: DT_INT8 }
output_arg { name: "input_backprop" type_attr: "T" }
output_arg { name: "input_h_backprop" type_attr: "T" }
output_arg { name: "input_c_backprop" type_attr: "T" }
output_arg { name: "params_backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "num_proj" type: "int" default_value { i: 0 } }
attr { name: "time_major" type: "bool" default_value { b: true } }
-}
-- | 
cudnnRNNCanonicalToParams :: forall v'1 v'2 v'3 v'4 v'5
                             t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
                             Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                             -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                             -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                             -> [Tensor v'4 t] -- ^ __weights__
                             -> [Tensor v'5 t] -- ^ __biases__
                             -> Tensor Build t -- ^ __params__
cudnnRNNCanonicalToParams :: Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParams = OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParams' OpParams
forall a. a -> a
id
cudnnRNNCanonicalToParams' :: forall v'1 v'2 v'3 v'4 v'5
                              t . (OneOf '[Data.Word.Word16, Double,
                                           Float] t) => OpParams ->
                              Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                              -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                              -> [Tensor v'4 t] -- ^ __weights__
                              -> [Tensor v'5 t] -- ^ __biases__
                              -> Tensor Build t -- ^ __params__
cudnnRNNCanonicalToParams' :: OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParams' op'options :: OpParams
op'options num_layers :: Tensor v'1 Int32
num_layers num_units :: Tensor v'2 Int32
num_units input_size :: Tensor v'3 Int32
input_size weights :: [Tensor v'4 t]
weights
                           biases :: [Tensor v'5 t]
biases | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_params", [("weights", [Tensor v'4 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 t]
weights),
                                                                   ("biases", [Tensor v'5 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 t]
biases)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
num_layers,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_units,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
input_size,
                                                             [Tensor v'4 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 t]
weights,
                                                             [Tensor v'5 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'5 t]
biases]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CudnnRNNCanonicalToParams"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_params :: Int64
num_params = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 t]
weights) :: Int64
{-
input_arg { name: "num_layers" type: DT_INT32 }
input_arg { name: "num_units" type: DT_INT32 }
input_arg { name: "input_size" type: DT_INT32 }
input_arg {
  name: "weights" type_attr: "T" number_attr: "num_params"
}
input_arg {
  name: "biases" type_attr: "T" number_attr: "num_params"
}
output_arg { name: "params" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "num_params" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNCanonicalToParamsV2 :: forall v'1 v'2 v'3 v'4 v'5
                               t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) =>
                               Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                               -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                               -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                               -> [Tensor v'4 t] -- ^ __weights__
                               -> [Tensor v'5 t] -- ^ __biases__
                               -> Tensor Build t -- ^ __params__
cudnnRNNCanonicalToParamsV2 :: Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParamsV2 = OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParamsV2' OpParams
forall a. a -> a
id
cudnnRNNCanonicalToParamsV2' :: forall v'1 v'2 v'3 v'4 v'5
                                t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                                Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                                -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                                -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                                -> [Tensor v'4 t] -- ^ __weights__
                                -> [Tensor v'5 t] -- ^ __biases__
                                -> Tensor Build t -- ^ __params__
cudnnRNNCanonicalToParamsV2' :: OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> [Tensor v'4 t]
-> [Tensor v'5 t]
-> Tensor Build t
cudnnRNNCanonicalToParamsV2' op'options :: OpParams
op'options num_layers :: Tensor v'1 Int32
num_layers num_units :: Tensor v'2 Int32
num_units input_size :: Tensor v'3 Int32
input_size weights :: [Tensor v'4 t]
weights
                             biases :: [Tensor v'5 t]
biases | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_params_weights", [("weights", [Tensor v'4 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 t]
weights)]),
                                                     ("num_params_biases", [("biases", [Tensor v'5 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 t]
biases)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
num_layers,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_units,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
input_size,
                                                             [Tensor v'4 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 t]
weights,
                                                             [Tensor v'5 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'5 t]
biases]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CudnnRNNCanonicalToParamsV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params_weights" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params_weights
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params_biases" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params_biases
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_params_weights :: Int64
num_params_weights = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 t]
weights) :: Int64
    num_params_biases :: Int64
num_params_biases = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'5 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 t]
biases) :: Int64
{-
input_arg { name: "num_layers" type: DT_INT32 }
input_arg { name: "num_units" type: DT_INT32 }
input_arg { name: "input_size" type: DT_INT32 }
input_arg {
  name: "weights" type_attr: "T" number_attr: "num_params_weights"
}
input_arg {
  name: "biases" type_attr: "T" number_attr: "num_params_biases"
}
output_arg { name: "params" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "num_params_weights" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_params_biases" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "num_proj" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNParamsSize :: forall v'1 v'2 v'3 s . (OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] s) =>
                      DataType -- ^ __T__
                      -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                      -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                      -> Tensor Build s -- ^ __params_size__
cudnnRNNParamsSize :: DataType
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build s
cudnnRNNParamsSize = OpParams
-> DataType
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build s
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) s.
OneOf '[Int32, Int64] s =>
OpParams
-> DataType
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build s
cudnnRNNParamsSize' OpParams
forall a. a -> a
id
cudnnRNNParamsSize' :: forall v'1 v'2 v'3 s . (OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] s) =>
                       OpParams ->
                       DataType -- ^ __T__
                       -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                       -> Tensor Build s -- ^ __params_size__
cudnnRNNParamsSize' :: OpParams
-> DataType
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build s
cudnnRNNParamsSize' op'options :: OpParams
op'options t :: DataType
t num_layers :: Tensor v'1 Int32
num_layers num_units :: Tensor v'2 Int32
num_units
                    input_size :: Tensor v'3 Int32
input_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build s
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build s) -> Build OpDef -> Tensor Build s
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
num_layers,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_units,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
input_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CudnnRNNParamsSize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
t
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "num_layers" type: DT_INT32 }
input_arg { name: "num_units" type: DT_INT32 }
input_arg { name: "input_size" type: DT_INT32 }
output_arg { name: "params_size" type_attr: "S" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "num_proj" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNParamsToCanonical :: forall v'1 v'2 v'3 v'4
                             t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
                             Data.Int.Int64 -- ^ __num_params__
                             -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                             -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                             -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                             -> Tensor v'4 t -- ^ __params__
                             -> ([Tensor Build t], [Tensor Build t])
                             -- ^ (__weights__, __biases__)
                             --
                             -- * __weights__
                             --
                             -- * __biases__
cudnnRNNParamsToCanonical :: Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonical = OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonical' OpParams
forall a. a -> a
id
cudnnRNNParamsToCanonical' :: forall v'1 v'2 v'3 v'4
                              t . (OneOf '[Data.Word.Word16, Double,
                                           Float] t) => OpParams ->
                              Data.Int.Int64 -- ^ __num_params__
                              -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                              -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                              -> Tensor v'4 t -- ^ __params__
                              -> ([Tensor Build t], [Tensor Build t])
                              -- ^ (__weights__, __biases__)
                              --
                              -- * __weights__
                              --
                              -- * __biases__
cudnnRNNParamsToCanonical' :: OpParams
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonical' op'options :: OpParams
op'options num_params :: Int64
num_params num_layers :: Tensor v'1 Int32
num_layers num_units :: Tensor v'2 Int32
num_units input_size :: Tensor v'3 Int32
input_size
                           params :: Tensor v'4 t
params | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> ([Tensor Build t], [Tensor Build t])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_params, Int64
num_params] (Build OpDef -> ([Tensor Build t], [Tensor Build t]))
-> Build OpDef -> ([Tensor Build t], [Tensor Build t])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
num_layers,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_units,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
input_size,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CudnnRNNParamsToCanonical"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "num_layers" type: DT_INT32 }
input_arg { name: "num_units" type: DT_INT32 }
input_arg { name: "input_size" type: DT_INT32 }
input_arg { name: "params" type_attr: "T" }
output_arg {
  name: "weights" type_attr: "T" number_attr: "num_params"
}
output_arg {
  name: "biases" type_attr: "T" number_attr: "num_params"
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "num_params" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNParamsToCanonicalV2 :: forall v'1 v'2 v'3 v'4
                               t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) =>
                               Data.Int.Int64 -- ^ __num_params_biases__
                               -> Data.Int.Int64 -- ^ __num_params_weights__
                               -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                               -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                               -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                               -> Tensor v'4 t -- ^ __params__
                               -> ([Tensor Build t], [Tensor Build t])
                               -- ^ (__weights__, __biases__)
                               --
                               -- * __weights__
                               --
                               -- * __biases__
cudnnRNNParamsToCanonicalV2 :: Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonicalV2 = OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonicalV2' OpParams
forall a. a -> a
id
cudnnRNNParamsToCanonicalV2' :: forall v'1 v'2 v'3 v'4
                                t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                                Data.Int.Int64 -- ^ __num_params_biases__
                                -> Data.Int.Int64 -- ^ __num_params_weights__
                                -> Tensor v'1 Data.Int.Int32 -- ^ __num_layers__
                                -> Tensor v'2 Data.Int.Int32 -- ^ __num_units__
                                -> Tensor v'3 Data.Int.Int32 -- ^ __input_size__
                                -> Tensor v'4 t -- ^ __params__
                                -> ([Tensor Build t], [Tensor Build t])
                                -- ^ (__weights__, __biases__)
                                --
                                -- * __weights__
                                --
                                -- * __biases__
cudnnRNNParamsToCanonicalV2' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> ([Tensor Build t], [Tensor Build t])
cudnnRNNParamsToCanonicalV2' op'options :: OpParams
op'options num_params_biases :: Int64
num_params_biases num_params_weights :: Int64
num_params_weights
                             num_layers :: Tensor v'1 Int32
num_layers num_units :: Tensor v'2 Int32
num_units input_size :: Tensor v'3 Int32
input_size
                             params :: Tensor v'4 t
params | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> ([Tensor Build t], [Tensor Build t])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_params_weights, Int64
num_params_biases] (Build OpDef -> ([Tensor Build t], [Tensor Build t]))
-> Build OpDef -> ([Tensor Build t], [Tensor Build t])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
num_layers,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_units,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
input_size,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CudnnRNNParamsToCanonicalV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params_biases" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params_biases
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_params_weights" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_params_weights
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "num_layers" type: DT_INT32 }
input_arg { name: "num_units" type: DT_INT32 }
input_arg { name: "input_size" type: DT_INT32 }
input_arg { name: "params" type_attr: "T" }
output_arg {
  name: "weights" type_attr: "T" number_attr: "num_params_weights"
}
output_arg {
  name: "biases" type_attr: "T" number_attr: "num_params_biases"
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "num_params_weights" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "num_params_biases" type: "int" has_minimum: true minimum: 1
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "num_proj" type: "int" default_value { i: 0 } }
-}
-- | 
cudnnRNNV2 :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                             OneOf '[Data.Word.Word16, Double,
                                                     Float] t) =>
              Tensor v'1 t -- ^ __input__
              -> Tensor v'2 t -- ^ __input_h__
              -> Tensor v'3 t -- ^ __input_c__
              -> Tensor v'4 t -- ^ __params__
              -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                      Tensor Value t, Tensor Value Data.Int.Int8))
              -- ^ (__output__, __output_h__, __output_c__, __reserve_space__, __host_reserved__)
              --
              -- * __output__
              --
              -- * __output_h__
              --
              -- * __output_c__
              --
              -- * __reserve_space__
              --
              -- * __host_reserved__
cudnnRNNV2 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV2' OpParams
forall a. a -> a
id
cudnnRNNV2' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                              OneOf '[Data.Word.Word16, Double,
                                                      Float] t) => OpParams ->
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 t -- ^ __input_h__
               -> Tensor v'3 t -- ^ __input_c__
               -> Tensor v'4 t -- ^ __params__
               -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                       Tensor Value t, Tensor Value Data.Int.Int8))
               -- ^ (__output__, __output_h__, __output_c__, __reserve_space__, __host_reserved__)
               --
               -- * __output__
               --
               -- * __output_h__
               --
               -- * __output_c__
               --
               -- * __reserve_space__
               --
               -- * __host_reserved__
cudnnRNNV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
   Tensor Value Int8)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
    Tensor Value Int8)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
       Tensor Value Int8))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNNV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_h" type_attr: "T" }
output_arg { name: "output_c" type_attr: "T" }
output_arg { name: "reserve_space" type_attr: "T" }
output_arg { name: "host_reserved" type: DT_INT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
cudnnRNNV3 :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                 OneOf '[Data.Word.Word16,
                                                         Double, Float] t) =>
              Tensor v'1 t -- ^ __input__
              -> Tensor v'2 t -- ^ __input_h__
              -> Tensor v'3 t -- ^ __input_c__
              -> Tensor v'4 t -- ^ __params__
              -> Tensor v'5 Data.Int.Int32 -- ^ __sequence_lengths__
              -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                      Tensor Value t, Tensor Value Data.Int.Int8))
              -- ^ (__output__, __output_h__, __output_c__, __reserve_space__, __host_reserved__)
              --
              -- * __output__
              --
              -- * __output_h__
              --
              -- * __output_c__
              --
              -- * __reserve_space__
              --
              -- * __host_reserved__
cudnnRNNV3 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV3' OpParams
forall a. a -> a
id
cudnnRNNV3' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                  OneOf '[Data.Word.Word16,
                                                          Double, Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 t -- ^ __input_h__
               -> Tensor v'3 t -- ^ __input_c__
               -> Tensor v'4 t -- ^ __params__
               -> Tensor v'5 Data.Int.Int32 -- ^ __sequence_lengths__
               -> m' ((Tensor Value t, Tensor Value t, Tensor Value t,
                       Tensor Value t, Tensor Value Data.Int.Int8))
               -- ^ (__output__, __output_h__, __output_c__, __reserve_space__, __host_reserved__)
               --
               -- * __output__
               --
               -- * __output_h__
               --
               -- * __output_c__
               --
               -- * __reserve_space__
               --
               -- * __host_reserved__
cudnnRNNV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 Int32
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
cudnnRNNV3' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_h :: Tensor v'2 t
input_h input_c :: Tensor v'3 t
input_c params :: Tensor v'4 t
params
            sequence_lengths :: Tensor v'5 Int32
sequence_lengths | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
   Tensor Value Int8)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
    Tensor Value Int8)
 -> m'
      (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
       Tensor Value Int8))
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
-> m'
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_h,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_c,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
params,
                                                             Tensor v'5 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int32
sequence_lengths]
        [Int64]
-> OpDef
-> Build
     (Tensor Value t, Tensor Value t, Tensor Value t, Tensor Value t,
      Tensor Value Int8)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "CudnnRNNV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_h" type_attr: "T" }
input_arg { name: "input_c" type_attr: "T" }
input_arg { name: "params" type_attr: "T" }
input_arg { name: "sequence_lengths" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_h" type_attr: "T" }
output_arg { name: "output_c" type_attr: "T" }
output_arg { name: "reserve_space" type_attr: "T" }
output_arg { name: "host_reserved" type: DT_INT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "rnn_mode"
  type: "string"
  default_value { s: "lstm" }
  allowed_values {
    list { s: "rnn_relu" s: "rnn_tanh" s: "lstm" s: "gru" }
  }
}
attr {
  name: "input_mode"
  type: "string"
  default_value { s: "linear_input" }
  allowed_values {
    list { s: "linear_input" s: "skip_input" s: "auto_select" }
  }
}
attr {
  name: "direction"
  type: "string"
  default_value { s: "unidirectional" }
  allowed_values { list { s: "unidirectional" s: "bidirectional" } }
}
attr { name: "dropout" type: "float" default_value { f: 0.0 } }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "num_proj" type: "int" default_value { i: 0 } }
attr { name: "is_training" type: "bool" default_value { b: true } }
attr { name: "time_major" type: "bool" default_value { b: true } }
-}
-- | 
cumprod :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tidx) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 tidx -- ^ __axis__
           -> Tensor Build t -- ^ __out__
cumprod :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumprod = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumprod' OpParams
forall a. a -> a
id
cumprod' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tidx) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 tidx -- ^ __axis__
            -> Tensor Build t -- ^ __out__
cumprod' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumprod' op'options :: OpParams
op'options x :: Tensor v'1 t
x axis :: Tensor v'2 tidx
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cumprod"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "axis" type_attr: "Tidx" }
output_arg { name: "out" type_attr: "T" }
attr { name: "exclusive" type: "bool" default_value { b: false } }
attr { name: "reverse" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
cumsum :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tidx) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor v'2 tidx -- ^ __axis__
          -> Tensor Build t -- ^ __out__
cumsum :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumsum = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumsum' OpParams
forall a. a -> a
id
cumsum' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tidx) => OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 tidx -- ^ __axis__
           -> Tensor Build t -- ^ __out__
cumsum' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumsum' op'options :: OpParams
op'options x :: Tensor v'1 t
x axis :: Tensor v'2 tidx
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Cumsum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "axis" type_attr: "Tidx" }
output_arg { name: "out" type_attr: "T" }
attr { name: "exclusive" type: "bool" default_value { b: false } }
attr { name: "reverse" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
cumulativeLogsumexp :: forall v'1 v'2 t tidx . (OneOf '[Data.Word.Word16,
                                                        Double, Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tidx) =>
                       Tensor v'1 t -- ^ __x__
                       -> Tensor v'2 tidx -- ^ __axis__
                       -> Tensor Build t -- ^ __out__
cumulativeLogsumexp :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumulativeLogsumexp = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumulativeLogsumexp' OpParams
forall a. a -> a
id
cumulativeLogsumexp' :: forall v'1 v'2 t tidx . (OneOf '[Data.Word.Word16,
                                                         Double, Float] t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tidx) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __x__
                        -> Tensor v'2 tidx -- ^ __axis__
                        -> Tensor Build t -- ^ __out__
cumulativeLogsumexp' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
cumulativeLogsumexp' op'options :: OpParams
op'options x :: Tensor v'1 t
x axis :: Tensor v'2 tidx
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "CumulativeLogsumexp"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "axis" type_attr: "Tidx" }
output_arg { name: "out" type_attr: "T" }
attr { name: "exclusive" type: "bool" default_value { b: false } }
attr { name: "reverse" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
dataFormatDimMap :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] t) =>
                    Tensor v'1 t -- ^ __x__
                    -> Tensor Build t -- ^ __y__
dataFormatDimMap :: Tensor v'1 t -> Tensor Build t
dataFormatDimMap = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
dataFormatDimMap' OpParams
forall a. a -> a
id
dataFormatDimMap' :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] t) => OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> Tensor Build t -- ^ __y__
dataFormatDimMap' :: OpParams -> Tensor v'1 t -> Tensor Build t
dataFormatDimMap' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DataFormatDimMap"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "src_format" type: "string" default_value { s: "NHWC" }
}
attr {
  name: "dst_format" type: "string" default_value { s: "NCHW" }
}
-}
-- | 
dataFormatVecPermute :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] t) =>
                        Tensor v'1 t -- ^ __x__
                        -> Tensor Build t -- ^ __y__
dataFormatVecPermute :: Tensor v'1 t -> Tensor Build t
dataFormatVecPermute = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
dataFormatVecPermute' OpParams
forall a. a -> a
id
dataFormatVecPermute' :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] t) =>
                         OpParams ->
                         Tensor v'1 t -- ^ __x__
                         -> Tensor Build t -- ^ __y__
dataFormatVecPermute' :: OpParams -> Tensor v'1 t -> Tensor Build t
dataFormatVecPermute' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DataFormatVecPermute"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "src_format" type: "string" default_value { s: "NHWC" }
}
attr {
  name: "dst_format" type: "string" default_value { s: "NCHW" }
}
-}
-- | 
dataServiceDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m' . (MonadBuild m') =>
                      [DataType] -- ^ __output_types__
                      -> Tensor v'1 Data.Int.Int64 -- ^ __dataset_id__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __processing_mode__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __address__
                      -> Tensor v'4 Data.ByteString.ByteString -- ^ __protocol__
                      -> Tensor v'5 Data.ByteString.ByteString -- ^ __job_name__
                      -> Tensor v'6 Data.Int.Int64 -- ^ __max_outstanding_requests__
                      -> Tensor v'7 ResourceHandle -- ^ __iteration_counter__
                      -> m' (Tensor Value Variant) -- ^ __handle__
dataServiceDataset :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> Tensor v'6 Int64
-> Tensor v'7 ResourceHandle
-> m' (Tensor Value Variant)
dataServiceDataset = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> Tensor v'6 Int64
-> Tensor v'7 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> Tensor v'6 Int64
-> Tensor v'7 ResourceHandle
-> m' (Tensor Value Variant)
dataServiceDataset' OpParams
forall a. a -> a
id
dataServiceDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7
                       m' . (MonadBuild m') => OpParams ->
                       [DataType] -- ^ __output_types__
                       -> Tensor v'1 Data.Int.Int64 -- ^ __dataset_id__
                       -> Tensor v'2 Data.ByteString.ByteString -- ^ __processing_mode__
                       -> Tensor v'3 Data.ByteString.ByteString -- ^ __address__
                       -> Tensor v'4 Data.ByteString.ByteString -- ^ __protocol__
                       -> Tensor v'5 Data.ByteString.ByteString -- ^ __job_name__
                       -> Tensor v'6 Data.Int.Int64 -- ^ __max_outstanding_requests__
                       -> Tensor v'7 ResourceHandle -- ^ __iteration_counter__
                       -> m' (Tensor Value Variant) -- ^ __handle__
dataServiceDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> Tensor v'6 Int64
-> Tensor v'7 ResourceHandle
-> m' (Tensor Value Variant)
dataServiceDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types dataset_id :: Tensor v'1 Int64
dataset_id processing_mode :: Tensor v'2 ByteString
processing_mode address :: Tensor v'3 ByteString
address
                    protocol :: Tensor v'4 ByteString
protocol job_name :: Tensor v'5 ByteString
job_name max_outstanding_requests :: Tensor v'6 Int64
max_outstanding_requests
                    iteration_counter :: Tensor v'7 ResourceHandle
iteration_counter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
dataset_id,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
processing_mode,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
address,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
protocol,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
job_name,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
max_outstanding_requests,
                                                             Tensor v'7 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 ResourceHandle
iteration_counter]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DataServiceDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dataset_id" type: DT_INT64 }
input_arg { name: "processing_mode" type: DT_STRING }
input_arg { name: "address" type: DT_STRING }
input_arg { name: "protocol" type: DT_STRING }
input_arg { name: "job_name" type: DT_STRING }
input_arg { name: "max_outstanding_requests" type: DT_INT64 }
input_arg { name: "iteration_counter" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "task_refresh_interval_hint_ms"
  type: "int"
  default_value { i: -1 }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
datasetCardinality :: Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor Build Data.Int.Int64 -- ^ __cardinality__
datasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
datasetCardinality = OpParams -> Tensor v'1 Variant -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Int64
datasetCardinality' OpParams
forall a. a -> a
id
datasetCardinality' :: OpParams ->
                       Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor Build Data.Int.Int64 -- ^ __cardinality__
datasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
datasetCardinality' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DatasetCardinality"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "cardinality" type: DT_INT64 }
-}
-- | 
datasetFromGraph :: Tensor v'1 Data.ByteString.ByteString -- ^ __graph_def__
                    -> Tensor Build Variant -- ^ __handle__
datasetFromGraph :: Tensor v'1 ByteString -> Tensor Build Variant
datasetFromGraph = OpParams -> Tensor v'1 ByteString -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Variant
datasetFromGraph' OpParams
forall a. a -> a
id
datasetFromGraph' :: OpParams ->
                     Tensor v'1 Data.ByteString.ByteString -- ^ __graph_def__
                     -> Tensor Build Variant -- ^ __handle__
datasetFromGraph' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Variant
datasetFromGraph' op'options :: OpParams
op'options graph_def :: Tensor v'1 ByteString
graph_def | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
graph_def]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DatasetFromGraph"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "graph_def" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
datasetToGraph :: Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __graph__
datasetToGraph :: Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraph = OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraph' OpParams
forall a. a -> a
id
datasetToGraph' :: OpParams ->
                   Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __graph__
datasetToGraph' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraph' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DatasetToGraph"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "graph" type: DT_STRING }
attr {
  name: "stateful_whitelist"
  type: "list(string)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "allow_stateful" type: "bool" default_value { b: false }
}
attr {
  name: "strip_device_assignment"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
datasetToGraphV2 :: Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __graph__
datasetToGraphV2 :: Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraphV2 = OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraphV2' OpParams
forall a. a -> a
id
datasetToGraphV2' :: OpParams ->
                     Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor Build Data.ByteString.ByteString -- ^ __graph__
datasetToGraphV2' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
datasetToGraphV2' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DatasetToGraphV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "graph" type: DT_STRING }
attr {
  name: "external_state_policy" type: "int" default_value { i: 0 }
}
attr {
  name: "strip_device_assignment"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
datasetToSingleElement :: forall v'1 output_types m' . (MonadBuild m',
                                                        TensorTypes output_types) =>
                          Tensor v'1 Variant -- ^ __dataset__
                          -> m' (TensorList (Value) output_types) -- ^ __components__
datasetToSingleElement :: Tensor v'1 Variant -> m' (TensorList Value output_types)
datasetToSingleElement = OpParams
-> Tensor v'1 Variant -> m' (TensorList Value output_types)
forall (v'1 :: * -> *) (output_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes output_types) =>
OpParams
-> Tensor v'1 Variant -> m' (TensorList Value output_types)
datasetToSingleElement' OpParams
forall a. a -> a
id
datasetToSingleElement' :: forall v'1 output_types m' . (MonadBuild m',
                                                         TensorTypes output_types) =>
                           OpParams ->
                           Tensor v'1 Variant -- ^ __dataset__
                           -> m' (TensorList (Value) output_types) -- ^ __components__
datasetToSingleElement' :: OpParams
-> Tensor v'1 Variant -> m' (TensorList Value output_types)
datasetToSingleElement' op'options :: OpParams
op'options dataset :: Tensor v'1 Variant
dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value output_types)
 -> m' (TensorList Value output_types))
-> Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
dataset]
        [Int64] -> OpDef -> Build (TensorList Value output_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DatasetToSingleElement"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dataset" type: DT_VARIANT }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
datasetToTFRecord :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                     Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                     -> Tensor v'3 Data.ByteString.ByteString -- ^ __compression_type__
                     -> m' (ControlNode)
datasetToTFRecord :: Tensor v'1 Variant
-> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
datasetToTFRecord = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
datasetToTFRecord' OpParams
forall a. a -> a
id
datasetToTFRecord' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __compression_type__
                      -> m' (ControlNode)
datasetToTFRecord' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
datasetToTFRecord' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset filename :: Tensor v'2 ByteString
filename
                   compression_type :: Tensor v'3 ByteString
compression_type | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
compression_type]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DatasetToTFRecord"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
-}
-- | 
dawsn :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
dawsn :: Tensor v'1 t -> Tensor Build t
dawsn = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
dawsn' OpParams
forall a. a -> a
id
dawsn' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
dawsn' :: OpParams -> Tensor v'1 t -> Tensor Build t
dawsn' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Dawsn"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
debugGradientIdentity :: forall v'1 t . (TensorType t) =>
                         Tensor v'1 t -- ^ __input__
                         -> Tensor Build t -- ^ __output__
debugGradientIdentity :: Tensor v'1 t -> Tensor Build t
debugGradientIdentity = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
debugGradientIdentity' OpParams
forall a. a -> a
id
debugGradientIdentity' :: forall v'1 t . (TensorType t) => OpParams ->
                          Tensor v'1 t -- ^ __input__
                          -> Tensor Build t -- ^ __output__
debugGradientIdentity' :: OpParams -> Tensor v'1 t -> Tensor Build t
debugGradientIdentity' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DebugGradientIdentity"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
debugGradientRefIdentity :: forall t m' . (MonadBuild m', TensorType t) =>
                            Tensor Ref t -- ^ __input__
                            -> m' (Tensor Ref t) -- ^ __output__
debugGradientRefIdentity :: Tensor Ref t -> m' (Tensor Ref t)
debugGradientRefIdentity = OpParams -> Tensor Ref t -> m' (Tensor Ref t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor Ref t -> m' (Tensor Ref t)
debugGradientRefIdentity' OpParams
forall a. a -> a
id
debugGradientRefIdentity' :: forall t m' . (MonadBuild m', TensorType t) =>
                             OpParams ->
                             Tensor Ref t -- ^ __input__
                             -> m' (Tensor Ref t) -- ^ __output__
debugGradientRefIdentity' :: OpParams -> Tensor Ref t -> m' (Tensor Ref t)
debugGradientRefIdentity' op'options :: OpParams
op'options input :: Tensor Ref t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
input]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DebugGradientRefIdentity"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
-}
-- | 
debugIdentity :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
debugIdentity :: Tensor v'1 t -> Tensor Build t
debugIdentity = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
debugIdentity' OpParams
forall a. a -> a
id
debugIdentity' :: forall v'1 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor Build t -- ^ __output__
debugIdentity' :: OpParams -> Tensor v'1 t -> Tensor Build t
debugIdentity' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DebugIdentity"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "device_name" type: "string" default_value { s: "" } }
attr { name: "tensor_name" type: "string" default_value { s: "" } }
attr {
  name: "debug_urls" type: "list(string)" default_value { list { } }
}
attr { name: "gated_grpc" type: "bool" default_value { b: false } }
-}
-- | 
debugIdentityV2 :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                   Tensor v'1 t -- ^ __input__
                   -> m' (Tensor Value t) -- ^ __output__
debugIdentityV2 :: Tensor v'1 t -> m' (Tensor Value t)
debugIdentityV2 = OpParams -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value t)
debugIdentityV2' OpParams
forall a. a -> a
id
debugIdentityV2' :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                    OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> m' (Tensor Value t) -- ^ __output__
debugIdentityV2' :: OpParams -> Tensor v'1 t -> m' (Tensor Value t)
debugIdentityV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DebugIdentityV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "tfdbg_context_id" type: "string" default_value { s: "" }
}
attr { name: "op_name" type: "string" default_value { s: "" } }
attr { name: "output_slot" type: "int" default_value { i: -1 } }
attr {
  name: "tensor_debug_mode" type: "int" default_value { i: -1 }
}
attr {
  name: "debug_urls" type: "list(string)" default_value { list { } }
}
attr {
  name: "circular_buffer_size" type: "int" default_value { i: 1000 }
}
attr {
  name: "tfdbg_run_id" type: "string" default_value { s: "" }
}
-}
-- | 
debugNanCount :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                 -> Tensor Build Data.Int.Int64 -- ^ __output__
debugNanCount :: Tensor v'1 t -> Tensor Build Int64
debugNanCount = OpParams -> Tensor v'1 t -> Tensor Build Int64
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build Int64
debugNanCount' OpParams
forall a. a -> a
id
debugNanCount' :: forall v'1 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor Build Data.Int.Int64 -- ^ __output__
debugNanCount' :: OpParams -> Tensor v'1 t -> Tensor Build Int64
debugNanCount' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DebugNanCount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_INT64 }
attr { name: "T" type: "type" }
attr { name: "device_name" type: "string" default_value { s: "" } }
attr { name: "tensor_name" type: "string" default_value { s: "" } }
attr {
  name: "debug_urls" type: "list(string)" default_value { list { } }
}
attr { name: "gated_grpc" type: "bool" default_value { b: false } }
-}
-- | 
debugNumericSummary :: forall v'1 t . (TensorType t) =>
                       Tensor v'1 t -- ^ __input__
                       -> Tensor Build Double -- ^ __output__
debugNumericSummary :: Tensor v'1 t -> Tensor Build Double
debugNumericSummary = OpParams -> Tensor v'1 t -> Tensor Build Double
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build Double
debugNumericSummary' OpParams
forall a. a -> a
id
debugNumericSummary' :: forall v'1 t . (TensorType t) => OpParams ->
                        Tensor v'1 t -- ^ __input__
                        -> Tensor Build Double -- ^ __output__
debugNumericSummary' :: OpParams -> Tensor v'1 t -> Tensor Build Double
debugNumericSummary' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Double
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Double)
-> Build OpDef -> Tensor Build Double
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DebugNumericSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_DOUBLE }
attr { name: "T" type: "type" }
attr { name: "device_name" type: "string" default_value { s: "" } }
attr { name: "tensor_name" type: "string" default_value { s: "" } }
attr {
  name: "debug_urls" type: "list(string)" default_value { list { } }
}
attr {
  name: "lower_bound" type: "float" default_value { f: -Infinity }
}
attr {
  name: "upper_bound" type: "float" default_value { f: Infinity }
}
attr {
  name: "mute_if_healthy" type: "bool" default_value { b: false }
}
attr { name: "gated_grpc" type: "bool" default_value { b: false } }
-}
-- | 
debugNumericSummaryV2 :: forall v'1 output_dtype t . (OneOf '[Double,
                                                              Float] output_dtype,
                                                      TensorType t) =>
                         Tensor v'1 t -- ^ __input__
                         -> Tensor Build output_dtype -- ^ __output__
debugNumericSummaryV2 :: Tensor v'1 t -> Tensor Build output_dtype
debugNumericSummaryV2 = OpParams -> Tensor v'1 t -> Tensor Build output_dtype
forall (v'1 :: * -> *) output_dtype t.
(OneOf '[Double, Float] output_dtype, TensorType t) =>
OpParams -> Tensor v'1 t -> Tensor Build output_dtype
debugNumericSummaryV2' OpParams
forall a. a -> a
id
debugNumericSummaryV2' :: forall v'1 output_dtype t . (OneOf '[Double,
                                                               Float] output_dtype,
                                                       TensorType t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __input__
                          -> Tensor Build output_dtype -- ^ __output__
debugNumericSummaryV2' :: OpParams -> Tensor v'1 t -> Tensor Build output_dtype
debugNumericSummaryV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build output_dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build output_dtype)
-> Build OpDef -> Tensor Build output_dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DebugNumericSummaryV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_dtype
forall a. HasCallStack => a
undefined :: output_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "output_dtype" }
attr {
  name: "output_dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr { name: "T" type: "type" }
attr {
  name: "tensor_debug_mode" type: "int" default_value { i: -1 }
}
attr { name: "tensor_id" type: "int" default_value { i: -1 } }
-}
-- | 
decodeAndCropJpeg :: Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __crop_window__
                     -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeAndCropJpeg :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
decodeAndCropJpeg = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
decodeAndCropJpeg' OpParams
forall a. a -> a
id
decodeAndCropJpeg' :: OpParams ->
                      Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __crop_window__
                      -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeAndCropJpeg' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build Word8
decodeAndCropJpeg' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents crop_window :: Tensor v'2 Int32
crop_window | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
crop_window]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeAndCropJpeg"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
input_arg { name: "crop_window" type: DT_INT32 }
output_arg { name: "image" type: DT_UINT8 }
attr { name: "channels" type: "int" default_value { i: 0 } }
attr { name: "ratio" type: "int" default_value { i: 1 } }
attr {
  name: "fancy_upscaling" type: "bool" default_value { b: true }
}
attr {
  name: "try_recover_truncated"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "acceptable_fraction" type: "float" default_value { f: 1.0 }
}
attr { name: "dct_method" type: "string" default_value { s: "" } }
-}
-- | 
decodeBase64 :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
decodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
decodeBase64 = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeBase64' OpParams
forall a. a -> a
id
decodeBase64' :: OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __output__
decodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeBase64' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeBase64"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
-}
-- | 
decodeBmp :: Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
             -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeBmp :: Tensor v'1 ByteString -> Tensor Build Word8
decodeBmp = OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeBmp' OpParams
forall a. a -> a
id
decodeBmp' :: OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
              -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeBmp' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeBmp' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeBmp"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "image" type: DT_UINT8 }
attr { name: "channels" type: "int" default_value { i: 0 } }
-}
-- | 
decodeCSV :: forall v'1 v'2 oUT_TYPE . (OneOfs '[Data.ByteString.ByteString,
                                                 Data.Int.Int32, Data.Int.Int64,
                                                 Double, Float] oUT_TYPE) =>
             Tensor v'1 Data.ByteString.ByteString -- ^ __records__
             -> TensorList (v'2) oUT_TYPE -- ^ __record_defaults__
             -> TensorList (Build) oUT_TYPE -- ^ __output__
decodeCSV :: Tensor v'1 ByteString
-> TensorList v'2 oUT_TYPE -> TensorList Build oUT_TYPE
decodeCSV = OpParams
-> Tensor v'1 ByteString
-> TensorList v'2 oUT_TYPE
-> TensorList Build oUT_TYPE
forall (v'1 :: * -> *) (v'2 :: * -> *) (oUT_TYPE :: [*]).
OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE =>
OpParams
-> Tensor v'1 ByteString
-> TensorList v'2 oUT_TYPE
-> TensorList Build oUT_TYPE
decodeCSV' OpParams
forall a. a -> a
id
decodeCSV' :: forall v'1 v'2 oUT_TYPE . (OneOfs '[Data.ByteString.ByteString,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Double,
                                                  Float] oUT_TYPE) =>
              OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __records__
              -> TensorList (v'2) oUT_TYPE -- ^ __record_defaults__
              -> TensorList (Build) oUT_TYPE -- ^ __output__
decodeCSV' :: OpParams
-> Tensor v'1 ByteString
-> TensorList v'2 oUT_TYPE
-> TensorList Build oUT_TYPE
decodeCSV' op'options :: OpParams
op'options records :: Tensor v'1 ByteString
records record_defaults :: TensorList v'2 oUT_TYPE
record_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build oUT_TYPE
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build oUT_TYPE)
-> Build OpDef -> TensorList Build oUT_TYPE
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
records,
                                                             TensorList v'2 oUT_TYPE -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 oUT_TYPE
record_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeCSV"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "OUT_TYPE" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy oUT_TYPE -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy oUT_TYPE
forall k (t :: k). Proxy t
Proxy :: Proxy oUT_TYPE)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "records" type: DT_STRING }
input_arg { name: "record_defaults" type_list_attr: "OUT_TYPE" }
output_arg { name: "output" type_list_attr: "OUT_TYPE" }
attr {
  name: "OUT_TYPE"
  type: "list(type)"
  has_minimum: true
  minimum: 1
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_STRING
    }
  }
}
attr {
  name: "field_delim" type: "string" default_value { s: "," }
}
attr {
  name: "use_quote_delim" type: "bool" default_value { b: true }
}
attr { name: "na_value" type: "string" default_value { s: "" } }
attr {
  name: "select_cols" type: "list(int)" default_value { list { } }
}
-}
-- | 
decodeCompressed :: Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __output__
decodeCompressed :: Tensor v'1 ByteString -> Tensor Build ByteString
decodeCompressed = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeCompressed' OpParams
forall a. a -> a
id
decodeCompressed' :: OpParams ->
                     Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
                     -> Tensor Build Data.ByteString.ByteString -- ^ __output__
decodeCompressed' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeCompressed' op'options :: OpParams
op'options bytes :: Tensor v'1 ByteString
bytes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
bytes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeCompressed"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "bytes" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr {
  name: "compression_type" type: "string" default_value { s: "" }
}
-}
-- | 
decodeGif :: Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
             -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeGif :: Tensor v'1 ByteString -> Tensor Build Word8
decodeGif = OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeGif' OpParams
forall a. a -> a
id
decodeGif' :: OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
              -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeGif' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeGif' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeGif"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "image" type: DT_UINT8 }
-}
-- | 
decodeJSONExample :: Tensor v'1 Data.ByteString.ByteString -- ^ __json_examples__
                     -> Tensor Build Data.ByteString.ByteString -- ^ __binary_examples__
decodeJSONExample :: Tensor v'1 ByteString -> Tensor Build ByteString
decodeJSONExample = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeJSONExample' OpParams
forall a. a -> a
id
decodeJSONExample' :: OpParams ->
                      Tensor v'1 Data.ByteString.ByteString -- ^ __json_examples__
                      -> Tensor Build Data.ByteString.ByteString -- ^ __binary_examples__
decodeJSONExample' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
decodeJSONExample' op'options :: OpParams
op'options json_examples :: Tensor v'1 ByteString
json_examples | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
json_examples]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeJSONExample"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "json_examples" type: DT_STRING }
output_arg { name: "binary_examples" type: DT_STRING }
-}
-- | 
decodeJpeg :: Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
              -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeJpeg :: Tensor v'1 ByteString -> Tensor Build Word8
decodeJpeg = OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeJpeg' OpParams
forall a. a -> a
id
decodeJpeg' :: OpParams ->
               Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
               -> Tensor Build Data.Word.Word8 -- ^ __image__
decodeJpeg' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
decodeJpeg' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeJpeg"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "image" type: DT_UINT8 }
attr { name: "channels" type: "int" default_value { i: 0 } }
attr { name: "ratio" type: "int" default_value { i: 1 } }
attr {
  name: "fancy_upscaling" type: "bool" default_value { b: true }
}
attr {
  name: "try_recover_truncated"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "acceptable_fraction" type: "float" default_value { f: 1.0 }
}
attr { name: "dct_method" type: "string" default_value { s: "" } }
-}
-- | 
decodePaddedRaw :: forall v'1 v'2 out_type . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8, Double,
                                                      Float] out_type) =>
                   Tensor v'1 Data.ByteString.ByteString -- ^ __input_bytes__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __fixed_length__
                   -> Tensor Build out_type -- ^ __output__
decodePaddedRaw :: Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build out_type
decodePaddedRaw = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build out_type
forall (v'1 :: * -> *) (v'2 :: * -> *) out_type.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float]
  out_type =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build out_type
decodePaddedRaw' OpParams
forall a. a -> a
id
decodePaddedRaw' :: forall v'1 v'2 out_type . (OneOf '[Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word8, Double,
                                                       Float] out_type) =>
                    OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __input_bytes__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __fixed_length__
                    -> Tensor Build out_type -- ^ __output__
decodePaddedRaw' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build out_type
decodePaddedRaw' op'options :: OpParams
op'options input_bytes :: Tensor v'1 ByteString
input_bytes fixed_length :: Tensor v'2 Int32
fixed_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input_bytes,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fixed_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodePaddedRaw"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_bytes" type: DT_STRING }
input_arg { name: "fixed_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "out_type" }
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT16
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
    }
  }
}
attr {
  name: "little_endian" type: "bool" default_value { b: true }
}
-}
-- | 
decodePng :: forall v'1 dtype . (OneOf '[Data.Word.Word16,
                                         Data.Word.Word8] dtype) =>
             Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
             -> Tensor Build dtype -- ^ __image__
decodePng :: Tensor v'1 ByteString -> Tensor Build dtype
decodePng = OpParams -> Tensor v'1 ByteString -> Tensor Build dtype
forall (v'1 :: * -> *) dtype.
OneOf '[Word16, Word8] dtype =>
OpParams -> Tensor v'1 ByteString -> Tensor Build dtype
decodePng' OpParams
forall a. a -> a
id
decodePng' :: forall v'1 dtype . (OneOf '[Data.Word.Word16,
                                          Data.Word.Word8] dtype) => OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
              -> Tensor Build dtype -- ^ __image__
decodePng' :: OpParams -> Tensor v'1 ByteString -> Tensor Build dtype
decodePng' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodePng"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "image" type_attr: "dtype" }
attr { name: "channels" type: "int" default_value { i: 0 } }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_UINT8 }
  allowed_values { list { type: DT_UINT8 type: DT_UINT16 } }
}
-}
-- | 
decodeProtoV2 :: forall v'1 output_types . (TensorTypes output_types) =>
                 ByteString -- ^ __message_type__
                 -> Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
                 -> (Tensor Build Data.Int.Int32,
                     TensorList (Build) output_types)
                 -- ^ (__sizes__, __values__)
                 --
                 -- * __sizes__
                 --
                 -- * __values__
decodeProtoV2 :: ByteString
-> Tensor v'1 ByteString
-> (Tensor Build Int32, TensorList Build output_types)
decodeProtoV2 = OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build Int32, TensorList Build output_types)
forall (v'1 :: * -> *) (output_types :: [*]).
TensorTypes output_types =>
OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build Int32, TensorList Build output_types)
decodeProtoV2' OpParams
forall a. a -> a
id
decodeProtoV2' :: forall v'1 output_types . (TensorTypes output_types) =>
                  OpParams ->
                  ByteString -- ^ __message_type__
                  -> Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
                  -> (Tensor Build Data.Int.Int32,
                      TensorList (Build) output_types)
                  -- ^ (__sizes__, __values__)
                  --
                  -- * __sizes__
                  --
                  -- * __values__
decodeProtoV2' :: OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build Int32, TensorList Build output_types)
decodeProtoV2' op'options :: OpParams
op'options message_type :: ByteString
message_type bytes :: Tensor v'1 ByteString
bytes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, TensorList Build output_types)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, TensorList Build output_types))
-> Build OpDef
-> (Tensor Build Int32, TensorList Build output_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
bytes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeProtoV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "message_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
message_type
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "bytes" type: DT_STRING }
output_arg { name: "sizes" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "output_types" }
attr { name: "message_type" type: "string" }
attr { name: "field_names" type: "list(string)" }
attr { name: "output_types" type: "list(type)" has_minimum: true }
attr {
  name: "descriptor_source"
  type: "string"
  default_value { s: "local://" }
}
attr {
  name: "message_format" type: "string" default_value { s: "binary" }
}
attr { name: "sanitize" type: "bool" default_value { b: false } }
-}
-- | 
decodeRaw :: forall v'1 out_type . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float), Bool,
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word8,
                                            Double, Float] out_type) =>
             Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
             -> Tensor Build out_type -- ^ __output__
decodeRaw :: Tensor v'1 ByteString -> Tensor Build out_type
decodeRaw = OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
forall (v'1 :: * -> *) out_type.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Word16, Word8, Double, Float]
  out_type =>
OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
decodeRaw' OpParams
forall a. a -> a
id
decodeRaw' :: forall v'1 out_type . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float), Bool,
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word8,
                                             Double, Float] out_type) =>
              OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __bytes__
              -> Tensor Build out_type -- ^ __output__
decodeRaw' :: OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
decodeRaw' op'options :: OpParams
op'options bytes :: Tensor v'1 ByteString
bytes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
bytes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeRaw"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "bytes" type: DT_STRING }
output_arg { name: "output" type_attr: "out_type" }
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT16
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_BOOL
    }
  }
}
attr {
  name: "little_endian" type: "bool" default_value { b: true }
}
-}
-- | 
decodeWav :: Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
             -> (Tensor Build Float, Tensor Build Data.Int.Int32)
             -- ^ (__audio__, __sample_rate__)
             --
             -- * __audio__
             --
             -- * __sample_rate__
decodeWav :: Tensor v'1 ByteString -> (Tensor Build Float, Tensor Build Int32)
decodeWav = OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Float, Tensor Build Int32)
forall (v'1 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Float, Tensor Build Int32)
decodeWav' OpParams
forall a. a -> a
id
decodeWav' :: OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
              -> (Tensor Build Float, Tensor Build Data.Int.Int32)
              -- ^ (__audio__, __sample_rate__)
              --
              -- * __audio__
              --
              -- * __sample_rate__
decodeWav' :: OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Float, Tensor Build Int32)
decodeWav' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Float, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Float, Tensor Build Int32))
-> Build OpDef -> (Tensor Build Float, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DecodeWav"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "audio" type: DT_FLOAT }
output_arg { name: "sample_rate" type: DT_INT32 }
attr {
  name: "desired_channels" type: "int" default_value { i: -1 }
}
attr {
  name: "desired_samples" type: "int" default_value { i: -1 }
}
-}
-- | 
deepCopy :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
            Tensor v'1 t -- ^ __x__
            -> m' (Tensor Value t) -- ^ __y__
deepCopy :: Tensor v'1 t -> m' (Tensor Value t)
deepCopy = OpParams -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value t)
deepCopy' OpParams
forall a. a -> a
id
deepCopy' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
             Tensor v'1 t -- ^ __x__
             -> m' (Tensor Value t) -- ^ __y__
deepCopy' :: OpParams -> Tensor v'1 t -> m' (Tensor Value t)
deepCopy' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeepCopy"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
deleteIterator :: forall v'1 v'2 m' . (MonadBuild m') =>
                  Tensor v'1 ResourceHandle -- ^ __handle__
                  -> Tensor v'2 Variant -- ^ __deleter__
                  -> m' (ControlNode)
deleteIterator :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteIterator = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteIterator' OpParams
forall a. a -> a
id
deleteIterator' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                   Tensor v'1 ResourceHandle -- ^ __handle__
                   -> Tensor v'2 Variant -- ^ __deleter__
                   -> m' (ControlNode)
deleteIterator' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteIterator' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle deleter :: Tensor v'2 Variant
deleter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
deleter]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
deleteMemoryCache :: forall v'1 v'2 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __handle__
                     -> Tensor v'2 Variant -- ^ __deleter__
                     -> m' (ControlNode)
deleteMemoryCache :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteMemoryCache = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteMemoryCache' OpParams
forall a. a -> a
id
deleteMemoryCache' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Variant -- ^ __deleter__
                      -> m' (ControlNode)
deleteMemoryCache' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteMemoryCache' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle deleter :: Tensor v'2 Variant
deleter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
deleter]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteMemoryCache"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
deleteMultiDeviceIterator :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                             Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                             -> [Tensor v'2 ResourceHandle] -- ^ __iterators__
                             -> Tensor v'3 Variant -- ^ __deleter__
                             -> m' (ControlNode)
deleteMultiDeviceIterator :: Tensor v'1 ResourceHandle
-> [Tensor v'2 ResourceHandle]
-> Tensor v'3 Variant
-> m' ControlNode
deleteMultiDeviceIterator = OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 ResourceHandle]
-> Tensor v'3 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 ResourceHandle]
-> Tensor v'3 Variant
-> m' ControlNode
deleteMultiDeviceIterator' OpParams
forall a. a -> a
id
deleteMultiDeviceIterator' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                              OpParams ->
                              Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                              -> [Tensor v'2 ResourceHandle] -- ^ __iterators__
                              -> Tensor v'3 Variant -- ^ __deleter__
                              -> m' (ControlNode)
deleteMultiDeviceIterator' :: OpParams
-> Tensor v'1 ResourceHandle
-> [Tensor v'2 ResourceHandle]
-> Tensor v'3 Variant
-> m' ControlNode
deleteMultiDeviceIterator' op'options :: OpParams
op'options multi_device_iterator :: Tensor v'1 ResourceHandle
multi_device_iterator iterators :: [Tensor v'2 ResourceHandle]
iterators
                           deleter :: Tensor v'3 Variant
deleter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("iterators", [Tensor v'2 ResourceHandle] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 ResourceHandle]
iterators)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
multi_device_iterator,
                                                             [Tensor v'2 ResourceHandle] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 ResourceHandle]
iterators,
                                                             Tensor v'3 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Variant
deleter]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteMultiDeviceIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 ResourceHandle] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 ResourceHandle]
iterators) :: Int64
{-
input_arg { name: "multi_device_iterator" type: DT_RESOURCE }
input_arg { name: "iterators" type: DT_RESOURCE number_attr: "N" }
input_arg { name: "deleter" type: DT_VARIANT }
attr { name: "N" type: "int" has_minimum: true }
-}
-- | 
deleteRandomSeedGenerator :: forall v'1 v'2 m' . (MonadBuild m') =>
                             Tensor v'1 ResourceHandle -- ^ __handle__
                             -> Tensor v'2 Variant -- ^ __deleter__
                             -> m' (ControlNode)
deleteRandomSeedGenerator :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteRandomSeedGenerator = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteRandomSeedGenerator' OpParams
forall a. a -> a
id
deleteRandomSeedGenerator' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                              Tensor v'1 ResourceHandle -- ^ __handle__
                              -> Tensor v'2 Variant -- ^ __deleter__
                              -> m' (ControlNode)
deleteRandomSeedGenerator' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteRandomSeedGenerator' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle deleter :: Tensor v'2 Variant
deleter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
deleter]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteRandomSeedGenerator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
deleteSeedGenerator :: forall v'1 v'2 m' . (MonadBuild m') =>
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Variant -- ^ __deleter__
                       -> m' (ControlNode)
deleteSeedGenerator :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deleteSeedGenerator = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteSeedGenerator' OpParams
forall a. a -> a
id
deleteSeedGenerator' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __handle__
                        -> Tensor v'2 Variant -- ^ __deleter__
                        -> m' (ControlNode)
deleteSeedGenerator' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deleteSeedGenerator' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle deleter :: Tensor v'2 Variant
deleter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
deleter]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteSeedGenerator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "deleter" type: DT_VARIANT }
-}
-- | 
deleteSessionTensor :: forall v'1 m' . (MonadBuild m') =>
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> m' (ControlNode)
deleteSessionTensor :: Tensor v'1 ByteString -> m' ControlNode
deleteSessionTensor = OpParams -> Tensor v'1 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' ControlNode
deleteSessionTensor' OpParams
forall a. a -> a
id
deleteSessionTensor' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                        Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                        -> m' (ControlNode)
deleteSessionTensor' :: OpParams -> Tensor v'1 ByteString -> m' ControlNode
deleteSessionTensor' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeleteSessionTensor"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
-}
-- | 
denseBincount :: forall v'1 v'2 v'3 tidx t . (OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tidx,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64, Double,
                                                      Float] t) =>
                 Tensor v'1 tidx -- ^ __input__
                 -> Tensor v'2 tidx -- ^ __size__
                 -> Tensor v'3 t -- ^ __weights__
                 -> Tensor Build t -- ^ __output__
denseBincount :: Tensor v'1 tidx
-> Tensor v'2 tidx -> Tensor v'3 t -> Tensor Build t
denseBincount = OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tidx t.
(OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64, Double, Float] t) =>
OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 t
-> Tensor Build t
denseBincount' OpParams
forall a. a -> a
id
denseBincount' :: forall v'1 v'2 v'3 tidx t . (OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tidx,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64, Double,
                                                       Float] t) => OpParams ->
                  Tensor v'1 tidx -- ^ __input__
                  -> Tensor v'2 tidx -- ^ __size__
                  -> Tensor v'3 t -- ^ __weights__
                  -> Tensor Build t -- ^ __output__
denseBincount' :: OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 t
-> Tensor Build t
denseBincount' op'options :: OpParams
op'options input :: Tensor v'1 tidx
input size :: Tensor v'2 tidx
size weights :: Tensor v'3 t
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tidx
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
size,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseBincount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tidx" }
input_arg { name: "size" type_attr: "Tidx" }
input_arg { name: "weights" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "Tidx"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "binary_output" type: "bool" default_value { b: false }
}
-}
-- | 
denseCountSparseOutput :: forall v'1 v'2 t
                          output_type . (OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] t,
                                         OneOf '[Data.Int.Int32, Data.Int.Int64,
                                                 Double, Float] output_type) =>
                          Bool -- ^ __binary_output__
                          -> Tensor v'1 t -- ^ __values__
                          -> Tensor v'2 output_type -- ^ __weights__
                          -> (Tensor Build Data.Int.Int64,
                              Tensor Build output_type,
                              Tensor Build Data.Int.Int64)
                          -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                          --
                          -- * __output_indices__
                          --
                          -- * __output_values__
                          --
                          -- * __output_dense_shape__
denseCountSparseOutput :: Bool
-> Tensor v'1 t
-> Tensor v'2 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
denseCountSparseOutput = OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) t output_type.
(OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64, Double, Float] output_type) =>
OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
denseCountSparseOutput' OpParams
forall a. a -> a
id
denseCountSparseOutput' :: forall v'1 v'2 t
                           output_type . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64, Double,
                                                  Float] output_type) =>
                           OpParams ->
                           Bool -- ^ __binary_output__
                           -> Tensor v'1 t -- ^ __values__
                           -> Tensor v'2 output_type -- ^ __weights__
                           -> (Tensor Build Data.Int.Int64,
                               Tensor Build output_type,
                               Tensor Build Data.Int.Int64)
                           -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                           --
                           -- * __output_indices__
                           --
                           -- * __output_values__
                           --
                           -- * __output_dense_shape__
denseCountSparseOutput' :: OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
denseCountSparseOutput' op'options :: OpParams
op'options binary_output :: Bool
binary_output values :: Tensor v'1 t
values
                        weights :: Tensor v'2 output_type
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build output_type,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
values,
                                                             Tensor v'2 output_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 output_type
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseCountSparseOutput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "binary_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
binary_output
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "values" type_attr: "T" }
input_arg { name: "weights" type_attr: "output_type" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "output_type" }
output_arg { name: "output_dense_shape" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "minlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr {
  name: "maxlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr { name: "binary_output" type: "bool" }
attr {
  name: "output_type"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
denseToCSRSparseMatrix :: forall v'1 v'2
                          t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float), Double,
                                       Float] t) =>
                          Tensor v'1 t -- ^ __dense_input__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __indices__
                          -> Tensor Build Variant -- ^ __sparse_output__
denseToCSRSparseMatrix :: Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
denseToCSRSparseMatrix = OpParams
-> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
denseToCSRSparseMatrix' OpParams
forall a. a -> a
id
denseToCSRSparseMatrix' :: forall v'1 v'2
                           t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float), Double,
                                        Float] t) => OpParams ->
                           Tensor v'1 t -- ^ __dense_input__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __indices__
                           -> Tensor Build Variant -- ^ __sparse_output__
denseToCSRSparseMatrix' :: OpParams
-> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
denseToCSRSparseMatrix' op'options :: OpParams
op'options dense_input :: Tensor v'1 t
dense_input indices :: Tensor v'2 Int64
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
dense_input,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseToCSRSparseMatrix"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dense_input" type_attr: "T" }
input_arg { name: "indices" type: DT_INT64 }
output_arg { name: "sparse_output" type: DT_VARIANT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
denseToDenseSetOperation :: forall v'1 v'2
                            t . (OneOf '[Data.ByteString.ByteString,
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16,
                                         Data.Word.Word8] t) =>
                            ByteString -- ^ __set_operation__
                            -> Tensor v'1 t -- ^ __set1__
                            -> Tensor v'2 t -- ^ __set2__
                            -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                Tensor Build Data.Int.Int64)
                            -- ^ (__result_indices__, __result_values__, __result_shape__)
                            --
                            -- * __result_indices__
                            --
                            -- * __result_values__
                            --
                            -- * __result_shape__
denseToDenseSetOperation :: ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToDenseSetOperation = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToDenseSetOperation' OpParams
forall a. a -> a
id
denseToDenseSetOperation' :: forall v'1 v'2
                             t . (OneOf '[Data.ByteString.ByteString,
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16,
                                          Data.Word.Word8] t) => OpParams ->
                             ByteString -- ^ __set_operation__
                             -> Tensor v'1 t -- ^ __set1__
                             -> Tensor v'2 t -- ^ __set2__
                             -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                 Tensor Build Data.Int.Int64)
                             -- ^ (__result_indices__, __result_values__, __result_shape__)
                             --
                             -- * __result_indices__
                             --
                             -- * __result_values__
                             --
                             -- * __result_shape__
denseToDenseSetOperation' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToDenseSetOperation' op'options :: OpParams
op'options set_operation :: ByteString
set_operation set1 :: Tensor v'1 t
set1
                          set2 :: Tensor v'2 t
set2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
set1,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
set2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseToDenseSetOperation"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "set_operation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
set_operation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "set1" type_attr: "T" }
input_arg { name: "set2" type_attr: "T" }
output_arg { name: "result_indices" type: DT_INT64 }
output_arg { name: "result_values" type_attr: "T" }
output_arg { name: "result_shape" type: DT_INT64 }
attr { name: "set_operation" type: "string" }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_STRING
    }
  }
}
-}
-- | 
denseToSparseBatchDataset :: [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                             -> Tensor v'3 Data.Int.Int64 -- ^ __row_shape__
                             -> Tensor Build Variant -- ^ __handle__
denseToSparseBatchDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
denseToSparseBatchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
denseToSparseBatchDataset' OpParams
forall a. a -> a
id
denseToSparseBatchDataset' :: OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                              -> Tensor v'3 Data.Int.Int64 -- ^ __row_shape__
                              -> Tensor Build Variant -- ^ __handle__
denseToSparseBatchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
denseToSparseBatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset batch_size :: Tensor v'2 Int64
batch_size
                           row_shape :: Tensor v'3 Int64
row_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
row_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseToSparseBatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
input_arg { name: "row_shape" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
denseToSparseSetOperation :: forall v'1 v'2 v'3 v'4
                             t . (OneOf '[Data.ByteString.ByteString,
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16,
                                          Data.Word.Word8] t) =>
                             ByteString -- ^ __set_operation__
                             -> Tensor v'1 t -- ^ __set1__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __set2_indices__
                             -> Tensor v'3 t -- ^ __set2_values__
                             -> Tensor v'4 Data.Int.Int64 -- ^ __set2_shape__
                             -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                 Tensor Build Data.Int.Int64)
                             -- ^ (__result_indices__, __result_values__, __result_shape__)
                             --
                             -- * __result_indices__
                             --
                             -- * __result_values__
                             --
                             -- * __result_shape__
denseToSparseSetOperation :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToSparseSetOperation = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToSparseSetOperation' OpParams
forall a. a -> a
id
denseToSparseSetOperation' :: forall v'1 v'2 v'3 v'4
                              t . (OneOf '[Data.ByteString.ByteString,
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16,
                                           Data.Word.Word8] t) => OpParams ->
                              ByteString -- ^ __set_operation__
                              -> Tensor v'1 t -- ^ __set1__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __set2_indices__
                              -> Tensor v'3 t -- ^ __set2_values__
                              -> Tensor v'4 Data.Int.Int64 -- ^ __set2_shape__
                              -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                  Tensor Build Data.Int.Int64)
                              -- ^ (__result_indices__, __result_values__, __result_shape__)
                              --
                              -- * __result_indices__
                              --
                              -- * __result_values__
                              --
                              -- * __result_shape__
denseToSparseSetOperation' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
denseToSparseSetOperation' op'options :: OpParams
op'options set_operation :: ByteString
set_operation set1 :: Tensor v'1 t
set1 set2_indices :: Tensor v'2 Int64
set2_indices
                           set2_values :: Tensor v'3 t
set2_values set2_shape :: Tensor v'4 Int64
set2_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
set1,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
set2_indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
set2_values,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
set2_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DenseToSparseSetOperation"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "set_operation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
set_operation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "set1" type_attr: "T" }
input_arg { name: "set2_indices" type: DT_INT64 }
input_arg { name: "set2_values" type_attr: "T" }
input_arg { name: "set2_shape" type: DT_INT64 }
output_arg { name: "result_indices" type: DT_INT64 }
output_arg { name: "result_values" type_attr: "T" }
output_arg { name: "result_shape" type: DT_INT64 }
attr { name: "set_operation" type: "string" }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_STRING
    }
  }
}
-}
-- | 
depthToSpace :: forall v'1 t . (TensorType t) =>
                Data.Int.Int64 -- ^ __block_size__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor Build t -- ^ __output__
depthToSpace :: Int64 -> Tensor v'1 t -> Tensor Build t
depthToSpace = OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
depthToSpace' OpParams
forall a. a -> a
id
depthToSpace' :: forall v'1 t . (TensorType t) => OpParams ->
                 Data.Int.Int64 -- ^ __block_size__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
depthToSpace' :: OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
depthToSpace' op'options :: OpParams
op'options block_size :: Int64
block_size input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DepthToSpace"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "block_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
block_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "block_size" type: "int" has_minimum: true minimum: 2
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" s: "NCHW_VECT_C" } }
}
-}
-- | 
depthwiseConv2dNative :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                                     Float] t) =>
                         ByteString -- ^ __padding__
                         -> Tensor v'1 t -- ^ __input__
                         -> Tensor v'2 t -- ^ __filter__
                         -> Tensor Build t -- ^ __output__
depthwiseConv2dNative :: ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
depthwiseConv2dNative = OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
depthwiseConv2dNative' OpParams
forall a. a -> a
id
depthwiseConv2dNative' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                                      Float] t) => OpParams ->
                          ByteString -- ^ __padding__
                          -> Tensor v'1 t -- ^ __input__
                          -> Tensor v'2 t -- ^ __filter__
                          -> Tensor Build t -- ^ __output__
depthwiseConv2dNative' :: OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
depthwiseConv2dNative' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DepthwiseConv2dNative"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
depthwiseConv2dNativeBackpropFilter :: forall v'1 v'2 v'3
                                       t . (OneOf '[Data.Word.Word16, Double,
                                                    Float] t) =>
                                       ByteString -- ^ __padding__
                                       -> Tensor v'1 t -- ^ __input__
                                       -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                                       -> Tensor v'3 t -- ^ __out_backprop__
                                       -> Tensor Build t -- ^ __output__
depthwiseConv2dNativeBackpropFilter :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropFilter = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropFilter' OpParams
forall a. a -> a
id
depthwiseConv2dNativeBackpropFilter' :: forall v'1 v'2 v'3
                                        t . (OneOf '[Data.Word.Word16, Double,
                                                     Float] t) => OpParams ->
                                        ByteString -- ^ __padding__
                                        -> Tensor v'1 t -- ^ __input__
                                        -> Tensor v'2 Data.Int.Int32 -- ^ __filter_sizes__
                                        -> Tensor v'3 t -- ^ __out_backprop__
                                        -> Tensor Build t -- ^ __output__
depthwiseConv2dNativeBackpropFilter' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropFilter' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter_sizes :: Tensor v'2 Int32
filter_sizes
                                     out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
filter_sizes,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DepthwiseConv2dNativeBackpropFilter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter_sizes" type: DT_INT32 }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
depthwiseConv2dNativeBackpropInput :: forall v'1 v'2 v'3
                                      t . (OneOf '[Data.Word.Word16, Double,
                                                   Float] t) =>
                                      ByteString -- ^ __padding__
                                      -> Tensor v'1 Data.Int.Int32 -- ^ __input_sizes__
                                      -> Tensor v'2 t -- ^ __filter__
                                      -> Tensor v'3 t -- ^ __out_backprop__
                                      -> Tensor Build t -- ^ __output__
depthwiseConv2dNativeBackpropInput :: ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropInput = OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropInput' OpParams
forall a. a -> a
id
depthwiseConv2dNativeBackpropInput' :: forall v'1 v'2 v'3
                                       t . (OneOf '[Data.Word.Word16, Double,
                                                    Float] t) => OpParams ->
                                       ByteString -- ^ __padding__
                                       -> Tensor v'1 Data.Int.Int32 -- ^ __input_sizes__
                                       -> Tensor v'2 t -- ^ __filter__
                                       -> Tensor v'3 t -- ^ __out_backprop__
                                       -> Tensor Build t -- ^ __output__
depthwiseConv2dNativeBackpropInput' :: OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
depthwiseConv2dNativeBackpropInput' op'options :: OpParams
op'options padding :: ByteString
padding input_sizes :: Tensor v'1 Int32
input_sizes filter :: Tensor v'2 t
filter
                                    out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
input_sizes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DepthwiseConv2dNativeBackpropInput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_sizes" type: DT_INT32 }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
dequantize :: forall v'1 v'2 v'3 t dtype . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Word.Word16,
                                                    Data.Word.Word8] t,
                                            OneOf '[Data.Word.Word16,
                                                    Float] dtype) =>
              Tensor v'1 t -- ^ __input__
              -> Tensor v'2 Float -- ^ __min_range__
              -> Tensor v'3 Float -- ^ __max_range__
              -> Tensor Build dtype -- ^ __output__
dequantize :: Tensor v'1 t
-> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build dtype
dequantize = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t dtype.
(OneOf '[Int16, Int32, Word16, Word8] t,
 OneOf '[Word16, Float] dtype) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build dtype
dequantize' OpParams
forall a. a -> a
id
dequantize' :: forall v'1 v'2 v'3 t dtype . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] t,
                                             OneOf '[Data.Word.Word16,
                                                     Float] dtype) =>
               OpParams ->
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 Float -- ^ __min_range__
               -> Tensor v'3 Float -- ^ __max_range__
               -> Tensor Build dtype -- ^ __output__
dequantize' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build dtype
dequantize' op'options :: OpParams
op'options input :: Tensor v'1 t
input min_range :: Tensor v'2 Float
min_range max_range :: Tensor v'3 Float
max_range | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_range,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_range]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Dequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "min_range" type: DT_FLOAT }
input_arg { name: "max_range" type: DT_FLOAT }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "mode"
  type: "string"
  default_value { s: "MIN_COMBINED" }
  allowed_values {
    list { s: "MIN_COMBINED" s: "MIN_FIRST" s: "SCALED" }
  }
}
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
attr { name: "axis" type: "int" default_value { i: -1 } }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_BFLOAT16 type: DT_FLOAT } }
}
-}
-- | 
deserializeIterator :: forall v'1 v'2 m' . (MonadBuild m') =>
                       Tensor v'1 ResourceHandle -- ^ __resource_handle__
                       -> Tensor v'2 Variant -- ^ __serialized__
                       -> m' (ControlNode)
deserializeIterator :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
deserializeIterator = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deserializeIterator' OpParams
forall a. a -> a
id
deserializeIterator' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __resource_handle__
                        -> Tensor v'2 Variant -- ^ __serialized__
                        -> m' (ControlNode)
deserializeIterator' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
deserializeIterator' op'options :: OpParams
op'options resource_handle :: Tensor v'1 ResourceHandle
resource_handle serialized :: Tensor v'2 Variant
serialized | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource_handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
serialized]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DeserializeIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource_handle" type: DT_RESOURCE }
input_arg { name: "serialized" type: DT_VARIANT }
-}
-- | 
deserializeManySparse :: forall v'1 dtype . (TensorType dtype) =>
                         Tensor v'1 Data.ByteString.ByteString -- ^ __serialized_sparse__
                         -> (Tensor Build Data.Int.Int64, Tensor Build dtype,
                             Tensor Build Data.Int.Int64)
                         -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                         --
                         -- * __sparse_indices__
                         --
                         -- * __sparse_values__
                         --
                         -- * __sparse_shape__
deserializeManySparse :: Tensor v'1 ByteString
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeManySparse = OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall (v'1 :: * -> *) dtype.
TensorType dtype =>
OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeManySparse' OpParams
forall a. a -> a
id
deserializeManySparse' :: forall v'1 dtype . (TensorType dtype) => OpParams ->
                          Tensor v'1 Data.ByteString.ByteString -- ^ __serialized_sparse__
                          -> (Tensor Build Data.Int.Int64, Tensor Build dtype,
                              Tensor Build Data.Int.Int64)
                          -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                          --
                          -- * __sparse_indices__
                          --
                          -- * __sparse_values__
                          --
                          -- * __sparse_shape__
deserializeManySparse' :: OpParams
-> Tensor v'1 ByteString
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeManySparse' op'options :: OpParams
op'options serialized_sparse :: Tensor v'1 ByteString
serialized_sparse | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
serialized_sparse]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DeserializeManySparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "serialized_sparse" type: DT_STRING }
output_arg { name: "sparse_indices" type: DT_INT64 }
output_arg { name: "sparse_values" type_attr: "dtype" }
output_arg { name: "sparse_shape" type: DT_INT64 }
attr { name: "dtype" type: "type" }
-}
-- | 
deserializeSparse :: forall v'1 dtype tserialized . (TensorType dtype,
                                                     OneOf '[Data.ByteString.ByteString,
                                                             Variant] tserialized) =>
                     Tensor v'1 tserialized -- ^ __serialized_sparse__
                     -> (Tensor Build Data.Int.Int64, Tensor Build dtype,
                         Tensor Build Data.Int.Int64)
                     -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                     --
                     -- * __sparse_indices__
                     --
                     -- * __sparse_values__
                     --
                     -- * __sparse_shape__
deserializeSparse :: Tensor v'1 tserialized
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeSparse = OpParams
-> Tensor v'1 tserialized
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall (v'1 :: * -> *) dtype tserialized.
(TensorType dtype, OneOf '[ByteString, Variant] tserialized) =>
OpParams
-> Tensor v'1 tserialized
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeSparse' OpParams
forall a. a -> a
id
deserializeSparse' :: forall v'1 dtype tserialized . (TensorType dtype,
                                                      OneOf '[Data.ByteString.ByteString,
                                                              Variant] tserialized) =>
                      OpParams ->
                      Tensor v'1 tserialized -- ^ __serialized_sparse__
                      -> (Tensor Build Data.Int.Int64, Tensor Build dtype,
                          Tensor Build Data.Int.Int64)
                      -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                      --
                      -- * __sparse_indices__
                      --
                      -- * __sparse_values__
                      --
                      -- * __sparse_shape__
deserializeSparse' :: OpParams
-> Tensor v'1 tserialized
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
deserializeSparse' op'options :: OpParams
op'options serialized_sparse :: Tensor v'1 tserialized
serialized_sparse | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tserialized -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tserialized
serialized_sparse]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DeserializeSparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tserialized" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tserialized -> DataType
forall a. TensorType a => a -> DataType
tensorType (tserialized
forall a. HasCallStack => a
undefined :: tserialized)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "serialized_sparse" type_attr: "Tserialized" }
output_arg { name: "sparse_indices" type: DT_INT64 }
output_arg { name: "sparse_values" type_attr: "dtype" }
output_arg { name: "sparse_shape" type: DT_INT64 }
attr { name: "dtype" type: "type" }
attr {
  name: "Tserialized"
  type: "type"
  default_value { type: DT_STRING }
  allowed_values { list { type: DT_STRING type: DT_VARIANT } }
}
-}
-- | 
destroyResourceOp :: forall v'1 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __resource__
                     -> m' (ControlNode)
destroyResourceOp :: Tensor v'1 ResourceHandle -> m' ControlNode
destroyResourceOp = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
destroyResourceOp' OpParams
forall a. a -> a
id
destroyResourceOp' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> m' (ControlNode)
destroyResourceOp' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
destroyResourceOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DestroyResourceOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
attr {
  name: "ignore_lookup_error" type: "bool" default_value { b: true }
}
-}
-- | 
destroyTemporaryVariable :: forall t m' . (MonadBuild m', TensorType t) =>
                            ByteString -- ^ __var_name__
                            -> Tensor Ref t -- ^ __ref__
                            -> m' (Tensor Value t) -- ^ __value__
destroyTemporaryVariable :: ByteString -> Tensor Ref t -> m' (Tensor Value t)
destroyTemporaryVariable = OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Value t)
destroyTemporaryVariable' OpParams
forall a. a -> a
id
destroyTemporaryVariable' :: forall t m' . (MonadBuild m', TensorType t) =>
                             OpParams ->
                             ByteString -- ^ __var_name__
                             -> Tensor Ref t -- ^ __ref__
                             -> m' (Tensor Value t) -- ^ __value__
destroyTemporaryVariable' :: OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Value t)
destroyTemporaryVariable' op'options :: OpParams
op'options var_name :: ByteString
var_name ref :: Tensor Ref t
ref | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DestroyTemporaryVariable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "var_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
var_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
output_arg { name: "value" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "var_name" type: "string" }
-}
-- | 
deviceIndex :: Tensor Build Data.Int.Int32 -- ^ __index__
deviceIndex :: Tensor Build Int32
deviceIndex = OpParams -> Tensor Build Int32
deviceIndex' OpParams
forall a. a -> a
id
deviceIndex' :: OpParams ->
                Tensor Build Data.Int.Int32 -- ^ __index__
deviceIndex' :: OpParams -> Tensor Build Int32
deviceIndex' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DeviceIndex"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "index" type: DT_INT32 }
attr { name: "device_names" type: "list(string)" }
-}
-- | 
diag :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int32,
                                Data.Int.Int64, Data.Word.Word16, Double,
                                Float] t) => Tensor v'1 t -- ^ __diagonal__
        -> Tensor Build t -- ^ __output__
diag :: Tensor v'1 t -> Tensor Build t
diag = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
diag' OpParams
forall a. a -> a
id
diag' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int32,
                                 Data.Int.Int64, Data.Word.Word16, Double,
                                 Float] t) => OpParams ->
         Tensor v'1 t -- ^ __diagonal__
         -> Tensor Build t -- ^ __output__
diag' :: OpParams -> Tensor v'1 t -> Tensor Build t
diag' op'options :: OpParams
op'options diagonal :: Tensor v'1 t
diagonal | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonal]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Diag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonal" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
diagPart :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int32, Data.Int.Int64,
                                    Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __diagonal__
diagPart :: Tensor v'1 t -> Tensor Build t
diagPart = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
diagPart' OpParams
forall a. a -> a
id
diagPart' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int32, Data.Int.Int64,
                                     Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __diagonal__
diagPart' :: OpParams -> Tensor v'1 t -> Tensor Build t
diagPart' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DiagPart"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "diagonal" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
digamma :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
digamma :: Tensor v'1 t -> Tensor Build t
digamma = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
digamma' OpParams
forall a. a -> a
id
digamma' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
digamma' :: OpParams -> Tensor v'1 t -> Tensor Build t
digamma' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Digamma"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
dilation2D :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) =>
              ByteString -- ^ __padding__
              -> Tensor v'1 t -- ^ __input__
              -> Tensor v'2 t -- ^ __filter__
              -> Tensor Build t -- ^ __output__
dilation2D :: ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
dilation2D = OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
dilation2D' OpParams
forall a. a -> a
id
dilation2D' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t) => OpParams ->
               ByteString -- ^ __padding__
               -> Tensor v'1 t -- ^ __input__
               -> Tensor v'2 t -- ^ __filter__
               -> Tensor Build t -- ^ __output__
dilation2D' :: OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
dilation2D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Dilation2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "rates" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
dilation2DBackpropFilter :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word16,
                                                            Data.Word.Word32,
                                                            Data.Word.Word64,
                                                            Data.Word.Word8,
                                                            Double, Float] t) =>
                            ByteString -- ^ __padding__
                            -> Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 t -- ^ __filter__
                            -> Tensor v'3 t -- ^ __out_backprop__
                            -> Tensor Build t -- ^ __filter_backprop__
dilation2DBackpropFilter :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
dilation2DBackpropFilter = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
dilation2DBackpropFilter' OpParams
forall a. a -> a
id
dilation2DBackpropFilter' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                             OpParams ->
                             ByteString -- ^ __padding__
                             -> Tensor v'1 t -- ^ __input__
                             -> Tensor v'2 t -- ^ __filter__
                             -> Tensor v'3 t -- ^ __out_backprop__
                             -> Tensor Build t -- ^ __filter_backprop__
dilation2DBackpropFilter' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
dilation2DBackpropFilter' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
                          out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Dilation2DBackpropFilter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "filter_backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "rates" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
dilation2DBackpropInput :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double, Float] t) =>
                           ByteString -- ^ __padding__
                           -> Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 t -- ^ __filter__
                           -> Tensor v'3 t -- ^ __out_backprop__
                           -> Tensor Build t -- ^ __in_backprop__
dilation2DBackpropInput :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
dilation2DBackpropInput = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
dilation2DBackpropInput' OpParams
forall a. a -> a
id
dilation2DBackpropInput' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word16,
                                                            Data.Word.Word32,
                                                            Data.Word.Word64,
                                                            Data.Word.Word8,
                                                            Double, Float] t) =>
                            OpParams ->
                            ByteString -- ^ __padding__
                            -> Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 t -- ^ __filter__
                            -> Tensor v'3 t -- ^ __out_backprop__
                            -> Tensor Build t -- ^ __in_backprop__
dilation2DBackpropInput' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
dilation2DBackpropInput' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
                         out_backprop :: Tensor v'3 t
out_backprop | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Dilation2DBackpropInput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
output_arg { name: "in_backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "rates" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
directedInterleaveDataset :: [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __selector_input_dataset__
                             -> [Tensor v'2 Variant] -- ^ __data_input_datasets__
                             -> Tensor Build Variant -- ^ __handle__
directedInterleaveDataset :: [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
directedInterleaveDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
directedInterleaveDataset' OpParams
forall a. a -> a
id
directedInterleaveDataset' :: OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __selector_input_dataset__
                              -> [Tensor v'2 Variant] -- ^ __data_input_datasets__
                              -> Tensor Build Variant -- ^ __handle__
directedInterleaveDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
directedInterleaveDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types selector_input_dataset :: Tensor v'1 Variant
selector_input_dataset
                           data_input_datasets :: [Tensor v'2 Variant]
data_input_datasets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("data_input_datasets", [Tensor v'2 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Variant]
data_input_datasets)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
selector_input_dataset,
                                                             [Tensor v'2 Variant] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Variant]
data_input_datasets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DirectedInterleaveDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Variant]
data_input_datasets) :: Int64
{-
input_arg { name: "selector_input_dataset" type: DT_VARIANT }
input_arg {
  name: "data_input_datasets" type: DT_VARIANT number_attr: "N"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
div :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word8, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
div :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
div = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
div' OpParams
forall a. a -> a
id
div' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word8, Double,
                                    Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
div' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
div' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Div"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
divNoNan :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
divNoNan :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
divNoNan = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
divNoNan' OpParams
forall a. a -> a
id
divNoNan' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
divNoNan' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
divNoNan' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DivNoNan"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
drawBoundingBoxes :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                     Tensor v'1 t -- ^ __images__
                     -> Tensor v'2 Float -- ^ __boxes__
                     -> Tensor Build t -- ^ __output__
drawBoundingBoxes :: Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
drawBoundingBoxes = OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
drawBoundingBoxes' OpParams
forall a. a -> a
id
drawBoundingBoxes' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Float] t) =>
                      OpParams ->
                      Tensor v'1 t -- ^ __images__
                      -> Tensor v'2 Float -- ^ __boxes__
                      -> Tensor Build t -- ^ __output__
drawBoundingBoxes' :: OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
drawBoundingBoxes' op'options :: OpParams
op'options images :: Tensor v'1 t
images boxes :: Tensor v'2 Float
boxes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
boxes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DrawBoundingBoxes"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "boxes" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_HALF } }
}
-}
-- | 
drawBoundingBoxesV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                       Float] t) =>
                       Tensor v'1 t -- ^ __images__
                       -> Tensor v'2 Float -- ^ __boxes__
                       -> Tensor v'3 Float -- ^ __colors__
                       -> Tensor Build t -- ^ __output__
drawBoundingBoxesV2 :: Tensor v'1 t
-> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build t
drawBoundingBoxesV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build t
drawBoundingBoxesV2' OpParams
forall a. a -> a
id
drawBoundingBoxesV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                        Float] t) => OpParams ->
                        Tensor v'1 t -- ^ __images__
                        -> Tensor v'2 Float -- ^ __boxes__
                        -> Tensor v'3 Float -- ^ __colors__
                        -> Tensor Build t -- ^ __output__
drawBoundingBoxesV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build t
drawBoundingBoxesV2' op'options :: OpParams
op'options images :: Tensor v'1 t
images boxes :: Tensor v'2 Float
boxes colors :: Tensor v'3 Float
colors | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
boxes,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
colors]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DrawBoundingBoxesV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "colors" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_HALF } }
}
-}
-- | 
dummyIterationCounter :: forall m' . (MonadBuild m') =>
                         m' (Tensor Value ResourceHandle) -- ^ __handle__
dummyIterationCounter :: m' (Tensor Value ResourceHandle)
dummyIterationCounter = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
dummyIterationCounter' OpParams
forall a. a -> a
id
dummyIterationCounter' :: forall m' . (MonadBuild m') => OpParams ->
                          m' (Tensor Value ResourceHandle) -- ^ __handle__
dummyIterationCounter' :: OpParams -> m' (Tensor Value ResourceHandle)
dummyIterationCounter' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DummyIterationCounter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
-}
-- | 
dummyMemoryCache :: forall m' . (MonadBuild m') =>
                    m' (Tensor Value ResourceHandle) -- ^ __handle__
dummyMemoryCache :: m' (Tensor Value ResourceHandle)
dummyMemoryCache = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
dummyMemoryCache' OpParams
forall a. a -> a
id
dummyMemoryCache' :: forall m' . (MonadBuild m') => OpParams ->
                     m' (Tensor Value ResourceHandle) -- ^ __handle__
dummyMemoryCache' :: OpParams -> m' (Tensor Value ResourceHandle)
dummyMemoryCache' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DummyMemoryCache"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
-}
-- | 
dummySeedGenerator :: forall m' . (MonadBuild m') =>
                      m' (Tensor Value ResourceHandle) -- ^ __handle__
dummySeedGenerator :: m' (Tensor Value ResourceHandle)
dummySeedGenerator = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
dummySeedGenerator' OpParams
forall a. a -> a
id
dummySeedGenerator' :: forall m' . (MonadBuild m') => OpParams ->
                       m' (Tensor Value ResourceHandle) -- ^ __handle__
dummySeedGenerator' :: OpParams -> m' (Tensor Value ResourceHandle)
dummySeedGenerator' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "DummySeedGenerator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
-}
-- | 
dynamicPartition :: forall v'1 v'2 t . (TensorType t) =>
                    Data.Int.Int64 -- ^ __num_partitions__
                    -> Tensor v'1 t -- ^ __data__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __partitions__
                    -> [Tensor Build t] -- ^ __outputs__
dynamicPartition :: Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
dynamicPartition = OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
dynamicPartition' OpParams
forall a. a -> a
id
dynamicPartition' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                     Data.Int.Int64 -- ^ __num_partitions__
                     -> Tensor v'1 t -- ^ __data__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __partitions__
                     -> [Tensor Build t] -- ^ __outputs__
dynamicPartition' :: OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
dynamicPartition' op'options :: OpParams
op'options num_partitions :: Int64
num_partitions data' :: Tensor v'1 t
data'
                  partitions :: Tensor v'2 Int32
partitions | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_partitions] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
partitions]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DynamicPartition"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_partitions" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_partitions
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "partitions" type: DT_INT32 }
output_arg {
  name: "outputs" type_attr: "T" number_attr: "num_partitions"
}
attr {
  name: "num_partitions" type: "int" has_minimum: true minimum: 1
}
attr { name: "T" type: "type" }
-}
-- | 
dynamicStitch :: forall v'1 v'2 t . (TensorType t) =>
                 [Tensor v'1 Data.Int.Int32] -- ^ __indices__
                 -> [Tensor v'2 t] -- ^ __data__
                 -> Tensor Build t -- ^ __merged__
dynamicStitch :: [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
dynamicStitch = OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
dynamicStitch' OpParams
forall a. a -> a
id
dynamicStitch' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                  [Tensor v'1 Data.Int.Int32] -- ^ __indices__
                  -> [Tensor v'2 t] -- ^ __data__
                  -> Tensor Build t -- ^ __merged__
dynamicStitch' :: OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
dynamicStitch' op'options :: OpParams
op'options indices :: [Tensor v'1 Int32]
indices
               data' :: [Tensor v'2 t]
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
indices),
                                             ("data", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
data')])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int32]
indices,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
data']
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "DynamicStitch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT32 number_attr: "N" }
input_arg { name: "data" type_attr: "T" number_attr: "N" }
output_arg { name: "merged" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | 
eagerPyFunc :: forall v'1 tin tout m' . (MonadBuild m', TensorTypes tin,
                                         TensorTypes tout) =>
               ByteString -- ^ __token__
               -> TensorList (v'1) tin -- ^ __input__
               -> m' (TensorList (Value) tout) -- ^ __output__
eagerPyFunc :: ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
eagerPyFunc = OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
forall (v'1 :: * -> *) (tin :: [*]) (tout :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes tin, TensorTypes tout) =>
OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
eagerPyFunc' OpParams
forall a. a -> a
id
eagerPyFunc' :: forall v'1 tin tout m' . (MonadBuild m', TensorTypes tin,
                                          TensorTypes tout) => OpParams ->
                ByteString -- ^ __token__
                -> TensorList (v'1) tin -- ^ __input__
                -> m' (TensorList (Value) tout) -- ^ __output__
eagerPyFunc' :: OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
eagerPyFunc' op'options :: OpParams
op'options token :: ByteString
token input :: TensorList v'1 tin
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value tout) -> m' (TensorList Value tout)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value tout) -> m' (TensorList Value tout))
-> Build (TensorList Value tout) -> m' (TensorList Value tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tin
input]
        [Int64] -> OpDef -> Build (TensorList Value tout)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "EagerPyFunc"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tin -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tin
forall k (t :: k). Proxy t
Proxy :: Proxy tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tout -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tout
forall k (t :: k). Proxy t
Proxy :: Proxy tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "token" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
token
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_list_attr: "Tin" }
output_arg { name: "output" type_list_attr: "Tout" }
attr { name: "token" type: "string" }
attr { name: "is_async" type: "bool" default_value { b: false } }
attr { name: "Tin" type: "list(type)" has_minimum: true }
attr { name: "Tout" type: "list(type)" has_minimum: true }
-}
-- | 
editDistance :: forall v'1 v'2 v'3 v'4 v'5 v'6 t . (TensorType t) =>
                Tensor v'1 Data.Int.Int64 -- ^ __hypothesis_indices__
                -> Tensor v'2 t -- ^ __hypothesis_values__
                -> Tensor v'3 Data.Int.Int64 -- ^ __hypothesis_shape__
                -> Tensor v'4 Data.Int.Int64 -- ^ __truth_indices__
                -> Tensor v'5 t -- ^ __truth_values__
                -> Tensor v'6 Data.Int.Int64 -- ^ __truth_shape__
                -> Tensor Build Float -- ^ __output__
editDistance :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor Build Float
editDistance = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor Build Float
editDistance' OpParams
forall a. a -> a
id
editDistance' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t . (TensorType t) =>
                 OpParams ->
                 Tensor v'1 Data.Int.Int64 -- ^ __hypothesis_indices__
                 -> Tensor v'2 t -- ^ __hypothesis_values__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __hypothesis_shape__
                 -> Tensor v'4 Data.Int.Int64 -- ^ __truth_indices__
                 -> Tensor v'5 t -- ^ __truth_values__
                 -> Tensor v'6 Data.Int.Int64 -- ^ __truth_shape__
                 -> Tensor Build Float -- ^ __output__
editDistance' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor Build Float
editDistance' op'options :: OpParams
op'options hypothesis_indices :: Tensor v'1 Int64
hypothesis_indices hypothesis_values :: Tensor v'2 t
hypothesis_values hypothesis_shape :: Tensor v'3 Int64
hypothesis_shape
              truth_indices :: Tensor v'4 Int64
truth_indices truth_values :: Tensor v'5 t
truth_values truth_shape :: Tensor v'6 Int64
truth_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
hypothesis_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
hypothesis_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
hypothesis_shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
truth_indices,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
truth_values,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
truth_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EditDistance"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "hypothesis_indices" type: DT_INT64 }
input_arg { name: "hypothesis_values" type_attr: "T" }
input_arg { name: "hypothesis_shape" type: DT_INT64 }
input_arg { name: "truth_indices" type: DT_INT64 }
input_arg { name: "truth_values" type_attr: "T" }
input_arg { name: "truth_shape" type: DT_INT64 }
output_arg { name: "output" type: DT_FLOAT }
attr { name: "normalize" type: "bool" default_value { b: true } }
attr { name: "T" type: "type" }
-}
-- | 
eig :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float), Double,
                                    Float] t,
                            OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float)] tout) =>
       Tensor v'1 t -- ^ __input__
       -> (Tensor Build tout, Tensor Build tout) -- ^ (__e__, __v__)
       --
       -- * __e__
       --
       -- * __v__
eig :: Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
eig = OpParams -> Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
forall (v'1 :: * -> *) t tout.
(OneOf '[Complex Double, Complex Float, Double, Float] t,
 OneOf '[Complex Double, Complex Float] tout) =>
OpParams -> Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
eig' OpParams
forall a. a -> a
id
eig' :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float), Double,
                                     Float] t,
                             OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float)] tout) =>
        OpParams ->
        Tensor v'1 t -- ^ __input__
        -> (Tensor Build tout, Tensor Build tout) -- ^ (__e__, __v__)
        --
        -- * __e__
        --
        -- * __v__
eig' :: OpParams -> Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
eig' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build tout, Tensor Build tout)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build tout, Tensor Build tout))
-> Build OpDef -> (Tensor Build tout, Tensor Build tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Eig"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "e" type_attr: "Tout" }
output_arg { name: "v" type_attr: "Tout" }
attr { name: "compute_v" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr {
  name: "Tout"
  type: "type"
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
einsum :: forall v'1 t . (TensorType t) => ByteString -- ^ __equation__
          -> [Tensor v'1 t] -- ^ __inputs__
          -> Tensor Build t -- ^ __output__
einsum :: ByteString -> [Tensor v'1 t] -> Tensor Build t
einsum = OpParams -> ByteString -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> ByteString -> [Tensor v'1 t] -> Tensor Build t
einsum' OpParams
forall a. a -> a
id
einsum' :: forall v'1 t . (TensorType t) => OpParams ->
           ByteString -- ^ __equation__
           -> [Tensor v'1 t] -- ^ __inputs__
           -> Tensor Build t -- ^ __output__
einsum' :: OpParams -> ByteString -> [Tensor v'1 t] -> Tensor Build t
einsum' op'options :: OpParams
op'options equation :: ByteString
equation
        inputs :: [Tensor v'1 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Einsum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "equation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
equation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs) :: Int64
{-
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "equation" type: "string" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | 
elu :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __features__
       -> Tensor Build t -- ^ __activations__
elu :: Tensor v'1 t -> Tensor Build t
elu = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
elu' OpParams
forall a. a -> a
id
elu' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __features__
        -> Tensor Build t -- ^ __activations__
elu' :: OpParams -> Tensor v'1 t -> Tensor Build t
elu' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Elu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
eluGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __gradients__
           -> Tensor v'2 t -- ^ __outputs__
           -> Tensor Build t -- ^ __backprops__
eluGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
eluGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
eluGrad' OpParams
forall a. a -> a
id
eluGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __gradients__
            -> Tensor v'2 t -- ^ __outputs__
            -> Tensor Build t -- ^ __backprops__
eluGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
eluGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients outputs :: Tensor v'2 t
outputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
outputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EluGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "outputs" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
empty :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
         Tensor v'1 Data.Int.Int32 -- ^ __shape__
         -> m' (Tensor Value dtype) -- ^ __output__
empty :: Tensor v'1 Int32 -> m' (Tensor Value dtype)
empty = OpParams -> Tensor v'1 Int32 -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 Int32 -> m' (Tensor Value dtype)
empty' OpParams
forall a. a -> a
id
empty' :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) => OpParams ->
          Tensor v'1 Data.Int.Int32 -- ^ __shape__
          -> m' (Tensor Value dtype) -- ^ __output__
empty' :: OpParams -> Tensor v'1 Int32 -> m' (Tensor Value dtype)
empty' op'options :: OpParams
op'options shape :: Tensor v'1 Int32
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Empty"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type: DT_INT32 }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "init" type: "bool" default_value { b: false } }
-}
-- | 
emptyTensorList :: forall v'1 v'2 shape_type . (OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] shape_type) =>
                   DataType -- ^ __element_dtype__
                   -> Tensor v'1 shape_type -- ^ __element_shape__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __max_num_elements__
                   -> Tensor Build Variant -- ^ __handle__
emptyTensorList :: DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
emptyTensorList = OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) shape_type.
OneOf '[Int32, Int64] shape_type =>
OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
emptyTensorList' OpParams
forall a. a -> a
id
emptyTensorList' :: forall v'1 v'2 shape_type . (OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] shape_type) =>
                    OpParams ->
                    DataType -- ^ __element_dtype__
                    -> Tensor v'1 shape_type -- ^ __element_shape__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __max_num_elements__
                    -> Tensor Build Variant -- ^ __handle__
emptyTensorList' :: OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
emptyTensorList' op'options :: OpParams
op'options element_dtype :: DataType
element_dtype element_shape :: Tensor v'1 shape_type
element_shape
                 max_num_elements :: Tensor v'2 Int32
max_num_elements | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 shape_type
element_shape,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
max_num_elements]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EmptyTensorList"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
element_dtype
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "element_shape" type_attr: "shape_type" }
input_arg { name: "max_num_elements" type: DT_INT32 }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
encodeBase64 :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
encodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
encodeBase64 = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
encodeBase64' OpParams
forall a. a -> a
id
encodeBase64' :: OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __output__
encodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
encodeBase64' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodeBase64"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr { name: "pad" type: "bool" default_value { b: false } }
-}
-- | 
encodeJpeg :: Tensor v'1 Data.Word.Word8 -- ^ __image__
              -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeJpeg :: Tensor v'1 Word8 -> Tensor Build ByteString
encodeJpeg = OpParams -> Tensor v'1 Word8 -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Word8 -> Tensor Build ByteString
encodeJpeg' OpParams
forall a. a -> a
id
encodeJpeg' :: OpParams ->
               Tensor v'1 Data.Word.Word8 -- ^ __image__
               -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeJpeg' :: OpParams -> Tensor v'1 Word8 -> Tensor Build ByteString
encodeJpeg' op'options :: OpParams
op'options image :: Tensor v'1 Word8
image | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Word8
image]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodeJpeg"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image" type: DT_UINT8 }
output_arg { name: "contents" type: DT_STRING }
attr {
  name: "format"
  type: "string"
  default_value { s: "" }
  allowed_values { list { s: "" s: "grayscale" s: "rgb" } }
}
attr { name: "quality" type: "int" default_value { i: 95 } }
attr {
  name: "progressive" type: "bool" default_value { b: false }
}
attr {
  name: "optimize_size" type: "bool" default_value { b: false }
}
attr {
  name: "chroma_downsampling" type: "bool" default_value { b: true }
}
attr {
  name: "density_unit"
  type: "string"
  default_value { s: "in" }
  allowed_values { list { s: "in" s: "cm" } }
}
attr { name: "x_density" type: "int" default_value { i: 300 } }
attr { name: "y_density" type: "int" default_value { i: 300 } }
attr {
  name: "xmp_metadata" type: "string" default_value { s: "" }
}
-}
-- | 
encodeJpegVariableQuality :: Tensor v'1 Data.Word.Word8 -- ^ __images__
                             -> Tensor v'2 Data.Int.Int32 -- ^ __quality__
                             -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeJpegVariableQuality :: Tensor v'1 Word8 -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeJpegVariableQuality = OpParams
-> Tensor v'1 Word8 -> Tensor v'2 Int32 -> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Word8 -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeJpegVariableQuality' OpParams
forall a. a -> a
id
encodeJpegVariableQuality' :: OpParams ->
                              Tensor v'1 Data.Word.Word8 -- ^ __images__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __quality__
                              -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeJpegVariableQuality' :: OpParams
-> Tensor v'1 Word8 -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeJpegVariableQuality' op'options :: OpParams
op'options images :: Tensor v'1 Word8
images quality :: Tensor v'2 Int32
quality | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Word8
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
quality]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodeJpegVariableQuality"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type: DT_UINT8 }
input_arg { name: "quality" type: DT_INT32 }
output_arg { name: "contents" type: DT_STRING }
-}
-- | 
encodePng :: forall v'1 t . (OneOf '[Data.Word.Word16, Data.Word.Word8] t) =>
             Tensor v'1 t -- ^ __image__
             -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodePng :: Tensor v'1 t -> Tensor Build ByteString
encodePng = OpParams -> Tensor v'1 t -> Tensor Build ByteString
forall (v'1 :: * -> *) t.
OneOf '[Word16, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor Build ByteString
encodePng' OpParams
forall a. a -> a
id
encodePng' :: forall v'1 t . (OneOf '[Data.Word.Word16, Data.Word.Word8] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __image__
              -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodePng' :: OpParams -> Tensor v'1 t -> Tensor Build ByteString
encodePng' op'options :: OpParams
op'options image :: Tensor v'1 t
image | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
image]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodePng"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image" type_attr: "T" }
output_arg { name: "contents" type: DT_STRING }
attr { name: "compression" type: "int" default_value { i: -1 } }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_UINT8 }
  allowed_values { list { type: DT_UINT8 type: DT_UINT16 } }
}
-}
-- | 
encodeProto :: forall v'1 v'2 tinput_types . (TensorTypes tinput_types) =>
               ByteString -- ^ __message_type__
               -> Tensor v'1 Data.Int.Int32 -- ^ __sizes__
               -> TensorList (v'2) tinput_types -- ^ __values__
               -> Tensor Build Data.ByteString.ByteString -- ^ __bytes__
encodeProto :: ByteString
-> Tensor v'1 Int32
-> TensorList v'2 tinput_types
-> Tensor Build ByteString
encodeProto = OpParams
-> ByteString
-> Tensor v'1 Int32
-> TensorList v'2 tinput_types
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (tinput_types :: [*]).
TensorTypes tinput_types =>
OpParams
-> ByteString
-> Tensor v'1 Int32
-> TensorList v'2 tinput_types
-> Tensor Build ByteString
encodeProto' OpParams
forall a. a -> a
id
encodeProto' :: forall v'1 v'2 tinput_types . (TensorTypes tinput_types) =>
                OpParams ->
                ByteString -- ^ __message_type__
                -> Tensor v'1 Data.Int.Int32 -- ^ __sizes__
                -> TensorList (v'2) tinput_types -- ^ __values__
                -> Tensor Build Data.ByteString.ByteString -- ^ __bytes__
encodeProto' :: OpParams
-> ByteString
-> Tensor v'1 Int32
-> TensorList v'2 tinput_types
-> Tensor Build ByteString
encodeProto' op'options :: OpParams
op'options message_type :: ByteString
message_type sizes :: Tensor v'1 Int32
sizes values :: TensorList v'2 tinput_types
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
sizes,
                                                             TensorList v'2 tinput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tinput_types
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodeProto"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tinput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tinput_types
forall k (t :: k). Proxy t
Proxy :: Proxy tinput_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "message_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
message_type
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sizes" type: DT_INT32 }
input_arg { name: "values" type_list_attr: "Tinput_types" }
output_arg { name: "bytes" type: DT_STRING }
attr { name: "field_names" type: "list(string)" }
attr { name: "message_type" type: "string" }
attr {
  name: "descriptor_source"
  type: "string"
  default_value { s: "local://" }
}
attr {
  name: "Tinput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
encodeWav :: Tensor v'1 Float -- ^ __audio__
             -> Tensor v'2 Data.Int.Int32 -- ^ __sample_rate__
             -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeWav :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeWav = OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeWav' OpParams
forall a. a -> a
id
encodeWav' :: OpParams ->
              Tensor v'1 Float -- ^ __audio__
              -> Tensor v'2 Data.Int.Int32 -- ^ __sample_rate__
              -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
encodeWav' :: OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build ByteString
encodeWav' op'options :: OpParams
op'options audio :: Tensor v'1 Float
audio sample_rate :: Tensor v'2 Int32
sample_rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
audio,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
sample_rate]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EncodeWav"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "audio" type: DT_FLOAT }
input_arg { name: "sample_rate" type: DT_INT32 }
output_arg { name: "contents" type: DT_STRING }
-}
-- | 
enqueueTPUEmbeddingIntegerBatch :: forall v'1 v'2 m' . (MonadBuild m') =>
                                   [Tensor v'1 Data.Int.Int32] -- ^ __batch__
                                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __mode_override__
                                   -> m' (ControlNode)
enqueueTPUEmbeddingIntegerBatch :: [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
enqueueTPUEmbeddingIntegerBatch = OpParams
-> [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
enqueueTPUEmbeddingIntegerBatch' OpParams
forall a. a -> a
id
enqueueTPUEmbeddingIntegerBatch' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                    OpParams ->
                                    [Tensor v'1 Data.Int.Int32] -- ^ __batch__
                                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __mode_override__
                                    -> m' (ControlNode)
enqueueTPUEmbeddingIntegerBatch' :: OpParams
-> [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
enqueueTPUEmbeddingIntegerBatch' op'options :: OpParams
op'options batch :: [Tensor v'1 Int32]
batch
                                 mode_override :: Tensor v'2 ByteString
mode_override | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("batch", [Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
batch)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int32]
batch,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
mode_override]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "EnqueueTPUEmbeddingIntegerBatch"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
batch) :: Int64
{-
input_arg { name: "batch" type: DT_INT32 number_attr: "N" }
input_arg { name: "mode_override" type: DT_STRING }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
enqueueTPUEmbeddingRaggedTensorBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                        m' . (MonadBuild m',
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] t1,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] t2,
                                              OneOf '[Double, Float] t3) =>
                                        [Tensor v'1 t1] -- ^ __sample_splits__
                                        -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                        -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                        -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                        -> m' (ControlNode)
enqueueTPUEmbeddingRaggedTensorBatch :: [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingRaggedTensorBatch = OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t1 t2 t3 (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2,
 OneOf '[Double, Float] t3) =>
OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingRaggedTensorBatch' OpParams
forall a. a -> a
id
enqueueTPUEmbeddingRaggedTensorBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                         m' . (MonadBuild m',
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] t1,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] t2,
                                               OneOf '[Double, Float] t3) =>
                                         OpParams ->
                                         [Tensor v'1 t1] -- ^ __sample_splits__
                                         -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                         -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                         -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                         -> m' (ControlNode)
enqueueTPUEmbeddingRaggedTensorBatch' :: OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingRaggedTensorBatch' op'options :: OpParams
op'options sample_splits :: [Tensor v'1 t1]
sample_splits embedding_indices :: [Tensor v'2 t2]
embedding_indices
                                      aggregation_weights :: [Tensor v'3 t3]
aggregation_weights
                                      mode_override :: Tensor v'4 ByteString
mode_override | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("sample_splits", [Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_splits),
                                                                            ("embedding_indices", [Tensor v'2 t2] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t2]
embedding_indices),
                                                                            ("aggregation_weights", [Tensor v'3 t3] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t3]
aggregation_weights)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t1] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t1]
sample_splits,
                                                             [Tensor v'2 t2] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t2]
embedding_indices,
                                                             [Tensor v'3 t3] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t3]
aggregation_weights,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
mode_override]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "EnqueueTPUEmbeddingRaggedTensorBatch"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T3" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t3 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t3
forall a. HasCallStack => a
undefined :: t3)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_splits) :: Int64
{-
input_arg {
  name: "sample_splits" type_attr: "T1" number_attr: "N"
}
input_arg {
  name: "embedding_indices" type_attr: "T2" number_attr: "N"
}
input_arg {
  name: "aggregation_weights" type_attr: "T3" number_attr: "N"
}
input_arg { name: "mode_override" type: DT_STRING }
attr {
  name: "T1"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T2"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T3"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
attr {
  name: "combiners" type: "list(string)" default_value { list { } }
}
attr { name: "table_ids" type: "list(int)" }
attr {
  name: "max_sequence_lengths"
  type: "list(int)"
  default_value { list { } }
}
-}
-- | 
enqueueTPUEmbeddingSparseBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                  m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] t1,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] t2,
                                        OneOf '[Double, Float] t3) =>
                                  [Tensor v'1 t1] -- ^ __sample_indices__
                                  -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                  -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                  -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                  -> m' (ControlNode)
enqueueTPUEmbeddingSparseBatch :: [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseBatch = OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t1 t2 t3 (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2,
 OneOf '[Double, Float] t3) =>
OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseBatch' OpParams
forall a. a -> a
id
enqueueTPUEmbeddingSparseBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                   m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] t1,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] t2,
                                         OneOf '[Double, Float] t3) =>
                                   OpParams ->
                                   [Tensor v'1 t1] -- ^ __sample_indices__
                                   -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                   -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                   -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                   -> m' (ControlNode)
enqueueTPUEmbeddingSparseBatch' :: OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseBatch' op'options :: OpParams
op'options sample_indices :: [Tensor v'1 t1]
sample_indices embedding_indices :: [Tensor v'2 t2]
embedding_indices
                                aggregation_weights :: [Tensor v'3 t3]
aggregation_weights
                                mode_override :: Tensor v'4 ByteString
mode_override | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("sample_indices", [Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_indices),
                                                                      ("embedding_indices", [Tensor v'2 t2] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t2]
embedding_indices),
                                                                      ("aggregation_weights", [Tensor v'3 t3] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t3]
aggregation_weights)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t1] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t1]
sample_indices,
                                                             [Tensor v'2 t2] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t2]
embedding_indices,
                                                             [Tensor v'3 t3] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t3]
aggregation_weights,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
mode_override]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "EnqueueTPUEmbeddingSparseBatch"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T3" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t3 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t3
forall a. HasCallStack => a
undefined :: t3)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_indices) :: Int64
{-
input_arg {
  name: "sample_indices" type_attr: "T1" number_attr: "N"
}
input_arg {
  name: "embedding_indices" type_attr: "T2" number_attr: "N"
}
input_arg {
  name: "aggregation_weights" type_attr: "T3" number_attr: "N"
}
input_arg { name: "mode_override" type: DT_STRING }
attr {
  name: "T1"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T2"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T3"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
attr {
  name: "combiners" type: "list(string)" default_value { list { } }
}
-}
-- | 
enqueueTPUEmbeddingSparseTensorBatch :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                        m' . (MonadBuild m',
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] t1,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] t2,
                                              OneOf '[Double, Float] t3) =>
                                        [Tensor v'1 t1] -- ^ __sample_indices__
                                        -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                        -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                        -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                        -> m' (ControlNode)
enqueueTPUEmbeddingSparseTensorBatch :: [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseTensorBatch = OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t1 t2 t3 (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] t1, OneOf '[Int32, Int64] t2,
 OneOf '[Double, Float] t3) =>
OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseTensorBatch' OpParams
forall a. a -> a
id
enqueueTPUEmbeddingSparseTensorBatch' :: forall v'1 v'2 v'3 v'4 t1 t2 t3
                                         m' . (MonadBuild m',
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] t1,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] t2,
                                               OneOf '[Double, Float] t3) =>
                                         OpParams ->
                                         [Tensor v'1 t1] -- ^ __sample_indices__
                                         -> [Tensor v'2 t2] -- ^ __embedding_indices__
                                         -> [Tensor v'3 t3] -- ^ __aggregation_weights__
                                         -> Tensor v'4 Data.ByteString.ByteString -- ^ __mode_override__
                                         -> m' (ControlNode)
enqueueTPUEmbeddingSparseTensorBatch' :: OpParams
-> [Tensor v'1 t1]
-> [Tensor v'2 t2]
-> [Tensor v'3 t3]
-> Tensor v'4 ByteString
-> m' ControlNode
enqueueTPUEmbeddingSparseTensorBatch' op'options :: OpParams
op'options sample_indices :: [Tensor v'1 t1]
sample_indices
                                      embedding_indices :: [Tensor v'2 t2]
embedding_indices aggregation_weights :: [Tensor v'3 t3]
aggregation_weights
                                      mode_override :: Tensor v'4 ByteString
mode_override | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("sample_indices", [Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_indices),
                                                                            ("embedding_indices", [Tensor v'2 t2] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t2]
embedding_indices),
                                                                            ("aggregation_weights", [Tensor v'3 t3] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t3]
aggregation_weights)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t1] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t1]
sample_indices,
                                                             [Tensor v'2 t2] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t2]
embedding_indices,
                                                             [Tensor v'3 t3] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t3]
aggregation_weights,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
mode_override]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "EnqueueTPUEmbeddingSparseTensorBatch"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T3" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t3 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t3
forall a. HasCallStack => a
undefined :: t3)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t1] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t1]
sample_indices) :: Int64
{-
input_arg {
  name: "sample_indices" type_attr: "T1" number_attr: "N"
}
input_arg {
  name: "embedding_indices" type_attr: "T2" number_attr: "N"
}
input_arg {
  name: "aggregation_weights" type_attr: "T3" number_attr: "N"
}
input_arg { name: "mode_override" type: DT_STRING }
attr {
  name: "T1"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T2"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T3"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
attr {
  name: "combiners" type: "list(string)" default_value { list { } }
}
attr { name: "table_ids" type: "list(int)" }
attr {
  name: "max_sequence_lengths"
  type: "list(int)"
  default_value { list { } }
}
-}
-- | 
ensureShape :: forall v'1 t . (TensorType t) => Shape -- ^ __shape__
               -> Tensor v'1 t -- ^ __input__
               -> Tensor Build t -- ^ __output__
ensureShape :: Shape -> Tensor v'1 t -> Tensor Build t
ensureShape = OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
ensureShape' OpParams
forall a. a -> a
id
ensureShape' :: forall v'1 t . (TensorType t) => OpParams ->
                Shape -- ^ __shape__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor Build t -- ^ __output__
ensureShape' :: OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
ensureShape' op'options :: OpParams
op'options shape :: Shape
shape input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EnsureShape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "shape" type: "shape" }
attr { name: "T" type: "type" }
-}
-- | 
enter :: forall v'1 t . (TensorType t) => ByteString -- ^ __frame_name__
         -> Tensor v'1 t -- ^ __data__
         -> Tensor Build t -- ^ __output__
enter :: ByteString -> Tensor v'1 t -> Tensor Build t
enter = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
enter' OpParams
forall a. a -> a
id
enter' :: forall v'1 t . (TensorType t) => OpParams ->
          ByteString -- ^ __frame_name__
          -> Tensor v'1 t -- ^ __data__
          -> Tensor Build t -- ^ __output__
enter' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
enter' op'options :: OpParams
op'options frame_name :: ByteString
frame_name data' :: Tensor v'1 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data']
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Enter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "frame_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
frame_name
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "frame_name" type: "string" }
attr {
  name: "is_constant" type: "bool" default_value { b: false }
}
attr {
  name: "parallel_iterations" type: "int" default_value { i: 10 }
}
-}
-- | 
equal :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float), Bool,
                                     Data.ByteString.ByteString, Data.Int.Int16,
                                     Data.Int.Int32, Data.Int.Int64,
                                     Data.Int.Int8, Data.Word.Word16,
                                     Data.Word.Word32, Data.Word.Word64,
                                     Data.Word.Word8, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __y__
         -> Tensor Build Bool -- ^ __z__
equal :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
equal = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32,
    Int64, Int8, Word16, Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
equal' OpParams
forall a. a -> a
id
equal' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float), Bool,
                                      Data.ByteString.ByteString,
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) => OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor v'2 t -- ^ __y__
          -> Tensor Build Bool -- ^ __z__
equal' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
equal' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Equal"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
      type: DT_COMPLEX64
      type: DT_QUINT8
      type: DT_QINT8
      type: DT_QINT32
      type: DT_STRING
      type: DT_BOOL
      type: DT_COMPLEX128
    }
  }
}
attr {
  name: "incompatible_shape_error"
  type: "bool"
  default_value { b: true }
}
-}
-- | 
erf :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
erf :: Tensor v'1 t -> Tensor Build t
erf = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
erf' OpParams
forall a. a -> a
id
erf' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
erf' :: OpParams -> Tensor v'1 t -> Tensor Build t
erf' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Erf"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
erfc :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
erfc :: Tensor v'1 t -> Tensor Build t
erfc = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
erfc' OpParams
forall a. a -> a
id
erfc' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
erfc' :: OpParams -> Tensor v'1 t -> Tensor Build t
erfc' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Erfc"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
erfinv :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
erfinv :: Tensor v'1 t -> Tensor Build t
erfinv = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
erfinv' OpParams
forall a. a -> a
id
erfinv' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
erfinv' :: OpParams -> Tensor v'1 t -> Tensor Build t
erfinv' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Erfinv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
euclideanNorm :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tidx) =>
                 Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 tidx -- ^ __reduction_indices__
                 -> Tensor Build t -- ^ __output__
euclideanNorm :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
euclideanNorm = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
euclideanNorm' OpParams
forall a. a -> a
id
euclideanNorm' :: forall v'1 v'2 t
                  tidx . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t, OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tidx) =>
                  OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 tidx -- ^ __reduction_indices__
                  -> Tensor Build t -- ^ __output__
euclideanNorm' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
euclideanNorm' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "EuclideanNorm"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
exit :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __data__
        -> Tensor Build t -- ^ __output__
exit :: Tensor v'1 t -> Tensor Build t
exit = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
exit' OpParams
forall a. a -> a
id
exit' :: forall v'1 t . (TensorType t) => OpParams ->
         Tensor v'1 t -- ^ __data__
         -> Tensor Build t -- ^ __output__
exit' :: OpParams -> Tensor v'1 t -> Tensor Build t
exit' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data']
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Exit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
exp :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
exp :: Tensor v'1 t -> Tensor Build t
exp = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
exp' OpParams
forall a. a -> a
id
exp' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
exp' :: OpParams -> Tensor v'1 t -> Tensor Build t
exp' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Exp"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
expandDims :: forall v'1 v'2 t tdim . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tdim) =>
              Tensor v'1 t -- ^ __input__
              -> Tensor v'2 tdim -- ^ __dim__
              -> Tensor Build t -- ^ __output__
expandDims :: Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
expandDims = OpParams -> Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
expandDims' OpParams
forall a. a -> a
id
expandDims' :: forall v'1 v'2 t tdim . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tdim) =>
               OpParams ->
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 tdim -- ^ __dim__
               -> Tensor Build t -- ^ __output__
expandDims' :: OpParams -> Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
expandDims' op'options :: OpParams
op'options input :: Tensor v'1 t
input dim :: Tensor v'2 tdim
dim | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tdim -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tdim
dim]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExpandDims"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdim" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tdim -> DataType
forall a. TensorType a => a -> DataType
tensorType (tdim
forall a. HasCallStack => a
undefined :: tdim)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "dim" type_attr: "Tdim" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tdim"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
experimentalAssertNextDataset :: [DataType] -- ^ __output_types__
                                 -> Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __transformations__
                                 -> Tensor Build Variant -- ^ __handle__
experimentalAssertNextDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalAssertNextDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalAssertNextDataset' OpParams
forall a. a -> a
id
experimentalAssertNextDataset' :: OpParams ->
                                  [DataType] -- ^ __output_types__
                                  -> Tensor v'1 Variant -- ^ __input_dataset__
                                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __transformations__
                                  -> Tensor Build Variant -- ^ __handle__
experimentalAssertNextDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalAssertNextDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                               transformations :: Tensor v'2 ByteString
transformations | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
transformations]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalAssertNextDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "transformations" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalAutoShardDataset :: [DataType] -- ^ __output_types__
                                -> Tensor v'1 Variant -- ^ __input_dataset__
                                -> Tensor v'2 Data.Int.Int64 -- ^ __num_workers__
                                -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                                -> Tensor Build Variant -- ^ __handle__
experimentalAutoShardDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalAutoShardDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalAutoShardDataset' OpParams
forall a. a -> a
id
experimentalAutoShardDataset' :: OpParams ->
                                 [DataType] -- ^ __output_types__
                                 -> Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor v'2 Data.Int.Int64 -- ^ __num_workers__
                                 -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                                 -> Tensor Build Variant -- ^ __handle__
experimentalAutoShardDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalAutoShardDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset num_workers :: Tensor v'2 Int64
num_workers
                              index :: Tensor v'3 Int64
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_workers,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
index]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalAutoShardDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_workers" type: DT_INT64 }
input_arg { name: "index" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "auto_shard_policy" type: "int" default_value { i: 0 }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalBytesProducedStatsDataset :: [DataType] -- ^ __output_types__
                                         -> Tensor v'1 Variant -- ^ __input_dataset__
                                         -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                                         -> Tensor Build Variant -- ^ __handle__
experimentalBytesProducedStatsDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalBytesProducedStatsDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalBytesProducedStatsDataset' OpParams
forall a. a -> a
id
experimentalBytesProducedStatsDataset' :: OpParams ->
                                          [DataType] -- ^ __output_types__
                                          -> Tensor v'1 Variant -- ^ __input_dataset__
                                          -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                                          -> Tensor Build Variant -- ^ __handle__
experimentalBytesProducedStatsDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalBytesProducedStatsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                       tag :: Tensor v'2 ByteString
tag | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tag]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalBytesProducedStatsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "tag" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalCSVDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                          output_types m' . (MonadBuild m',
                                             OneOfs '[Data.ByteString.ByteString,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64, Double,
                                                      Float] output_types) =>
                          Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                          -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                          -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                          -> Tensor v'4 Bool -- ^ __header__
                          -> Tensor v'5 Data.ByteString.ByteString -- ^ __field_delim__
                          -> Tensor v'6 Bool -- ^ __use_quote_delim__
                          -> Tensor v'7 Data.ByteString.ByteString -- ^ __na_value__
                          -> Tensor v'8 Data.Int.Int64 -- ^ __select_cols__
                          -> TensorList (v'9) output_types -- ^ __record_defaults__
                          -> m' (Tensor Value Variant) -- ^ __handle__
experimentalCSVDataset :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
experimentalCSVDataset = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (output_types :: [*])
       (m' :: * -> *).
(MonadBuild m',
 OneOfs '[ByteString, Int32, Int64, Double, Float] output_types) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
experimentalCSVDataset' OpParams
forall a. a -> a
id
experimentalCSVDataset' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                           output_types m' . (MonadBuild m',
                                              OneOfs '[Data.ByteString.ByteString,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64, Double,
                                                       Float] output_types) =>
                           OpParams ->
                           Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                           -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                           -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                           -> Tensor v'4 Bool -- ^ __header__
                           -> Tensor v'5 Data.ByteString.ByteString -- ^ __field_delim__
                           -> Tensor v'6 Bool -- ^ __use_quote_delim__
                           -> Tensor v'7 Data.ByteString.ByteString -- ^ __na_value__
                           -> Tensor v'8 Data.Int.Int64 -- ^ __select_cols__
                           -> TensorList (v'9) output_types -- ^ __record_defaults__
                           -> m' (Tensor Value Variant) -- ^ __handle__
experimentalCSVDataset' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Bool
-> Tensor v'5 ByteString
-> Tensor v'6 Bool
-> Tensor v'7 ByteString
-> Tensor v'8 Int64
-> TensorList v'9 output_types
-> m' (Tensor Value Variant)
experimentalCSVDataset' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames compression_type :: Tensor v'2 ByteString
compression_type buffer_size :: Tensor v'3 Int64
buffer_size header :: Tensor v'4 Bool
header
                        field_delim :: Tensor v'5 ByteString
field_delim use_quote_delim :: Tensor v'6 Bool
use_quote_delim na_value :: Tensor v'7 ByteString
na_value select_cols :: Tensor v'8 Int64
select_cols
                        record_defaults :: TensorList v'9 output_types
record_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
compression_type,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
buffer_size,
                                                             Tensor v'4 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Bool
header,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
field_delim,
                                                             Tensor v'6 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Bool
use_quote_delim,
                                                             Tensor v'7 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 ByteString
na_value,
                                                             Tensor v'8 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Int64
select_cols,
                                                             TensorList v'9 output_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'9 output_types
record_defaults]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalCSVDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "header" type: DT_BOOL }
input_arg { name: "field_delim" type: DT_STRING }
input_arg { name: "use_quote_delim" type: DT_BOOL }
input_arg { name: "na_value" type: DT_STRING }
input_arg { name: "select_cols" type: DT_INT64 }
input_arg {
  name: "record_defaults" type_list_attr: "output_types"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_STRING
    }
  }
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalChooseFastestDataset :: Data.Int.Int64 -- ^ __num_experiments__
                                    -> [DataType] -- ^ __output_types__
                                    -> [Tensor v'1 Variant] -- ^ __input_datasets__
                                    -> Tensor Build Variant -- ^ __handle__
experimentalChooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
experimentalChooseFastestDataset = OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
experimentalChooseFastestDataset' OpParams
forall a. a -> a
id
experimentalChooseFastestDataset' :: OpParams ->
                                     Data.Int.Int64 -- ^ __num_experiments__
                                     -> [DataType] -- ^ __output_types__
                                     -> [Tensor v'1 Variant] -- ^ __input_datasets__
                                     -> Tensor Build Variant -- ^ __handle__
experimentalChooseFastestDataset' :: OpParams
-> Int64
-> [DataType]
-> [Tensor v'1 Variant]
-> Tensor Build Variant
experimentalChooseFastestDataset' op'options :: OpParams
op'options num_experiments :: Int64
num_experiments output_types :: [DataType]
output_types
                                  input_datasets :: [Tensor v'1 Variant]
input_datasets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("input_datasets", [Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Variant] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Variant]
input_datasets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalChooseFastestDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_experiments" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_experiments
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets) :: Int64
{-
input_arg {
  name: "input_datasets" type: DT_VARIANT number_attr: "N"
}
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "num_experiments" type: "int" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalDatasetCardinality :: Tensor v'1 Variant -- ^ __input_dataset__
                                  -> Tensor Build Data.Int.Int64 -- ^ __cardinality__
experimentalDatasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
experimentalDatasetCardinality = OpParams -> Tensor v'1 Variant -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Int64
experimentalDatasetCardinality' OpParams
forall a. a -> a
id
experimentalDatasetCardinality' :: OpParams ->
                                   Tensor v'1 Variant -- ^ __input_dataset__
                                   -> Tensor Build Data.Int.Int64 -- ^ __cardinality__
experimentalDatasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
experimentalDatasetCardinality' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalDatasetCardinality"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "cardinality" type: DT_INT64 }
-}
-- | 
experimentalDatasetToTFRecord :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                 Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                                 -> Tensor v'3 Data.ByteString.ByteString -- ^ __compression_type__
                                 -> m' (ControlNode)
experimentalDatasetToTFRecord :: Tensor v'1 Variant
-> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
experimentalDatasetToTFRecord = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
experimentalDatasetToTFRecord' OpParams
forall a. a -> a
id
experimentalDatasetToTFRecord' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                  OpParams ->
                                  Tensor v'1 Variant -- ^ __input_dataset__
                                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                                  -> Tensor v'3 Data.ByteString.ByteString -- ^ __compression_type__
                                  -> m' (ControlNode)
experimentalDatasetToTFRecord' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' ControlNode
experimentalDatasetToTFRecord' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset filename :: Tensor v'2 ByteString
filename
                               compression_type :: Tensor v'3 ByteString
compression_type | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
compression_type]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalDatasetToTFRecord"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
-}
-- | 
experimentalDenseToSparseBatchDataset :: [DataType] -- ^ __output_types__
                                         -> Tensor v'1 Variant -- ^ __input_dataset__
                                         -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                                         -> Tensor v'3 Data.Int.Int64 -- ^ __row_shape__
                                         -> Tensor Build Variant -- ^ __handle__
experimentalDenseToSparseBatchDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalDenseToSparseBatchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalDenseToSparseBatchDataset' OpParams
forall a. a -> a
id
experimentalDenseToSparseBatchDataset' :: OpParams ->
                                          [DataType] -- ^ __output_types__
                                          -> Tensor v'1 Variant -- ^ __input_dataset__
                                          -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                                          -> Tensor v'3 Data.Int.Int64 -- ^ __row_shape__
                                          -> Tensor Build Variant -- ^ __handle__
experimentalDenseToSparseBatchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
experimentalDenseToSparseBatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                       batch_size :: Tensor v'2 Int64
batch_size row_shape :: Tensor v'3 Int64
row_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
row_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalDenseToSparseBatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
input_arg { name: "row_shape" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalDirectedInterleaveDataset :: [DataType] -- ^ __output_types__
                                         -> Tensor v'1 Variant -- ^ __selector_input_dataset__
                                         -> [Tensor v'2 Variant] -- ^ __data_input_datasets__
                                         -> Tensor Build Variant -- ^ __handle__
experimentalDirectedInterleaveDataset :: [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
experimentalDirectedInterleaveDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
experimentalDirectedInterleaveDataset' OpParams
forall a. a -> a
id
experimentalDirectedInterleaveDataset' :: OpParams ->
                                          [DataType] -- ^ __output_types__
                                          -> Tensor v'1 Variant -- ^ __selector_input_dataset__
                                          -> [Tensor v'2 Variant] -- ^ __data_input_datasets__
                                          -> Tensor Build Variant -- ^ __handle__
experimentalDirectedInterleaveDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> [Tensor v'2 Variant]
-> Tensor Build Variant
experimentalDirectedInterleaveDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                                       selector_input_dataset :: Tensor v'1 Variant
selector_input_dataset
                                       data_input_datasets :: [Tensor v'2 Variant]
data_input_datasets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("data_input_datasets", [Tensor v'2 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Variant]
data_input_datasets)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
selector_input_dataset,
                                                             [Tensor v'2 Variant] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Variant]
data_input_datasets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalDirectedInterleaveDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Variant]
data_input_datasets) :: Int64
{-
input_arg { name: "selector_input_dataset" type: DT_VARIANT }
input_arg {
  name: "data_input_datasets" type: DT_VARIANT number_attr: "N"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
experimentalIgnoreErrorsDataset :: [DataType] -- ^ __output_types__
                                   -> Tensor v'1 Variant -- ^ __input_dataset__
                                   -> Tensor Build Variant -- ^ __handle__
experimentalIgnoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalIgnoreErrorsDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalIgnoreErrorsDataset' OpParams
forall a. a -> a
id
experimentalIgnoreErrorsDataset' :: OpParams ->
                                    [DataType] -- ^ __output_types__
                                    -> Tensor v'1 Variant -- ^ __input_dataset__
                                    -> Tensor Build Variant -- ^ __handle__
experimentalIgnoreErrorsDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalIgnoreErrorsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                                 input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalIgnoreErrorsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalIteratorGetDevice :: forall v'1 m' . (MonadBuild m') =>
                                 Tensor v'1 ResourceHandle -- ^ __resource__
                                 -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __device__
experimentalIteratorGetDevice :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalIteratorGetDevice = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalIteratorGetDevice' OpParams
forall a. a -> a
id
experimentalIteratorGetDevice' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                                  Tensor v'1 ResourceHandle -- ^ __resource__
                                  -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __device__
experimentalIteratorGetDevice' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalIteratorGetDevice' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalIteratorGetDevice"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "device" type: DT_STRING }
-}
-- | 
experimentalLMDBDataset :: forall v'1 m' . (MonadBuild m') =>
                           [DataType] -- ^ __output_types__
                           -> Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                           -> m' (Tensor Value Variant) -- ^ __handle__
experimentalLMDBDataset :: [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalLMDBDataset = OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalLMDBDataset' OpParams
forall a. a -> a
id
experimentalLMDBDataset' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                            [DataType] -- ^ __output_types__
                            -> Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                            -> m' (Tensor Value Variant) -- ^ __handle__
experimentalLMDBDataset' :: OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalLMDBDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types filenames :: Tensor v'1 ByteString
filenames | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalLMDBDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalLatencyStatsDataset :: [DataType] -- ^ __output_types__
                                   -> Tensor v'1 Variant -- ^ __input_dataset__
                                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                                   -> Tensor Build Variant -- ^ __handle__
experimentalLatencyStatsDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalLatencyStatsDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalLatencyStatsDataset' OpParams
forall a. a -> a
id
experimentalLatencyStatsDataset' :: OpParams ->
                                    [DataType] -- ^ __output_types__
                                    -> Tensor v'1 Variant -- ^ __input_dataset__
                                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                                    -> Tensor Build Variant -- ^ __handle__
experimentalLatencyStatsDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
experimentalLatencyStatsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                 tag :: Tensor v'2 ByteString
tag | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tag]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalLatencyStatsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "tag" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalMatchingFilesDataset :: forall v'1 m' . (MonadBuild m') =>
                                    Tensor v'1 Data.ByteString.ByteString -- ^ __patterns__
                                    -> m' (Tensor Value Variant) -- ^ __handle__
experimentalMatchingFilesDataset :: Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalMatchingFilesDataset = OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalMatchingFilesDataset' OpParams
forall a. a -> a
id
experimentalMatchingFilesDataset' :: forall v'1 m' . (MonadBuild m') =>
                                     OpParams ->
                                     Tensor v'1 Data.ByteString.ByteString -- ^ __patterns__
                                     -> m' (Tensor Value Variant) -- ^ __handle__
experimentalMatchingFilesDataset' :: OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
experimentalMatchingFilesDataset' op'options :: OpParams
op'options patterns :: Tensor v'1 ByteString
patterns | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
patterns]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalMatchingFilesDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "patterns" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
experimentalMaxIntraOpParallelismDataset :: [DataType] -- ^ __output_types__
                                            -> Tensor v'1 Variant -- ^ __input_dataset__
                                            -> Tensor v'2 Data.Int.Int64 -- ^ __max_intra_op_parallelism__
                                            -> Tensor Build Variant -- ^ __handle__
experimentalMaxIntraOpParallelismDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalMaxIntraOpParallelismDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalMaxIntraOpParallelismDataset' OpParams
forall a. a -> a
id
experimentalMaxIntraOpParallelismDataset' :: OpParams ->
                                             [DataType] -- ^ __output_types__
                                             -> Tensor v'1 Variant -- ^ __input_dataset__
                                             -> Tensor v'2 Data.Int.Int64 -- ^ __max_intra_op_parallelism__
                                             -> Tensor Build Variant -- ^ __handle__
experimentalMaxIntraOpParallelismDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalMaxIntraOpParallelismDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                          max_intra_op_parallelism :: Tensor v'2 Int64
max_intra_op_parallelism | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
max_intra_op_parallelism]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalMaxIntraOpParallelismDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "max_intra_op_parallelism" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalNonSerializableDataset :: [DataType] -- ^ __output_types__
                                      -> Tensor v'1 Variant -- ^ __input_dataset__
                                      -> Tensor Build Variant -- ^ __handle__
experimentalNonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalNonSerializableDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalNonSerializableDataset' OpParams
forall a. a -> a
id
experimentalNonSerializableDataset' :: OpParams ->
                                       [DataType] -- ^ __output_types__
                                       -> Tensor v'1 Variant -- ^ __input_dataset__
                                       -> Tensor Build Variant -- ^ __handle__
experimentalNonSerializableDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalNonSerializableDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                                    input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalNonSerializableDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalParseExampleDataset :: forall v'1 v'2 v'3
                                   tdense . (OneOfs '[Data.ByteString.ByteString,
                                                      Data.Int.Int64,
                                                      Float] tdense) =>
                                   [DataType] -- ^ __output_types__
                                   -> [DataType] -- ^ __sparse_types__
                                   -> Tensor v'1 Variant -- ^ __input_dataset__
                                   -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                                   -> TensorList (v'3) tdense -- ^ __dense_defaults__
                                   -> Tensor Build Variant -- ^ __handle__
experimentalParseExampleDataset :: [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
experimentalParseExampleDataset = OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (tdense :: [*]).
OneOfs '[ByteString, Int64, Float] tdense =>
OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
experimentalParseExampleDataset' OpParams
forall a. a -> a
id
experimentalParseExampleDataset' :: forall v'1 v'2 v'3
                                    tdense . (OneOfs '[Data.ByteString.ByteString,
                                                       Data.Int.Int64,
                                                       Float] tdense) =>
                                    OpParams ->
                                    [DataType] -- ^ __output_types__
                                    -> [DataType] -- ^ __sparse_types__
                                    -> Tensor v'1 Variant -- ^ __input_dataset__
                                    -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                                    -> TensorList (v'3) tdense -- ^ __dense_defaults__
                                    -> Tensor Build Variant -- ^ __handle__
experimentalParseExampleDataset' :: OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
experimentalParseExampleDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types sparse_types :: [DataType]
sparse_types
                                 input_dataset :: Tensor v'1 Variant
input_dataset num_parallel_calls :: Tensor v'2 Int64
num_parallel_calls
                                 dense_defaults :: TensorList v'3 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_parallel_calls,
                                                             TensorList v'3 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalParseExampleDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
sparse_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_parallel_calls" type: DT_INT64 }
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "sparse_keys" type: "list(string)" has_minimum: true }
attr { name: "dense_keys" type: "list(string)" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "sloppy" type: "bool" default_value { b: false } }
-}
-- | 
experimentalPrivateThreadPoolDataset :: [DataType] -- ^ __output_types__
                                        -> Tensor v'1 Variant -- ^ __input_dataset__
                                        -> Tensor v'2 Data.Int.Int64 -- ^ __num_threads__
                                        -> Tensor Build Variant -- ^ __handle__
experimentalPrivateThreadPoolDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalPrivateThreadPoolDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalPrivateThreadPoolDataset' OpParams
forall a. a -> a
id
experimentalPrivateThreadPoolDataset' :: OpParams ->
                                         [DataType] -- ^ __output_types__
                                         -> Tensor v'1 Variant -- ^ __input_dataset__
                                         -> Tensor v'2 Data.Int.Int64 -- ^ __num_threads__
                                         -> Tensor Build Variant -- ^ __handle__
experimentalPrivateThreadPoolDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalPrivateThreadPoolDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                      num_threads :: Tensor v'2 Int64
num_threads | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_threads]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalPrivateThreadPoolDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_threads" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalRandomDataset :: forall v'1 v'2 m' . (MonadBuild m') =>
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Data.Int.Int64 -- ^ __seed__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                             -> m' (Tensor Value Variant) -- ^ __handle__
experimentalRandomDataset :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
experimentalRandomDataset = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
experimentalRandomDataset' OpParams
forall a. a -> a
id
experimentalRandomDataset' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Data.Int.Int64 -- ^ __seed__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                              -> m' (Tensor Value Variant) -- ^ __handle__
experimentalRandomDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
experimentalRandomDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types seed :: Tensor v'1 Int64
seed
                           seed2 :: Tensor v'2 Int64
seed2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seed,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
seed2]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalRandomDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalRebatchDataset :: [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __num_replicas__
                              -> Tensor Build Variant -- ^ __handle__
experimentalRebatchDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalRebatchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalRebatchDataset' OpParams
forall a. a -> a
id
experimentalRebatchDataset' :: OpParams ->
                               [DataType] -- ^ __output_types__
                               -> Tensor v'1 Variant -- ^ __input_dataset__
                               -> Tensor v'2 Data.Int.Int64 -- ^ __num_replicas__
                               -> Tensor Build Variant -- ^ __handle__
experimentalRebatchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalRebatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                            num_replicas :: Tensor v'2 Int64
num_replicas | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_replicas]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalRebatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_replicas" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "use_fallback" type: "bool" default_value { b: true }
}
-}
-- | 
experimentalSetStatsAggregatorDataset :: forall v'1 v'2 v'3 v'4
                                         m' . (MonadBuild m') =>
                                         [DataType] -- ^ __output_types__
                                         -> Tensor v'1 Variant -- ^ __input_dataset__
                                         -> Tensor v'2 ResourceHandle -- ^ __stats_aggregator__
                                         -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                                         -> Tensor v'4 Data.ByteString.ByteString -- ^ __counter_prefix__
                                         -> m' (Tensor Value Variant) -- ^ __handle__
experimentalSetStatsAggregatorDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
experimentalSetStatsAggregatorDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
experimentalSetStatsAggregatorDataset' OpParams
forall a. a -> a
id
experimentalSetStatsAggregatorDataset' :: forall v'1 v'2 v'3 v'4
                                          m' . (MonadBuild m') => OpParams ->
                                          [DataType] -- ^ __output_types__
                                          -> Tensor v'1 Variant -- ^ __input_dataset__
                                          -> Tensor v'2 ResourceHandle -- ^ __stats_aggregator__
                                          -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                                          -> Tensor v'4 Data.ByteString.ByteString -- ^ __counter_prefix__
                                          -> m' (Tensor Value Variant) -- ^ __handle__
experimentalSetStatsAggregatorDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
experimentalSetStatsAggregatorDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                       stats_aggregator :: Tensor v'2 ResourceHandle
stats_aggregator tag :: Tensor v'3 ByteString
tag
                                       counter_prefix :: Tensor v'4 ByteString
counter_prefix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
stats_aggregator,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
counter_prefix]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalSetStatsAggregatorDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "stats_aggregator" type: DT_RESOURCE }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "counter_prefix" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalSleepDataset :: [DataType] -- ^ __output_types__
                            -> Tensor v'1 Variant -- ^ __input_dataset__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __sleep_microseconds__
                            -> Tensor Build Variant -- ^ __handle__
experimentalSleepDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
experimentalSleepDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalSleepDataset' OpParams
forall a. a -> a
id
experimentalSleepDataset' :: OpParams ->
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __sleep_microseconds__
                             -> Tensor Build Variant -- ^ __handle__
experimentalSleepDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
experimentalSleepDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                          sleep_microseconds :: Tensor v'2 Int64
sleep_microseconds | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
sleep_microseconds]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalSleepDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "sleep_microseconds" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalSlidingWindowDataset :: [DataType] -- ^ __output_types__
                                    -> Tensor v'1 Variant -- ^ __input_dataset__
                                    -> Tensor v'2 Data.Int.Int64 -- ^ __window_size__
                                    -> Tensor v'3 Data.Int.Int64 -- ^ __window_shift__
                                    -> Tensor v'4 Data.Int.Int64 -- ^ __window_stride__
                                    -> Tensor Build Variant -- ^ __handle__
experimentalSlidingWindowDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
experimentalSlidingWindowDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
experimentalSlidingWindowDataset' OpParams
forall a. a -> a
id
experimentalSlidingWindowDataset' :: OpParams ->
                                     [DataType] -- ^ __output_types__
                                     -> Tensor v'1 Variant -- ^ __input_dataset__
                                     -> Tensor v'2 Data.Int.Int64 -- ^ __window_size__
                                     -> Tensor v'3 Data.Int.Int64 -- ^ __window_shift__
                                     -> Tensor v'4 Data.Int.Int64 -- ^ __window_stride__
                                     -> Tensor Build Variant -- ^ __handle__
experimentalSlidingWindowDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
experimentalSlidingWindowDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                                  window_size :: Tensor v'2 Int64
window_size window_shift :: Tensor v'3 Int64
window_shift
                                  window_stride :: Tensor v'4 Int64
window_stride | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
window_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
window_shift,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
window_stride]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalSlidingWindowDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "window_size" type: DT_INT64 }
input_arg { name: "window_shift" type: DT_INT64 }
input_arg { name: "window_stride" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalSqlDataset :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                          [DataType] -- ^ __output_types__
                          -> Tensor v'1 Data.ByteString.ByteString -- ^ __driver_name__
                          -> Tensor v'2 Data.ByteString.ByteString -- ^ __data_source_name__
                          -> Tensor v'3 Data.ByteString.ByteString -- ^ __query__
                          -> m' (Tensor Value Variant) -- ^ __handle__
experimentalSqlDataset :: [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
experimentalSqlDataset = OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
experimentalSqlDataset' OpParams
forall a. a -> a
id
experimentalSqlDataset' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                           OpParams ->
                           [DataType] -- ^ __output_types__
                           -> Tensor v'1 Data.ByteString.ByteString -- ^ __driver_name__
                           -> Tensor v'2 Data.ByteString.ByteString -- ^ __data_source_name__
                           -> Tensor v'3 Data.ByteString.ByteString -- ^ __query__
                           -> m' (Tensor Value Variant) -- ^ __handle__
experimentalSqlDataset' :: OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
experimentalSqlDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types driver_name :: Tensor v'1 ByteString
driver_name data_source_name :: Tensor v'2 ByteString
data_source_name
                        query :: Tensor v'3 ByteString
query | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
driver_name,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
data_source_name,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
query]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalSqlDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "driver_name" type: DT_STRING }
input_arg { name: "data_source_name" type: DT_STRING }
input_arg { name: "query" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalStatsAggregatorHandle :: forall m' . (MonadBuild m') =>
                                     m' (Tensor Value ResourceHandle) -- ^ __handle__
experimentalStatsAggregatorHandle :: m' (Tensor Value ResourceHandle)
experimentalStatsAggregatorHandle = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
experimentalStatsAggregatorHandle' OpParams
forall a. a -> a
id
experimentalStatsAggregatorHandle' :: forall m' . (MonadBuild m') => OpParams ->
                                      m' (Tensor Value ResourceHandle) -- ^ __handle__
experimentalStatsAggregatorHandle' :: OpParams -> m' (Tensor Value ResourceHandle)
experimentalStatsAggregatorHandle' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalStatsAggregatorHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
experimentalStatsAggregatorSummary :: forall v'1 m' . (MonadBuild m') =>
                                      Tensor v'1 ResourceHandle -- ^ __iterator__
                                      -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __summary__
experimentalStatsAggregatorSummary :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalStatsAggregatorSummary = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalStatsAggregatorSummary' OpParams
forall a. a -> a
id
experimentalStatsAggregatorSummary' :: forall v'1 m' . (MonadBuild m') =>
                                       OpParams ->
                                       Tensor v'1 ResourceHandle -- ^ __iterator__
                                       -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __summary__
experimentalStatsAggregatorSummary' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
experimentalStatsAggregatorSummary' op'options :: OpParams
op'options iterator :: Tensor v'1 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalStatsAggregatorSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "iterator" type: DT_RESOURCE }
output_arg { name: "summary" type: DT_STRING }
-}
-- | 
experimentalThreadPoolDataset :: forall v'1 v'2 m' . (MonadBuild m') =>
                                 [DataType] -- ^ __output_types__
                                 -> Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor v'2 ResourceHandle -- ^ __thread_pool__
                                 -> m' (Tensor Value Variant) -- ^ __handle__
experimentalThreadPoolDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
experimentalThreadPoolDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
experimentalThreadPoolDataset' OpParams
forall a. a -> a
id
experimentalThreadPoolDataset' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                  OpParams ->
                                  [DataType] -- ^ __output_types__
                                  -> Tensor v'1 Variant -- ^ __input_dataset__
                                  -> Tensor v'2 ResourceHandle -- ^ __thread_pool__
                                  -> m' (Tensor Value Variant) -- ^ __handle__
experimentalThreadPoolDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
experimentalThreadPoolDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                               thread_pool :: Tensor v'2 ResourceHandle
thread_pool | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
thread_pool]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalThreadPoolDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "thread_pool" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalThreadPoolHandle :: forall m' . (MonadBuild m') =>
                                ByteString -- ^ __display_name__
                                -> Data.Int.Int64 -- ^ __num_threads__
                                -> m' (Tensor Value ResourceHandle) -- ^ __handle__
experimentalThreadPoolHandle :: ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
experimentalThreadPoolHandle = OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
experimentalThreadPoolHandle' OpParams
forall a. a -> a
id
experimentalThreadPoolHandle' :: forall m' . (MonadBuild m') => OpParams ->
                                 ByteString -- ^ __display_name__
                                 -> Data.Int.Int64 -- ^ __num_threads__
                                 -> m' (Tensor Value ResourceHandle) -- ^ __handle__
experimentalThreadPoolHandle' :: OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
experimentalThreadPoolHandle' op'options :: OpParams
op'options display_name :: ByteString
display_name
                              num_threads :: Int64
num_threads | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ExperimentalThreadPoolHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "display_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
display_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_threads" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_threads
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "num_threads" type: "int" }
attr {
  name: "max_intra_op_parallelism" type: "int" default_value { i: 1 }
}
attr { name: "display_name" type: "string" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
experimentalUnbatchDataset :: [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor Build Variant -- ^ __handle__
experimentalUnbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUnbatchDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUnbatchDataset' OpParams
forall a. a -> a
id
experimentalUnbatchDataset' :: OpParams ->
                               [DataType] -- ^ __output_types__
                               -> Tensor v'1 Variant -- ^ __input_dataset__
                               -> Tensor Build Variant -- ^ __handle__
experimentalUnbatchDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUnbatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                            input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalUnbatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
experimentalUniqueDataset :: [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor Build Variant -- ^ __handle__
experimentalUniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUniqueDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUniqueDataset' OpParams
forall a. a -> a
id
experimentalUniqueDataset' :: OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor Build Variant -- ^ __handle__
experimentalUniqueDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
experimentalUniqueDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                           input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExperimentalUniqueDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
expint :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
expint :: Tensor v'1 t -> Tensor Build t
expint = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
expint' OpParams
forall a. a -> a
id
expint' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
expint' :: OpParams -> Tensor v'1 t -> Tensor Build t
expint' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Expint"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
expm1 :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
expm1 :: Tensor v'1 t -> Tensor Build t
expm1 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
expm1' OpParams
forall a. a -> a
id
expm1' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
expm1' :: OpParams -> Tensor v'1 t -> Tensor Build t
expm1' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Expm1"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
extractGlimpse :: Tensor v'1 Float -- ^ __input__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                  -> Tensor v'3 Float -- ^ __offsets__
                  -> Tensor Build Float -- ^ __glimpse__
extractGlimpse :: Tensor v'1 Float
-> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
extractGlimpse = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
extractGlimpse' OpParams
forall a. a -> a
id
extractGlimpse' :: OpParams ->
                   Tensor v'1 Float -- ^ __input__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                   -> Tensor v'3 Float -- ^ __offsets__
                   -> Tensor Build Float -- ^ __glimpse__
extractGlimpse' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
extractGlimpse' op'options :: OpParams
op'options input :: Tensor v'1 Float
input size :: Tensor v'2 Int32
size offsets :: Tensor v'3 Float
offsets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
offsets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExtractGlimpse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_FLOAT }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "offsets" type: DT_FLOAT }
output_arg { name: "glimpse" type: DT_FLOAT }
attr { name: "centered" type: "bool" default_value { b: true } }
attr { name: "normalized" type: "bool" default_value { b: true } }
attr {
  name: "uniform_noise" type: "bool" default_value { b: true }
}
attr {
  name: "noise" type: "string" default_value { s: "uniform" }
}
-}
-- | 
extractGlimpseV2 :: Tensor v'1 Float -- ^ __input__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                    -> Tensor v'3 Float -- ^ __offsets__
                    -> Tensor Build Float -- ^ __glimpse__
extractGlimpseV2 :: Tensor v'1 Float
-> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build Float
extractGlimpseV2 = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
extractGlimpseV2' OpParams
forall a. a -> a
id
extractGlimpseV2' :: OpParams ->
                     Tensor v'1 Float -- ^ __input__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                     -> Tensor v'3 Float -- ^ __offsets__
                     -> Tensor Build Float -- ^ __glimpse__
extractGlimpseV2' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build Float
extractGlimpseV2' op'options :: OpParams
op'options input :: Tensor v'1 Float
input size :: Tensor v'2 Int32
size offsets :: Tensor v'3 Float
offsets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
offsets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExtractGlimpseV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_FLOAT }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "offsets" type: DT_FLOAT }
output_arg { name: "glimpse" type: DT_FLOAT }
attr { name: "centered" type: "bool" default_value { b: true } }
attr { name: "normalized" type: "bool" default_value { b: true } }
attr {
  name: "uniform_noise" type: "bool" default_value { b: true }
}
attr {
  name: "noise" type: "string" default_value { s: "uniform" }
}
-}
-- | 
extractImagePatches :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Bool, Data.Int.Int16,
                                               Data.Int.Int32, Data.Int.Int64,
                                               Data.Int.Int8, Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) =>
                       ByteString -- ^ __padding__
                       -> Tensor v'1 t -- ^ __images__
                       -> Tensor Build t -- ^ __patches__
extractImagePatches :: ByteString -> Tensor v'1 t -> Tensor Build t
extractImagePatches = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Word16, Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
extractImagePatches' OpParams
forall a. a -> a
id
extractImagePatches' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Bool, Data.Int.Int16,
                                                Data.Int.Int32, Data.Int.Int64,
                                                Data.Int.Int8, Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) => OpParams ->
                        ByteString -- ^ __padding__
                        -> Tensor v'1 t -- ^ __images__
                        -> Tensor Build t -- ^ __patches__
extractImagePatches' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
extractImagePatches' op'options :: OpParams
op'options padding :: ByteString
padding images :: Tensor v'1 t
images | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExtractImagePatches"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
output_arg { name: "patches" type_attr: "T" }
attr {
  name: "ksizes" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "rates" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_BOOL
    }
  }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
extractJpegShape :: forall v'1 output_type . (OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] output_type) =>
                    Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
                    -> Tensor Build output_type -- ^ __image_shape__
extractJpegShape :: Tensor v'1 ByteString -> Tensor Build output_type
extractJpegShape = OpParams -> Tensor v'1 ByteString -> Tensor Build output_type
forall (v'1 :: * -> *) output_type.
OneOf '[Int32, Int64] output_type =>
OpParams -> Tensor v'1 ByteString -> Tensor Build output_type
extractJpegShape' OpParams
forall a. a -> a
id
extractJpegShape' :: forall v'1 output_type . (OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] output_type) =>
                     OpParams ->
                     Tensor v'1 Data.ByteString.ByteString -- ^ __contents__
                     -> Tensor Build output_type -- ^ __image_shape__
extractJpegShape' :: OpParams -> Tensor v'1 ByteString -> Tensor Build output_type
extractJpegShape' op'options :: OpParams
op'options contents :: Tensor v'1 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build output_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build output_type)
-> Build OpDef -> Tensor Build output_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
contents]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExtractJpegShape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "contents" type: DT_STRING }
output_arg { name: "image_shape" type_attr: "output_type" }
attr {
  name: "output_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
extractVolumePatches :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) =>
                        ByteString -- ^ __padding__
                        -> Tensor v'1 t -- ^ __input__
                        -> Tensor Build t -- ^ __patches__
extractVolumePatches :: ByteString -> Tensor v'1 t -> Tensor Build t
extractVolumePatches = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
extractVolumePatches' OpParams
forall a. a -> a
id
extractVolumePatches' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t) => OpParams ->
                         ByteString -- ^ __padding__
                         -> Tensor v'1 t -- ^ __input__
                         -> Tensor Build t -- ^ __patches__
extractVolumePatches' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
extractVolumePatches' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ExtractVolumePatches"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "patches" type_attr: "T" }
attr {
  name: "ksizes" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
fFT :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float)] tcomplex) =>
       Tensor v'1 tcomplex -- ^ __input__
       -> Tensor Build tcomplex -- ^ __output__
fFT :: Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT' OpParams
forall a. a -> a
id
fFT' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float)] tcomplex) =>
        OpParams ->
        Tensor v'1 tcomplex -- ^ __input__
        -> Tensor Build tcomplex -- ^ __output__
fFT' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
fFT2D :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float)] tcomplex) =>
         Tensor v'1 tcomplex -- ^ __input__
         -> Tensor Build tcomplex -- ^ __output__
fFT2D :: Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT2D = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT2D' OpParams
forall a. a -> a
id
fFT2D' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float)] tcomplex) =>
          OpParams ->
          Tensor v'1 tcomplex -- ^ __input__
          -> Tensor Build tcomplex -- ^ __output__
fFT2D' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT2D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
fFT3D :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float)] tcomplex) =>
         Tensor v'1 tcomplex -- ^ __input__
         -> Tensor Build tcomplex -- ^ __output__
fFT3D :: Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT3D = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT3D' OpParams
forall a. a -> a
id
fFT3D' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float)] tcomplex) =>
          OpParams ->
          Tensor v'1 tcomplex -- ^ __input__
          -> Tensor Build tcomplex -- ^ __output__
fFT3D' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
fFT3D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
fIFOQueue :: forall m' . (MonadBuild m') => [DataType] -- ^ __component_types__
             -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
fIFOQueue :: [DataType] -> m' (Tensor Ref ByteString)
fIFOQueue = OpParams -> [DataType] -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Ref ByteString)
fIFOQueue' OpParams
forall a. a -> a
id
fIFOQueue' :: forall m' . (MonadBuild m') => OpParams ->
              [DataType] -- ^ __component_types__
              -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
fIFOQueue' :: OpParams -> [DataType] -> m' (Tensor Ref ByteString)
fIFOQueue' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FIFOQueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
fIFOQueueV2 :: forall m' . (MonadBuild m') =>
               [DataType] -- ^ __component_types__
               -> m' (Tensor Value ResourceHandle) -- ^ __handle__
fIFOQueueV2 :: [DataType] -> m' (Tensor Value ResourceHandle)
fIFOQueueV2 = OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
fIFOQueueV2' OpParams
forall a. a -> a
id
fIFOQueueV2' :: forall m' . (MonadBuild m') => OpParams ->
                [DataType] -- ^ __component_types__
                -> m' (Tensor Value ResourceHandle) -- ^ __handle__
fIFOQueueV2' :: OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
fIFOQueueV2' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FIFOQueueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
fact :: Tensor Build Data.ByteString.ByteString -- ^ __fact__
fact :: Tensor Build ByteString
fact = OpParams -> Tensor Build ByteString
fact' OpParams
forall a. a -> a
id
fact' :: OpParams ->
         Tensor Build Data.ByteString.ByteString -- ^ __fact__
fact' :: OpParams -> Tensor Build ByteString
fact' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Fact"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "fact" type: DT_STRING }
-}
-- | 
fakeParam :: forall dtype . (TensorType dtype) => Shape -- ^ __shape__
             -> Tensor Build dtype -- ^ __output__
fakeParam :: Shape -> Tensor Build dtype
fakeParam = OpParams -> Shape -> Tensor Build dtype
forall dtype.
TensorType dtype =>
OpParams -> Shape -> Tensor Build dtype
fakeParam' OpParams
forall a. a -> a
id
fakeParam' :: forall dtype . (TensorType dtype) => OpParams ->
              Shape -- ^ __shape__
              -> Tensor Build dtype -- ^ __output__
fakeParam' :: OpParams -> Shape -> Tensor Build dtype
fakeParam' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeParam"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
-}
-- | 
fakeQuantWithMinMaxArgs :: Tensor v'1 Float -- ^ __inputs__
                           -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxArgs :: Tensor v'1 Float -> Tensor Build Float
fakeQuantWithMinMaxArgs = OpParams -> Tensor v'1 Float -> Tensor Build Float
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Float -> Tensor Build Float
fakeQuantWithMinMaxArgs' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxArgs' :: OpParams ->
                            Tensor v'1 Float -- ^ __inputs__
                            -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxArgs' :: OpParams -> Tensor v'1 Float -> Tensor Build Float
fakeQuantWithMinMaxArgs' op'options :: OpParams
op'options inputs :: Tensor v'1 Float
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxArgs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_FLOAT }
output_arg { name: "outputs" type: DT_FLOAT }
attr { name: "min" type: "float" default_value { f: -6.0 } }
attr { name: "max" type: "float" default_value { f: 6.0 } }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQuantWithMinMaxArgsGradient :: Tensor v'1 Float -- ^ __gradients__
                                   -> Tensor v'2 Float -- ^ __inputs__
                                   -> Tensor Build Float -- ^ __backprops__
fakeQuantWithMinMaxArgsGradient :: Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
fakeQuantWithMinMaxArgsGradient = OpParams
-> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
fakeQuantWithMinMaxArgsGradient' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxArgsGradient' :: OpParams ->
                                    Tensor v'1 Float -- ^ __gradients__
                                    -> Tensor v'2 Float -- ^ __inputs__
                                    -> Tensor Build Float -- ^ __backprops__
fakeQuantWithMinMaxArgsGradient' :: OpParams
-> Tensor v'1 Float -> Tensor v'2 Float -> Tensor Build Float
fakeQuantWithMinMaxArgsGradient' op'options :: OpParams
op'options gradients :: Tensor v'1 Float
gradients
                                 inputs :: Tensor v'2 Float
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
gradients,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxArgsGradient"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "inputs" type: DT_FLOAT }
output_arg { name: "backprops" type: DT_FLOAT }
attr { name: "min" type: "float" default_value { f: -6.0 } }
attr { name: "max" type: "float" default_value { f: 6.0 } }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQuantWithMinMaxVars :: Tensor v'1 Float -- ^ __inputs__
                           -> Tensor v'2 Float -- ^ __min__
                           -> Tensor v'3 Float -- ^ __max__
                           -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxVars :: Tensor v'1 Float
-> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
fakeQuantWithMinMaxVars = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
fakeQuantWithMinMaxVars' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxVars' :: OpParams ->
                            Tensor v'1 Float -- ^ __inputs__
                            -> Tensor v'2 Float -- ^ __min__
                            -> Tensor v'3 Float -- ^ __max__
                            -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxVars' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
fakeQuantWithMinMaxVars' op'options :: OpParams
op'options inputs :: Tensor v'1 Float
inputs min :: Tensor v'2 Float
min max :: Tensor v'3 Float
max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
inputs,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxVars"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_FLOAT }
input_arg { name: "min" type: DT_FLOAT }
input_arg { name: "max" type: DT_FLOAT }
output_arg { name: "outputs" type: DT_FLOAT }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQuantWithMinMaxVarsGradient :: Tensor v'1 Float -- ^ __gradients__
                                   -> Tensor v'2 Float -- ^ __inputs__
                                   -> Tensor v'3 Float -- ^ __min__
                                   -> Tensor v'4 Float -- ^ __max__
                                   -> (Tensor Build Float, Tensor Build Float,
                                       Tensor Build Float)
                                   -- ^ (__backprops_wrt_input__, __backprop_wrt_min__, __backprop_wrt_max__)
                                   --
                                   -- * __backprops_wrt_input__
                                   --
                                   -- * __backprop_wrt_min__
                                   --
                                   -- * __backprop_wrt_max__
fakeQuantWithMinMaxVarsGradient :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsGradient = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsGradient' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxVarsGradient' :: OpParams ->
                                    Tensor v'1 Float -- ^ __gradients__
                                    -> Tensor v'2 Float -- ^ __inputs__
                                    -> Tensor v'3 Float -- ^ __min__
                                    -> Tensor v'4 Float -- ^ __max__
                                    -> (Tensor Build Float, Tensor Build Float,
                                        Tensor Build Float)
                                    -- ^ (__backprops_wrt_input__, __backprop_wrt_min__, __backprop_wrt_max__)
                                    --
                                    -- * __backprops_wrt_input__
                                    --
                                    -- * __backprop_wrt_min__
                                    --
                                    -- * __backprop_wrt_max__
fakeQuantWithMinMaxVarsGradient' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsGradient' op'options :: OpParams
op'options gradients :: Tensor v'1 Float
gradients inputs :: Tensor v'2 Float
inputs min :: Tensor v'3 Float
min
                                 max :: Tensor v'4 Float
max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Float, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
gradients,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
inputs,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxVarsGradient"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "inputs" type: DT_FLOAT }
input_arg { name: "min" type: DT_FLOAT }
input_arg { name: "max" type: DT_FLOAT }
output_arg { name: "backprops_wrt_input" type: DT_FLOAT }
output_arg { name: "backprop_wrt_min" type: DT_FLOAT }
output_arg { name: "backprop_wrt_max" type: DT_FLOAT }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQuantWithMinMaxVarsPerChannel :: Tensor v'1 Float -- ^ __inputs__
                                     -> Tensor v'2 Float -- ^ __min__
                                     -> Tensor v'3 Float -- ^ __max__
                                     -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxVarsPerChannel :: Tensor v'1 Float
-> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build Float
fakeQuantWithMinMaxVarsPerChannel = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
fakeQuantWithMinMaxVarsPerChannel' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxVarsPerChannel' :: OpParams ->
                                      Tensor v'1 Float -- ^ __inputs__
                                      -> Tensor v'2 Float -- ^ __min__
                                      -> Tensor v'3 Float -- ^ __max__
                                      -> Tensor Build Float -- ^ __outputs__
fakeQuantWithMinMaxVarsPerChannel' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor Build Float
fakeQuantWithMinMaxVarsPerChannel' op'options :: OpParams
op'options inputs :: Tensor v'1 Float
inputs min :: Tensor v'2 Float
min
                                   max :: Tensor v'3 Float
max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
inputs,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxVarsPerChannel"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_FLOAT }
input_arg { name: "min" type: DT_FLOAT }
input_arg { name: "max" type: DT_FLOAT }
output_arg { name: "outputs" type: DT_FLOAT }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQuantWithMinMaxVarsPerChannelGradient :: Tensor v'1 Float -- ^ __gradients__
                                             -> Tensor v'2 Float -- ^ __inputs__
                                             -> Tensor v'3 Float -- ^ __min__
                                             -> Tensor v'4 Float -- ^ __max__
                                             -> (Tensor Build Float,
                                                 Tensor Build Float,
                                                 Tensor Build Float)
                                             -- ^ (__backprops_wrt_input__, __backprop_wrt_min__, __backprop_wrt_max__)
                                             --
                                             -- * __backprops_wrt_input__
                                             --
                                             -- * __backprop_wrt_min__
                                             --
                                             -- * __backprop_wrt_max__
fakeQuantWithMinMaxVarsPerChannelGradient :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsPerChannelGradient = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsPerChannelGradient' OpParams
forall a. a -> a
id
fakeQuantWithMinMaxVarsPerChannelGradient' :: OpParams ->
                                              Tensor v'1 Float -- ^ __gradients__
                                              -> Tensor v'2 Float -- ^ __inputs__
                                              -> Tensor v'3 Float -- ^ __min__
                                              -> Tensor v'4 Float -- ^ __max__
                                              -> (Tensor Build Float,
                                                  Tensor Build Float,
                                                  Tensor Build Float)
                                              -- ^ (__backprops_wrt_input__, __backprop_wrt_min__, __backprop_wrt_max__)
                                              --
                                              -- * __backprops_wrt_input__
                                              --
                                              -- * __backprop_wrt_min__
                                              --
                                              -- * __backprop_wrt_max__
fakeQuantWithMinMaxVarsPerChannelGradient' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
fakeQuantWithMinMaxVarsPerChannelGradient' op'options :: OpParams
op'options gradients :: Tensor v'1 Float
gradients inputs :: Tensor v'2 Float
inputs min :: Tensor v'3 Float
min
                                           max :: Tensor v'4 Float
max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Float, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build Float, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
gradients,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
inputs,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FakeQuantWithMinMaxVarsPerChannelGradient"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type: DT_FLOAT }
input_arg { name: "inputs" type: DT_FLOAT }
input_arg { name: "min" type: DT_FLOAT }
input_arg { name: "max" type: DT_FLOAT }
output_arg { name: "backprops_wrt_input" type: DT_FLOAT }
output_arg { name: "backprop_wrt_min" type: DT_FLOAT }
output_arg { name: "backprop_wrt_max" type: DT_FLOAT }
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
-}
-- | 
fakeQueue :: forall v'1 m' . (MonadBuild m') =>
             Tensor v'1 ResourceHandle -- ^ __resource__
             -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
fakeQueue :: Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
fakeQueue = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
fakeQueue' OpParams
forall a. a -> a
id
fakeQueue' :: forall v'1 m' . (MonadBuild m') => OpParams ->
              Tensor v'1 ResourceHandle -- ^ __resource__
              -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
fakeQueue' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
fakeQueue' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FakeQueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_STRING is_ref: true }
-}
-- | 
fill :: forall v'1 v'2 t index_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] index_type) =>
        Tensor v'1 index_type -- ^ __dims__
        -> Tensor v'2 t -- ^ __value__
        -> Tensor Build t -- ^ __output__
fill :: Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
fill = OpParams -> Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t index_type.
(TensorType t, OneOf '[Int32, Int64] index_type) =>
OpParams -> Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
fill' OpParams
forall a. a -> a
id
fill' :: forall v'1 v'2 t index_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] index_type) =>
         OpParams ->
         Tensor v'1 index_type -- ^ __dims__
         -> Tensor v'2 t -- ^ __value__
         -> Tensor Build t -- ^ __output__
fill' :: OpParams -> Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
fill' op'options :: OpParams
op'options dims :: Tensor v'1 index_type
dims value :: Tensor v'2 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 index_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 index_type
dims,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Fill"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "index_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (index_type
forall a. HasCallStack => a
undefined :: index_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dims" type_attr: "index_type" }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "index_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
filterByLastComponentDataset :: [DataType] -- ^ __output_types__
                                -> Tensor v'1 Variant -- ^ __input_dataset__
                                -> Tensor Build Variant -- ^ __output__
filterByLastComponentDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
filterByLastComponentDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
filterByLastComponentDataset' OpParams
forall a. a -> a
id
filterByLastComponentDataset' :: OpParams ->
                                 [DataType] -- ^ __output_types__
                                 -> Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor Build Variant -- ^ __output__
filterByLastComponentDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
filterByLastComponentDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                              input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FilterByLastComponentDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "output" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
fingerprint :: forall v'1 v'2 t . (TensorType t) => Tensor v'1 t -- ^ __data__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
               -> Tensor Build Data.Word.Word8 -- ^ __fingerprint__
fingerprint :: Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
fingerprint = OpParams
-> Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
fingerprint' OpParams
forall a. a -> a
id
fingerprint' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                Tensor v'1 t -- ^ __data__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
                -> Tensor Build Data.Word.Word8 -- ^ __fingerprint__
fingerprint' :: OpParams
-> Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
fingerprint' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' method :: Tensor v'2 ByteString
method | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
method]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Fingerprint"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "method" type: DT_STRING }
output_arg { name: "fingerprint" type: DT_UINT8 }
attr { name: "T" type: "type" }
-}
-- | 
fixedLengthRecordDataset :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                            Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __header_bytes__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __record_bytes__
                            -> Tensor v'4 Data.Int.Int64 -- ^ __footer_bytes__
                            -> Tensor v'5 Data.Int.Int64 -- ^ __buffer_size__
                            -> m' (Tensor Value Variant) -- ^ __handle__
fixedLengthRecordDataset :: Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> m' (Tensor Value Variant)
fixedLengthRecordDataset = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> m' (Tensor Value Variant)
fixedLengthRecordDataset' OpParams
forall a. a -> a
id
fixedLengthRecordDataset' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                             OpParams ->
                             Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __header_bytes__
                             -> Tensor v'3 Data.Int.Int64 -- ^ __record_bytes__
                             -> Tensor v'4 Data.Int.Int64 -- ^ __footer_bytes__
                             -> Tensor v'5 Data.Int.Int64 -- ^ __buffer_size__
                             -> m' (Tensor Value Variant) -- ^ __handle__
fixedLengthRecordDataset' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> m' (Tensor Value Variant)
fixedLengthRecordDataset' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames header_bytes :: Tensor v'2 Int64
header_bytes record_bytes :: Tensor v'3 Int64
record_bytes
                          footer_bytes :: Tensor v'4 Int64
footer_bytes buffer_size :: Tensor v'5 Int64
buffer_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
header_bytes,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
record_bytes,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
footer_bytes,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
buffer_size]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FixedLengthRecordDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "header_bytes" type: DT_INT64 }
input_arg { name: "record_bytes" type: DT_INT64 }
input_arg { name: "footer_bytes" type: DT_INT64 }
input_arg { name: "buffer_size" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
fixedLengthRecordDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6
                              m' . (MonadBuild m') =>
                              Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __header_bytes__
                              -> Tensor v'3 Data.Int.Int64 -- ^ __record_bytes__
                              -> Tensor v'4 Data.Int.Int64 -- ^ __footer_bytes__
                              -> Tensor v'5 Data.Int.Int64 -- ^ __buffer_size__
                              -> Tensor v'6 Data.ByteString.ByteString -- ^ __compression_type__
                              -> m' (Tensor Value Variant) -- ^ __handle__
fixedLengthRecordDatasetV2 :: Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ByteString
-> m' (Tensor Value Variant)
fixedLengthRecordDatasetV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ByteString
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ByteString
-> m' (Tensor Value Variant)
fixedLengthRecordDatasetV2' OpParams
forall a. a -> a
id
fixedLengthRecordDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                               m' . (MonadBuild m') => OpParams ->
                               Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                               -> Tensor v'2 Data.Int.Int64 -- ^ __header_bytes__
                               -> Tensor v'3 Data.Int.Int64 -- ^ __record_bytes__
                               -> Tensor v'4 Data.Int.Int64 -- ^ __footer_bytes__
                               -> Tensor v'5 Data.Int.Int64 -- ^ __buffer_size__
                               -> Tensor v'6 Data.ByteString.ByteString -- ^ __compression_type__
                               -> m' (Tensor Value Variant) -- ^ __handle__
fixedLengthRecordDatasetV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ByteString
-> m' (Tensor Value Variant)
fixedLengthRecordDatasetV2' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames header_bytes :: Tensor v'2 Int64
header_bytes record_bytes :: Tensor v'3 Int64
record_bytes
                            footer_bytes :: Tensor v'4 Int64
footer_bytes buffer_size :: Tensor v'5 Int64
buffer_size
                            compression_type :: Tensor v'6 ByteString
compression_type | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
header_bytes,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
record_bytes,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
footer_bytes,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
buffer_size,
                                                             Tensor v'6 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 ByteString
compression_type]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FixedLengthRecordDatasetV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "header_bytes" type: DT_INT64 }
input_arg { name: "record_bytes" type: DT_INT64 }
input_arg { name: "footer_bytes" type: DT_INT64 }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "compression_type" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
fixedLengthRecordReader :: forall m' . (MonadBuild m') =>
                           Data.Int.Int64 -- ^ __record_bytes__
                           -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
fixedLengthRecordReader :: Int64 -> m' (Tensor Ref ByteString)
fixedLengthRecordReader = OpParams -> Int64 -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> m' (Tensor Ref ByteString)
fixedLengthRecordReader' OpParams
forall a. a -> a
id
fixedLengthRecordReader' :: forall m' . (MonadBuild m') => OpParams ->
                            Data.Int.Int64 -- ^ __record_bytes__
                            -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
fixedLengthRecordReader' :: OpParams -> Int64 -> m' (Tensor Ref ByteString)
fixedLengthRecordReader' op'options :: OpParams
op'options record_bytes :: Int64
record_bytes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FixedLengthRecordReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "record_bytes" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
record_bytes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr { name: "header_bytes" type: "int" default_value { i: 0 } }
attr { name: "record_bytes" type: "int" }
attr { name: "footer_bytes" type: "int" default_value { i: 0 } }
attr { name: "hop_bytes" type: "int" default_value { i: 0 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
fixedLengthRecordReaderV2 :: forall m' . (MonadBuild m') =>
                             Data.Int.Int64 -- ^ __record_bytes__
                             -> m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
fixedLengthRecordReaderV2 :: Int64 -> m' (Tensor Value ResourceHandle)
fixedLengthRecordReaderV2 = OpParams -> Int64 -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> m' (Tensor Value ResourceHandle)
fixedLengthRecordReaderV2' OpParams
forall a. a -> a
id
fixedLengthRecordReaderV2' :: forall m' . (MonadBuild m') => OpParams ->
                              Data.Int.Int64 -- ^ __record_bytes__
                              -> m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
fixedLengthRecordReaderV2' :: OpParams -> Int64 -> m' (Tensor Value ResourceHandle)
fixedLengthRecordReaderV2' op'options :: OpParams
op'options record_bytes :: Int64
record_bytes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FixedLengthRecordReaderV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "record_bytes" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
record_bytes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_RESOURCE }
attr { name: "header_bytes" type: "int" default_value { i: 0 } }
attr { name: "record_bytes" type: "int" }
attr { name: "footer_bytes" type: "int" default_value { i: 0 } }
attr { name: "hop_bytes" type: "int" default_value { i: 0 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr { name: "encoding" type: "string" default_value { s: "" } }
-}
-- | 
fixedUnigramCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                                Data.Int.Int64 -- ^ __num_sampled__
                                -> Data.Int.Int64 -- ^ __num_true__
                                -> Data.Int.Int64 -- ^ __range_max__
                                -> Bool -- ^ __unique__
                                -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                -> m' ((Tensor Value Data.Int.Int64,
                                        Tensor Value Float, Tensor Value Float))
                                -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                --
                                -- * __sampled_candidates__
                                --
                                -- * __true_expected_count__
                                --
                                -- * __sampled_expected_count__
fixedUnigramCandidateSampler :: Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
fixedUnigramCandidateSampler = OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
fixedUnigramCandidateSampler' OpParams
forall a. a -> a
id
fixedUnigramCandidateSampler' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                                 Data.Int.Int64 -- ^ __num_sampled__
                                 -> Data.Int.Int64 -- ^ __num_true__
                                 -> Data.Int.Int64 -- ^ __range_max__
                                 -> Bool -- ^ __unique__
                                 -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                 -> m' ((Tensor Value Data.Int.Int64,
                                         Tensor Value Float,
                                         Tensor Value Float))
                                 -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                 --
                                 -- * __sampled_candidates__
                                 --
                                 -- * __true_expected_count__
                                 --
                                 -- * __sampled_expected_count__
fixedUnigramCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
fixedUnigramCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true range_max :: Int64
range_max unique :: Bool
unique
                              true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FixedUnigramCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "range_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
range_max
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "range_max" type: "int" has_minimum: true minimum: 1 }
attr { name: "vocab_file" type: "string" default_value { s: "" } }
attr { name: "distortion" type: "float" default_value { f: 1.0 } }
attr {
  name: "num_reserved_ids" type: "int" default_value { i: 0 }
}
attr {
  name: "num_shards"
  type: "int"
  default_value { i: 1 }
  has_minimum: true
  minimum: 1
}
attr {
  name: "shard" type: "int" default_value { i: 0 } has_minimum: true
}
attr {
  name: "unigrams" type: "list(float)" default_value { list { } }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
floor :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
floor :: Tensor v'1 t -> Tensor Build t
floor = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
floor' OpParams
forall a. a -> a
id
floor' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
floor' :: OpParams -> Tensor v'1 t -> Tensor Build t
floor' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Floor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
floorDiv :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word8,
                                        Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
floorDiv :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorDiv = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorDiv' OpParams
forall a. a -> a
id
floorDiv' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
floorDiv' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorDiv' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FloorDiv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
floorMod :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                        Data.Word.Word16, Data.Word.Word64,
                                        Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
floorMod :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorMod = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64, Word16, Word64, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorMod' OpParams
forall a. a -> a
id
floorMod' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                         Data.Word.Word16, Data.Word.Word64,
                                         Double, Float] t) => OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
floorMod' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
floorMod' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FloorMod"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT64
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
-}
-- | 
flushSummaryWriter :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> m' (ControlNode)
flushSummaryWriter :: Tensor v'1 ResourceHandle -> m' ControlNode
flushSummaryWriter = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
flushSummaryWriter' OpParams
forall a. a -> a
id
flushSummaryWriter' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __writer__
                       -> m' (ControlNode)
flushSummaryWriter' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
flushSummaryWriter' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "FlushSummaryWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
-}
-- | 
fractionalAvgPool :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Double, Float] t) =>
                     Tensor v'1 t -- ^ __value__
                     -> (Tensor Build t, Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64)
                     -- ^ (__output__, __row_pooling_sequence__, __col_pooling_sequence__)
                     --
                     -- * __output__
                     --
                     -- * __row_pooling_sequence__
                     --
                     -- * __col_pooling_sequence__
fractionalAvgPool :: Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalAvgPool = OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalAvgPool' OpParams
forall a. a -> a
id
fractionalAvgPool' :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                              Double, Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __value__
                      -> (Tensor Build t, Tensor Build Data.Int.Int64,
                          Tensor Build Data.Int.Int64)
                      -- ^ (__output__, __row_pooling_sequence__, __col_pooling_sequence__)
                      --
                      -- * __output__
                      --
                      -- * __row_pooling_sequence__
                      --
                      -- * __col_pooling_sequence__
fractionalAvgPool' :: OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalAvgPool' op'options :: OpParams
op'options value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FractionalAvgPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "row_pooling_sequence" type: DT_INT64 }
output_arg { name: "col_pooling_sequence" type: DT_INT64 }
attr {
  name: "pooling_ratio"
  type: "list(float)"
  has_minimum: true
  minimum: 4
}
attr {
  name: "pseudo_random" type: "bool" default_value { b: false }
}
attr {
  name: "overlapping" type: "bool" default_value { b: false }
}
attr {
  name: "deterministic" type: "bool" default_value { b: false }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT type: DT_DOUBLE type: DT_INT32 type: DT_INT64
    }
  }
}
-}
-- | 
fractionalAvgPoolGrad :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Double,
                                                             Float] t) =>
                         Tensor v'1 Data.Int.Int64 -- ^ __orig_input_tensor_shape__
                         -> Tensor v'2 t -- ^ __out_backprop__
                         -> Tensor v'3 Data.Int.Int64 -- ^ __row_pooling_sequence__
                         -> Tensor v'4 Data.Int.Int64 -- ^ __col_pooling_sequence__
                         -> Tensor Build t -- ^ __output__
fractionalAvgPoolGrad :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
fractionalAvgPoolGrad = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
fractionalAvgPoolGrad' OpParams
forall a. a -> a
id
fractionalAvgPoolGrad' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Double,
                                                              Float] t) =>
                          OpParams ->
                          Tensor v'1 Data.Int.Int64 -- ^ __orig_input_tensor_shape__
                          -> Tensor v'2 t -- ^ __out_backprop__
                          -> Tensor v'3 Data.Int.Int64 -- ^ __row_pooling_sequence__
                          -> Tensor v'4 Data.Int.Int64 -- ^ __col_pooling_sequence__
                          -> Tensor Build t -- ^ __output__
fractionalAvgPoolGrad' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
fractionalAvgPoolGrad' op'options :: OpParams
op'options orig_input_tensor_shape :: Tensor v'1 Int64
orig_input_tensor_shape out_backprop :: Tensor v'2 t
out_backprop
                       row_pooling_sequence :: Tensor v'3 Int64
row_pooling_sequence
                       col_pooling_sequence :: Tensor v'4 Int64
col_pooling_sequence | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
orig_input_tensor_shape,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
out_backprop,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
row_pooling_sequence,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
col_pooling_sequence]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FractionalAvgPoolGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input_tensor_shape" type: DT_INT64 }
input_arg { name: "out_backprop" type_attr: "T" }
input_arg { name: "row_pooling_sequence" type: DT_INT64 }
input_arg { name: "col_pooling_sequence" type: DT_INT64 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "overlapping" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT type: DT_DOUBLE type: DT_INT32 type: DT_INT64
    }
  }
}
-}
-- | 
fractionalMaxPool :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Double, Float] t) =>
                     Tensor v'1 t -- ^ __value__
                     -> (Tensor Build t, Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64)
                     -- ^ (__output__, __row_pooling_sequence__, __col_pooling_sequence__)
                     --
                     -- * __output__
                     --
                     -- * __row_pooling_sequence__
                     --
                     -- * __col_pooling_sequence__
fractionalMaxPool :: Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalMaxPool = OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalMaxPool' OpParams
forall a. a -> a
id
fractionalMaxPool' :: forall v'1 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                              Double, Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __value__
                      -> (Tensor Build t, Tensor Build Data.Int.Int64,
                          Tensor Build Data.Int.Int64)
                      -- ^ (__output__, __row_pooling_sequence__, __col_pooling_sequence__)
                      --
                      -- * __output__
                      --
                      -- * __row_pooling_sequence__
                      --
                      -- * __col_pooling_sequence__
fractionalMaxPool' :: OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
fractionalMaxPool' op'options :: OpParams
op'options value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FractionalMaxPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "row_pooling_sequence" type: DT_INT64 }
output_arg { name: "col_pooling_sequence" type: DT_INT64 }
attr {
  name: "pooling_ratio"
  type: "list(float)"
  has_minimum: true
  minimum: 4
}
attr {
  name: "pseudo_random" type: "bool" default_value { b: false }
}
attr {
  name: "overlapping" type: "bool" default_value { b: false }
}
attr {
  name: "deterministic" type: "bool" default_value { b: false }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT type: DT_DOUBLE type: DT_INT32 type: DT_INT64
    }
  }
}
-}
-- | 
fractionalMaxPoolGrad :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Double,
                                                                 Float] t) =>
                         Tensor v'1 t -- ^ __orig_input__
                         -> Tensor v'2 t -- ^ __orig_output__
                         -> Tensor v'3 t -- ^ __out_backprop__
                         -> Tensor v'4 Data.Int.Int64 -- ^ __row_pooling_sequence__
                         -> Tensor v'5 Data.Int.Int64 -- ^ __col_pooling_sequence__
                         -> Tensor Build t -- ^ __output__
fractionalMaxPoolGrad :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build t
fractionalMaxPoolGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf '[Int32, Int64, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build t
fractionalMaxPoolGrad' OpParams
forall a. a -> a
id
fractionalMaxPoolGrad' :: forall v'1 v'2 v'3 v'4 v'5
                          t . (OneOf '[Data.Int.Int32, Data.Int.Int64, Double,
                                       Float] t) => OpParams ->
                          Tensor v'1 t -- ^ __orig_input__
                          -> Tensor v'2 t -- ^ __orig_output__
                          -> Tensor v'3 t -- ^ __out_backprop__
                          -> Tensor v'4 Data.Int.Int64 -- ^ __row_pooling_sequence__
                          -> Tensor v'5 Data.Int.Int64 -- ^ __col_pooling_sequence__
                          -> Tensor Build t -- ^ __output__
fractionalMaxPoolGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build t
fractionalMaxPoolGrad' op'options :: OpParams
op'options orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output out_backprop :: Tensor v'3 t
out_backprop
                       row_pooling_sequence :: Tensor v'4 Int64
row_pooling_sequence
                       col_pooling_sequence :: Tensor v'5 Int64
col_pooling_sequence | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
out_backprop,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
row_pooling_sequence,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
col_pooling_sequence]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FractionalMaxPoolGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "out_backprop" type_attr: "T" }
input_arg { name: "row_pooling_sequence" type: DT_INT64 }
input_arg { name: "col_pooling_sequence" type: DT_INT64 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "overlapping" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT type: DT_DOUBLE type: DT_INT32 type: DT_INT64
    }
  }
}
-}
-- | 
fresnelCos :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
fresnelCos :: Tensor v'1 t -> Tensor Build t
fresnelCos = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
fresnelCos' OpParams
forall a. a -> a
id
fresnelCos' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor Build t -- ^ __y__
fresnelCos' :: OpParams -> Tensor v'1 t -> Tensor Build t
fresnelCos' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FresnelCos"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
fresnelSin :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
fresnelSin :: Tensor v'1 t -> Tensor Build t
fresnelSin = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
fresnelSin' OpParams
forall a. a -> a
id
fresnelSin' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor Build t -- ^ __y__
fresnelSin' :: OpParams -> Tensor v'1 t -> Tensor Build t
fresnelSin' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FresnelSin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
fusedBatchNorm :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Float] t) =>
                  Tensor v'1 t -- ^ __x__
                  -> Tensor v'2 t -- ^ __scale__
                  -> Tensor v'3 t -- ^ __offset__
                  -> Tensor v'4 t -- ^ __mean__
                  -> Tensor v'5 t -- ^ __variance__
                  -> (Tensor Build t, Tensor Build t, Tensor Build t,
                      Tensor Build t, Tensor Build t)
                  -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__)
                  --
                  -- * __y__
                  --
                  -- * __batch_mean__
                  --
                  -- * __batch_variance__
                  --
                  -- * __reserve_space_1__
                  --
                  -- * __reserve_space_2__
fusedBatchNorm :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNorm = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNorm' OpParams
forall a. a -> a
id
fusedBatchNorm' :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Float] t) =>
                   OpParams ->
                   Tensor v'1 t -- ^ __x__
                   -> Tensor v'2 t -- ^ __scale__
                   -> Tensor v'3 t -- ^ __offset__
                   -> Tensor v'4 t -- ^ __mean__
                   -> Tensor v'5 t -- ^ __variance__
                   -> (Tensor Build t, Tensor Build t, Tensor Build t,
                       Tensor Build t, Tensor Build t)
                   -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__)
                   --
                   -- * __y__
                   --
                   -- * __batch_mean__
                   --
                   -- * __batch_variance__
                   --
                   -- * __reserve_space_1__
                   --
                   -- * __reserve_space_2__
fusedBatchNorm' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNorm' op'options :: OpParams
op'options x :: Tensor v'1 t
x scale :: Tensor v'2 t
scale offset :: Tensor v'3 t
offset mean :: Tensor v'4 t
mean variance :: Tensor v'5 t
variance | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
scale,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
offset,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
mean,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
variance]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNorm"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type_attr: "T" }
input_arg { name: "offset" type_attr: "T" }
input_arg { name: "mean" type_attr: "T" }
input_arg { name: "variance" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "batch_mean" type_attr: "T" }
output_arg { name: "batch_variance" type_attr: "T" }
output_arg { name: "reserve_space_1" type_attr: "T" }
output_arg { name: "reserve_space_2" type_attr: "T" }
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "exponential_avg_factor"
  type: "float"
  default_value { f: 1.0 }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedBatchNormGrad :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Float] t) =>
                      Tensor v'1 t -- ^ __y_backprop__
                      -> Tensor v'2 t -- ^ __x__
                      -> Tensor v'3 t -- ^ __scale__
                      -> Tensor v'4 t -- ^ __reserve_space_1__
                      -> Tensor v'5 t -- ^ __reserve_space_2__
                      -> (Tensor Build t, Tensor Build t, Tensor Build t,
                          Tensor Build t, Tensor Build t)
                      -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_3__, __reserve_space_4__)
                      --
                      -- * __x_backprop__
                      --
                      -- * __scale_backprop__
                      --
                      -- * __offset_backprop__
                      --
                      -- * __reserve_space_3__
                      --
                      -- * __reserve_space_4__
fusedBatchNormGrad :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNormGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNormGrad' OpParams
forall a. a -> a
id
fusedBatchNormGrad' :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Float] t) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __y_backprop__
                       -> Tensor v'2 t -- ^ __x__
                       -> Tensor v'3 t -- ^ __scale__
                       -> Tensor v'4 t -- ^ __reserve_space_1__
                       -> Tensor v'5 t -- ^ __reserve_space_2__
                       -> (Tensor Build t, Tensor Build t, Tensor Build t,
                           Tensor Build t, Tensor Build t)
                       -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_3__, __reserve_space_4__)
                       --
                       -- * __x_backprop__
                       --
                       -- * __scale_backprop__
                       --
                       -- * __offset_backprop__
                       --
                       -- * __reserve_space_3__
                       --
                       -- * __reserve_space_4__
fusedBatchNormGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
fusedBatchNormGrad' op'options :: OpParams
op'options y_backprop :: Tensor v'1 t
y_backprop x :: Tensor v'2 t
x scale :: Tensor v'3 t
scale reserve_space_1 :: Tensor v'4 t
reserve_space_1
                    reserve_space_2 :: Tensor v'5 t
reserve_space_2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y_backprop,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
scale,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
reserve_space_1,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
reserve_space_2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNormGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y_backprop" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type_attr: "T" }
input_arg { name: "reserve_space_1" type_attr: "T" }
input_arg { name: "reserve_space_2" type_attr: "T" }
output_arg { name: "x_backprop" type_attr: "T" }
output_arg { name: "scale_backprop" type_attr: "T" }
output_arg { name: "offset_backprop" type_attr: "T" }
output_arg { name: "reserve_space_3" type_attr: "T" }
output_arg { name: "reserve_space_4" type_attr: "T" }
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedBatchNormGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t
                        u . (OneOf '[Data.Word.Word16, Float] t,
                             OneOf '[Float] u) =>
                        Tensor v'1 t -- ^ __y_backprop__
                        -> Tensor v'2 t -- ^ __x__
                        -> Tensor v'3 Float -- ^ __scale__
                        -> Tensor v'4 u -- ^ __reserve_space_1__
                        -> Tensor v'5 u -- ^ __reserve_space_2__
                        -> (Tensor Build t, Tensor Build u, Tensor Build u,
                            Tensor Build u, Tensor Build u)
                        -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_3__, __reserve_space_4__)
                        --
                        -- * __x_backprop__
                        --
                        -- * __scale_backprop__
                        --
                        -- * __offset_backprop__
                        --
                        -- * __reserve_space_3__
                        --
                        -- * __reserve_space_4__
fusedBatchNormGradV2 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t u.
(OneOf '[Word16, Float] t, OneOf '[Float] u) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV2' OpParams
forall a. a -> a
id
fusedBatchNormGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t
                         u . (OneOf '[Data.Word.Word16, Float] t,
                              OneOf '[Float] u) => OpParams ->
                         Tensor v'1 t -- ^ __y_backprop__
                         -> Tensor v'2 t -- ^ __x__
                         -> Tensor v'3 Float -- ^ __scale__
                         -> Tensor v'4 u -- ^ __reserve_space_1__
                         -> Tensor v'5 u -- ^ __reserve_space_2__
                         -> (Tensor Build t, Tensor Build u, Tensor Build u,
                             Tensor Build u, Tensor Build u)
                         -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_3__, __reserve_space_4__)
                         --
                         -- * __x_backprop__
                         --
                         -- * __scale_backprop__
                         --
                         -- * __offset_backprop__
                         --
                         -- * __reserve_space_3__
                         --
                         -- * __reserve_space_4__
fusedBatchNormGradV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV2' op'options :: OpParams
op'options y_backprop :: Tensor v'1 t
y_backprop x :: Tensor v'2 t
x scale :: Tensor v'3 Float
scale reserve_space_1 :: Tensor v'4 u
reserve_space_1
                      reserve_space_2 :: Tensor v'5 u
reserve_space_2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
     Tensor Build u))
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y_backprop,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
scale,
                                                             Tensor v'4 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 u
reserve_space_1,
                                                             Tensor v'5 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 u
reserve_space_2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNormGradV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ u -> DataType
forall a. TensorType a => a -> DataType
tensorType (u
forall a. HasCallStack => a
undefined :: u)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y_backprop" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type: DT_FLOAT }
input_arg { name: "reserve_space_1" type_attr: "U" }
input_arg { name: "reserve_space_2" type_attr: "U" }
output_arg { name: "x_backprop" type_attr: "T" }
output_arg { name: "scale_backprop" type_attr: "U" }
output_arg { name: "offset_backprop" type_attr: "U" }
output_arg { name: "reserve_space_3" type_attr: "U" }
output_arg { name: "reserve_space_4" type_attr: "U" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
attr {
  name: "U" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedBatchNormGradV3 :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                        u . (OneOf '[Data.Word.Word16, Float] t,
                             OneOf '[Float] u) =>
                        Tensor v'1 t -- ^ __y_backprop__
                        -> Tensor v'2 t -- ^ __x__
                        -> Tensor v'3 Float -- ^ __scale__
                        -> Tensor v'4 u -- ^ __reserve_space_1__
                        -> Tensor v'5 u -- ^ __reserve_space_2__
                        -> Tensor v'6 u -- ^ __reserve_space_3__
                        -> (Tensor Build t, Tensor Build u, Tensor Build u,
                            Tensor Build u, Tensor Build u)
                        -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_4__, __reserve_space_5__)
                        --
                        -- * __x_backprop__
                        --
                        -- * __scale_backprop__
                        --
                        -- * __offset_backprop__
                        --
                        -- * __reserve_space_4__
                        --
                        -- * __reserve_space_5__
fusedBatchNormGradV3 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> Tensor v'6 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> Tensor v'6 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t u.
(OneOf '[Word16, Float] t, OneOf '[Float] u) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> Tensor v'6 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV3' OpParams
forall a. a -> a
id
fusedBatchNormGradV3' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                         u . (OneOf '[Data.Word.Word16, Float] t,
                              OneOf '[Float] u) => OpParams ->
                         Tensor v'1 t -- ^ __y_backprop__
                         -> Tensor v'2 t -- ^ __x__
                         -> Tensor v'3 Float -- ^ __scale__
                         -> Tensor v'4 u -- ^ __reserve_space_1__
                         -> Tensor v'5 u -- ^ __reserve_space_2__
                         -> Tensor v'6 u -- ^ __reserve_space_3__
                         -> (Tensor Build t, Tensor Build u, Tensor Build u,
                             Tensor Build u, Tensor Build u)
                         -- ^ (__x_backprop__, __scale_backprop__, __offset_backprop__, __reserve_space_4__, __reserve_space_5__)
                         --
                         -- * __x_backprop__
                         --
                         -- * __scale_backprop__
                         --
                         -- * __offset_backprop__
                         --
                         -- * __reserve_space_4__
                         --
                         -- * __reserve_space_5__
fusedBatchNormGradV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 u
-> Tensor v'5 u
-> Tensor v'6 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormGradV3' op'options :: OpParams
op'options y_backprop :: Tensor v'1 t
y_backprop x :: Tensor v'2 t
x scale :: Tensor v'3 Float
scale reserve_space_1 :: Tensor v'4 u
reserve_space_1
                      reserve_space_2 :: Tensor v'5 u
reserve_space_2 reserve_space_3 :: Tensor v'6 u
reserve_space_3 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
     Tensor Build u))
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y_backprop,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
scale,
                                                             Tensor v'4 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 u
reserve_space_1,
                                                             Tensor v'5 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 u
reserve_space_2,
                                                             Tensor v'6 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 u
reserve_space_3]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNormGradV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ u -> DataType
forall a. TensorType a => a -> DataType
tensorType (u
forall a. HasCallStack => a
undefined :: u)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y_backprop" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type: DT_FLOAT }
input_arg { name: "reserve_space_1" type_attr: "U" }
input_arg { name: "reserve_space_2" type_attr: "U" }
input_arg { name: "reserve_space_3" type_attr: "U" }
output_arg { name: "x_backprop" type_attr: "T" }
output_arg { name: "scale_backprop" type_attr: "U" }
output_arg { name: "offset_backprop" type_attr: "U" }
output_arg { name: "reserve_space_4" type_attr: "U" }
output_arg { name: "reserve_space_5" type_attr: "U" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
attr {
  name: "U" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedBatchNormV2 :: forall v'1 v'2 v'3 v'4 v'5 t u . (OneOf '[Data.Word.Word16,
                                                              Float] t,
                                                      OneOf '[Float] u) =>
                    Tensor v'1 t -- ^ __x__
                    -> Tensor v'2 u -- ^ __scale__
                    -> Tensor v'3 u -- ^ __offset__
                    -> Tensor v'4 u -- ^ __mean__
                    -> Tensor v'5 u -- ^ __variance__
                    -> (Tensor Build t, Tensor Build u, Tensor Build u,
                        Tensor Build u, Tensor Build u)
                    -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__)
                    --
                    -- * __y__
                    --
                    -- * __batch_mean__
                    --
                    -- * __batch_variance__
                    --
                    -- * __reserve_space_1__
                    --
                    -- * __reserve_space_2__
fusedBatchNormV2 :: Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t u.
(OneOf '[Word16, Float] t, OneOf '[Float] u) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormV2' OpParams
forall a. a -> a
id
fusedBatchNormV2' :: forall v'1 v'2 v'3 v'4 v'5 t u . (OneOf '[Data.Word.Word16,
                                                               Float] t,
                                                       OneOf '[Float] u) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 u -- ^ __scale__
                     -> Tensor v'3 u -- ^ __offset__
                     -> Tensor v'4 u -- ^ __mean__
                     -> Tensor v'5 u -- ^ __variance__
                     -> (Tensor Build t, Tensor Build u, Tensor Build u,
                         Tensor Build u, Tensor Build u)
                     -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__)
                     --
                     -- * __y__
                     --
                     -- * __batch_mean__
                     --
                     -- * __batch_variance__
                     --
                     -- * __reserve_space_1__
                     --
                     -- * __reserve_space_2__
fusedBatchNormV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
fusedBatchNormV2' op'options :: OpParams
op'options x :: Tensor v'1 t
x scale :: Tensor v'2 u
scale offset :: Tensor v'3 u
offset mean :: Tensor v'4 u
mean variance :: Tensor v'5 u
variance | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
     Tensor Build u))
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 u
scale,
                                                             Tensor v'3 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 u
offset,
                                                             Tensor v'4 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 u
mean,
                                                             Tensor v'5 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 u
variance]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNormV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ u -> DataType
forall a. TensorType a => a -> DataType
tensorType (u
forall a. HasCallStack => a
undefined :: u)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type_attr: "U" }
input_arg { name: "offset" type_attr: "U" }
input_arg { name: "mean" type_attr: "U" }
input_arg { name: "variance" type_attr: "U" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "batch_mean" type_attr: "U" }
output_arg { name: "batch_variance" type_attr: "U" }
output_arg { name: "reserve_space_1" type_attr: "U" }
output_arg { name: "reserve_space_2" type_attr: "U" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
attr {
  name: "U" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "exponential_avg_factor"
  type: "float"
  default_value { f: 1.0 }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedBatchNormV3 :: forall v'1 v'2 v'3 v'4 v'5 t u . (OneOf '[Data.Word.Word16,
                                                              Float] t,
                                                      OneOf '[Float] u) =>
                    Tensor v'1 t -- ^ __x__
                    -> Tensor v'2 u -- ^ __scale__
                    -> Tensor v'3 u -- ^ __offset__
                    -> Tensor v'4 u -- ^ __mean__
                    -> Tensor v'5 u -- ^ __variance__
                    -> (Tensor Build t, Tensor Build u, Tensor Build u,
                        Tensor Build u, Tensor Build u, Tensor Build u)
                    -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__, __reserve_space_3__)
                    --
                    -- * __y__
                    --
                    -- * __batch_mean__
                    --
                    -- * __batch_variance__
                    --
                    -- * __reserve_space_1__
                    --
                    -- * __reserve_space_2__
                    --
                    -- * __reserve_space_3__
fusedBatchNormV3 :: Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
fusedBatchNormV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t u.
(OneOf '[Word16, Float] t, OneOf '[Float] u) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
fusedBatchNormV3' OpParams
forall a. a -> a
id
fusedBatchNormV3' :: forall v'1 v'2 v'3 v'4 v'5 t u . (OneOf '[Data.Word.Word16,
                                                               Float] t,
                                                       OneOf '[Float] u) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 u -- ^ __scale__
                     -> Tensor v'3 u -- ^ __offset__
                     -> Tensor v'4 u -- ^ __mean__
                     -> Tensor v'5 u -- ^ __variance__
                     -> (Tensor Build t, Tensor Build u, Tensor Build u,
                         Tensor Build u, Tensor Build u, Tensor Build u)
                     -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__, __reserve_space_3__)
                     --
                     -- * __y__
                     --
                     -- * __batch_mean__
                     --
                     -- * __batch_variance__
                     --
                     -- * __reserve_space_1__
                     --
                     -- * __reserve_space_2__
                     --
                     -- * __reserve_space_3__
fusedBatchNormV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
fusedBatchNormV3' op'options :: OpParams
op'options x :: Tensor v'1 t
x scale :: Tensor v'2 u
scale offset :: Tensor v'3 u
offset mean :: Tensor v'4 u
mean variance :: Tensor v'5 u
variance | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
     Tensor Build u, Tensor Build u))
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 u
scale,
                                                             Tensor v'3 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 u
offset,
                                                             Tensor v'4 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 u
mean,
                                                             Tensor v'5 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 u
variance]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedBatchNormV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ u -> DataType
forall a. TensorType a => a -> DataType
tensorType (u
forall a. HasCallStack => a
undefined :: u)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type_attr: "U" }
input_arg { name: "offset" type_attr: "U" }
input_arg { name: "mean" type_attr: "U" }
input_arg { name: "variance" type_attr: "U" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "batch_mean" type_attr: "U" }
output_arg { name: "batch_variance" type_attr: "U" }
output_arg { name: "reserve_space_1" type_attr: "U" }
output_arg { name: "reserve_space_2" type_attr: "U" }
output_arg { name: "reserve_space_3" type_attr: "U" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
attr {
  name: "U" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "exponential_avg_factor"
  type: "float"
  default_value { f: 1.0 }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | 
fusedPadConv2D :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16, Double,
                                                  Float] t) =>
                  ByteString -- ^ __mode__
                  -> ByteString -- ^ __padding__
                  -> Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __paddings__
                  -> Tensor v'3 t -- ^ __filter__
                  -> Tensor Build t -- ^ __output__
fusedPadConv2D :: ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
fusedPadConv2D = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
fusedPadConv2D' OpParams
forall a. a -> a
id
fusedPadConv2D' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16, Double,
                                                   Float] t) => OpParams ->
                   ByteString -- ^ __mode__
                   -> ByteString -- ^ __padding__
                   -> Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __paddings__
                   -> Tensor v'3 t -- ^ __filter__
                   -> Tensor Build t -- ^ __output__
fusedPadConv2D' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
fusedPadConv2D' op'options :: OpParams
op'options mode :: ByteString
mode padding :: ByteString
padding input :: Tensor v'1 t
input paddings :: Tensor v'2 Int32
paddings
                filter :: Tensor v'3 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
paddings,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedPadConv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mode
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type: DT_INT32 }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "mode"
  type: "string"
  allowed_values { list { s: "REFLECT" s: "SYMMETRIC" } }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
fusedResizeAndPadConv2D :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                           ByteString -- ^ __mode__
                           -> ByteString -- ^ __padding__
                           -> Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                           -> Tensor v'3 Data.Int.Int32 -- ^ __paddings__
                           -> Tensor v'4 t -- ^ __filter__
                           -> Tensor Build t -- ^ __output__
fusedResizeAndPadConv2D :: ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor Build t
fusedResizeAndPadConv2D = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor Build t
fusedResizeAndPadConv2D' OpParams
forall a. a -> a
id
fusedResizeAndPadConv2D' :: forall v'1 v'2 v'3 v'4
                            t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
                            OpParams ->
                            ByteString -- ^ __mode__
                            -> ByteString -- ^ __padding__
                            -> Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                            -> Tensor v'3 Data.Int.Int32 -- ^ __paddings__
                            -> Tensor v'4 t -- ^ __filter__
                            -> Tensor Build t -- ^ __output__
fusedResizeAndPadConv2D' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor Build t
fusedResizeAndPadConv2D' op'options :: OpParams
op'options mode :: ByteString
mode padding :: ByteString
padding input :: Tensor v'1 t
input size :: Tensor v'2 Int32
size paddings :: Tensor v'3 Int32
paddings
                         filter :: Tensor v'4 t
filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
paddings,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "FusedResizeAndPadConv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mode
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "paddings" type: DT_INT32 }
input_arg { name: "filter" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "resize_align_corners"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "mode"
  type: "string"
  allowed_values { list { s: "REFLECT" s: "SYMMETRIC" } }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
gRUBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 t . (OneOf '[Float] t) =>
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __h_prev__
                -> Tensor v'3 t -- ^ __w_ru__
                -> Tensor v'4 t -- ^ __w_c__
                -> Tensor v'5 t -- ^ __b_ru__
                -> Tensor v'6 t -- ^ __b_c__
                -> (Tensor Build t, Tensor Build t, Tensor Build t,
                    Tensor Build t) -- ^ (__r__, __u__, __c__, __h__)
                --
                -- * __r__
                --
                -- * __u__
                --
                -- * __c__
                --
                -- * __h__
gRUBlockCell :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCell = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCell' OpParams
forall a. a -> a
id
gRUBlockCell' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t . (OneOf '[Float] t) =>
                 OpParams ->
                 Tensor v'1 t -- ^ __x__
                 -> Tensor v'2 t -- ^ __h_prev__
                 -> Tensor v'3 t -- ^ __w_ru__
                 -> Tensor v'4 t -- ^ __w_c__
                 -> Tensor v'5 t -- ^ __b_ru__
                 -> Tensor v'6 t -- ^ __b_c__
                 -> (Tensor Build t, Tensor Build t, Tensor Build t,
                     Tensor Build t) -- ^ (__r__, __u__, __c__, __h__)
                 --
                 -- * __r__
                 --
                 -- * __u__
                 --
                 -- * __c__
                 --
                 -- * __h__
gRUBlockCell' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCell' op'options :: OpParams
op'options x :: Tensor v'1 t
x h_prev :: Tensor v'2 t
h_prev w_ru :: Tensor v'3 t
w_ru w_c :: Tensor v'4 t
w_c b_ru :: Tensor v'5 t
b_ru b_c :: Tensor v'6 t
b_c | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
h_prev,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
w_ru,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
w_c,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
b_ru,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
b_c]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GRUBlockCell"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w_ru" type_attr: "T" }
input_arg { name: "w_c" type_attr: "T" }
input_arg { name: "b_ru" type_attr: "T" }
input_arg { name: "b_c" type_attr: "T" }
output_arg { name: "r" type_attr: "T" }
output_arg { name: "u" type_attr: "T" }
output_arg { name: "c" type_attr: "T" }
output_arg { name: "h" type_attr: "T" }
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
-}
-- | 
gRUBlockCellGrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                    t . (OneOf '[Float] t) => Tensor v'1 t -- ^ __x__
                    -> Tensor v'2 t -- ^ __h_prev__
                    -> Tensor v'3 t -- ^ __w_ru__
                    -> Tensor v'4 t -- ^ __w_c__
                    -> Tensor v'5 t -- ^ __b_ru__
                    -> Tensor v'6 t -- ^ __b_c__
                    -> Tensor v'7 t -- ^ __r__
                    -> Tensor v'8 t -- ^ __u__
                    -> Tensor v'9 t -- ^ __c__
                    -> Tensor v'10 t -- ^ __d_h__
                    -> (Tensor Build t, Tensor Build t, Tensor Build t,
                        Tensor Build t)
                    -- ^ (__d_x__, __d_h_prev__, __d_c_bar__, __d_r_bar_u_bar__)
                    --
                    -- * __d_x__
                    --
                    -- * __d_h_prev__
                    --
                    -- * __d_c_bar__
                    --
                    -- * __d_r_bar_u_bar__
gRUBlockCellGrad :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCellGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCellGrad' OpParams
forall a. a -> a
id
gRUBlockCellGrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                     t . (OneOf '[Float] t) => OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 t -- ^ __h_prev__
                     -> Tensor v'3 t -- ^ __w_ru__
                     -> Tensor v'4 t -- ^ __w_c__
                     -> Tensor v'5 t -- ^ __b_ru__
                     -> Tensor v'6 t -- ^ __b_c__
                     -> Tensor v'7 t -- ^ __r__
                     -> Tensor v'8 t -- ^ __u__
                     -> Tensor v'9 t -- ^ __c__
                     -> Tensor v'10 t -- ^ __d_h__
                     -> (Tensor Build t, Tensor Build t, Tensor Build t,
                         Tensor Build t)
                     -- ^ (__d_x__, __d_h_prev__, __d_c_bar__, __d_r_bar_u_bar__)
                     --
                     -- * __d_x__
                     --
                     -- * __d_h_prev__
                     --
                     -- * __d_c_bar__
                     --
                     -- * __d_r_bar_u_bar__
gRUBlockCellGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
gRUBlockCellGrad' op'options :: OpParams
op'options x :: Tensor v'1 t
x h_prev :: Tensor v'2 t
h_prev w_ru :: Tensor v'3 t
w_ru w_c :: Tensor v'4 t
w_c b_ru :: Tensor v'5 t
b_ru b_c :: Tensor v'6 t
b_c r :: Tensor v'7 t
r u :: Tensor v'8 t
u c :: Tensor v'9 t
c
                  d_h :: Tensor v'10 t
d_h | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
h_prev,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
w_ru,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
w_c,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
b_ru,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
b_c,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
r,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
u,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
c,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
d_h]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GRUBlockCellGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w_ru" type_attr: "T" }
input_arg { name: "w_c" type_attr: "T" }
input_arg { name: "b_ru" type_attr: "T" }
input_arg { name: "b_c" type_attr: "T" }
input_arg { name: "r" type_attr: "T" }
input_arg { name: "u" type_attr: "T" }
input_arg { name: "c" type_attr: "T" }
input_arg { name: "d_h" type_attr: "T" }
output_arg { name: "d_x" type_attr: "T" }
output_arg { name: "d_h_prev" type_attr: "T" }
output_arg { name: "d_c_bar" type_attr: "T" }
output_arg { name: "d_r_bar_u_bar" type_attr: "T" }
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
-}
-- | 
gather :: forall v'1 v'2 tparams tindices . (TensorType tparams,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tindices) =>
          Tensor v'1 tparams -- ^ __params__
          -> Tensor v'2 tindices -- ^ __indices__
          -> Tensor Build tparams -- ^ __output__
gather :: Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
gather = OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor Build tparams
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
gather' OpParams
forall a. a -> a
id
gather' :: forall v'1 v'2 tparams tindices . (TensorType tparams,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
           OpParams ->
           Tensor v'1 tparams -- ^ __params__
           -> Tensor v'2 tindices -- ^ __indices__
           -> Tensor Build tparams -- ^ __output__
gather' :: OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor Build tparams
gather' op'options :: OpParams
op'options params :: Tensor v'1 tparams
params indices :: Tensor v'2 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tparams
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tparams)
-> Build OpDef -> Tensor Build tparams
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tparams -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tparams
params,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Gather"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tparams" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tparams -> DataType
forall a. TensorType a => a -> DataType
tensorType (tparams
forall a. HasCallStack => a
undefined :: tparams)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "params" type_attr: "Tparams" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "Tparams" }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr { name: "Tparams" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
gatherNd :: forall v'1 v'2 tparams tindices . (TensorType tparams,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
            Tensor v'1 tparams -- ^ __params__
            -> Tensor v'2 tindices -- ^ __indices__
            -> Tensor Build tparams -- ^ __output__
gatherNd :: Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
gatherNd = OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor Build tparams
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
gatherNd' OpParams
forall a. a -> a
id
gatherNd' :: forall v'1 v'2 tparams tindices . (TensorType tparams,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
             OpParams ->
             Tensor v'1 tparams -- ^ __params__
             -> Tensor v'2 tindices -- ^ __indices__
             -> Tensor Build tparams -- ^ __output__
gatherNd' :: OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor Build tparams
gatherNd' op'options :: OpParams
op'options params :: Tensor v'1 tparams
params indices :: Tensor v'2 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tparams
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tparams)
-> Build OpDef -> Tensor Build tparams
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tparams -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tparams
params,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GatherNd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tparams" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tparams -> DataType
forall a. TensorType a => a -> DataType
tensorType (tparams
forall a. HasCallStack => a
undefined :: tparams)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "params" type_attr: "Tparams" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "Tparams" }
attr { name: "Tparams" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
gatherV2 :: forall v'1 v'2 v'3 tparams tindices taxis . (TensorType tparams,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] taxis) =>
            Tensor v'1 tparams -- ^ __params__
            -> Tensor v'2 tindices -- ^ __indices__
            -> Tensor v'3 taxis -- ^ __axis__
            -> Tensor Build tparams -- ^ __output__
gatherV2 :: Tensor v'1 tparams
-> Tensor v'2 tindices -> Tensor v'3 taxis -> Tensor Build tparams
gatherV2 = OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor v'3 taxis
-> Tensor Build tparams
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       tblock_shape tcrops.
(TensorType t, OneOf '[Int32, Int64] tblock_shape,
 OneOf '[Int32, Int64] tcrops) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
gatherV2' OpParams
forall a. a -> a
id
gatherV2' :: forall v'1 v'2 v'3 tparams tindices taxis . (TensorType tparams,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] tindices,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] taxis) =>
             OpParams ->
             Tensor v'1 tparams -- ^ __params__
             -> Tensor v'2 tindices -- ^ __indices__
             -> Tensor v'3 taxis -- ^ __axis__
             -> Tensor Build tparams -- ^ __output__
gatherV2' :: OpParams
-> Tensor v'1 tparams
-> Tensor v'2 tindices
-> Tensor v'3 taxis
-> Tensor Build tparams
gatherV2' op'options :: OpParams
op'options params :: Tensor v'1 tparams
params indices :: Tensor v'2 tindices
indices axis :: Tensor v'3 taxis
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tparams
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tparams)
-> Build OpDef -> Tensor Build tparams
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tparams -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tparams
params,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 taxis -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 taxis
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GatherV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tparams" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tparams -> DataType
forall a. TensorType a => a -> DataType
tensorType (tparams
forall a. HasCallStack => a
undefined :: tparams)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Taxis" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ taxis -> DataType
forall a. TensorType a => a -> DataType
tensorType (taxis
forall a. HasCallStack => a
undefined :: taxis)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "params" type_attr: "Tparams" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "axis" type_attr: "Taxis" }
output_arg { name: "output" type_attr: "Tparams" }
attr { name: "batch_dims" type: "int" default_value { i: 0 } }
attr { name: "Tparams" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Taxis"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
generateBoundingBoxProposals :: Tensor v'1 Float -- ^ __scores__
                                -> Tensor v'2 Float -- ^ __bbox_deltas__
                                -> Tensor v'3 Float -- ^ __image_info__
                                -> Tensor v'4 Float -- ^ __anchors__
                                -> Tensor v'5 Float -- ^ __nms_threshold__
                                -> Tensor v'6 Data.Int.Int32 -- ^ __pre_nms_topn__
                                -> Tensor v'7 Float -- ^ __min_size__
                                -> (Tensor Build Float, Tensor Build Float)
                                -- ^ (__rois__, __roi_probabilities__)
                                --
                                -- * __rois__
                                --
                                -- * __roi_probabilities__
generateBoundingBoxProposals :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Int32
-> Tensor v'7 Float
-> (Tensor Build Float, Tensor Build Float)
generateBoundingBoxProposals = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Int32
-> Tensor v'7 Float
-> (Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Int32
-> Tensor v'7 Float
-> (Tensor Build Float, Tensor Build Float)
generateBoundingBoxProposals' OpParams
forall a. a -> a
id
generateBoundingBoxProposals' :: OpParams ->
                                 Tensor v'1 Float -- ^ __scores__
                                 -> Tensor v'2 Float -- ^ __bbox_deltas__
                                 -> Tensor v'3 Float -- ^ __image_info__
                                 -> Tensor v'4 Float -- ^ __anchors__
                                 -> Tensor v'5 Float -- ^ __nms_threshold__
                                 -> Tensor v'6 Data.Int.Int32 -- ^ __pre_nms_topn__
                                 -> Tensor v'7 Float -- ^ __min_size__
                                 -> (Tensor Build Float, Tensor Build Float)
                                 -- ^ (__rois__, __roi_probabilities__)
                                 --
                                 -- * __rois__
                                 --
                                 -- * __roi_probabilities__
generateBoundingBoxProposals' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Int32
-> Tensor v'7 Float
-> (Tensor Build Float, Tensor Build Float)
generateBoundingBoxProposals' op'options :: OpParams
op'options scores :: Tensor v'1 Float
scores bbox_deltas :: Tensor v'2 Float
bbox_deltas image_info :: Tensor v'3 Float
image_info anchors :: Tensor v'4 Float
anchors
                              nms_threshold :: Tensor v'5 Float
nms_threshold pre_nms_topn :: Tensor v'6 Int32
pre_nms_topn
                              min_size :: Tensor v'7 Float
min_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Float, Tensor Build Float))
-> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
scores,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
bbox_deltas,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
image_info,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
anchors,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
nms_threshold,
                                                             Tensor v'6 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int32
pre_nms_topn,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
min_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GenerateBoundingBoxProposals"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "scores" type: DT_FLOAT }
input_arg { name: "bbox_deltas" type: DT_FLOAT }
input_arg { name: "image_info" type: DT_FLOAT }
input_arg { name: "anchors" type: DT_FLOAT }
input_arg { name: "nms_threshold" type: DT_FLOAT }
input_arg { name: "pre_nms_topn" type: DT_INT32 }
input_arg { name: "min_size" type: DT_FLOAT }
output_arg { name: "rois" type: DT_FLOAT }
output_arg { name: "roi_probabilities" type: DT_FLOAT }
attr { name: "post_nms_topn" type: "int" default_value { i: 300 } }
-}
-- | 
generateVocabRemapping :: Data.Int.Int64 -- ^ __new_vocab_offset__
                          -> Data.Int.Int64 -- ^ __num_new_vocab__
                          -> Tensor v'1 Data.ByteString.ByteString -- ^ __new_vocab_file__
                          -> Tensor v'2 Data.ByteString.ByteString -- ^ __old_vocab_file__
                          -> (Tensor Build Data.Int.Int64,
                              Tensor Build Data.Int.Int32)
                          -- ^ (__remapping__, __num_present__)
                          --
                          -- * __remapping__
                          --
                          -- * __num_present__
generateVocabRemapping :: Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build Int32)
generateVocabRemapping = OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build Int32)
generateVocabRemapping' OpParams
forall a. a -> a
id
generateVocabRemapping' :: OpParams ->
                           Data.Int.Int64 -- ^ __new_vocab_offset__
                           -> Data.Int.Int64 -- ^ __num_new_vocab__
                           -> Tensor v'1 Data.ByteString.ByteString -- ^ __new_vocab_file__
                           -> Tensor v'2 Data.ByteString.ByteString -- ^ __old_vocab_file__
                           -> (Tensor Build Data.Int.Int64,
                               Tensor Build Data.Int.Int32)
                           -- ^ (__remapping__, __num_present__)
                           --
                           -- * __remapping__
                           --
                           -- * __num_present__
generateVocabRemapping' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build Int32)
generateVocabRemapping' op'options :: OpParams
op'options new_vocab_offset :: Int64
new_vocab_offset num_new_vocab :: Int64
num_new_vocab new_vocab_file :: Tensor v'1 ByteString
new_vocab_file
                        old_vocab_file :: Tensor v'2 ByteString
old_vocab_file | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build Int32))
-> Build OpDef -> (Tensor Build Int64, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
new_vocab_file,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
old_vocab_file]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GenerateVocabRemapping"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "new_vocab_offset" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
new_vocab_offset
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_new_vocab" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_new_vocab
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "new_vocab_file" type: DT_STRING }
input_arg { name: "old_vocab_file" type: DT_STRING }
output_arg { name: "remapping" type: DT_INT64 }
output_arg { name: "num_present" type: DT_INT32 }
attr { name: "new_vocab_offset" type: "int" has_minimum: true }
attr { name: "num_new_vocab" type: "int" has_minimum: true }
attr {
  name: "old_vocab_size"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
-}
-- | 
getSessionHandle :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                    Tensor v'1 t -- ^ __value__
                    -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __handle__
getSessionHandle :: Tensor v'1 t -> m' (Tensor Value ByteString)
getSessionHandle = OpParams -> Tensor v'1 t -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value ByteString)
getSessionHandle' OpParams
forall a. a -> a
id
getSessionHandle' :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __value__
                     -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __handle__
getSessionHandle' :: OpParams -> Tensor v'1 t -> m' (Tensor Value ByteString)
getSessionHandle' op'options :: OpParams
op'options value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "GetSessionHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "handle" type: DT_STRING }
attr { name: "T" type: "type" }
-}
-- | 
getSessionHandleV2 :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                      Tensor v'1 t -- ^ __value__
                      -> m' (Tensor Value ResourceHandle) -- ^ __handle__
getSessionHandleV2 :: Tensor v'1 t -> m' (Tensor Value ResourceHandle)
getSessionHandleV2 = OpParams -> Tensor v'1 t -> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value ResourceHandle)
getSessionHandleV2' OpParams
forall a. a -> a
id
getSessionHandleV2' :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __value__
                       -> m' (Tensor Value ResourceHandle) -- ^ __handle__
getSessionHandleV2' :: OpParams -> Tensor v'1 t -> m' (Tensor Value ResourceHandle)
getSessionHandleV2' op'options :: OpParams
op'options value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "GetSessionHandleV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "T" type: "type" }
-}
-- | 
getSessionTensor :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                    Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                    -> m' (Tensor Value dtype) -- ^ __value__
getSessionTensor :: Tensor v'1 ByteString -> m' (Tensor Value dtype)
getSessionTensor = OpParams -> Tensor v'1 ByteString -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 ByteString -> m' (Tensor Value dtype)
getSessionTensor' OpParams
forall a. a -> a
id
getSessionTensor' :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                     OpParams ->
                     Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                     -> m' (Tensor Value dtype) -- ^ __value__
getSessionTensor' :: OpParams -> Tensor v'1 ByteString -> m' (Tensor Value dtype)
getSessionTensor' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "GetSessionTensor"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
greater :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor Build Bool -- ^ __z__
greater :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greater = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greater' OpParams
forall a. a -> a
id
greater' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build Bool -- ^ __z__
greater' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greater' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Greater"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
greaterEqual :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t) =>
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __y__
                -> Tensor Build Bool -- ^ __z__
greaterEqual :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greaterEqual = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greaterEqual' OpParams
forall a. a -> a
id
greaterEqual' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t) => OpParams ->
                 Tensor v'1 t -- ^ __x__
                 -> Tensor v'2 t -- ^ __y__
                 -> Tensor Build Bool -- ^ __z__
greaterEqual' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
greaterEqual' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "GreaterEqual"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
guaranteeConst :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                  Tensor v'1 t -- ^ __input__
                  -> m' (Tensor Value t) -- ^ __output__
guaranteeConst :: Tensor v'1 t -> m' (Tensor Value t)
guaranteeConst = OpParams -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value t)
guaranteeConst' OpParams
forall a. a -> a
id
guaranteeConst' :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                   OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> m' (Tensor Value t) -- ^ __output__
guaranteeConst' :: OpParams -> Tensor v'1 t -> m' (Tensor Value t)
guaranteeConst' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "GuaranteeConst"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
hSVToRGB :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __images__
            -> Tensor Build t -- ^ __output__
hSVToRGB :: Tensor v'1 t -> Tensor Build t
hSVToRGB = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
hSVToRGB' OpParams
forall a. a -> a
id
hSVToRGB' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __images__
             -> Tensor Build t -- ^ __output__
hSVToRGB' :: OpParams -> Tensor v'1 t -> Tensor Build t
hSVToRGB' op'options :: OpParams
op'options images :: Tensor v'1 t
images | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "HSVToRGB"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
hashTable :: forall m' . (MonadBuild m') => DataType -- ^ __key_dtype__
             -> DataType -- ^ __value_dtype__
             -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
hashTable :: DataType -> DataType -> m' (Tensor Ref ByteString)
hashTable = OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
hashTable' OpParams
forall a. a -> a
id
hashTable' :: forall m' . (MonadBuild m') => OpParams ->
              DataType -- ^ __key_dtype__
              -> DataType -- ^ __value_dtype__
              -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
hashTable' :: OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
hashTable' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "HashTable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
-}
-- | 
hashTableV2 :: forall m' . (MonadBuild m') => DataType -- ^ __key_dtype__
               -> DataType -- ^ __value_dtype__
               -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
hashTableV2 :: DataType -> DataType -> m' (Tensor Value ResourceHandle)
hashTableV2 = OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
hashTableV2' OpParams
forall a. a -> a
id
hashTableV2' :: forall m' . (MonadBuild m') => OpParams ->
                DataType -- ^ __key_dtype__
                -> DataType -- ^ __value_dtype__
                -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
hashTableV2' :: OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
hashTableV2' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "HashTableV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
-}
-- | 
histogramFixedWidth :: forall v'1 v'2 v'3 t dtype . (OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Double, Float] t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] dtype) =>
                       Tensor v'1 t -- ^ __values__
                       -> Tensor v'2 t -- ^ __value_range__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __nbins__
                       -> Tensor Build dtype -- ^ __out__
histogramFixedWidth :: Tensor v'1 t
-> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build dtype
histogramFixedWidth = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t dtype.
(OneOf '[Int32, Int64, Double, Float] t,
 OneOf '[Int32, Int64] dtype) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build dtype
histogramFixedWidth' OpParams
forall a. a -> a
id
histogramFixedWidth' :: forall v'1 v'2 v'3 t dtype . (OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Double, Float] t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] dtype) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __values__
                        -> Tensor v'2 t -- ^ __value_range__
                        -> Tensor v'3 Data.Int.Int32 -- ^ __nbins__
                        -> Tensor Build dtype -- ^ __out__
histogramFixedWidth' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build dtype
histogramFixedWidth' op'options :: OpParams
op'options values :: Tensor v'1 t
values value_range :: Tensor v'2 t
value_range nbins :: Tensor v'3 Int32
nbins | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
values,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value_range,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
nbins]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "HistogramFixedWidth"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "values" type_attr: "T" }
input_arg { name: "value_range" type_attr: "T" }
input_arg { name: "nbins" type: DT_INT32 }
output_arg { name: "out" type_attr: "dtype" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
histogramSummary :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) =>
                    Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                    -> Tensor v'2 t -- ^ __values__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
histogramSummary :: Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
histogramSummary = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
histogramSummary' OpParams
forall a. a -> a
id
histogramSummary' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t) => OpParams ->
                     Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                     -> Tensor v'2 t -- ^ __values__
                     -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
histogramSummary' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
histogramSummary' op'options :: OpParams
op'options tag :: Tensor v'1 ByteString
tag values :: Tensor v'2 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tag,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "HistogramSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "values" type_attr: "T" }
output_arg { name: "summary" type: DT_STRING }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
hostConst :: forall dtype . (TensorType dtype) =>
             Tensor Build dtype -- ^ __output__
hostConst :: Tensor Build dtype
hostConst = OpParams -> Tensor Build dtype
forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
hostConst' OpParams
forall a. a -> a
id
hostConst' :: forall dtype . (TensorType dtype) => OpParams ->
              Tensor Build dtype -- ^ __output__
hostConst' :: OpParams -> Tensor Build dtype
hostConst' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "HostConst"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "value" type: "tensor" }
attr { name: "dtype" type: "type" }
-}
-- | 
iFFT :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float)] tcomplex) =>
        Tensor v'1 tcomplex -- ^ __input__
        -> Tensor Build tcomplex -- ^ __output__
iFFT :: Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT' OpParams
forall a. a -> a
id
iFFT' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float)] tcomplex) =>
         OpParams ->
         Tensor v'1 tcomplex -- ^ __input__
         -> Tensor Build tcomplex -- ^ __output__
iFFT' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IFFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
iFFT2D :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float)] tcomplex) =>
          Tensor v'1 tcomplex -- ^ __input__
          -> Tensor Build tcomplex -- ^ __output__
iFFT2D :: Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT2D = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT2D' OpParams
forall a. a -> a
id
iFFT2D' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float)] tcomplex) =>
           OpParams ->
           Tensor v'1 tcomplex -- ^ __input__
           -> Tensor Build tcomplex -- ^ __output__
iFFT2D' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT2D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IFFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
iFFT3D :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float)] tcomplex) =>
          Tensor v'1 tcomplex -- ^ __input__
          -> Tensor Build tcomplex -- ^ __output__
iFFT3D :: Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT3D = OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT3D' OpParams
forall a. a -> a
id
iFFT3D' :: forall v'1 tcomplex . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float)] tcomplex) =>
           OpParams ->
           Tensor v'1 tcomplex -- ^ __input__
           -> Tensor Build tcomplex -- ^ __output__
iFFT3D' :: OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
iFFT3D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IFFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
iRFFT :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                          OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float)] tcomplex) =>
         Tensor v'1 tcomplex -- ^ __input__
         -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
         -> Tensor Build treal -- ^ __output__
iRFFT :: Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT = OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT' OpParams
forall a. a -> a
id
iRFFT' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float)] tcomplex) =>
          OpParams ->
          Tensor v'1 tcomplex -- ^ __input__
          -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
          -> Tensor Build treal -- ^ __output__
iRFFT' :: OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build treal
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build treal)
-> Build OpDef -> Tensor Build treal
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IRFFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Treal" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
iRFFT2D :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float)] tcomplex) =>
           Tensor v'1 tcomplex -- ^ __input__
           -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
           -> Tensor Build treal -- ^ __output__
iRFFT2D :: Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT2D = OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT2D' OpParams
forall a. a -> a
id
iRFFT2D' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float)] tcomplex) =>
            OpParams ->
            Tensor v'1 tcomplex -- ^ __input__
            -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
            -> Tensor Build treal -- ^ __output__
iRFFT2D' :: OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT2D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build treal
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build treal)
-> Build OpDef -> Tensor Build treal
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IRFFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Treal" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
iRFFT3D :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float)] tcomplex) =>
           Tensor v'1 tcomplex -- ^ __input__
           -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
           -> Tensor Build treal -- ^ __output__
iRFFT3D :: Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT3D = OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT3D' OpParams
forall a. a -> a
id
iRFFT3D' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float)] tcomplex) =>
            OpParams ->
            Tensor v'1 tcomplex -- ^ __input__
            -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
            -> Tensor Build treal -- ^ __output__
iRFFT3D' :: OpParams
-> Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
iRFFT3D' op'options :: OpParams
op'options input :: Tensor v'1 tcomplex
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build treal
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build treal)
-> Build OpDef -> Tensor Build treal
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tcomplex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tcomplex
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IRFFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tcomplex" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Treal" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
identity :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
identity :: Tensor v'1 t -> Tensor Build t
identity = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
identity' OpParams
forall a. a -> a
id
identity' :: forall v'1 t . (TensorType t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
identity' :: OpParams -> Tensor v'1 t -> Tensor Build t
identity' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Identity"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
identityN :: forall v'1 t . (TensorTypes t) => TensorList (v'1) t -- ^ __input__
             -> TensorList (Build) t -- ^ __output__
identityN :: TensorList v'1 t -> TensorList Build t
identityN = OpParams -> TensorList v'1 t -> TensorList Build t
forall (v'1 :: * -> *) (t :: [*]).
TensorTypes t =>
OpParams -> TensorList v'1 t -> TensorList Build t
identityN' OpParams
forall a. a -> a
id
identityN' :: forall v'1 t . (TensorTypes t) => OpParams ->
              TensorList (v'1) t -- ^ __input__
              -> TensorList (Build) t -- ^ __output__
identityN' :: OpParams -> TensorList v'1 t -> TensorList Build t
identityN' op'options :: OpParams
op'options input :: TensorList v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build t)
-> Build OpDef -> TensorList Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IdentityN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_list_attr: "T" }
output_arg { name: "output" type_list_attr: "T" }
attr { name: "T" type: "list(type)" has_minimum: true minimum: 1 }
-}
-- | 
identityReader :: forall m' . (MonadBuild m') =>
                  m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
identityReader :: m' (Tensor Ref ByteString)
identityReader = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
identityReader' OpParams
forall a. a -> a
id
identityReader' :: forall m' . (MonadBuild m') => OpParams ->
                   m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
identityReader' :: OpParams -> m' (Tensor Ref ByteString)
identityReader' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IdentityReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
identityReaderV2 :: forall m' . (MonadBuild m') =>
                    m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
identityReaderV2 :: m' (Tensor Value ResourceHandle)
identityReaderV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
identityReaderV2' OpParams
forall a. a -> a
id
identityReaderV2' :: forall m' . (MonadBuild m') => OpParams ->
                     m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
identityReaderV2' :: OpParams -> m' (Tensor Value ResourceHandle)
identityReaderV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IdentityReaderV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
igamma :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
          Tensor v'1 t -- ^ __a__
          -> Tensor v'2 t -- ^ __x__
          -> Tensor Build t -- ^ __z__
igamma :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igamma = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igamma' OpParams
forall a. a -> a
id
igamma' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
           Tensor v'1 t -- ^ __a__
           -> Tensor v'2 t -- ^ __x__
           -> Tensor Build t -- ^ __z__
igamma' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igamma' op'options :: OpParams
op'options a :: Tensor v'1 t
a x :: Tensor v'2 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Igamma"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
igammaGradA :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
               Tensor v'1 t -- ^ __a__
               -> Tensor v'2 t -- ^ __x__
               -> Tensor Build t -- ^ __z__
igammaGradA :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammaGradA = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammaGradA' OpParams
forall a. a -> a
id
igammaGradA' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
                Tensor v'1 t -- ^ __a__
                -> Tensor v'2 t -- ^ __x__
                -> Tensor Build t -- ^ __z__
igammaGradA' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammaGradA' op'options :: OpParams
op'options a :: Tensor v'1 t
a x :: Tensor v'2 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IgammaGradA"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
igammac :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
           Tensor v'1 t -- ^ __a__
           -> Tensor v'2 t -- ^ __x__
           -> Tensor Build t -- ^ __z__
igammac :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammac = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammac' OpParams
forall a. a -> a
id
igammac' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __a__
            -> Tensor v'2 t -- ^ __x__
            -> Tensor Build t -- ^ __z__
igammac' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
igammac' op'options :: OpParams
op'options a :: Tensor v'1 t
a x :: Tensor v'2 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Igammac"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
ignoreErrorsDataset :: [DataType] -- ^ __output_types__
                       -> Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor Build Variant -- ^ __handle__
ignoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
ignoreErrorsDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
ignoreErrorsDataset' OpParams
forall a. a -> a
id
ignoreErrorsDataset' :: OpParams ->
                        [DataType] -- ^ __output_types__
                        -> Tensor v'1 Variant -- ^ __input_dataset__
                        -> Tensor Build Variant -- ^ __handle__
ignoreErrorsDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
ignoreErrorsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IgnoreErrorsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
imag :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float)] t,
                             OneOf '[Double, Float] tout) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor Build tout -- ^ __output__
imag :: Tensor v'1 t -> Tensor Build tout
imag = OpParams -> Tensor v'1 t -> Tensor Build tout
forall (v'1 :: * -> *) t tout.
(OneOf '[Complex Double, Complex Float] t,
 OneOf '[Double, Float] tout) =>
OpParams -> Tensor v'1 t -> Tensor Build tout
imag' OpParams
forall a. a -> a
id
imag' :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float)] t,
                              OneOf '[Double, Float] tout) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build tout -- ^ __output__
imag' :: OpParams -> Tensor v'1 t -> Tensor Build tout
imag' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tout)
-> Build OpDef -> Tensor Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Imag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "Tout" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
attr {
  name: "Tout"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
imageProjectiveTransformV2 :: forall v'1 v'2 v'3
                              dtype . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                               Data.Word.Word16,
                                               Data.Word.Word8, Double,
                                               Float] dtype) =>
                              ByteString -- ^ __interpolation__
                              -> Tensor v'1 dtype -- ^ __images__
                              -> Tensor v'2 Float -- ^ __transforms__
                              -> Tensor v'3 Data.Int.Int32 -- ^ __output_shape__
                              -> Tensor Build dtype -- ^ __transformed_images__
imageProjectiveTransformV2 :: ByteString
-> Tensor v'1 dtype
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build dtype
imageProjectiveTransformV2 = OpParams
-> ByteString
-> Tensor v'1 dtype
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype.
OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype =>
OpParams
-> ByteString
-> Tensor v'1 dtype
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build dtype
imageProjectiveTransformV2' OpParams
forall a. a -> a
id
imageProjectiveTransformV2' :: forall v'1 v'2 v'3
                               dtype . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                                Data.Word.Word16,
                                                Data.Word.Word8, Double,
                                                Float] dtype) => OpParams ->
                               ByteString -- ^ __interpolation__
                               -> Tensor v'1 dtype -- ^ __images__
                               -> Tensor v'2 Float -- ^ __transforms__
                               -> Tensor v'3 Data.Int.Int32 -- ^ __output_shape__
                               -> Tensor Build dtype -- ^ __transformed_images__
imageProjectiveTransformV2' :: OpParams
-> ByteString
-> Tensor v'1 dtype
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build dtype
imageProjectiveTransformV2' op'options :: OpParams
op'options interpolation :: ByteString
interpolation images :: Tensor v'1 dtype
images transforms :: Tensor v'2 Float
transforms
                            output_shape :: Tensor v'3 Int32
output_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 dtype
images,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
transforms,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
output_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ImageProjectiveTransformV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "interpolation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
interpolation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "dtype" }
input_arg { name: "transforms" type: DT_FLOAT }
input_arg { name: "output_shape" type: DT_INT32 }
output_arg { name: "transformed_images" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr { name: "interpolation" type: "string" }
attr {
  name: "fill_mode" type: "string" default_value { s: "CONSTANT" }
}
-}
-- | 
imageSummary :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Data.Word.Word8,
                                            Double, Float] t) =>
                Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                -> Tensor v'2 t -- ^ __tensor__
                -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
imageSummary :: Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
imageSummary = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Word8, Double, Float] t =>
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
imageSummary' OpParams
forall a. a -> a
id
imageSummary' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Data.Word.Word8,
                                             Double, Float] t) => OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                 -> Tensor v'2 t -- ^ __tensor__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
imageSummary' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
imageSummary' op'options :: OpParams
op'options tag :: Tensor v'1 ByteString
tag tensor :: Tensor v'2 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tag,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ImageSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type_attr: "T" }
output_arg { name: "summary" type: DT_STRING }
attr {
  name: "max_images"
  type: "int"
  default_value { i: 3 }
  has_minimum: true
  minimum: 1
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_UINT8 type: DT_FLOAT type: DT_HALF type: DT_DOUBLE
    }
  }
}
attr {
  name: "bad_color"
  type: "tensor"
  default_value {
    tensor {
      dtype: DT_UINT8
      tensor_shape { dim { size: 4 } }
      int_val: 255
      int_val: 0
      int_val: 0
      int_val: 255
    }
  }
}
-}
-- | 
immutableConst :: forall dtype . (TensorType dtype) =>
                  ByteString -- ^ __memory_region_name__
                  -> Shape -- ^ __shape__
                  -> Tensor Build dtype -- ^ __tensor__
immutableConst :: ByteString -> Shape -> Tensor Build dtype
immutableConst = OpParams -> ByteString -> Shape -> Tensor Build dtype
forall dtype.
TensorType dtype =>
OpParams -> ByteString -> Shape -> Tensor Build dtype
immutableConst' OpParams
forall a. a -> a
id
immutableConst' :: forall dtype . (TensorType dtype) => OpParams ->
                   ByteString -- ^ __memory_region_name__
                   -> Shape -- ^ __shape__
                   -> Tensor Build dtype -- ^ __tensor__
immutableConst' :: OpParams -> ByteString -> Shape -> Tensor Build dtype
immutableConst' op'options :: OpParams
op'options memory_region_name :: ByteString
memory_region_name shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ImmutableConst"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "memory_region_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
memory_region_name
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "tensor" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
attr { name: "memory_region_name" type: "string" }
-}
-- | 
importEvent :: forall v'1 v'2 m' . (MonadBuild m') =>
               Tensor v'1 ResourceHandle -- ^ __writer__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __event__
               -> m' (ControlNode)
importEvent :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString -> m' ControlNode
importEvent = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
importEvent' OpParams
forall a. a -> a
id
importEvent' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                Tensor v'1 ResourceHandle -- ^ __writer__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __event__
                -> m' (ControlNode)
importEvent' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
importEvent' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer event :: Tensor v'2 ByteString
event | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
event]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ImportEvent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "event" type: DT_STRING }
-}
-- | 
inTopK :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64] t) =>
          Data.Int.Int64 -- ^ __k__
          -> Tensor v'1 Float -- ^ __predictions__
          -> Tensor v'2 t -- ^ __targets__
          -> Tensor Build Bool -- ^ __precision__
inTopK :: Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
inTopK = OpParams
-> Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams
-> Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
inTopK' OpParams
forall a. a -> a
id
inTopK' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64] t) =>
           OpParams ->
           Data.Int.Int64 -- ^ __k__
           -> Tensor v'1 Float -- ^ __predictions__
           -> Tensor v'2 t -- ^ __targets__
           -> Tensor Build Bool -- ^ __precision__
inTopK' :: OpParams
-> Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
inTopK' op'options :: OpParams
op'options k :: Int64
k predictions :: Tensor v'1 Float
predictions targets :: Tensor v'2 t
targets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
predictions,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
targets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InTopK"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "k" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
k
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "predictions" type: DT_FLOAT }
input_arg { name: "targets" type_attr: "T" }
output_arg { name: "precision" type: DT_BOOL }
attr { name: "k" type: "int" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
inTopKV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] t) =>
            Tensor v'1 Float -- ^ __predictions__
            -> Tensor v'2 t -- ^ __targets__
            -> Tensor v'3 t -- ^ __k__
            -> Tensor Build Bool -- ^ __precision__
inTopKV2 :: Tensor v'1 Float
-> Tensor v'2 t -> Tensor v'3 t -> Tensor Build Bool
inTopKV2 = OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build Bool
inTopKV2' OpParams
forall a. a -> a
id
inTopKV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] t) => OpParams ->
             Tensor v'1 Float -- ^ __predictions__
             -> Tensor v'2 t -- ^ __targets__
             -> Tensor v'3 t -- ^ __k__
             -> Tensor Build Bool -- ^ __precision__
inTopKV2' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build Bool
inTopKV2' op'options :: OpParams
op'options predictions :: Tensor v'1 Float
predictions targets :: Tensor v'2 t
targets k :: Tensor v'3 t
k | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
predictions,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
targets,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
k]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InTopKV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "predictions" type: DT_FLOAT }
input_arg { name: "targets" type_attr: "T" }
input_arg { name: "k" type_attr: "T" }
output_arg { name: "precision" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
infeedDequeue :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                 Shape -- ^ __shape__
                 -> m' (Tensor Value dtype) -- ^ __output__
infeedDequeue :: Shape -> m' (Tensor Value dtype)
infeedDequeue = OpParams -> Shape -> m' (Tensor Value dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Value dtype)
infeedDequeue' OpParams
forall a. a -> a
id
infeedDequeue' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                  OpParams ->
                  Shape -- ^ __shape__
                  -> m' (Tensor Value dtype) -- ^ __output__
infeedDequeue' :: OpParams -> Shape -> m' (Tensor Value dtype)
infeedDequeue' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InfeedDequeue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
-}
-- | 
infeedDequeueTuple :: forall dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
                      m' (TensorList (Value) dtypes) -- ^ __outputs__
infeedDequeueTuple :: m' (TensorList Value dtypes)
infeedDequeueTuple = OpParams -> m' (TensorList Value dtypes)
forall (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> m' (TensorList Value dtypes)
infeedDequeueTuple' OpParams
forall a. a -> a
id
infeedDequeueTuple' :: forall dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
                       OpParams ->
                       m' (TensorList (Value) dtypes) -- ^ __outputs__
infeedDequeueTuple' :: OpParams -> m' (TensorList Value dtypes)
infeedDequeueTuple' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InfeedDequeueTuple"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "outputs" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "shapes" type: "list(shape)" }
-}
-- | 
infeedEnqueue :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                 Tensor v'1 dtype -- ^ __input__
                 -> m' (ControlNode)
infeedEnqueue :: Tensor v'1 dtype -> m' ControlNode
infeedEnqueue = OpParams -> Tensor v'1 dtype -> m' ControlNode
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 dtype -> m' ControlNode
infeedEnqueue' OpParams
forall a. a -> a
id
infeedEnqueue' :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                  OpParams ->
                  Tensor v'1 dtype -- ^ __input__
                  -> m' (ControlNode)
infeedEnqueue' :: OpParams -> Tensor v'1 dtype -> m' ControlNode
infeedEnqueue' op'options :: OpParams
op'options input :: Tensor v'1 dtype
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 dtype
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InfeedEnqueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" default_value { shape { } } }
attr {
  name: "layout" type: "list(int)" default_value { list { } }
}
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
infeedEnqueuePrelinearizedBuffer :: forall v'1 m' . (MonadBuild m') =>
                                    Tensor v'1 Variant -- ^ __input__
                                    -> m' (ControlNode)
infeedEnqueuePrelinearizedBuffer :: Tensor v'1 Variant -> m' ControlNode
infeedEnqueuePrelinearizedBuffer = OpParams -> Tensor v'1 Variant -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 Variant -> m' ControlNode
infeedEnqueuePrelinearizedBuffer' OpParams
forall a. a -> a
id
infeedEnqueuePrelinearizedBuffer' :: forall v'1 m' . (MonadBuild m') =>
                                     OpParams ->
                                     Tensor v'1 Variant -- ^ __input__
                                     -> m' (ControlNode)
infeedEnqueuePrelinearizedBuffer' :: OpParams -> Tensor v'1 Variant -> m' ControlNode
infeedEnqueuePrelinearizedBuffer' op'options :: OpParams
op'options input :: Tensor v'1 Variant
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InfeedEnqueuePrelinearizedBuffer"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_VARIANT }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
infeedEnqueueTuple :: forall v'1 dtypes m' . (MonadBuild m',
                                              TensorTypes dtypes) =>
                      TensorList (v'1) dtypes -- ^ __inputs__
                      -> m' (ControlNode)
infeedEnqueueTuple :: TensorList v'1 dtypes -> m' ControlNode
infeedEnqueueTuple = OpParams -> TensorList v'1 dtypes -> m' ControlNode
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> TensorList v'1 dtypes -> m' ControlNode
infeedEnqueueTuple' OpParams
forall a. a -> a
id
infeedEnqueueTuple' :: forall v'1 dtypes m' . (MonadBuild m',
                                               TensorTypes dtypes) =>
                       OpParams ->
                       TensorList (v'1) dtypes -- ^ __inputs__
                       -> m' (ControlNode)
infeedEnqueueTuple' :: OpParams -> TensorList v'1 dtypes -> m' ControlNode
infeedEnqueueTuple' op'options :: OpParams
op'options inputs :: TensorList v'1 dtypes
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 dtypes
inputs]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InfeedEnqueueTuple"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "shapes" type: "list(shape)" }
attr {
  name: "layouts" type: "list(int)" default_value { list { } }
}
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
initializeTable :: forall v'2 v'3 tkey tval m' . (MonadBuild m',
                                                  TensorType tkey,
                                                  TensorType tval) =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                   -> Tensor v'2 tkey -- ^ __keys__
                   -> Tensor v'3 tval -- ^ __values__
                   -> m' (ControlNode)
initializeTable :: Tensor Ref ByteString
-> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
initializeTable = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) tkey tval (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
initializeTable' OpParams
forall a. a -> a
id
initializeTable' :: forall v'2 v'3 tkey tval m' . (MonadBuild m',
                                                   TensorType tkey,
                                                   TensorType tval) =>
                    OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                    -> Tensor v'2 tkey -- ^ __keys__
                    -> Tensor v'3 tval -- ^ __values__
                    -> m' (ControlNode)
initializeTable' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
initializeTable' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle keys :: Tensor v'2 tkey
keys values :: Tensor v'3 tval
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle,
                                                             Tensor v'2 tkey -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tkey
keys,
                                                             Tensor v'3 tval -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tval
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InitializeTable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tkey" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tkey -> DataType
forall a. TensorType a => a -> DataType
tensorType (tkey
forall a. HasCallStack => a
undefined :: tkey)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tval" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tval -> DataType
forall a. TensorType a => a -> DataType
tensorType (tval
forall a. HasCallStack => a
undefined :: tval)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
input_arg { name: "keys" type_attr: "Tkey" }
input_arg { name: "values" type_attr: "Tval" }
attr { name: "Tkey" type: "type" }
attr { name: "Tval" type: "type" }
-}
-- | 
initializeTableFromDataset :: forall v'1 v'2 m' . (MonadBuild m') =>
                              Tensor v'1 ResourceHandle -- ^ __table_handle__
                              -> Tensor v'2 Variant -- ^ __dataset__
                              -> m' (ControlNode)
initializeTableFromDataset :: Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
initializeTableFromDataset = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
initializeTableFromDataset' OpParams
forall a. a -> a
id
initializeTableFromDataset' :: forall v'1 v'2 m' . (MonadBuild m') =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __table_handle__
                               -> Tensor v'2 Variant -- ^ __dataset__
                               -> m' (ControlNode)
initializeTableFromDataset' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Variant
-> m' ControlNode
initializeTableFromDataset' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle dataset :: Tensor v'2 Variant
dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
dataset]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InitializeTableFromDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "dataset" type: DT_VARIANT }
-}
-- | 
initializeTableFromTextFile :: forall v'2 m' . (MonadBuild m') =>
                               Data.Int.Int64 -- ^ __key_index__
                               -> Data.Int.Int64 -- ^ __value_index__
                               -> Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                               -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                               -> m' (ControlNode)
initializeTableFromTextFile :: Int64
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFile = OpParams
-> Int64
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFile' OpParams
forall a. a -> a
id
initializeTableFromTextFile' :: forall v'2 m' . (MonadBuild m') => OpParams ->
                                Data.Int.Int64 -- ^ __key_index__
                                -> Data.Int.Int64 -- ^ __value_index__
                                -> Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                                -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                                -> m' (ControlNode)
initializeTableFromTextFile' :: OpParams
-> Int64
-> Int64
-> Tensor Ref ByteString
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFile' op'options :: OpParams
op'options key_index :: Int64
key_index value_index :: Int64
value_index table_handle :: Tensor Ref ByteString
table_handle
                             filename :: Tensor v'2 ByteString
filename | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InitializeTableFromTextFile"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
key_index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
value_index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
input_arg { name: "filename" type: DT_STRING }
attr {
  name: "key_index" type: "int" has_minimum: true minimum: -2
}
attr {
  name: "value_index" type: "int" has_minimum: true minimum: -2
}
attr {
  name: "vocab_size"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr { name: "delimiter" type: "string" default_value { s: "\t" } }
-}
-- | 
initializeTableFromTextFileV2 :: forall v'1 v'2 m' . (MonadBuild m') =>
                                 Data.Int.Int64 -- ^ __key_index__
                                 -> Data.Int.Int64 -- ^ __value_index__
                                 -> Tensor v'1 ResourceHandle -- ^ __table_handle__
                                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                                 -> m' (ControlNode)
initializeTableFromTextFileV2 :: Int64
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFileV2 = OpParams
-> Int64
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFileV2' OpParams
forall a. a -> a
id
initializeTableFromTextFileV2' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                  OpParams ->
                                  Data.Int.Int64 -- ^ __key_index__
                                  -> Data.Int.Int64 -- ^ __value_index__
                                  -> Tensor v'1 ResourceHandle -- ^ __table_handle__
                                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __filename__
                                  -> m' (ControlNode)
initializeTableFromTextFileV2' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
initializeTableFromTextFileV2' op'options :: OpParams
op'options key_index :: Int64
key_index value_index :: Int64
value_index table_handle :: Tensor v'1 ResourceHandle
table_handle
                               filename :: Tensor v'2 ByteString
filename | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
filename]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InitializeTableFromTextFileV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
key_index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
value_index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "filename" type: DT_STRING }
attr {
  name: "key_index" type: "int" has_minimum: true minimum: -2
}
attr {
  name: "value_index" type: "int" has_minimum: true minimum: -2
}
attr {
  name: "vocab_size"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr { name: "delimiter" type: "string" default_value { s: "\t" } }
-}
-- | 
initializeTableV2 :: forall v'1 v'2 v'3 tkey tval m' . (MonadBuild m',
                                                        TensorType tkey,
                                                        TensorType tval) =>
                     Tensor v'1 ResourceHandle -- ^ __table_handle__
                     -> Tensor v'2 tkey -- ^ __keys__
                     -> Tensor v'3 tval -- ^ __values__
                     -> m' (ControlNode)
initializeTableV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
initializeTableV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tkey tval
       (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
initializeTableV2' OpParams
forall a. a -> a
id
initializeTableV2' :: forall v'1 v'2 v'3 tkey tval m' . (MonadBuild m',
                                                         TensorType tkey,
                                                         TensorType tval) =>
                      OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __table_handle__
                      -> Tensor v'2 tkey -- ^ __keys__
                      -> Tensor v'3 tval -- ^ __values__
                      -> m' (ControlNode)
initializeTableV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
initializeTableV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle keys :: Tensor v'2 tkey
keys values :: Tensor v'3 tval
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 tkey -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tkey
keys,
                                                             Tensor v'3 tval -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tval
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "InitializeTableV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tkey" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tkey -> DataType
forall a. TensorType a => a -> DataType
tensorType (tkey
forall a. HasCallStack => a
undefined :: tkey)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tval" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tval -> DataType
forall a. TensorType a => a -> DataType
tensorType (tval
forall a. HasCallStack => a
undefined :: tval)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "keys" type_attr: "Tkey" }
input_arg { name: "values" type_attr: "Tval" }
attr { name: "Tkey" type: "type" }
attr { name: "Tval" type: "type" }
-}
-- | 
inplaceAdd :: forall v'1 v'2 v'3 t . (TensorType t) => Tensor v'1 t -- ^ __x__
              -> Tensor v'2 Data.Int.Int32 -- ^ __i__
              -> Tensor v'3 t -- ^ __v__
              -> Tensor Build t -- ^ __y__
inplaceAdd :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
inplaceAdd = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceAdd' OpParams
forall a. a -> a
id
inplaceAdd' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 Data.Int.Int32 -- ^ __i__
               -> Tensor v'3 t -- ^ __v__
               -> Tensor Build t -- ^ __y__
inplaceAdd' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceAdd' op'options :: OpParams
op'options x :: Tensor v'1 t
x i :: Tensor v'2 Int32
i v :: Tensor v'3 t
v | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
i,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
v]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InplaceAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "i" type: DT_INT32 }
input_arg { name: "v" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
inplaceSub :: forall v'1 v'2 v'3 t . (TensorType t) => Tensor v'1 t -- ^ __x__
              -> Tensor v'2 Data.Int.Int32 -- ^ __i__
              -> Tensor v'3 t -- ^ __v__
              -> Tensor Build t -- ^ __y__
inplaceSub :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
inplaceSub = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceSub' OpParams
forall a. a -> a
id
inplaceSub' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 Data.Int.Int32 -- ^ __i__
               -> Tensor v'3 t -- ^ __v__
               -> Tensor Build t -- ^ __y__
inplaceSub' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceSub' op'options :: OpParams
op'options x :: Tensor v'1 t
x i :: Tensor v'2 Int32
i v :: Tensor v'3 t
v | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
i,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
v]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InplaceSub"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "i" type: DT_INT32 }
input_arg { name: "v" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
inplaceUpdate :: forall v'1 v'2 v'3 t . (TensorType t) =>
                 Tensor v'1 t -- ^ __x__
                 -> Tensor v'2 Data.Int.Int32 -- ^ __i__
                 -> Tensor v'3 t -- ^ __v__
                 -> Tensor Build t -- ^ __y__
inplaceUpdate :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
inplaceUpdate = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceUpdate' OpParams
forall a. a -> a
id
inplaceUpdate' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __x__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __i__
                  -> Tensor v'3 t -- ^ __v__
                  -> Tensor Build t -- ^ __y__
inplaceUpdate' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
inplaceUpdate' op'options :: OpParams
op'options x :: Tensor v'1 t
x i :: Tensor v'2 Int32
i v :: Tensor v'3 t
v | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
i,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
v]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InplaceUpdate"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "i" type: DT_INT32 }
input_arg { name: "v" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
inv :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
inv :: Tensor v'1 t -> Tensor Build t
inv = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
inv' OpParams
forall a. a -> a
id
inv' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
inv' :: OpParams -> Tensor v'1 t -> Tensor Build t
inv' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Inv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
invGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __y__
           -> Tensor v'2 t -- ^ __dy__
           -> Tensor Build t -- ^ __z__
invGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
invGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
invGrad' OpParams
forall a. a -> a
id
invGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __y__
            -> Tensor v'2 t -- ^ __dy__
            -> Tensor Build t -- ^ __z__
invGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
invGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InvGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
invert :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                  Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
invert :: Tensor v'1 t -> Tensor Build t
invert = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
invert' OpParams
forall a. a -> a
id
invert' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Int.Int64, Data.Int.Int8,
                                   Data.Word.Word16, Data.Word.Word32,
                                   Data.Word.Word64, Data.Word.Word8] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
invert' :: OpParams -> Tensor v'1 t -> Tensor Build t
invert' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Invert"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
invertPermutation :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] t) =>
                     Tensor v'1 t -- ^ __x__
                     -> Tensor Build t -- ^ __y__
invertPermutation :: Tensor v'1 t -> Tensor Build t
invertPermutation = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
invertPermutation' OpParams
forall a. a -> a
id
invertPermutation' :: forall v'1 t . (OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] t) => OpParams ->
                      Tensor v'1 t -- ^ __x__
                      -> Tensor Build t -- ^ __y__
invertPermutation' :: OpParams -> Tensor v'1 t -> Tensor Build t
invertPermutation' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "InvertPermutation"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
isBoostedTreesEnsembleInitialized :: forall v'1 m' . (MonadBuild m') =>
                                     Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                     -> m' (Tensor Value Bool) -- ^ __is_initialized__
isBoostedTreesEnsembleInitialized :: Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesEnsembleInitialized = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesEnsembleInitialized' OpParams
forall a. a -> a
id
isBoostedTreesEnsembleInitialized' :: forall v'1 m' . (MonadBuild m') =>
                                      OpParams ->
                                      Tensor v'1 ResourceHandle -- ^ __tree_ensemble_handle__
                                      -> m' (Tensor Value Bool) -- ^ __is_initialized__
isBoostedTreesEnsembleInitialized' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesEnsembleInitialized' op'options :: OpParams
op'options
                                   tree_ensemble_handle :: Tensor v'1 ResourceHandle
tree_ensemble_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_ensemble_handle]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IsBoostedTreesEnsembleInitialized"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_ensemble_handle" type: DT_RESOURCE }
output_arg { name: "is_initialized" type: DT_BOOL }
-}
-- | 
isBoostedTreesQuantileStreamResourceInitialized :: forall v'1
                                                   m' . (MonadBuild m') =>
                                                   Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                   -> m' (Tensor Value Bool) -- ^ __is_initialized__
isBoostedTreesQuantileStreamResourceInitialized :: Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesQuantileStreamResourceInitialized = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesQuantileStreamResourceInitialized' OpParams
forall a. a -> a
id
isBoostedTreesQuantileStreamResourceInitialized' :: forall v'1
                                                    m' . (MonadBuild m') =>
                                                    OpParams ->
                                                    Tensor v'1 ResourceHandle -- ^ __quantile_stream_resource_handle__
                                                    -> m' (Tensor Value Bool) -- ^ __is_initialized__
isBoostedTreesQuantileStreamResourceInitialized' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
isBoostedTreesQuantileStreamResourceInitialized' op'options :: OpParams
op'options
                                                 quantile_stream_resource_handle :: Tensor v'1 ResourceHandle
quantile_stream_resource_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
quantile_stream_resource_handle]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IsBoostedTreesQuantileStreamResourceInitialized"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "quantile_stream_resource_handle" type: DT_RESOURCE
}
output_arg { name: "is_initialized" type: DT_BOOL }
-}
-- | 
isFinite :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor Build Bool -- ^ __y__
isFinite :: Tensor v'1 t -> Tensor Build Bool
isFinite = OpParams -> Tensor v'1 t -> Tensor Build Bool
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build Bool
isFinite' OpParams
forall a. a -> a
id
isFinite' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build Bool -- ^ __y__
isFinite' :: OpParams -> Tensor v'1 t -> Tensor Build Bool
isFinite' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IsFinite"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
isInf :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build Bool -- ^ __y__
isInf :: Tensor v'1 t -> Tensor Build Bool
isInf = OpParams -> Tensor v'1 t -> Tensor Build Bool
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build Bool
isInf' OpParams
forall a. a -> a
id
isInf' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build Bool -- ^ __y__
isInf' :: OpParams -> Tensor v'1 t -> Tensor Build Bool
isInf' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IsInf"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
isNan :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build Bool -- ^ __y__
isNan :: Tensor v'1 t -> Tensor Build Bool
isNan = OpParams -> Tensor v'1 t -> Tensor Build Bool
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build Bool
isNan' OpParams
forall a. a -> a
id
isNan' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build Bool -- ^ __y__
isNan' :: OpParams -> Tensor v'1 t -> Tensor Build Bool
isNan' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "IsNan"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
isVariableInitialized :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                         Tensor Ref dtype -- ^ __ref__
                         -> m' (Tensor Value Bool) -- ^ __is_initialized__
isVariableInitialized :: Tensor Ref dtype -> m' (Tensor Value Bool)
isVariableInitialized = OpParams -> Tensor Ref dtype -> m' (Tensor Value Bool)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor Ref dtype -> m' (Tensor Value Bool)
isVariableInitialized' OpParams
forall a. a -> a
id
isVariableInitialized' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                          OpParams ->
                          Tensor Ref dtype -- ^ __ref__
                          -> m' (Tensor Value Bool) -- ^ __is_initialized__
isVariableInitialized' :: OpParams -> Tensor Ref dtype -> m' (Tensor Value Bool)
isVariableInitialized' op'options :: OpParams
op'options ref :: Tensor Ref dtype
ref | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref dtype
ref]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IsVariableInitialized"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "dtype" is_ref: true }
output_arg { name: "is_initialized" type: DT_BOOL }
attr { name: "dtype" type: "type" }
-}
-- | 
iterator :: forall m' . (MonadBuild m') => ByteString -- ^ __container__
            -> [DataType] -- ^ __output_types__
            -> ByteString -- ^ __shared_name__
            -> m' (Tensor Value ResourceHandle) -- ^ __handle__
iterator :: ByteString
-> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
iterator = OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
iterator' OpParams
forall a. a -> a
id
iterator' :: forall m' . (MonadBuild m') => OpParams ->
             ByteString -- ^ __container__
             -> [DataType] -- ^ __output_types__
             -> ByteString -- ^ __shared_name__
             -> m' (Tensor Value ResourceHandle) -- ^ __handle__
iterator' :: OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
iterator' op'options :: OpParams
op'options container :: ByteString
container output_types :: [DataType]
output_types shared_name :: ByteString
shared_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Iterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "container" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
container
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "shared_name" type: "string" }
attr { name: "container" type: "string" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
iteratorFromStringHandle :: forall v'1 m' . (MonadBuild m') =>
                            Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                            -> m' (Tensor Value ResourceHandle) -- ^ __resource_handle__
iteratorFromStringHandle :: Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandle = OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandle' OpParams
forall a. a -> a
id
iteratorFromStringHandle' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                             Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                             -> m' (Tensor Value ResourceHandle) -- ^ __resource_handle__
iteratorFromStringHandle' :: OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandle' op'options :: OpParams
op'options string_handle :: Tensor v'1 ByteString
string_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
string_handle]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorFromStringHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "string_handle" type: DT_STRING }
output_arg { name: "resource_handle" type: DT_RESOURCE }
attr {
  name: "output_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
-}
-- | 
iteratorFromStringHandleV2 :: forall v'1 m' . (MonadBuild m') =>
                              Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                              -> m' (Tensor Value ResourceHandle) -- ^ __resource_handle__
iteratorFromStringHandleV2 :: Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandleV2 = OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandleV2' OpParams
forall a. a -> a
id
iteratorFromStringHandleV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                               Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                               -> m' (Tensor Value ResourceHandle) -- ^ __resource_handle__
iteratorFromStringHandleV2' :: OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
iteratorFromStringHandleV2' op'options :: OpParams
op'options string_handle :: Tensor v'1 ByteString
string_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
string_handle]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorFromStringHandleV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "string_handle" type: DT_STRING }
output_arg { name: "resource_handle" type: DT_RESOURCE }
attr {
  name: "output_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
-}
-- | 
iteratorGetDevice :: forall v'1 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __resource__
                     -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __device__
iteratorGetDevice :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorGetDevice = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorGetDevice' OpParams
forall a. a -> a
id
iteratorGetDevice' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __device__
iteratorGetDevice' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorGetDevice' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorGetDevice"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "device" type: DT_STRING }
-}
-- | 
iteratorGetNext :: forall v'1 output_types m' . (MonadBuild m',
                                                 TensorTypes output_types) =>
                   Tensor v'1 ResourceHandle -- ^ __iterator__
                   -> m' (TensorList (Value) output_types) -- ^ __components__
iteratorGetNext :: Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNext = OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
forall (v'1 :: * -> *) (output_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes output_types) =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNext' OpParams
forall a. a -> a
id
iteratorGetNext' :: forall v'1 output_types m' . (MonadBuild m',
                                                  TensorTypes output_types) =>
                    OpParams ->
                    Tensor v'1 ResourceHandle -- ^ __iterator__
                    -> m' (TensorList (Value) output_types) -- ^ __components__
iteratorGetNext' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNext' op'options :: OpParams
op'options iterator :: Tensor v'1 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value output_types)
 -> m' (TensorList Value output_types))
-> Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build (TensorList Value output_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorGetNext"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "iterator" type: DT_RESOURCE }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
iteratorGetNextAsOptional :: forall v'1 m' . (MonadBuild m') =>
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 ResourceHandle -- ^ __iterator__
                             -> m' (Tensor Value Variant) -- ^ __optional__
iteratorGetNextAsOptional :: [DataType]
-> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
iteratorGetNextAsOptional = OpParams
-> [DataType]
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Variant)
iteratorGetNextAsOptional' OpParams
forall a. a -> a
id
iteratorGetNextAsOptional' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 ResourceHandle -- ^ __iterator__
                              -> m' (Tensor Value Variant) -- ^ __optional__
iteratorGetNextAsOptional' :: OpParams
-> [DataType]
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value Variant)
iteratorGetNextAsOptional' op'options :: OpParams
op'options output_types :: [DataType]
output_types iterator :: Tensor v'1 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorGetNextAsOptional"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "iterator" type: DT_RESOURCE }
output_arg { name: "optional" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
iteratorGetNextSync :: forall v'1 output_types m' . (MonadBuild m',
                                                     TensorTypes output_types) =>
                       Tensor v'1 ResourceHandle -- ^ __iterator__
                       -> m' (TensorList (Value) output_types) -- ^ __components__
iteratorGetNextSync :: Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNextSync = OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
forall (v'1 :: * -> *) (output_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes output_types) =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNextSync' OpParams
forall a. a -> a
id
iteratorGetNextSync' :: forall v'1 output_types m' . (MonadBuild m',
                                                      TensorTypes output_types) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __iterator__
                        -> m' (TensorList (Value) output_types) -- ^ __components__
iteratorGetNextSync' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
iteratorGetNextSync' op'options :: OpParams
op'options iterator :: Tensor v'1 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value output_types)
 -> m' (TensorList Value output_types))
-> Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build (TensorList Value output_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorGetNextSync"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "iterator" type: DT_RESOURCE }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
iteratorToStringHandle :: forall v'1 m' . (MonadBuild m') =>
                          Tensor v'1 ResourceHandle -- ^ __resource_handle__
                          -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __string_handle__
iteratorToStringHandle :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorToStringHandle = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorToStringHandle' OpParams
forall a. a -> a
id
iteratorToStringHandle' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __resource_handle__
                           -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __string_handle__
iteratorToStringHandle' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
iteratorToStringHandle' op'options :: OpParams
op'options resource_handle :: Tensor v'1 ResourceHandle
resource_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource_handle]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorToStringHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource_handle" type: DT_RESOURCE }
output_arg { name: "string_handle" type: DT_STRING }
-}
-- | 
iteratorV2 :: forall m' . (MonadBuild m') => ByteString -- ^ __container__
              -> [DataType] -- ^ __output_types__
              -> ByteString -- ^ __shared_name__
              -> m' (Tensor Value ResourceHandle) -- ^ __handle__
iteratorV2 :: ByteString
-> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
iteratorV2 = OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
iteratorV2' OpParams
forall a. a -> a
id
iteratorV2' :: forall m' . (MonadBuild m') => OpParams ->
               ByteString -- ^ __container__
               -> [DataType] -- ^ __output_types__
               -> ByteString -- ^ __shared_name__
               -> m' (Tensor Value ResourceHandle) -- ^ __handle__
iteratorV2' :: OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
iteratorV2' op'options :: OpParams
op'options container :: ByteString
container output_types :: [DataType]
output_types shared_name :: ByteString
shared_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "IteratorV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "container" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
container
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "shared_name" type: "string" }
attr { name: "container" type: "string" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
kMC2ChainInitialization :: Tensor v'1 Float -- ^ __distances__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __seed__
                           -> Tensor Build Data.Int.Int64 -- ^ __index__
kMC2ChainInitialization :: Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
kMC2ChainInitialization = OpParams
-> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
kMC2ChainInitialization' OpParams
forall a. a -> a
id
kMC2ChainInitialization' :: OpParams ->
                            Tensor v'1 Float -- ^ __distances__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __seed__
                            -> Tensor Build Data.Int.Int64 -- ^ __index__
kMC2ChainInitialization' :: OpParams
-> Tensor v'1 Float -> Tensor v'2 Int64 -> Tensor Build Int64
kMC2ChainInitialization' op'options :: OpParams
op'options distances :: Tensor v'1 Float
distances seed :: Tensor v'2 Int64
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
distances,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "KMC2ChainInitialization"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "distances" type: DT_FLOAT }
input_arg { name: "seed" type: DT_INT64 }
output_arg { name: "index" type: DT_INT64 }
-}
-- | 
kmeansPlusPlusInitialization :: Tensor v'1 Float -- ^ __points__
                                -> Tensor v'2 Data.Int.Int64 -- ^ __num_to_sample__
                                -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                                -> Tensor v'4 Data.Int.Int64 -- ^ __num_retries_per_sample__
                                -> Tensor Build Float -- ^ __samples__
kmeansPlusPlusInitialization :: Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Float
kmeansPlusPlusInitialization = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Float
kmeansPlusPlusInitialization' OpParams
forall a. a -> a
id
kmeansPlusPlusInitialization' :: OpParams ->
                                 Tensor v'1 Float -- ^ __points__
                                 -> Tensor v'2 Data.Int.Int64 -- ^ __num_to_sample__
                                 -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                                 -> Tensor v'4 Data.Int.Int64 -- ^ __num_retries_per_sample__
                                 -> Tensor Build Float -- ^ __samples__
kmeansPlusPlusInitialization' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Float
kmeansPlusPlusInitialization' op'options :: OpParams
op'options points :: Tensor v'1 Float
points num_to_sample :: Tensor v'2 Int64
num_to_sample seed :: Tensor v'3 Int64
seed
                              num_retries_per_sample :: Tensor v'4 Int64
num_retries_per_sample | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
points,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_to_sample,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
num_retries_per_sample]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "KmeansPlusPlusInitialization"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "points" type: DT_FLOAT }
input_arg { name: "num_to_sample" type: DT_INT64 }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "num_retries_per_sample" type: DT_INT64 }
output_arg { name: "samples" type: DT_FLOAT }
-}
-- | 
l2Loss :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __t__
          -> Tensor Build t -- ^ __output__
l2Loss :: Tensor v'1 t -> Tensor Build t
l2Loss = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
l2Loss' OpParams
forall a. a -> a
id
l2Loss' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __t__
           -> Tensor Build t -- ^ __output__
l2Loss' :: OpParams -> Tensor v'1 t -> Tensor Build t
l2Loss' op'options :: OpParams
op'options t :: Tensor v'1 t
t | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
t]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "L2Loss"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "t" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
lMDBDataset :: forall v'1 m' . (MonadBuild m') =>
               [DataType] -- ^ __output_types__
               -> Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
               -> m' (Tensor Value Variant) -- ^ __handle__
lMDBDataset :: [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
lMDBDataset = OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
lMDBDataset' OpParams
forall a. a -> a
id
lMDBDataset' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                [DataType] -- ^ __output_types__
                -> Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                -> m' (Tensor Value Variant) -- ^ __handle__
lMDBDataset' :: OpParams
-> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
lMDBDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types filenames :: Tensor v'1 ByteString
filenames | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LMDBDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
lMDBReader :: forall m' . (MonadBuild m') =>
              m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
lMDBReader :: m' (Tensor Ref ByteString)
lMDBReader = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
lMDBReader' OpParams
forall a. a -> a
id
lMDBReader' :: forall m' . (MonadBuild m') => OpParams ->
               m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
lMDBReader' :: OpParams -> m' (Tensor Ref ByteString)
lMDBReader' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LMDBReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
lRN :: forall v'1 t . (OneOf '[Data.Word.Word16, Float] t) =>
       Tensor v'1 t -- ^ __input__
       -> Tensor Build t -- ^ __output__
lRN :: Tensor v'1 t -> Tensor Build t
lRN = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
lRN' OpParams
forall a. a -> a
id
lRN' :: forall v'1 t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __input__
        -> Tensor Build t -- ^ __output__
lRN' :: OpParams -> Tensor v'1 t -> Tensor Build t
lRN' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LRN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "depth_radius" type: "int" default_value { i: 5 } }
attr { name: "bias" type: "float" default_value { f: 1.0 } }
attr { name: "alpha" type: "float" default_value { f: 1.0 } }
attr { name: "beta" type: "float" default_value { f: 0.5 } }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
-}
-- | 
lRNGrad :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16, Float] t) =>
           Tensor v'1 t -- ^ __input_grads__
           -> Tensor v'2 t -- ^ __input_image__
           -> Tensor v'3 t -- ^ __output_image__
           -> Tensor Build t -- ^ __output__
lRNGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
lRNGrad = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
lRNGrad' OpParams
forall a. a -> a
id
lRNGrad' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __input_grads__
            -> Tensor v'2 t -- ^ __input_image__
            -> Tensor v'3 t -- ^ __output_image__
            -> Tensor Build t -- ^ __output__
lRNGrad' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
lRNGrad' op'options :: OpParams
op'options input_grads :: Tensor v'1 t
input_grads input_image :: Tensor v'2 t
input_image output_image :: Tensor v'3 t
output_image | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input_grads,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_image,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
output_image]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LRNGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_grads" type_attr: "T" }
input_arg { name: "input_image" type_attr: "T" }
input_arg { name: "output_image" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "depth_radius" type: "int" default_value { i: 5 } }
attr { name: "bias" type: "float" default_value { f: 1.0 } }
attr { name: "alpha" type: "float" default_value { f: 1.0 } }
attr { name: "beta" type: "float" default_value { f: 0.5 } }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
-}
-- | 
lSTMBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                 t . (OneOf '[Data.Word.Word16, Float] t) =>
                 Tensor v'1 t -- ^ __x__
                 -> Tensor v'2 t -- ^ __cs_prev__
                 -> Tensor v'3 t -- ^ __h_prev__
                 -> Tensor v'4 t -- ^ __w__
                 -> Tensor v'5 t -- ^ __wci__
                 -> Tensor v'6 t -- ^ __wcf__
                 -> Tensor v'7 t -- ^ __wco__
                 -> Tensor v'8 t -- ^ __b__
                 -> (Tensor Build t, Tensor Build t, Tensor Build t,
                     Tensor Build t, Tensor Build t, Tensor Build t,
                     Tensor Build t)
                 -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
                 --
                 -- * __i__
                 --
                 -- * __cs__
                 --
                 -- * __f__
                 --
                 -- * __o__
                 --
                 -- * __ci__
                 --
                 -- * __co__
                 --
                 -- * __h__
lSTMBlockCell :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCell = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCell' OpParams
forall a. a -> a
id
lSTMBlockCell' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                  t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __x__
                  -> Tensor v'2 t -- ^ __cs_prev__
                  -> Tensor v'3 t -- ^ __h_prev__
                  -> Tensor v'4 t -- ^ __w__
                  -> Tensor v'5 t -- ^ __wci__
                  -> Tensor v'6 t -- ^ __wcf__
                  -> Tensor v'7 t -- ^ __wco__
                  -> Tensor v'8 t -- ^ __b__
                  -> (Tensor Build t, Tensor Build t, Tensor Build t,
                      Tensor Build t, Tensor Build t, Tensor Build t,
                      Tensor Build t)
                  -- ^ (__i__, __cs__, __f__, __o__, __ci__, __co__, __h__)
                  --
                  -- * __i__
                  --
                  -- * __cs__
                  --
                  -- * __f__
                  --
                  -- * __o__
                  --
                  -- * __ci__
                  --
                  -- * __co__
                  --
                  -- * __h__
lSTMBlockCell' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
lSTMBlockCell' op'options :: OpParams
op'options x :: Tensor v'1 t
x cs_prev :: Tensor v'2 t
cs_prev h_prev :: Tensor v'3 t
h_prev w :: Tensor v'4 t
w wci :: Tensor v'5 t
wci wcf :: Tensor v'6 t
wcf wco :: Tensor v'7 t
wco b :: Tensor v'8 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
cs_prev,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
h_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
w,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
wci,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wcf,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wco,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LSTMBlockCell"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "i" type_attr: "T" }
output_arg { name: "cs" type_attr: "T" }
output_arg { name: "f" type_attr: "T" }
output_arg { name: "o" type_attr: "T" }
output_arg { name: "ci" type_attr: "T" }
output_arg { name: "co" type_attr: "T" }
output_arg { name: "h" type_attr: "T" }
attr { name: "forget_bias" type: "float" default_value { f: 1.0 } }
attr { name: "cell_clip" type: "float" default_value { f: 3.0 } }
attr {
  name: "use_peephole" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
lSTMBlockCellGrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                     v'13 v'14 v'15 v'16 t . (OneOf '[Data.Word.Word16,
                                                      Float] t) =>
                     Bool -- ^ __use_peephole__
                     -> Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 t -- ^ __cs_prev__
                     -> Tensor v'3 t -- ^ __h_prev__
                     -> Tensor v'4 t -- ^ __w__
                     -> Tensor v'5 t -- ^ __wci__
                     -> Tensor v'6 t -- ^ __wcf__
                     -> Tensor v'7 t -- ^ __wco__
                     -> Tensor v'8 t -- ^ __b__
                     -> Tensor v'9 t -- ^ __i__
                     -> Tensor v'10 t -- ^ __cs__
                     -> Tensor v'11 t -- ^ __f__
                     -> Tensor v'12 t -- ^ __o__
                     -> Tensor v'13 t -- ^ __ci__
                     -> Tensor v'14 t -- ^ __co__
                     -> Tensor v'15 t -- ^ __cs_grad__
                     -> Tensor v'16 t -- ^ __h_grad__
                     -> (Tensor Build t, Tensor Build t, Tensor Build t,
                         Tensor Build t, Tensor Build t)
                     -- ^ (__cs_prev_grad__, __dicfo__, __wci_grad__, __wcf_grad__, __wco_grad__)
                     --
                     -- * __cs_prev_grad__
                     --
                     -- * __dicfo__
                     --
                     -- * __wci_grad__
                     --
                     -- * __wcf_grad__
                     --
                     -- * __wco_grad__
lSTMBlockCellGrad :: Bool
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
lSTMBlockCellGrad = OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (v'13 :: * -> *) (v'14 :: * -> *) (v'15 :: * -> *)
       (v'16 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
lSTMBlockCellGrad' OpParams
forall a. a -> a
id
lSTMBlockCellGrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 v'11 v'12
                      v'13 v'14 v'15 v'16 t . (OneOf '[Data.Word.Word16,
                                                       Float] t) => OpParams ->
                      Bool -- ^ __use_peephole__
                      -> Tensor v'1 t -- ^ __x__
                      -> Tensor v'2 t -- ^ __cs_prev__
                      -> Tensor v'3 t -- ^ __h_prev__
                      -> Tensor v'4 t -- ^ __w__
                      -> Tensor v'5 t -- ^ __wci__
                      -> Tensor v'6 t -- ^ __wcf__
                      -> Tensor v'7 t -- ^ __wco__
                      -> Tensor v'8 t -- ^ __b__
                      -> Tensor v'9 t -- ^ __i__
                      -> Tensor v'10 t -- ^ __cs__
                      -> Tensor v'11 t -- ^ __f__
                      -> Tensor v'12 t -- ^ __o__
                      -> Tensor v'13 t -- ^ __ci__
                      -> Tensor v'14 t -- ^ __co__
                      -> Tensor v'15 t -- ^ __cs_grad__
                      -> Tensor v'16 t -- ^ __h_grad__
                      -> (Tensor Build t, Tensor Build t, Tensor Build t,
                          Tensor Build t, Tensor Build t)
                      -- ^ (__cs_prev_grad__, __dicfo__, __wci_grad__, __wcf_grad__, __wco_grad__)
                      --
                      -- * __cs_prev_grad__
                      --
                      -- * __dicfo__
                      --
                      -- * __wci_grad__
                      --
                      -- * __wcf_grad__
                      --
                      -- * __wco_grad__
lSTMBlockCellGrad' :: OpParams
-> Bool
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> Tensor v'12 t
-> Tensor v'13 t
-> Tensor v'14 t
-> Tensor v'15 t
-> Tensor v'16 t
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
lSTMBlockCellGrad' op'options :: OpParams
op'options use_peephole :: Bool
use_peephole x :: Tensor v'1 t
x cs_prev :: Tensor v'2 t
cs_prev h_prev :: Tensor v'3 t
h_prev w :: Tensor v'4 t
w wci :: Tensor v'5 t
wci wcf :: Tensor v'6 t
wcf wco :: Tensor v'7 t
wco b :: Tensor v'8 t
b i :: Tensor v'9 t
i cs :: Tensor v'10 t
cs
                   f :: Tensor v'11 t
f o :: Tensor v'12 t
o ci :: Tensor v'13 t
ci co :: Tensor v'14 t
co cs_grad :: Tensor v'15 t
cs_grad h_grad :: Tensor v'16 t
h_grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
     Tensor Build t))
-> Build OpDef
-> (Tensor Build t, Tensor Build t, Tensor Build t, Tensor Build t,
    Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
cs_prev,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
h_prev,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
w,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
wci,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
wcf,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
wco,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
b,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
i,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
cs,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
f,
                                                             Tensor v'12 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 t
o,
                                                             Tensor v'13 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'13 t
ci,
                                                             Tensor v'14 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'14 t
co,
                                                             Tensor v'15 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'15 t
cs_grad,
                                                             Tensor v'16 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'16 t
h_grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LSTMBlockCellGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "use_peephole" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
use_peephole
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "cs_prev" type_attr: "T" }
input_arg { name: "h_prev" type_attr: "T" }
input_arg { name: "w" type_attr: "T" }
input_arg { name: "wci" type_attr: "T" }
input_arg { name: "wcf" type_attr: "T" }
input_arg { name: "wco" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
input_arg { name: "i" type_attr: "T" }
input_arg { name: "cs" type_attr: "T" }
input_arg { name: "f" type_attr: "T" }
input_arg { name: "o" type_attr: "T" }
input_arg { name: "ci" type_attr: "T" }
input_arg { name: "co" type_attr: "T" }
input_arg { name: "cs_grad" type_attr: "T" }
input_arg { name: "h_grad" type_attr: "T" }
output_arg { name: "cs_prev_grad" type_attr: "T" }
output_arg { name: "dicfo" type_attr: "T" }
output_arg { name: "wci_grad" type_attr: "T" }
output_arg { name: "wcf_grad" type_attr: "T" }
output_arg { name: "wco_grad" type_attr: "T" }
attr { name: "use_peephole" type: "bool" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
latencyStatsDataset :: [DataType] -- ^ __output_types__
                       -> Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                       -> Tensor Build Variant -- ^ __handle__
latencyStatsDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
latencyStatsDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
latencyStatsDataset' OpParams
forall a. a -> a
id
latencyStatsDataset' :: OpParams ->
                        [DataType] -- ^ __output_types__
                        -> Tensor v'1 Variant -- ^ __input_dataset__
                        -> Tensor v'2 Data.ByteString.ByteString -- ^ __tag__
                        -> Tensor Build Variant -- ^ __handle__
latencyStatsDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
latencyStatsDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                     tag :: Tensor v'2 ByteString
tag | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tag]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LatencyStatsDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "tag" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
leakyRelu :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __features__
             -> Tensor Build t -- ^ __activations__
leakyRelu :: Tensor v'1 t -> Tensor Build t
leakyRelu = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
leakyRelu' OpParams
forall a. a -> a
id
leakyRelu' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __features__
              -> Tensor Build t -- ^ __activations__
leakyRelu' :: OpParams -> Tensor v'1 t -> Tensor Build t
leakyRelu' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LeakyRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr { name: "alpha" type: "float" default_value { f: 0.2 } }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
leakyReluGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) =>
                 Tensor v'1 t -- ^ __gradients__
                 -> Tensor v'2 t -- ^ __features__
                 -> Tensor Build t -- ^ __backprops__
leakyReluGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leakyReluGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leakyReluGrad' OpParams
forall a. a -> a
id
leakyReluGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                              Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __gradients__
                  -> Tensor v'2 t -- ^ __features__
                  -> Tensor Build t -- ^ __backprops__
leakyReluGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leakyReluGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients features :: Tensor v'2 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LeakyReluGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "features" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr { name: "alpha" type: "float" default_value { f: 0.2 } }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
learnedUnigramCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                                  Data.Int.Int64 -- ^ __num_sampled__
                                  -> Data.Int.Int64 -- ^ __num_true__
                                  -> Data.Int.Int64 -- ^ __range_max__
                                  -> Bool -- ^ __unique__
                                  -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                  -> m' ((Tensor Value Data.Int.Int64,
                                          Tensor Value Float,
                                          Tensor Value Float))
                                  -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                  --
                                  -- * __sampled_candidates__
                                  --
                                  -- * __true_expected_count__
                                  --
                                  -- * __sampled_expected_count__
learnedUnigramCandidateSampler :: Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
learnedUnigramCandidateSampler = OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
learnedUnigramCandidateSampler' OpParams
forall a. a -> a
id
learnedUnigramCandidateSampler' :: forall v'1 m' . (MonadBuild m') =>
                                   OpParams ->
                                   Data.Int.Int64 -- ^ __num_sampled__
                                   -> Data.Int.Int64 -- ^ __num_true__
                                   -> Data.Int.Int64 -- ^ __range_max__
                                   -> Bool -- ^ __unique__
                                   -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                   -> m' ((Tensor Value Data.Int.Int64,
                                           Tensor Value Float,
                                           Tensor Value Float))
                                   -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                   --
                                   -- * __sampled_candidates__
                                   --
                                   -- * __true_expected_count__
                                   --
                                   -- * __sampled_expected_count__
learnedUnigramCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
learnedUnigramCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true range_max :: Int64
range_max unique :: Bool
unique
                                true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LearnedUnigramCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "range_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
range_max
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "range_max" type: "int" has_minimum: true minimum: 1 }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
leftShift :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64,
                                         Data.Word.Word8] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
leftShift :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leftShift = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leftShift' OpParams
forall a. a -> a
id
leftShift' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64,
                                          Data.Word.Word8] t) => OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build t -- ^ __z__
leftShift' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
leftShift' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LeftShift"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
less :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build Bool -- ^ __z__
less :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
less = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
less' OpParams
forall a. a -> a
id
less' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __y__
         -> Tensor Build Bool -- ^ __z__
less' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
less' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Less"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
lessEqual :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build Bool -- ^ __z__
lessEqual :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
lessEqual = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
lessEqual' OpParams
forall a. a -> a
id
lessEqual' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) => OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build Bool -- ^ __z__
lessEqual' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
lessEqual' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LessEqual"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
lgamma :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
lgamma :: Tensor v'1 t -> Tensor Build t
lgamma = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
lgamma' OpParams
forall a. a -> a
id
lgamma' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
lgamma' :: OpParams -> Tensor v'1 t -> Tensor Build t
lgamma' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Lgamma"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
linSpace :: forall v'1 v'2 v'3 t tidx . (OneOf '[Data.Word.Word16, Double,
                                                 Float] t,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tidx) =>
            Tensor v'1 t -- ^ __start__
            -> Tensor v'2 t -- ^ __stop__
            -> Tensor v'3 tidx -- ^ __num__
            -> Tensor Build t -- ^ __output__
linSpace :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tidx -> Tensor Build t
linSpace = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tidx
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tidx
-> Tensor Build t
linSpace' OpParams
forall a. a -> a
id
linSpace' :: forall v'1 v'2 v'3 t tidx . (OneOf '[Data.Word.Word16, Double,
                                                  Float] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tidx) =>
             OpParams ->
             Tensor v'1 t -- ^ __start__
             -> Tensor v'2 t -- ^ __stop__
             -> Tensor v'3 tidx -- ^ __num__
             -> Tensor Build t -- ^ __output__
linSpace' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tidx
-> Tensor Build t
linSpace' op'options :: OpParams
op'options start :: Tensor v'1 t
start stop :: Tensor v'2 t
stop num :: Tensor v'3 tidx
num | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
start,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
stop,
                                                             Tensor v'3 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tidx
num]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LinSpace"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "start" type_attr: "T" }
input_arg { name: "stop" type_attr: "T" }
input_arg { name: "num" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
listDiff :: forall v'1 v'2 t out_idx . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] out_idx) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> (Tensor Build t, Tensor Build out_idx) -- ^ (__out__, __idx__)
            --
            -- * __out__
            --
            -- * __idx__
listDiff :: Tensor v'1 t
-> Tensor v'2 t -> (Tensor Build t, Tensor Build out_idx)
listDiff = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build out_idx)
forall (v'1 :: * -> *) (v'2 :: * -> *) t out_idx.
(TensorType t, OneOf '[Int32, Int64] out_idx) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build out_idx)
listDiff' OpParams
forall a. a -> a
id
listDiff' :: forall v'1 v'2 t out_idx . (TensorType t, OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] out_idx) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> (Tensor Build t, Tensor Build out_idx) -- ^ (__out__, __idx__)
             --
             -- * __out__
             --
             -- * __idx__
listDiff' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build out_idx)
listDiff' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build out_idx))
-> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ListDiff"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_idx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_idx -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_idx
forall a. HasCallStack => a
undefined :: out_idx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "out" type_attr: "T" }
output_arg { name: "idx" type_attr: "out_idx" }
attr { name: "T" type: "type" }
attr {
  name: "out_idx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
loadAndRemapMatrix :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                      Data.Int.Int64 -- ^ __num_cols__
                      -> Data.Int.Int64 -- ^ __num_rows__
                      -> Tensor v'1 Data.ByteString.ByteString -- ^ __ckpt_path__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __old_tensor_name__
                      -> Tensor v'3 Data.Int.Int64 -- ^ __row_remapping__
                      -> Tensor v'4 Data.Int.Int64 -- ^ __col_remapping__
                      -> Tensor v'5 Float -- ^ __initializing_values__
                      -> m' (Tensor Value Float) -- ^ __output_matrix__
loadAndRemapMatrix :: Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Float
-> m' (Tensor Value Float)
loadAndRemapMatrix = OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Float
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Float
-> m' (Tensor Value Float)
loadAndRemapMatrix' OpParams
forall a. a -> a
id
loadAndRemapMatrix' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                       OpParams ->
                       Data.Int.Int64 -- ^ __num_cols__
                       -> Data.Int.Int64 -- ^ __num_rows__
                       -> Tensor v'1 Data.ByteString.ByteString -- ^ __ckpt_path__
                       -> Tensor v'2 Data.ByteString.ByteString -- ^ __old_tensor_name__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __row_remapping__
                       -> Tensor v'4 Data.Int.Int64 -- ^ __col_remapping__
                       -> Tensor v'5 Float -- ^ __initializing_values__
                       -> m' (Tensor Value Float) -- ^ __output_matrix__
loadAndRemapMatrix' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Float
-> m' (Tensor Value Float)
loadAndRemapMatrix' op'options :: OpParams
op'options num_cols :: Int64
num_cols num_rows :: Int64
num_rows ckpt_path :: Tensor v'1 ByteString
ckpt_path old_tensor_name :: Tensor v'2 ByteString
old_tensor_name
                    row_remapping :: Tensor v'3 Int64
row_remapping col_remapping :: Tensor v'4 Int64
col_remapping
                    initializing_values :: Tensor v'5 Float
initializing_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
ckpt_path,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
old_tensor_name,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
row_remapping,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
col_remapping,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
initializing_values]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadAndRemapMatrix"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_cols" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_cols
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_rows" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_rows
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ckpt_path" type: DT_STRING }
input_arg { name: "old_tensor_name" type: DT_STRING }
input_arg { name: "row_remapping" type: DT_INT64 }
input_arg { name: "col_remapping" type: DT_INT64 }
input_arg { name: "initializing_values" type: DT_FLOAT }
output_arg { name: "output_matrix" type: DT_FLOAT }
attr { name: "num_rows" type: "int" has_minimum: true }
attr { name: "num_cols" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "max_rows_in_memory" type: "int" default_value { i: -1 }
}
-}
-- | 
loadTPUEmbeddingADAMParameters :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                  Data.Int.Int64 -- ^ __num_shards__
                                  -> Data.Int.Int64 -- ^ __shard_id__
                                  -> Tensor v'1 Float -- ^ __parameters__
                                  -> Tensor v'2 Float -- ^ __momenta__
                                  -> Tensor v'3 Float -- ^ __velocities__
                                  -> m' (ControlNode)
loadTPUEmbeddingADAMParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingADAMParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingADAMParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingADAMParameters' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                   OpParams ->
                                   Data.Int.Int64 -- ^ __num_shards__
                                   -> Data.Int.Int64 -- ^ __shard_id__
                                   -> Tensor v'1 Float -- ^ __parameters__
                                   -> Tensor v'2 Float -- ^ __momenta__
                                   -> Tensor v'3 Float -- ^ __velocities__
                                   -> m' (ControlNode)
loadTPUEmbeddingADAMParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingADAMParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                momenta :: Tensor v'2 Float
momenta velocities :: Tensor v'3 Float
velocities | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
momenta,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
velocities]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingADAMParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "momenta" type: DT_FLOAT }
input_arg { name: "velocities" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingADAMParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4
                                                m' . (MonadBuild m') =>
                                                Data.Int.Int64 -- ^ __num_shards__
                                                -> Data.Int.Int64 -- ^ __shard_id__
                                                -> Tensor v'1 Float -- ^ __parameters__
                                                -> Tensor v'2 Float -- ^ __momenta__
                                                -> Tensor v'3 Float -- ^ __velocities__
                                                -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                -> m' (ControlNode)
loadTPUEmbeddingADAMParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingADAMParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingADAMParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingADAMParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4
                                                 m' . (MonadBuild m') =>
                                                 OpParams ->
                                                 Data.Int.Int64 -- ^ __num_shards__
                                                 -> Data.Int.Int64 -- ^ __shard_id__
                                                 -> Tensor v'1 Float -- ^ __parameters__
                                                 -> Tensor v'2 Float -- ^ __momenta__
                                                 -> Tensor v'3 Float -- ^ __velocities__
                                                 -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                 -> m' (ControlNode)
loadTPUEmbeddingADAMParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingADAMParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                              parameters :: Tensor v'1 Float
parameters momenta :: Tensor v'2 Float
momenta velocities :: Tensor v'3 Float
velocities
                                              gradient_accumulators :: Tensor v'4 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
momenta,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
velocities,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingADAMParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "momenta" type: DT_FLOAT }
input_arg { name: "velocities" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingAdadeltaParameters :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> Tensor v'1 Float -- ^ __parameters__
                                      -> Tensor v'2 Float -- ^ __accumulators__
                                      -> Tensor v'3 Float -- ^ __updates__
                                      -> m' (ControlNode)
loadTPUEmbeddingAdadeltaParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingAdadeltaParameters' :: forall v'1 v'2 v'3
                                       m' . (MonadBuild m') => OpParams ->
                                       Data.Int.Int64 -- ^ __num_shards__
                                       -> Data.Int.Int64 -- ^ __shard_id__
                                       -> Tensor v'1 Float -- ^ __parameters__
                                       -> Tensor v'2 Float -- ^ __accumulators__
                                       -> Tensor v'3 Float -- ^ __updates__
                                       -> m' (ControlNode)
loadTPUEmbeddingAdadeltaParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                    accumulators :: Tensor v'2 Float
accumulators updates :: Tensor v'3 Float
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingAdadeltaParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "updates" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4
                                                    m' . (MonadBuild m') =>
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> Tensor v'1 Float -- ^ __parameters__
                                                    -> Tensor v'2 Float -- ^ __accumulators__
                                                    -> Tensor v'3 Float -- ^ __updates__
                                                    -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                    -> m' (ControlNode)
loadTPUEmbeddingAdadeltaParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4
                                                     m' . (MonadBuild m') =>
                                                     OpParams ->
                                                     Data.Int.Int64 -- ^ __num_shards__
                                                     -> Data.Int.Int64 -- ^ __shard_id__
                                                     -> Tensor v'1 Float -- ^ __parameters__
                                                     -> Tensor v'2 Float -- ^ __accumulators__
                                                     -> Tensor v'3 Float -- ^ __updates__
                                                     -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                     -> m' (ControlNode)
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                                  parameters :: Tensor v'1 Float
parameters accumulators :: Tensor v'2 Float
accumulators
                                                  updates :: Tensor v'3 Float
updates
                                                  gradient_accumulators :: Tensor v'4 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
updates,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingAdadeltaParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "updates" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingAdagradParameters :: forall v'1 v'2 m' . (MonadBuild m') =>
                                     Data.Int.Int64 -- ^ __num_shards__
                                     -> Data.Int.Int64 -- ^ __shard_id__
                                     -> Tensor v'1 Float -- ^ __parameters__
                                     -> Tensor v'2 Float -- ^ __accumulators__
                                     -> m' (ControlNode)
loadTPUEmbeddingAdagradParameters :: Int64
-> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingAdagradParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingAdagradParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingAdagradParameters' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                      OpParams ->
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> Tensor v'1 Float -- ^ __parameters__
                                      -> Tensor v'2 Float -- ^ __accumulators__
                                      -> m' (ControlNode)
loadTPUEmbeddingAdagradParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingAdagradParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                   accumulators :: Tensor v'2 Float
accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingAdagradParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3
                                                   m' . (MonadBuild m') =>
                                                   Data.Int.Int64 -- ^ __num_shards__
                                                   -> Data.Int.Int64 -- ^ __shard_id__
                                                   -> Tensor v'1 Float -- ^ __parameters__
                                                   -> Tensor v'2 Float -- ^ __accumulators__
                                                   -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                   -> m' (ControlNode)
loadTPUEmbeddingAdagradParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdagradParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdagradParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingAdagradParametersGradAccumDebug' :: forall v'1 v'2 v'3
                                                    m' . (MonadBuild m') =>
                                                    OpParams ->
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> Tensor v'1 Float -- ^ __parameters__
                                                    -> Tensor v'2 Float -- ^ __accumulators__
                                                    -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                    -> m' (ControlNode)
loadTPUEmbeddingAdagradParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingAdagradParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                                 parameters :: Tensor v'1 Float
parameters accumulators :: Tensor v'2 Float
accumulators
                                                 gradient_accumulators :: Tensor v'3 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingAdagradParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingCenteredRMSPropParameters :: forall v'1 v'2 v'3 v'4
                                             m' . (MonadBuild m') =>
                                             Data.Int.Int64 -- ^ __num_shards__
                                             -> Data.Int.Int64 -- ^ __shard_id__
                                             -> Tensor v'1 Float -- ^ __parameters__
                                             -> Tensor v'2 Float -- ^ __ms__
                                             -> Tensor v'3 Float -- ^ __mom__
                                             -> Tensor v'4 Float -- ^ __mg__
                                             -> m' (ControlNode)
loadTPUEmbeddingCenteredRMSPropParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingCenteredRMSPropParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingCenteredRMSPropParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingCenteredRMSPropParameters' :: forall v'1 v'2 v'3 v'4
                                              m' . (MonadBuild m') =>
                                              OpParams ->
                                              Data.Int.Int64 -- ^ __num_shards__
                                              -> Data.Int.Int64 -- ^ __shard_id__
                                              -> Tensor v'1 Float -- ^ __parameters__
                                              -> Tensor v'2 Float -- ^ __ms__
                                              -> Tensor v'3 Float -- ^ __mom__
                                              -> Tensor v'4 Float -- ^ __mg__
                                              -> m' (ControlNode)
loadTPUEmbeddingCenteredRMSPropParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingCenteredRMSPropParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                           parameters :: Tensor v'1 Float
parameters ms :: Tensor v'2 Float
ms mom :: Tensor v'3 Float
mom
                                           mg :: Tensor v'4 Float
mg | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
ms,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
mom,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
mg]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingCenteredRMSPropParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "ms" type: DT_FLOAT }
input_arg { name: "mom" type: DT_FLOAT }
input_arg { name: "mg" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingFTRLParameters :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                  Data.Int.Int64 -- ^ __num_shards__
                                  -> Data.Int.Int64 -- ^ __shard_id__
                                  -> Tensor v'1 Float -- ^ __parameters__
                                  -> Tensor v'2 Float -- ^ __accumulators__
                                  -> Tensor v'3 Float -- ^ __linears__
                                  -> m' (ControlNode)
loadTPUEmbeddingFTRLParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingFTRLParameters' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                   OpParams ->
                                   Data.Int.Int64 -- ^ __num_shards__
                                   -> Data.Int.Int64 -- ^ __shard_id__
                                   -> Tensor v'1 Float -- ^ __parameters__
                                   -> Tensor v'2 Float -- ^ __accumulators__
                                   -> Tensor v'3 Float -- ^ __linears__
                                   -> m' (ControlNode)
loadTPUEmbeddingFTRLParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                accumulators :: Tensor v'2 Float
accumulators linears :: Tensor v'3 Float
linears | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
linears]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingFTRLParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "linears" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingFTRLParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4
                                                m' . (MonadBuild m') =>
                                                Data.Int.Int64 -- ^ __num_shards__
                                                -> Data.Int.Int64 -- ^ __shard_id__
                                                -> Tensor v'1 Float -- ^ __parameters__
                                                -> Tensor v'2 Float -- ^ __accumulators__
                                                -> Tensor v'3 Float -- ^ __linears__
                                                -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                -> m' (ControlNode)
loadTPUEmbeddingFTRLParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingFTRLParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4
                                                 m' . (MonadBuild m') =>
                                                 OpParams ->
                                                 Data.Int.Int64 -- ^ __num_shards__
                                                 -> Data.Int.Int64 -- ^ __shard_id__
                                                 -> Tensor v'1 Float -- ^ __parameters__
                                                 -> Tensor v'2 Float -- ^ __accumulators__
                                                 -> Tensor v'3 Float -- ^ __linears__
                                                 -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                 -> m' (ControlNode)
loadTPUEmbeddingFTRLParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingFTRLParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                              parameters :: Tensor v'1 Float
parameters accumulators :: Tensor v'2 Float
accumulators linears :: Tensor v'3 Float
linears
                                              gradient_accumulators :: Tensor v'4 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
linears,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingFTRLParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "linears" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingMDLAdagradLightParameters :: forall v'1 v'2 v'3 v'4
                                             m' . (MonadBuild m') =>
                                             Data.Int.Int64 -- ^ __num_shards__
                                             -> Data.Int.Int64 -- ^ __shard_id__
                                             -> Tensor v'1 Float -- ^ __parameters__
                                             -> Tensor v'2 Float -- ^ __accumulators__
                                             -> Tensor v'3 Float -- ^ __weights__
                                             -> Tensor v'4 Float -- ^ __benefits__
                                             -> m' (ControlNode)
loadTPUEmbeddingMDLAdagradLightParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingMDLAdagradLightParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingMDLAdagradLightParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingMDLAdagradLightParameters' :: forall v'1 v'2 v'3 v'4
                                              m' . (MonadBuild m') =>
                                              OpParams ->
                                              Data.Int.Int64 -- ^ __num_shards__
                                              -> Data.Int.Int64 -- ^ __shard_id__
                                              -> Tensor v'1 Float -- ^ __parameters__
                                              -> Tensor v'2 Float -- ^ __accumulators__
                                              -> Tensor v'3 Float -- ^ __weights__
                                              -> Tensor v'4 Float -- ^ __benefits__
                                              -> m' (ControlNode)
loadTPUEmbeddingMDLAdagradLightParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingMDLAdagradLightParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                           parameters :: Tensor v'1 Float
parameters accumulators :: Tensor v'2 Float
accumulators weights :: Tensor v'3 Float
weights
                                           benefits :: Tensor v'4 Float
benefits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
weights,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
benefits]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingMDLAdagradLightParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "weights" type: DT_FLOAT }
input_arg { name: "benefits" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingMomentumParameters :: forall v'1 v'2 m' . (MonadBuild m') =>
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> Tensor v'1 Float -- ^ __parameters__
                                      -> Tensor v'2 Float -- ^ __momenta__
                                      -> m' (ControlNode)
loadTPUEmbeddingMomentumParameters :: Int64
-> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingMomentumParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingMomentumParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingMomentumParameters' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                       OpParams ->
                                       Data.Int.Int64 -- ^ __num_shards__
                                       -> Data.Int.Int64 -- ^ __shard_id__
                                       -> Tensor v'1 Float -- ^ __parameters__
                                       -> Tensor v'2 Float -- ^ __momenta__
                                       -> m' (ControlNode)
loadTPUEmbeddingMomentumParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingMomentumParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                    momenta :: Tensor v'2 Float
momenta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
momenta]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingMomentumParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "momenta" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingMomentumParametersGradAccumDebug :: forall v'1 v'2 v'3
                                                    m' . (MonadBuild m') =>
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> Tensor v'1 Float -- ^ __parameters__
                                                    -> Tensor v'2 Float -- ^ __momenta__
                                                    -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                    -> m' (ControlNode)
loadTPUEmbeddingMomentumParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingMomentumParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingMomentumParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingMomentumParametersGradAccumDebug' :: forall v'1 v'2 v'3
                                                     m' . (MonadBuild m') =>
                                                     OpParams ->
                                                     Data.Int.Int64 -- ^ __num_shards__
                                                     -> Data.Int.Int64 -- ^ __shard_id__
                                                     -> Tensor v'1 Float -- ^ __parameters__
                                                     -> Tensor v'2 Float -- ^ __momenta__
                                                     -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                     -> m' (ControlNode)
loadTPUEmbeddingMomentumParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingMomentumParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                                  parameters :: Tensor v'1 Float
parameters momenta :: Tensor v'2 Float
momenta
                                                  gradient_accumulators :: Tensor v'3 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
momenta,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingMomentumParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "momenta" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingProximalAdagradParameters :: forall v'1 v'2
                                             m' . (MonadBuild m') =>
                                             Data.Int.Int64 -- ^ __num_shards__
                                             -> Data.Int.Int64 -- ^ __shard_id__
                                             -> Tensor v'1 Float -- ^ __parameters__
                                             -> Tensor v'2 Float -- ^ __accumulators__
                                             -> m' (ControlNode)
loadTPUEmbeddingProximalAdagradParameters :: Int64
-> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingProximalAdagradParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingProximalAdagradParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingProximalAdagradParameters' :: forall v'1 v'2
                                              m' . (MonadBuild m') =>
                                              OpParams ->
                                              Data.Int.Int64 -- ^ __num_shards__
                                              -> Data.Int.Int64 -- ^ __shard_id__
                                              -> Tensor v'1 Float -- ^ __parameters__
                                              -> Tensor v'2 Float -- ^ __accumulators__
                                              -> m' (ControlNode)
loadTPUEmbeddingProximalAdagradParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingProximalAdagradParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                           parameters :: Tensor v'1 Float
parameters
                                           accumulators :: Tensor v'2 Float
accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingProximalAdagradParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3
                                                           m' . (MonadBuild m') =>
                                                           Data.Int.Int64 -- ^ __num_shards__
                                                           -> Data.Int.Int64 -- ^ __shard_id__
                                                           -> Tensor v'1 Float -- ^ __parameters__
                                                           -> Tensor v'2 Float -- ^ __accumulators__
                                                           -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                           -> m' (ControlNode)
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: forall v'1 v'2 v'3
                                                            m' . (MonadBuild m') =>
                                                            OpParams ->
                                                            Data.Int.Int64 -- ^ __num_shards__
                                                            -> Data.Int.Int64 -- ^ __shard_id__
                                                            -> Tensor v'1 Float -- ^ __parameters__
                                                            -> Tensor v'2 Float -- ^ __accumulators__
                                                            -> Tensor v'3 Float -- ^ __gradient_accumulators__
                                                            -> m' (ControlNode)
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                         shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters
                                                         accumulators :: Tensor v'2 Float
accumulators
                                                         gradient_accumulators :: Tensor v'3 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
accumulators,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingProximalAdagradParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "accumulators" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingProximalYogiParameters :: forall v'1 v'2 v'3
                                          m' . (MonadBuild m') =>
                                          Data.Int.Int64 -- ^ __num_shards__
                                          -> Data.Int.Int64 -- ^ __shard_id__
                                          -> Tensor v'1 Float -- ^ __parameters__
                                          -> Tensor v'2 Float -- ^ __v__
                                          -> Tensor v'3 Float -- ^ __m__
                                          -> m' (ControlNode)
loadTPUEmbeddingProximalYogiParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingProximalYogiParameters' :: forall v'1 v'2 v'3
                                           m' . (MonadBuild m') => OpParams ->
                                           Data.Int.Int64 -- ^ __num_shards__
                                           -> Data.Int.Int64 -- ^ __shard_id__
                                           -> Tensor v'1 Float -- ^ __parameters__
                                           -> Tensor v'2 Float -- ^ __v__
                                           -> Tensor v'3 Float -- ^ __m__
                                           -> m' (ControlNode)
loadTPUEmbeddingProximalYogiParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                        parameters :: Tensor v'1 Float
parameters v :: Tensor v'2 Float
v m :: Tensor v'3 Float
m | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
v,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
m]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingProximalYogiParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "v" type: DT_FLOAT }
input_arg { name: "m" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4
                                                        m' . (MonadBuild m') =>
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> Tensor v'1 Float -- ^ __parameters__
                                                        -> Tensor v'2 Float -- ^ __v__
                                                        -> Tensor v'3 Float -- ^ __m__
                                                        -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                        -> m' (ControlNode)
loadTPUEmbeddingProximalYogiParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4
                                                         m' . (MonadBuild m') =>
                                                         OpParams ->
                                                         Data.Int.Int64 -- ^ __num_shards__
                                                         -> Data.Int.Int64 -- ^ __shard_id__
                                                         -> Tensor v'1 Float -- ^ __parameters__
                                                         -> Tensor v'2 Float -- ^ __v__
                                                         -> Tensor v'3 Float -- ^ __m__
                                                         -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                         -> m' (ControlNode)
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                      shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters v :: Tensor v'2 Float
v m :: Tensor v'3 Float
m
                                                      gradient_accumulators :: Tensor v'4 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
v,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
m,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingProximalYogiParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "v" type: DT_FLOAT }
input_arg { name: "m" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingRMSPropParameters :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                     Data.Int.Int64 -- ^ __num_shards__
                                     -> Data.Int.Int64 -- ^ __shard_id__
                                     -> Tensor v'1 Float -- ^ __parameters__
                                     -> Tensor v'2 Float -- ^ __ms__
                                     -> Tensor v'3 Float -- ^ __mom__
                                     -> m' (ControlNode)
loadTPUEmbeddingRMSPropParameters :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParameters = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingRMSPropParameters' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                                      OpParams ->
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> Tensor v'1 Float -- ^ __parameters__
                                      -> Tensor v'2 Float -- ^ __ms__
                                      -> Tensor v'3 Float -- ^ __mom__
                                      -> m' (ControlNode)
loadTPUEmbeddingRMSPropParameters' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id parameters :: Tensor v'1 Float
parameters ms :: Tensor v'2 Float
ms
                                   mom :: Tensor v'3 Float
mom | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
ms,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
mom]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingRMSPropParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "ms" type: DT_FLOAT }
input_arg { name: "mom" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingRMSPropParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4
                                                   m' . (MonadBuild m') =>
                                                   Data.Int.Int64 -- ^ __num_shards__
                                                   -> Data.Int.Int64 -- ^ __shard_id__
                                                   -> Tensor v'1 Float -- ^ __parameters__
                                                   -> Tensor v'2 Float -- ^ __ms__
                                                   -> Tensor v'3 Float -- ^ __mom__
                                                   -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                   -> m' (ControlNode)
loadTPUEmbeddingRMSPropParametersGradAccumDebug :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingRMSPropParametersGradAccumDebug' :: forall v'1 v'2 v'3 v'4
                                                    m' . (MonadBuild m') =>
                                                    OpParams ->
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> Tensor v'1 Float -- ^ __parameters__
                                                    -> Tensor v'2 Float -- ^ __ms__
                                                    -> Tensor v'3 Float -- ^ __mom__
                                                    -> Tensor v'4 Float -- ^ __gradient_accumulators__
                                                    -> m' (ControlNode)
loadTPUEmbeddingRMSPropParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> m' ControlNode
loadTPUEmbeddingRMSPropParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards shard_id :: Int64
shard_id
                                                 parameters :: Tensor v'1 Float
parameters ms :: Tensor v'2 Float
ms mom :: Tensor v'3 Float
mom
                                                 gradient_accumulators :: Tensor v'4 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
ms,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
mom,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingRMSPropParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "ms" type: DT_FLOAT }
input_arg { name: "mom" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingStochasticGradientDescentParameters :: forall v'1
                                                       m' . (MonadBuild m') =>
                                                       Data.Int.Int64 -- ^ __num_shards__
                                                       -> Data.Int.Int64 -- ^ __shard_id__
                                                       -> Tensor v'1 Float -- ^ __parameters__
                                                       -> m' (ControlNode)
loadTPUEmbeddingStochasticGradientDescentParameters :: Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParameters = OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParameters' OpParams
forall a. a -> a
id
loadTPUEmbeddingStochasticGradientDescentParameters' :: forall v'1
                                                        m' . (MonadBuild m') =>
                                                        OpParams ->
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> Tensor v'1 Float -- ^ __parameters__
                                                        -> m' (ControlNode)
loadTPUEmbeddingStochasticGradientDescentParameters' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                     shard_id :: Int64
shard_id
                                                     parameters :: Tensor v'1 Float
parameters | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingStochasticGradientDescentParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall v'1
                                                                     v'2
                                                                     m' . (MonadBuild m') =>
                                                                     Data.Int.Int64 -- ^ __num_shards__
                                                                     -> Data.Int.Int64 -- ^ __shard_id__
                                                                     -> Tensor v'1 Float -- ^ __parameters__
                                                                     -> Tensor v'2 Float -- ^ __gradient_accumulators__
                                                                     -> m' (ControlNode)
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: Int64
-> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' OpParams
forall a. a -> a
id
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall v'1
                                                                      v'2
                                                                      m' . (MonadBuild m') =>
                                                                      OpParams ->
                                                                      Data.Int.Int64 -- ^ __num_shards__
                                                                      -> Data.Int.Int64 -- ^ __shard_id__
                                                                      -> Tensor v'1 Float -- ^ __parameters__
                                                                      -> Tensor v'2 Float -- ^ __gradient_accumulators__
                                                                      -> m' (ControlNode)
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> m' ControlNode
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' op'options :: OpParams
op'options
                                                                   num_shards :: Int64
num_shards
                                                                   shard_id :: Int64
shard_id
                                                                   parameters :: Tensor v'1 Float
parameters
                                                                   gradient_accumulators :: Tensor v'2 Float
gradient_accumulators | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
parameters,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
gradient_accumulators]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LoadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "parameters" type: DT_FLOAT }
input_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
log :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
log :: Tensor v'1 t -> Tensor Build t
log = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
log' OpParams
forall a. a -> a
id
log' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
log' :: OpParams -> Tensor v'1 t -> Tensor Build t
log' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Log"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
log1p :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
log1p :: Tensor v'1 t -> Tensor Build t
log1p = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
log1p' OpParams
forall a. a -> a
id
log1p' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
log1p' :: OpParams -> Tensor v'1 t -> Tensor Build t
log1p' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Log1p"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
logMatrixDeterminant :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Word.Word16, Double,
                                                Float] t) =>
                        Tensor v'1 t -- ^ __input__
                        -> (Tensor Build t, Tensor Build t)
                        -- ^ (__sign__, __log_abs_determinant__)
                        --
                        -- * __sign__
                        --
                        -- * __log_abs_determinant__
logMatrixDeterminant :: Tensor v'1 t -> (Tensor Build t, Tensor Build t)
logMatrixDeterminant = OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
logMatrixDeterminant' OpParams
forall a. a -> a
id
logMatrixDeterminant' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Word.Word16, Double,
                                                 Float] t) => OpParams ->
                         Tensor v'1 t -- ^ __input__
                         -> (Tensor Build t, Tensor Build t)
                         -- ^ (__sign__, __log_abs_determinant__)
                         --
                         -- * __sign__
                         --
                         -- * __log_abs_determinant__
logMatrixDeterminant' :: OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
logMatrixDeterminant' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LogMatrixDeterminant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "sign" type_attr: "T" }
output_arg { name: "log_abs_determinant" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
logSoftmax :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
              Tensor v'1 t -- ^ __logits__
              -> Tensor Build t -- ^ __logsoftmax__
logSoftmax :: Tensor v'1 t -> Tensor Build t
logSoftmax = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
logSoftmax' OpParams
forall a. a -> a
id
logSoftmax' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __logits__
               -> Tensor Build t -- ^ __logsoftmax__
logSoftmax' :: OpParams -> Tensor v'1 t -> Tensor Build t
logSoftmax' op'options :: OpParams
op'options logits :: Tensor v'1 t
logits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
logits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LogSoftmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "logits" type_attr: "T" }
output_arg { name: "logsoftmax" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
logUniformCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                              Data.Int.Int64 -- ^ __num_sampled__
                              -> Data.Int.Int64 -- ^ __num_true__
                              -> Data.Int.Int64 -- ^ __range_max__
                              -> Bool -- ^ __unique__
                              -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                              -> m' ((Tensor Value Data.Int.Int64,
                                      Tensor Value Float, Tensor Value Float))
                              -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                              --
                              -- * __sampled_candidates__
                              --
                              -- * __true_expected_count__
                              --
                              -- * __sampled_expected_count__
logUniformCandidateSampler :: Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
logUniformCandidateSampler = OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
logUniformCandidateSampler' OpParams
forall a. a -> a
id
logUniformCandidateSampler' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                               Data.Int.Int64 -- ^ __num_sampled__
                               -> Data.Int.Int64 -- ^ __num_true__
                               -> Data.Int.Int64 -- ^ __range_max__
                               -> Bool -- ^ __unique__
                               -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                               -> m' ((Tensor Value Data.Int.Int64,
                                       Tensor Value Float, Tensor Value Float))
                               -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                               --
                               -- * __sampled_candidates__
                               --
                               -- * __true_expected_count__
                               --
                               -- * __sampled_expected_count__
logUniformCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
logUniformCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true range_max :: Int64
range_max unique :: Bool
unique
                            true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LogUniformCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "range_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
range_max
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "range_max" type: "int" has_minimum: true minimum: 1 }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
logicalAnd :: Tensor v'1 Bool -- ^ __x__
              -> Tensor v'2 Bool -- ^ __y__
              -> Tensor Build Bool -- ^ __z__
logicalAnd :: Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalAnd = OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalAnd' OpParams
forall a. a -> a
id
logicalAnd' :: OpParams ->
               Tensor v'1 Bool -- ^ __x__
               -> Tensor v'2 Bool -- ^ __y__
               -> Tensor Build Bool -- ^ __z__
logicalAnd' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalAnd' op'options :: OpParams
op'options x :: Tensor v'1 Bool
x y :: Tensor v'2 Bool
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
x,
                                                             Tensor v'2 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Bool
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LogicalAnd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type: DT_BOOL }
input_arg { name: "y" type: DT_BOOL }
output_arg { name: "z" type: DT_BOOL }
-}
-- | 
logicalNot :: Tensor v'1 Bool -- ^ __x__
              -> Tensor Build Bool -- ^ __y__
logicalNot :: Tensor v'1 Bool -> Tensor Build Bool
logicalNot = OpParams -> Tensor v'1 Bool -> Tensor Build Bool
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Bool -> Tensor Build Bool
logicalNot' OpParams
forall a. a -> a
id
logicalNot' :: OpParams ->
               Tensor v'1 Bool -- ^ __x__
               -> Tensor Build Bool -- ^ __y__
logicalNot' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
logicalNot' op'options :: OpParams
op'options x :: Tensor v'1 Bool
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LogicalNot"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type: DT_BOOL }
output_arg { name: "y" type: DT_BOOL }
-}
-- | 
logicalOr :: Tensor v'1 Bool -- ^ __x__
             -> Tensor v'2 Bool -- ^ __y__
             -> Tensor Build Bool -- ^ __z__
logicalOr :: Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalOr = OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalOr' OpParams
forall a. a -> a
id
logicalOr' :: OpParams ->
              Tensor v'1 Bool -- ^ __x__
              -> Tensor v'2 Bool -- ^ __y__
              -> Tensor Build Bool -- ^ __z__
logicalOr' :: OpParams -> Tensor v'1 Bool -> Tensor v'2 Bool -> Tensor Build Bool
logicalOr' op'options :: OpParams
op'options x :: Tensor v'1 Bool
x y :: Tensor v'2 Bool
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
x,
                                                             Tensor v'2 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Bool
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LogicalOr"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type: DT_BOOL }
input_arg { name: "y" type: DT_BOOL }
output_arg { name: "z" type: DT_BOOL }
-}
-- | 
lookupTableExport :: forall tkeys tvalues m' . (MonadBuild m', TensorType tkeys,
                                                TensorType tvalues) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                     -> m' ((Tensor Value tkeys, Tensor Value tvalues))
                     -- ^ (__keys__, __values__)
                     --
                     -- * __keys__
                     --
                     -- * __values__
lookupTableExport :: Tensor Ref ByteString
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExport = OpParams
-> Tensor Ref ByteString
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall tkeys tvalues (m' :: * -> *).
(MonadBuild m', TensorType tkeys, TensorType tvalues) =>
OpParams
-> Tensor Ref ByteString
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExport' OpParams
forall a. a -> a
id
lookupTableExport' :: forall tkeys tvalues m' . (MonadBuild m',
                                                 TensorType tkeys,
                                                 TensorType tvalues) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                      -> m' ((Tensor Value tkeys, Tensor Value tvalues))
                      -- ^ (__keys__, __values__)
                      --
                      -- * __keys__
                      --
                      -- * __values__
lookupTableExport' :: OpParams
-> Tensor Ref ByteString
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExport' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tkeys, Tensor Value tvalues)
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tkeys, Tensor Value tvalues)
 -> m' (Tensor Value tkeys, Tensor Value tvalues))
-> Build (Tensor Value tkeys, Tensor Value tvalues)
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle]
        [Int64]
-> OpDef -> Build (Tensor Value tkeys, Tensor Value tvalues)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableExport"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tkeys" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tkeys -> DataType
forall a. TensorType a => a -> DataType
tensorType (tkeys
forall a. HasCallStack => a
undefined :: tkeys)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
output_arg { name: "keys" type_attr: "Tkeys" }
output_arg { name: "values" type_attr: "Tvalues" }
attr { name: "Tkeys" type: "type" }
attr { name: "Tvalues" type: "type" }
-}
-- | 
lookupTableExportV2 :: forall v'1 tkeys tvalues m' . (MonadBuild m',
                                                      TensorType tkeys,
                                                      TensorType tvalues) =>
                       Tensor v'1 ResourceHandle -- ^ __table_handle__
                       -> m' ((Tensor Value tkeys, Tensor Value tvalues))
                       -- ^ (__keys__, __values__)
                       --
                       -- * __keys__
                       --
                       -- * __values__
lookupTableExportV2 :: Tensor v'1 ResourceHandle
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExportV2 = OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall (v'1 :: * -> *) tkeys tvalues (m' :: * -> *).
(MonadBuild m', TensorType tkeys, TensorType tvalues) =>
OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExportV2' OpParams
forall a. a -> a
id
lookupTableExportV2' :: forall v'1 tkeys tvalues m' . (MonadBuild m',
                                                       TensorType tkeys,
                                                       TensorType tvalues) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __table_handle__
                        -> m' ((Tensor Value tkeys, Tensor Value tvalues))
                        -- ^ (__keys__, __values__)
                        --
                        -- * __keys__
                        --
                        -- * __values__
lookupTableExportV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> m' (Tensor Value tkeys, Tensor Value tvalues)
lookupTableExportV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tkeys, Tensor Value tvalues)
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tkeys, Tensor Value tvalues)
 -> m' (Tensor Value tkeys, Tensor Value tvalues))
-> Build (Tensor Value tkeys, Tensor Value tvalues)
-> m' (Tensor Value tkeys, Tensor Value tvalues)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle]
        [Int64]
-> OpDef -> Build (Tensor Value tkeys, Tensor Value tvalues)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableExportV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tkeys" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tkeys -> DataType
forall a. TensorType a => a -> DataType
tensorType (tkeys
forall a. HasCallStack => a
undefined :: tkeys)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
output_arg { name: "keys" type_attr: "Tkeys" }
output_arg { name: "values" type_attr: "Tvalues" }
attr { name: "Tkeys" type: "type" }
attr { name: "Tvalues" type: "type" }
-}
-- | 
lookupTableFind :: forall v'2 v'3 tin tout m' . (MonadBuild m', TensorType tin,
                                                 TensorType tout) =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                   -> Tensor v'2 tin -- ^ __keys__
                   -> Tensor v'3 tout -- ^ __default_value__
                   -> m' (Tensor Value tout) -- ^ __values__
lookupTableFind :: Tensor Ref ByteString
-> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
lookupTableFind = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
forall (v'2 :: * -> *) (v'3 :: * -> *) tin tout (m' :: * -> *).
(MonadBuild m', TensorType tin, TensorType tout) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
lookupTableFind' OpParams
forall a. a -> a
id
lookupTableFind' :: forall v'2 v'3 tin tout m' . (MonadBuild m', TensorType tin,
                                                  TensorType tout) =>
                    OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                    -> Tensor v'2 tin -- ^ __keys__
                    -> Tensor v'3 tout -- ^ __default_value__
                    -> m' (Tensor Value tout) -- ^ __values__
lookupTableFind' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
lookupTableFind' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle keys :: Tensor v'2 tin
keys default_value :: Tensor v'3 tout
default_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tout) -> m' (Tensor Value tout)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tout) -> m' (Tensor Value tout))
-> Build (Tensor Value tout) -> m' (Tensor Value tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
default_value]
        [Int64] -> OpDef -> Build (Tensor Value tout)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableFind"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "default_value" type_attr: "Tout" }
output_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableFindV2 :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                       TensorType tin,
                                                       TensorType tout) =>
                     Tensor v'1 ResourceHandle -- ^ __table_handle__
                     -> Tensor v'2 tin -- ^ __keys__
                     -> Tensor v'3 tout -- ^ __default_value__
                     -> m' (Tensor Value tout) -- ^ __values__
lookupTableFindV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
lookupTableFindV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tin tout
       (m' :: * -> *).
(MonadBuild m', TensorType tin, TensorType tout) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
lookupTableFindV2' OpParams
forall a. a -> a
id
lookupTableFindV2' :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                        TensorType tin,
                                                        TensorType tout) =>
                      OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __table_handle__
                      -> Tensor v'2 tin -- ^ __keys__
                      -> Tensor v'3 tout -- ^ __default_value__
                      -> m' (Tensor Value tout) -- ^ __values__
lookupTableFindV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' (Tensor Value tout)
lookupTableFindV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle keys :: Tensor v'2 tin
keys
                   default_value :: Tensor v'3 tout
default_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tout) -> m' (Tensor Value tout)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tout) -> m' (Tensor Value tout))
-> Build (Tensor Value tout) -> m' (Tensor Value tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
default_value]
        [Int64] -> OpDef -> Build (Tensor Value tout)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableFindV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "default_value" type_attr: "Tout" }
output_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableImport :: forall v'2 v'3 tin tout m' . (MonadBuild m',
                                                   TensorType tin,
                                                   TensorType tout) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                     -> Tensor v'2 tin -- ^ __keys__
                     -> Tensor v'3 tout -- ^ __values__
                     -> m' (ControlNode)
lookupTableImport :: Tensor Ref ByteString
-> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
lookupTableImport = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) tkey tval (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
lookupTableImport' OpParams
forall a. a -> a
id
lookupTableImport' :: forall v'2 v'3 tin tout m' . (MonadBuild m',
                                                    TensorType tin,
                                                    TensorType tout) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                      -> Tensor v'2 tin -- ^ __keys__
                      -> Tensor v'3 tout -- ^ __values__
                      -> m' (ControlNode)
lookupTableImport' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
lookupTableImport' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle keys :: Tensor v'2 tin
keys values :: Tensor v'3 tout
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableImport"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableImportV2 :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                         TensorType tin,
                                                         TensorType tout) =>
                       Tensor v'1 ResourceHandle -- ^ __table_handle__
                       -> Tensor v'2 tin -- ^ __keys__
                       -> Tensor v'3 tout -- ^ __values__
                       -> m' (ControlNode)
lookupTableImportV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
lookupTableImportV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tkey tval
       (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
lookupTableImportV2' OpParams
forall a. a -> a
id
lookupTableImportV2' :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                          TensorType tin,
                                                          TensorType tout) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __table_handle__
                        -> Tensor v'2 tin -- ^ __keys__
                        -> Tensor v'3 tout -- ^ __values__
                        -> m' (ControlNode)
lookupTableImportV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
lookupTableImportV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle keys :: Tensor v'2 tin
keys values :: Tensor v'3 tout
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableImportV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableInsert :: forall v'2 v'3 tin tout m' . (MonadBuild m',
                                                   TensorType tin,
                                                   TensorType tout) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                     -> Tensor v'2 tin -- ^ __keys__
                     -> Tensor v'3 tout -- ^ __values__
                     -> m' (ControlNode)
lookupTableInsert :: Tensor Ref ByteString
-> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
lookupTableInsert = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) tkey tval (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
lookupTableInsert' OpParams
forall a. a -> a
id
lookupTableInsert' :: forall v'2 v'3 tin tout m' . (MonadBuild m',
                                                    TensorType tin,
                                                    TensorType tout) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                      -> Tensor v'2 tin -- ^ __keys__
                      -> Tensor v'3 tout -- ^ __values__
                      -> m' (ControlNode)
lookupTableInsert' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
lookupTableInsert' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle keys :: Tensor v'2 tin
keys values :: Tensor v'3 tout
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableInsert"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableInsertV2 :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                         TensorType tin,
                                                         TensorType tout) =>
                       Tensor v'1 ResourceHandle -- ^ __table_handle__
                       -> Tensor v'2 tin -- ^ __keys__
                       -> Tensor v'3 tout -- ^ __values__
                       -> m' (ControlNode)
lookupTableInsertV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
lookupTableInsertV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tkey tval
       (m' :: * -> *).
(MonadBuild m', TensorType tkey, TensorType tval) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tkey
-> Tensor v'3 tval
-> m' ControlNode
lookupTableInsertV2' OpParams
forall a. a -> a
id
lookupTableInsertV2' :: forall v'1 v'2 v'3 tin tout m' . (MonadBuild m',
                                                          TensorType tin,
                                                          TensorType tout) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __table_handle__
                        -> Tensor v'2 tin -- ^ __keys__
                        -> Tensor v'3 tout -- ^ __values__
                        -> m' (ControlNode)
lookupTableInsertV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tin
-> Tensor v'3 tout
-> m' ControlNode
lookupTableInsertV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle keys :: Tensor v'2 tin
keys values :: Tensor v'3 tout
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableInsertV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "keys" type_attr: "Tin" }
input_arg { name: "values" type_attr: "Tout" }
attr { name: "Tin" type: "type" }
attr { name: "Tout" type: "type" }
-}
-- | 
lookupTableRemoveV2 :: forall v'1 v'2 tin m' . (MonadBuild m',
                                                TensorType tin) =>
                       Tensor v'1 ResourceHandle -- ^ __table_handle__
                       -> Tensor v'2 tin -- ^ __keys__
                       -> m' (ControlNode)
lookupTableRemoveV2 :: Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
lookupTableRemoveV2 = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
lookupTableRemoveV2' OpParams
forall a. a -> a
id
lookupTableRemoveV2' :: forall v'1 v'2 tin m' . (MonadBuild m',
                                                 TensorType tin) => OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __table_handle__
                        -> Tensor v'2 tin -- ^ __keys__
                        -> m' (ControlNode)
lookupTableRemoveV2' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
lookupTableRemoveV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle keys :: Tensor v'2 tin
keys | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle,
                                                             Tensor v'2 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tin
keys]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableRemoveV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tin -> DataType
forall a. TensorType a => a -> DataType
tensorType (tin
forall a. HasCallStack => a
undefined :: tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
input_arg { name: "keys" type_attr: "Tin" }
attr { name: "Tin" type: "type" }
-}
-- | 
lookupTableSize :: forall m' . (MonadBuild m') =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                   -> m' (Tensor Value Data.Int.Int64) -- ^ __size__
lookupTableSize :: Tensor Ref ByteString -> m' (Tensor Value Int64)
lookupTableSize = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
lookupTableSize' OpParams
forall a. a -> a
id
lookupTableSize' :: forall m' . (MonadBuild m') => OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __table_handle__
                    -> m' (Tensor Value Data.Int.Int64) -- ^ __size__
lookupTableSize' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
lookupTableSize' op'options :: OpParams
op'options table_handle :: Tensor Ref ByteString
table_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
table_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_STRING is_ref: true }
output_arg { name: "size" type: DT_INT64 }
-}
-- | 
lookupTableSizeV2 :: forall v'1 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __table_handle__
                     -> m' (Tensor Value Data.Int.Int64) -- ^ __size__
lookupTableSizeV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
lookupTableSizeV2 = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
lookupTableSizeV2' OpParams
forall a. a -> a
id
lookupTableSizeV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __table_handle__
                      -> m' (Tensor Value Data.Int.Int64) -- ^ __size__
lookupTableSizeV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
lookupTableSizeV2' op'options :: OpParams
op'options table_handle :: Tensor v'1 ResourceHandle
table_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
table_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "LookupTableSizeV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "table_handle" type: DT_RESOURCE }
output_arg { name: "size" type: DT_INT64 }
-}
-- | 
loopCond :: Tensor v'1 Bool -- ^ __input__
            -> Tensor Build Bool -- ^ __output__
loopCond :: Tensor v'1 Bool -> Tensor Build Bool
loopCond = OpParams -> Tensor v'1 Bool -> Tensor Build Bool
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Bool -> Tensor Build Bool
loopCond' OpParams
forall a. a -> a
id
loopCond' :: OpParams ->
             Tensor v'1 Bool -- ^ __input__
             -> Tensor Build Bool -- ^ __output__
loopCond' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
loopCond' op'options :: OpParams
op'options input :: Tensor v'1 Bool
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LoopCond"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_BOOL }
output_arg { name: "output" type: DT_BOOL }
-}
-- | 
lowerBound :: forall v'1 v'2 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] out_type) =>
              Tensor v'1 t -- ^ __sorted_inputs__
              -> Tensor v'2 t -- ^ __values__
              -> Tensor Build out_type -- ^ __output__
lowerBound :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
lowerBound = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
forall (v'1 :: * -> *) (v'2 :: * -> *) t out_type.
(TensorType t, OneOf '[Int32, Int64] out_type) =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
lowerBound' OpParams
forall a. a -> a
id
lowerBound' :: forall v'1 v'2 t out_type . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] out_type) =>
               OpParams ->
               Tensor v'1 t -- ^ __sorted_inputs__
               -> Tensor v'2 t -- ^ __values__
               -> Tensor Build out_type -- ^ __output__
lowerBound' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
lowerBound' op'options :: OpParams
op'options sorted_inputs :: Tensor v'1 t
sorted_inputs values :: Tensor v'2 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
sorted_inputs,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "LowerBound"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sorted_inputs" type_attr: "T" }
input_arg { name: "values" type_attr: "T" }
output_arg { name: "output" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
lu :: forall v'1 t output_idx_type . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Word.Word16, Double,
                                              Float] t, OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] output_idx_type) =>
      Tensor v'1 t -- ^ __input__
      -> (Tensor Build t, Tensor Build output_idx_type) -- ^ (__lu__, __p__)
      --
      -- * __lu__
      --
      -- * __p__
lu :: Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
lu = OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
forall (v'1 :: * -> *) t output_idx_type.
(OneOf '[Complex Double, Complex Float, Word16, Double, Float] t,
 OneOf '[Int32, Int64] output_idx_type) =>
OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
lu' OpParams
forall a. a -> a
id
lu' :: forall v'1 t output_idx_type . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Word.Word16, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] output_idx_type) =>
       OpParams ->
       Tensor v'1 t -- ^ __input__
       -> (Tensor Build t, Tensor Build output_idx_type) -- ^ (__lu__, __p__)
       --
       -- * __lu__
       --
       -- * __p__
lu' :: OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
lu' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build t, Tensor Build output_idx_type)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build output_idx_type))
-> Build OpDef -> (Tensor Build t, Tensor Build output_idx_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Lu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_idx_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_idx_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_idx_type
forall a. HasCallStack => a
undefined :: output_idx_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "lu" type_attr: "T" }
output_arg { name: "p" type_attr: "output_idx_type" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr {
  name: "output_idx_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
makeIterator :: forall v'1 v'2 m' . (MonadBuild m') =>
                Tensor v'1 Variant -- ^ __dataset__
                -> Tensor v'2 ResourceHandle -- ^ __iterator__
                -> m' (ControlNode)
makeIterator :: Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' ControlNode
makeIterator = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' ControlNode
makeIterator' OpParams
forall a. a -> a
id
makeIterator' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                 Tensor v'1 Variant -- ^ __dataset__
                 -> Tensor v'2 ResourceHandle -- ^ __iterator__
                 -> m' (ControlNode)
makeIterator' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' ControlNode
makeIterator' op'options :: OpParams
op'options dataset :: Tensor v'1 Variant
dataset iterator :: Tensor v'2 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MakeIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dataset" type: DT_VARIANT }
input_arg { name: "iterator" type: DT_RESOURCE }
-}
-- | 
mapClear :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
            -> m' (ControlNode)
mapClear :: [DataType] -> m' ControlNode
mapClear = OpParams -> [DataType] -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' ControlNode
mapClear' OpParams
forall a. a -> a
id
mapClear' :: forall m' . (MonadBuild m') => OpParams ->
             [DataType] -- ^ __dtypes__
             -> m' (ControlNode)
mapClear' :: OpParams -> [DataType] -> m' ControlNode
mapClear' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapClear"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapIncompleteSize :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
                     -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
mapIncompleteSize :: [DataType] -> m' (Tensor Value Int32)
mapIncompleteSize = OpParams -> [DataType] -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value Int32)
mapIncompleteSize' OpParams
forall a. a -> a
id
mapIncompleteSize' :: forall m' . (MonadBuild m') => OpParams ->
                      [DataType] -- ^ __dtypes__
                      -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
mapIncompleteSize' :: OpParams -> [DataType] -> m' (Tensor Value Int32)
mapIncompleteSize' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapIncompleteSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapPeek :: forall v'1 v'2 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
           Tensor v'1 Data.Int.Int64 -- ^ __key__
           -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
           -> m' (TensorList (Value) dtypes) -- ^ __values__
mapPeek :: Tensor v'1 Int64
-> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
mapPeek = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (v'2 :: * -> *) (dtypes :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
mapPeek' OpParams
forall a. a -> a
id
mapPeek' :: forall v'1 v'2 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
            OpParams ->
            Tensor v'1 Data.Int.Int64 -- ^ __key__
            -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
            -> m' (TensorList (Value) dtypes) -- ^ __values__
mapPeek' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
mapPeek' op'options :: OpParams
op'options key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapPeek"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapSize :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
           -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
mapSize :: [DataType] -> m' (Tensor Value Int32)
mapSize = OpParams -> [DataType] -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value Int32)
mapSize' OpParams
forall a. a -> a
id
mapSize' :: forall m' . (MonadBuild m') => OpParams ->
            [DataType] -- ^ __dtypes__
            -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
mapSize' :: OpParams -> [DataType] -> m' (Tensor Value Int32)
mapSize' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapStage :: forall v'1 v'2 v'3 fake_dtypes m' . (MonadBuild m',
                                                 TensorTypes fake_dtypes) =>
            [DataType] -- ^ __dtypes__
            -> Tensor v'1 Data.Int.Int64 -- ^ __key__
            -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
            -> TensorList (v'3) fake_dtypes -- ^ __values__
            -> m' (ControlNode)
mapStage :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
mapStage = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (fake_dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes fake_dtypes) =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
mapStage' OpParams
forall a. a -> a
id
mapStage' :: forall v'1 v'2 v'3 fake_dtypes m' . (MonadBuild m',
                                                  TensorTypes fake_dtypes) =>
             OpParams ->
             [DataType] -- ^ __dtypes__
             -> Tensor v'1 Data.Int.Int64 -- ^ __key__
             -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
             -> TensorList (v'3) fake_dtypes -- ^ __values__
             -> m' (ControlNode)
mapStage' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
mapStage' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices values :: TensorList v'3 fake_dtypes
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             TensorList v'3 fake_dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 fake_dtypes
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapStage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "fake_dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy fake_dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy fake_dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy fake_dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "values" type_list_attr: "fake_dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr {
  name: "fake_dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapUnstage :: forall v'1 v'2 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
              Tensor v'1 Data.Int.Int64 -- ^ __key__
              -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
              -> m' (TensorList (Value) dtypes) -- ^ __values__
mapUnstage :: Tensor v'1 Int64
-> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
mapUnstage = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (v'2 :: * -> *) (dtypes :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
mapUnstage' OpParams
forall a. a -> a
id
mapUnstage' :: forall v'1 v'2 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
               OpParams ->
               Tensor v'1 Data.Int.Int64 -- ^ __key__
               -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
               -> m' (TensorList (Value) dtypes) -- ^ __values__
mapUnstage' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
mapUnstage' op'options :: OpParams
op'options key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapUnstage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
mapUnstageNoKey :: forall v'1 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
                   Tensor v'1 Data.Int.Int32 -- ^ __indices__
                   -> m' ((Tensor Value Data.Int.Int64,
                           TensorList (Value) dtypes))
                   -- ^ (__key__, __values__)
                   --
                   -- * __key__
                   --
                   -- * __values__
mapUnstageNoKey :: Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
mapUnstageNoKey = OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
mapUnstageNoKey' OpParams
forall a. a -> a
id
mapUnstageNoKey' :: forall v'1 dtypes m' . (MonadBuild m',
                                            TensorTypes dtypes) => OpParams ->
                    Tensor v'1 Data.Int.Int32 -- ^ __indices__
                    -> m' ((Tensor Value Data.Int.Int64,
                            TensorList (Value) dtypes))
                    -- ^ (__key__, __values__)
                    --
                    -- * __key__
                    --
                    -- * __values__
mapUnstageNoKey' :: OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
mapUnstageNoKey' op'options :: OpParams
op'options indices :: Tensor v'1 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, TensorList Value dtypes)
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, TensorList Value dtypes)
 -> m' (Tensor Value Int64, TensorList Value dtypes))
-> Build (Tensor Value Int64, TensorList Value dtypes)
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
indices]
        [Int64]
-> OpDef -> Build (Tensor Value Int64, TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MapUnstageNoKey"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "key" type: DT_INT64 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
matMul :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int32, Data.Int.Int64,
                                      Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __a__
          -> Tensor v'2 t -- ^ __b__
          -> Tensor Build t -- ^ __product__
matMul :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matMul = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matMul' OpParams
forall a. a -> a
id
matMul' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int32, Data.Int.Int64,
                                       Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __a__
           -> Tensor v'2 t -- ^ __b__
           -> Tensor Build t -- ^ __product__
matMul' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matMul' op'options :: OpParams
op'options a :: Tensor v'1 t
a b :: Tensor v'2 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "product" type_attr: "T" }
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matchingFiles :: Tensor v'1 Data.ByteString.ByteString -- ^ __pattern__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __filenames__
matchingFiles :: Tensor v'1 ByteString -> Tensor Build ByteString
matchingFiles = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
matchingFiles' OpParams
forall a. a -> a
id
matchingFiles' :: OpParams ->
                  Tensor v'1 Data.ByteString.ByteString -- ^ __pattern__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __filenames__
matchingFiles' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
matchingFiles' op'options :: OpParams
op'options pattern :: Tensor v'1 ByteString
pattern | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
pattern]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatchingFiles"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "pattern" type: DT_STRING }
output_arg { name: "filenames" type: DT_STRING }
-}
-- | 
matchingFilesDataset :: forall v'1 m' . (MonadBuild m') =>
                        Tensor v'1 Data.ByteString.ByteString -- ^ __patterns__
                        -> m' (Tensor Value Variant) -- ^ __handle__
matchingFilesDataset :: Tensor v'1 ByteString -> m' (Tensor Value Variant)
matchingFilesDataset = OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
matchingFilesDataset' OpParams
forall a. a -> a
id
matchingFilesDataset' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                         Tensor v'1 Data.ByteString.ByteString -- ^ __patterns__
                         -> m' (Tensor Value Variant) -- ^ __handle__
matchingFilesDataset' :: OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
matchingFilesDataset' op'options :: OpParams
op'options patterns :: Tensor v'1 ByteString
patterns | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
patterns]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MatchingFilesDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "patterns" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
matrixBandPart :: forall v'1 v'2 v'3 t tindex . (TensorType t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindex) =>
                  Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 tindex -- ^ __num_lower__
                  -> Tensor v'3 tindex -- ^ __num_upper__
                  -> Tensor Build t -- ^ __band__
matrixBandPart :: Tensor v'1 t
-> Tensor v'2 tindex -> Tensor v'3 tindex -> Tensor Build t
matrixBandPart = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindex
-> Tensor v'3 tindex
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindex.
(TensorType t, OneOf '[Int32, Int64] tindex) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindex
-> Tensor v'3 tindex
-> Tensor Build t
matrixBandPart' OpParams
forall a. a -> a
id
matrixBandPart' :: forall v'1 v'2 v'3 t tindex . (TensorType t,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tindex) =>
                   OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 tindex -- ^ __num_lower__
                   -> Tensor v'3 tindex -- ^ __num_upper__
                   -> Tensor Build t -- ^ __band__
matrixBandPart' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindex
-> Tensor v'3 tindex
-> Tensor Build t
matrixBandPart' op'options :: OpParams
op'options input :: Tensor v'1 t
input num_lower :: Tensor v'2 tindex
num_lower num_upper :: Tensor v'3 tindex
num_upper | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tindex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindex
num_lower,
                                                             Tensor v'3 tindex -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindex
num_upper]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixBandPart"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindex
forall a. HasCallStack => a
undefined :: tindex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "num_lower" type_attr: "Tindex" }
input_arg { name: "num_upper" type_attr: "Tindex" }
output_arg { name: "band" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindex"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
matrixDeterminant :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Word.Word16, Double,
                                             Float] t) =>
                     Tensor v'1 t -- ^ __input__
                     -> Tensor Build t -- ^ __output__
matrixDeterminant :: Tensor v'1 t -> Tensor Build t
matrixDeterminant = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixDeterminant' OpParams
forall a. a -> a
id
matrixDeterminant' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Word.Word16, Double,
                                              Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __input__
                      -> Tensor Build t -- ^ __output__
matrixDeterminant' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixDeterminant' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDeterminant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matrixDiag :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __diagonal__
              -> Tensor Build t -- ^ __output__
matrixDiag :: Tensor v'1 t -> Tensor Build t
matrixDiag = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixDiag' OpParams
forall a. a -> a
id
matrixDiag' :: forall v'1 t . (TensorType t) => OpParams ->
               Tensor v'1 t -- ^ __diagonal__
               -> Tensor Build t -- ^ __output__
matrixDiag' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixDiag' op'options :: OpParams
op'options diagonal :: Tensor v'1 t
diagonal | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonal]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonal" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixDiagPart :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                  -> Tensor Build t -- ^ __diagonal__
matrixDiagPart :: Tensor v'1 t -> Tensor Build t
matrixDiagPart = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixDiagPart' OpParams
forall a. a -> a
id
matrixDiagPart' :: forall v'1 t . (TensorType t) => OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> Tensor Build t -- ^ __diagonal__
matrixDiagPart' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixDiagPart' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiagPart"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "diagonal" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixDiagPartV2 :: forall v'1 v'2 v'3 t . (TensorType t) =>
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                    -> Tensor v'3 t -- ^ __padding_value__
                    -> Tensor Build t -- ^ __diagonal__
matrixDiagPartV2 :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
matrixDiagPartV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
matrixDiagPartV2' OpParams
forall a. a -> a
id
matrixDiagPartV2' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                     -> Tensor v'3 t -- ^ __padding_value__
                     -> Tensor Build t -- ^ __diagonal__
matrixDiagPartV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
matrixDiagPartV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input k :: Tensor v'2 Int32
k padding_value :: Tensor v'3 t
padding_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
k,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
padding_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiagPartV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
input_arg { name: "padding_value" type_attr: "T" }
output_arg { name: "diagonal" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixDiagPartV3 :: forall v'1 v'2 v'3 t . (TensorType t) =>
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                    -> Tensor v'3 t -- ^ __padding_value__
                    -> Tensor Build t -- ^ __diagonal__
matrixDiagPartV3 :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
matrixDiagPartV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
matrixDiagPartV3' OpParams
forall a. a -> a
id
matrixDiagPartV3' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                     -> Tensor v'3 t -- ^ __padding_value__
                     -> Tensor Build t -- ^ __diagonal__
matrixDiagPartV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor Build t
matrixDiagPartV3' op'options :: OpParams
op'options input :: Tensor v'1 t
input k :: Tensor v'2 Int32
k padding_value :: Tensor v'3 t
padding_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
k,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
padding_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiagPartV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
input_arg { name: "padding_value" type_attr: "T" }
output_arg { name: "diagonal" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "align"
  type: "string"
  default_value { s: "RIGHT_LEFT" }
  allowed_values {
    list {
      s: "LEFT_RIGHT" s: "RIGHT_LEFT" s: "LEFT_LEFT" s: "RIGHT_RIGHT"
    }
  }
}
-}
-- | 
matrixDiagV2 :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) =>
                Tensor v'1 t -- ^ __diagonal__
                -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                -> Tensor v'3 Data.Int.Int32 -- ^ __num_rows__
                -> Tensor v'4 Data.Int.Int32 -- ^ __num_cols__
                -> Tensor v'5 t -- ^ __padding_value__
                -> Tensor Build t -- ^ __output__
matrixDiagV2 :: Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV2' OpParams
forall a. a -> a
id
matrixDiagV2' :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) => OpParams ->
                 Tensor v'1 t -- ^ __diagonal__
                 -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                 -> Tensor v'3 Data.Int.Int32 -- ^ __num_rows__
                 -> Tensor v'4 Data.Int.Int32 -- ^ __num_cols__
                 -> Tensor v'5 t -- ^ __padding_value__
                 -> Tensor Build t -- ^ __output__
matrixDiagV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV2' op'options :: OpParams
op'options diagonal :: Tensor v'1 t
diagonal k :: Tensor v'2 Int32
k num_rows :: Tensor v'3 Int32
num_rows num_cols :: Tensor v'4 Int32
num_cols
              padding_value :: Tensor v'5 t
padding_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonal,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
k,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
num_rows,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
num_cols,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
padding_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiagV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonal" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
input_arg { name: "num_rows" type: DT_INT32 }
input_arg { name: "num_cols" type: DT_INT32 }
input_arg { name: "padding_value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixDiagV3 :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) =>
                Tensor v'1 t -- ^ __diagonal__
                -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                -> Tensor v'3 Data.Int.Int32 -- ^ __num_rows__
                -> Tensor v'4 Data.Int.Int32 -- ^ __num_cols__
                -> Tensor v'5 t -- ^ __padding_value__
                -> Tensor Build t -- ^ __output__
matrixDiagV3 :: Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV3' OpParams
forall a. a -> a
id
matrixDiagV3' :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) => OpParams ->
                 Tensor v'1 t -- ^ __diagonal__
                 -> Tensor v'2 Data.Int.Int32 -- ^ __k__
                 -> Tensor v'3 Data.Int.Int32 -- ^ __num_rows__
                 -> Tensor v'4 Data.Int.Int32 -- ^ __num_cols__
                 -> Tensor v'5 t -- ^ __padding_value__
                 -> Tensor Build t -- ^ __output__
matrixDiagV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 t
-> Tensor Build t
matrixDiagV3' op'options :: OpParams
op'options diagonal :: Tensor v'1 t
diagonal k :: Tensor v'2 Int32
k num_rows :: Tensor v'3 Int32
num_rows num_cols :: Tensor v'4 Int32
num_cols
              padding_value :: Tensor v'5 t
padding_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonal,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
k,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
num_rows,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
num_cols,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
padding_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixDiagV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonal" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
input_arg { name: "num_rows" type: DT_INT32 }
input_arg { name: "num_cols" type: DT_INT32 }
input_arg { name: "padding_value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "align"
  type: "string"
  default_value { s: "RIGHT_LEFT" }
  allowed_values {
    list {
      s: "LEFT_RIGHT" s: "RIGHT_LEFT" s: "LEFT_LEFT" s: "RIGHT_RIGHT"
    }
  }
}
-}
-- | 
matrixExponential :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Word.Word16, Double,
                                             Float] t) =>
                     Tensor v'1 t -- ^ __input__
                     -> Tensor Build t -- ^ __output__
matrixExponential :: Tensor v'1 t -> Tensor Build t
matrixExponential = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixExponential' OpParams
forall a. a -> a
id
matrixExponential' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Word.Word16, Double,
                                              Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __input__
                      -> Tensor Build t -- ^ __output__
matrixExponential' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixExponential' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixExponential"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matrixInverse :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
                 Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
matrixInverse :: Tensor v'1 t -> Tensor Build t
matrixInverse = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixInverse' OpParams
forall a. a -> a
id
matrixInverse' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Word.Word16, Double, Float] t) =>
                  OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor Build t -- ^ __output__
matrixInverse' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixInverse' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixInverse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matrixLogarithm :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float)] t) =>
                   Tensor v'1 t -- ^ __input__
                   -> Tensor Build t -- ^ __output__
matrixLogarithm :: Tensor v'1 t -> Tensor Build t
matrixLogarithm = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) tcomplex.
OneOf '[Complex Double, Complex Float] tcomplex =>
OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
matrixLogarithm' OpParams
forall a. a -> a
id
matrixLogarithm' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float)] t) =>
                    OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> Tensor Build t -- ^ __output__
matrixLogarithm' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixLogarithm' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixLogarithm"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
matrixSetDiag :: forall v'1 v'2 t . (TensorType t) =>
                 Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 t -- ^ __diagonal__
                 -> Tensor Build t -- ^ __output__
matrixSetDiag :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSetDiag = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSetDiag' OpParams
forall a. a -> a
id
matrixSetDiag' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 t -- ^ __diagonal__
                  -> Tensor Build t -- ^ __output__
matrixSetDiag' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSetDiag' op'options :: OpParams
op'options input :: Tensor v'1 t
input diagonal :: Tensor v'2 t
diagonal | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
diagonal]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSetDiag"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "diagonal" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixSetDiagV2 :: forall v'1 v'2 v'3 t . (TensorType t) =>
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 t -- ^ __diagonal__
                   -> Tensor v'3 Data.Int.Int32 -- ^ __k__
                   -> Tensor Build t -- ^ __output__
matrixSetDiagV2 :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSetDiagV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
matrixSetDiagV2' OpParams
forall a. a -> a
id
matrixSetDiagV2' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 t -- ^ __diagonal__
                    -> Tensor v'3 Data.Int.Int32 -- ^ __k__
                    -> Tensor Build t -- ^ __output__
matrixSetDiagV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
matrixSetDiagV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input diagonal :: Tensor v'2 t
diagonal k :: Tensor v'3 Int32
k | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
diagonal,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
k]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSetDiagV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "diagonal" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
matrixSetDiagV3 :: forall v'1 v'2 v'3 t . (TensorType t) =>
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 t -- ^ __diagonal__
                   -> Tensor v'3 Data.Int.Int32 -- ^ __k__
                   -> Tensor Build t -- ^ __output__
matrixSetDiagV3 :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
matrixSetDiagV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
matrixSetDiagV3' OpParams
forall a. a -> a
id
matrixSetDiagV3' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 t -- ^ __diagonal__
                    -> Tensor v'3 Data.Int.Int32 -- ^ __k__
                    -> Tensor Build t -- ^ __output__
matrixSetDiagV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor Build t
matrixSetDiagV3' op'options :: OpParams
op'options input :: Tensor v'1 t
input diagonal :: Tensor v'2 t
diagonal k :: Tensor v'3 Int32
k | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
diagonal,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
k]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSetDiagV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "diagonal" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "align"
  type: "string"
  default_value { s: "RIGHT_LEFT" }
  allowed_values {
    list {
      s: "LEFT_RIGHT" s: "RIGHT_LEFT" s: "LEFT_LEFT" s: "RIGHT_RIGHT"
    }
  }
}
-}
-- | 
matrixSolve :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Word.Word16, Double,
                                           Float] t) =>
               Tensor v'1 t -- ^ __matrix__
               -> Tensor v'2 t -- ^ __rhs__
               -> Tensor Build t -- ^ __output__
matrixSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSolve' OpParams
forall a. a -> a
id
matrixSolve' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Word.Word16, Double,
                                            Float] t) => OpParams ->
                Tensor v'1 t -- ^ __matrix__
                -> Tensor v'2 t -- ^ __rhs__
                -> Tensor Build t -- ^ __output__
matrixSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixSolve' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matrixSolveLs :: forall v'1 v'2 v'3 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Word.Word16, Double,
                                                 Float] t) =>
                 Tensor v'1 t -- ^ __matrix__
                 -> Tensor v'2 t -- ^ __rhs__
                 -> Tensor v'3 Double -- ^ __l2_regularizer__
                 -> Tensor Build t -- ^ __output__
matrixSolveLs :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
matrixSolveLs = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
matrixSolveLs' OpParams
forall a. a -> a
id
matrixSolveLs' :: forall v'1 v'2 v'3 t . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Word.Word16, Double,
                                                  Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __matrix__
                  -> Tensor v'2 t -- ^ __rhs__
                  -> Tensor v'3 Double -- ^ __l2_regularizer__
                  -> Tensor Build t -- ^ __output__
matrixSolveLs' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Double
-> Tensor Build t
matrixSolveLs' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs l2_regularizer :: Tensor v'3 Double
l2_regularizer | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs,
                                                             Tensor v'3 Double -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Double
l2_regularizer]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSolveLs"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
input_arg { name: "l2_regularizer" type: DT_DOUBLE }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr { name: "fast" type: "bool" default_value { b: true } }
-}
-- | 
matrixSquareRoot :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Word.Word16, Double,
                                            Float] t) =>
                    Tensor v'1 t -- ^ __input__
                    -> Tensor Build t -- ^ __output__
matrixSquareRoot :: Tensor v'1 t -> Tensor Build t
matrixSquareRoot = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
matrixSquareRoot' OpParams
forall a. a -> a
id
matrixSquareRoot' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> Tensor Build t -- ^ __output__
matrixSquareRoot' :: OpParams -> Tensor v'1 t -> Tensor Build t
matrixSquareRoot' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixSquareRoot"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
matrixTriangularSolve :: forall v'1 v'2
                         t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Word.Word16, Double, Float] t) =>
                         Tensor v'1 t -- ^ __matrix__
                         -> Tensor v'2 t -- ^ __rhs__
                         -> Tensor Build t -- ^ __output__
matrixTriangularSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixTriangularSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixTriangularSolve' OpParams
forall a. a -> a
id
matrixTriangularSolve' :: forall v'1 v'2
                          t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Word.Word16, Double, Float] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __matrix__
                          -> Tensor v'2 t -- ^ __rhs__
                          -> Tensor Build t -- ^ __output__
matrixTriangularSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
matrixTriangularSolve' op'options :: OpParams
op'options matrix :: Tensor v'1 t
matrix rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
matrix,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MatrixTriangularSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "matrix" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "lower" type: "bool" default_value { b: true } }
attr { name: "adjoint" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
max :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t,
                                OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
       Tensor v'1 t -- ^ __input__
       -> Tensor v'2 tidx -- ^ __reduction_indices__
       -> Tensor Build t -- ^ __output__
max :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
max = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
max' OpParams
forall a. a -> a
id
max' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tidx) => OpParams ->
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build t -- ^ __output__
max' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
max' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Max"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
maxIntraOpParallelismDataset :: [DataType] -- ^ __output_types__
                                -> Tensor v'1 Variant -- ^ __input_dataset__
                                -> Tensor v'2 Data.Int.Int64 -- ^ __max_intra_op_parallelism__
                                -> Tensor Build Variant -- ^ __handle__
maxIntraOpParallelismDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
maxIntraOpParallelismDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
maxIntraOpParallelismDataset' OpParams
forall a. a -> a
id
maxIntraOpParallelismDataset' :: OpParams ->
                                 [DataType] -- ^ __output_types__
                                 -> Tensor v'1 Variant -- ^ __input_dataset__
                                 -> Tensor v'2 Data.Int.Int64 -- ^ __max_intra_op_parallelism__
                                 -> Tensor Build Variant -- ^ __handle__
maxIntraOpParallelismDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
maxIntraOpParallelismDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                              max_intra_op_parallelism :: Tensor v'2 Int64
max_intra_op_parallelism | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
max_intra_op_parallelism]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxIntraOpParallelismDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "max_intra_op_parallelism" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
maxPool :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Int.Int64, Data.Int.Int8,
                                   Data.Word.Word16, Data.Word.Word8, Double,
                                   Float] t) => ByteString -- ^ __padding__
           -> Tensor v'1 t -- ^ __input__
           -> Tensor Build t -- ^ __output__
maxPool :: ByteString -> Tensor v'1 t -> Tensor Build t
maxPool = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
maxPool' OpParams
forall a. a -> a
id
maxPool' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word8, Double,
                                    Float] t) => OpParams ->
            ByteString -- ^ __padding__
            -> Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
maxPool' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
maxPool' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_UINT16
      type: DT_QINT8
    }
  }
}
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" s: "NCHW_VECT_C" } }
}
-}
-- | 
maxPool3D :: forall v'1 t . (OneOf '[Data.Word.Word16, Float] t) =>
             ByteString -- ^ __padding__
             -> Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
maxPool3D :: ByteString -> Tensor v'1 t -> Tensor Build t
maxPool3D = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
maxPool3D' OpParams
forall a. a -> a
id
maxPool3D' :: forall v'1 t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
              ByteString -- ^ __padding__
              -> Tensor v'1 t -- ^ __input__
              -> Tensor Build t -- ^ __output__
maxPool3D' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
maxPool3D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPool3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
-}
-- | 
maxPool3DGrad :: forall v'1 v'2 v'3 t tInput . (OneOf '[Data.Word.Word16,
                                                        Float] t,
                                                OneOf '[Data.Word.Word16,
                                                        Float] tInput) =>
                 ByteString -- ^ __padding__
                 -> Tensor v'1 tInput -- ^ __orig_input__
                 -> Tensor v'2 tInput -- ^ __orig_output__
                 -> Tensor v'3 t -- ^ __grad__
                 -> Tensor Build t -- ^ __output__
maxPool3DGrad :: ByteString
-> Tensor v'1 tInput
-> Tensor v'2 tInput
-> Tensor v'3 t
-> Tensor Build t
maxPool3DGrad = OpParams
-> ByteString
-> Tensor v'1 tInput
-> Tensor v'2 tInput
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tInput.
(OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput) =>
OpParams
-> ByteString
-> Tensor v'1 tInput
-> Tensor v'2 tInput
-> Tensor v'3 t
-> Tensor Build t
maxPool3DGrad' OpParams
forall a. a -> a
id
maxPool3DGrad' :: forall v'1 v'2 v'3 t tInput . (OneOf '[Data.Word.Word16,
                                                         Float] t,
                                                 OneOf '[Data.Word.Word16,
                                                         Float] tInput) =>
                  OpParams ->
                  ByteString -- ^ __padding__
                  -> Tensor v'1 tInput -- ^ __orig_input__
                  -> Tensor v'2 tInput -- ^ __orig_output__
                  -> Tensor v'3 t -- ^ __grad__
                  -> Tensor Build t -- ^ __output__
maxPool3DGrad' :: OpParams
-> ByteString
-> Tensor v'1 tInput
-> Tensor v'2 tInput
-> Tensor v'3 t
-> Tensor Build t
maxPool3DGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 tInput
orig_input orig_output :: Tensor v'2 tInput
orig_output
               grad :: Tensor v'3 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tInput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tInput
orig_input,
                                                             Tensor v'2 tInput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tInput
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPool3DGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "TInput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tInput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tInput
forall a. HasCallStack => a
undefined :: tInput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "TInput" }
input_arg { name: "orig_output" type_attr: "TInput" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
attr {
  name: "TInput"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list { type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
-}
-- | 
maxPool3DGradGrad :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                     ByteString -- ^ __padding__
                     -> Tensor v'1 t -- ^ __orig_input__
                     -> Tensor v'2 t -- ^ __orig_output__
                     -> Tensor v'3 t -- ^ __grad__
                     -> Tensor Build t -- ^ __output__
maxPool3DGradGrad :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
maxPool3DGradGrad = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPool3DGradGrad' OpParams
forall a. a -> a
id
maxPool3DGradGrad' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                      ByteString -- ^ __padding__
                      -> Tensor v'1 t -- ^ __orig_input__
                      -> Tensor v'2 t -- ^ __orig_output__
                      -> Tensor v'3 t -- ^ __grad__
                      -> Tensor Build t -- ^ __output__
maxPool3DGradGrad' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPool3DGradGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output
                   grad :: Tensor v'3 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPool3DGradGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 5
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NDHWC" }
  allowed_values { list { s: "NDHWC" s: "NCDHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGrad :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) =>
               ByteString -- ^ __padding__
               -> Tensor v'1 t -- ^ __orig_input__
               -> Tensor v'2 t -- ^ __orig_output__
               -> Tensor v'3 t -- ^ __grad__
               -> Tensor Build t -- ^ __output__
maxPoolGrad :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
maxPoolGrad = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPoolGrad' OpParams
forall a. a -> a
id
maxPoolGrad' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t) => OpParams ->
                ByteString -- ^ __padding__
                -> Tensor v'1 t -- ^ __orig_input__
                -> Tensor v'2 t -- ^ __orig_output__
                -> Tensor v'3 t -- ^ __grad__
                -> Tensor Build t -- ^ __output__
maxPoolGrad' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPoolGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output grad :: Tensor v'3 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGradGrad :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] t) =>
                   ByteString -- ^ __padding__
                   -> Tensor v'1 t -- ^ __orig_input__
                   -> Tensor v'2 t -- ^ __orig_output__
                   -> Tensor v'3 t -- ^ __grad__
                   -> Tensor Build t -- ^ __output__
maxPoolGradGrad :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
maxPoolGradGrad = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPoolGradGrad' OpParams
forall a. a -> a
id
maxPoolGradGrad' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t) => OpParams ->
                    ByteString -- ^ __padding__
                    -> Tensor v'1 t -- ^ __orig_input__
                    -> Tensor v'2 t -- ^ __orig_output__
                    -> Tensor v'3 t -- ^ __grad__
                    -> Tensor Build t -- ^ __output__
maxPoolGradGrad' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
maxPoolGradGrad' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output
                 grad :: Tensor v'3 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGradGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGradGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                     ByteString -- ^ __padding__
                     -> Tensor v'1 t -- ^ __orig_input__
                     -> Tensor v'2 t -- ^ __orig_output__
                     -> Tensor v'3 t -- ^ __grad__
                     -> Tensor v'4 Data.Int.Int32 -- ^ __ksize__
                     -> Tensor v'5 Data.Int.Int32 -- ^ __strides__
                     -> Tensor Build t -- ^ __output__
maxPoolGradGradV2 :: ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradGradV2 = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradGradV2' OpParams
forall a. a -> a
id
maxPoolGradGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                      OpParams ->
                      ByteString -- ^ __padding__
                      -> Tensor v'1 t -- ^ __orig_input__
                      -> Tensor v'2 t -- ^ __orig_output__
                      -> Tensor v'3 t -- ^ __grad__
                      -> Tensor v'4 Data.Int.Int32 -- ^ __ksize__
                      -> Tensor v'5 Data.Int.Int32 -- ^ __strides__
                      -> Tensor Build t -- ^ __output__
maxPoolGradGradV2' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradGradV2' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output grad :: Tensor v'3 t
grad ksize :: Tensor v'4 Int32
ksize
                   strides :: Tensor v'5 Int32
strides | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
ksize,
                                                             Tensor v'5 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int32
strides]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGradGradV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "ksize" type: DT_INT32 }
input_arg { name: "strides" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGradGradWithArgmax :: forall v'1 v'2 v'3 targmax
                             t . (OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] targmax,
                                  OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) =>
                             ByteString -- ^ __padding__
                             -> Tensor v'1 t -- ^ __input__
                             -> Tensor v'2 t -- ^ __grad__
                             -> Tensor v'3 targmax -- ^ __argmax__
                             -> Tensor Build t -- ^ __output__
maxPoolGradGradWithArgmax :: ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradGradWithArgmax = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) targmax t.
(OneOf '[Int32, Int64] targmax,
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t) =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradGradWithArgmax' OpParams
forall a. a -> a
id
maxPoolGradGradWithArgmax' :: forall v'1 v'2 v'3 targmax
                              t . (OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] targmax,
                                   OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t) => OpParams ->
                              ByteString -- ^ __padding__
                              -> Tensor v'1 t -- ^ __input__
                              -> Tensor v'2 t -- ^ __grad__
                              -> Tensor v'3 targmax -- ^ __argmax__
                              -> Tensor Build t -- ^ __output__
maxPoolGradGradWithArgmax' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradGradWithArgmax' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input grad :: Tensor v'2 t
grad
                           argmax :: Tensor v'3 targmax
argmax | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad,
                                                             Tensor v'3 targmax -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 targmax
argmax]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGradGradWithArgmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Targmax" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ targmax -> DataType
forall a. TensorType a => a -> DataType
tensorType (targmax
forall a. HasCallStack => a
undefined :: targmax)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "argmax" type_attr: "Targmax" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "include_batch_in_index"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "Targmax"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t) =>
                 ByteString -- ^ __padding__
                 -> Tensor v'1 t -- ^ __orig_input__
                 -> Tensor v'2 t -- ^ __orig_output__
                 -> Tensor v'3 t -- ^ __grad__
                 -> Tensor v'4 Data.Int.Int32 -- ^ __ksize__
                 -> Tensor v'5 Data.Int.Int32 -- ^ __strides__
                 -> Tensor Build t -- ^ __output__
maxPoolGradV2 :: ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradV2 = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradV2' OpParams
forall a. a -> a
id
maxPoolGradV2' :: forall v'1 v'2 v'3 v'4 v'5 t . (OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Int.Int8,
                                                          Data.Word.Word16,
                                                          Data.Word.Word32,
                                                          Data.Word.Word64,
                                                          Data.Word.Word8,
                                                          Double, Float] t) =>
                  OpParams ->
                  ByteString -- ^ __padding__
                  -> Tensor v'1 t -- ^ __orig_input__
                  -> Tensor v'2 t -- ^ __orig_output__
                  -> Tensor v'3 t -- ^ __grad__
                  -> Tensor v'4 Data.Int.Int32 -- ^ __ksize__
                  -> Tensor v'5 Data.Int.Int32 -- ^ __strides__
                  -> Tensor Build t -- ^ __output__
maxPoolGradV2' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor v'5 Int32
-> Tensor Build t
maxPoolGradV2' op'options :: OpParams
op'options padding :: ByteString
padding orig_input :: Tensor v'1 t
orig_input orig_output :: Tensor v'2 t
orig_output grad :: Tensor v'3 t
grad ksize :: Tensor v'4 Int32
ksize
               strides :: Tensor v'5 Int32
strides | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
orig_input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
orig_output,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
ksize,
                                                             Tensor v'5 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int32
strides]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGradV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "orig_input" type_attr: "T" }
input_arg { name: "orig_output" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "ksize" type: DT_INT32 }
input_arg { name: "strides" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolGradWithArgmax :: forall v'1 v'2 v'3 targmax t . (OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] targmax,
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Int.Int8,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word32,
                                                                 Data.Word.Word64,
                                                                 Data.Word.Word8,
                                                                 Double,
                                                                 Float] t) =>
                         ByteString -- ^ __padding__
                         -> Tensor v'1 t -- ^ __input__
                         -> Tensor v'2 t -- ^ __grad__
                         -> Tensor v'3 targmax -- ^ __argmax__
                         -> Tensor Build t -- ^ __output__
maxPoolGradWithArgmax :: ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradWithArgmax = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) targmax t.
(OneOf '[Int32, Int64] targmax,
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t) =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradWithArgmax' OpParams
forall a. a -> a
id
maxPoolGradWithArgmax' :: forall v'1 v'2 v'3 targmax
                          t . (OneOf '[Data.Int.Int32, Data.Int.Int64] targmax,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
                          ByteString -- ^ __padding__
                          -> Tensor v'1 t -- ^ __input__
                          -> Tensor v'2 t -- ^ __grad__
                          -> Tensor v'3 targmax -- ^ __argmax__
                          -> Tensor Build t -- ^ __output__
maxPoolGradWithArgmax' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 targmax
-> Tensor Build t
maxPoolGradWithArgmax' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input grad :: Tensor v'2 t
grad argmax :: Tensor v'3 targmax
argmax | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad,
                                                             Tensor v'3 targmax -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 targmax
argmax]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolGradWithArgmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Targmax" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ targmax -> DataType
forall a. TensorType a => a -> DataType
tensorType (targmax
forall a. HasCallStack => a
undefined :: targmax)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "argmax" type_attr: "Targmax" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "include_batch_in_index"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "Targmax"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maxPoolV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word8,
                                             Double, Float] t) =>
             ByteString -- ^ __padding__
             -> Tensor v'1 t -- ^ __input__
             -> Tensor v'2 Data.Int.Int32 -- ^ __ksize__
             -> Tensor v'3 Data.Int.Int32 -- ^ __strides__
             -> Tensor Build t -- ^ __output__
maxPoolV2 :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build t
maxPoolV2 = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build t
maxPoolV2' OpParams
forall a. a -> a
id
maxPoolV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16, Data.Word.Word8,
                                              Double, Float] t) => OpParams ->
              ByteString -- ^ __padding__
              -> Tensor v'1 t -- ^ __input__
              -> Tensor v'2 Data.Int.Int32 -- ^ __ksize__
              -> Tensor v'3 Data.Int.Int32 -- ^ __strides__
              -> Tensor Build t -- ^ __output__
maxPoolV2' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build t
maxPoolV2' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input ksize :: Tensor v'2 Int32
ksize strides :: Tensor v'3 Int32
strides | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
ksize,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
strides]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "ksize" type: DT_INT32 }
input_arg { name: "strides" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_UINT16
      type: DT_QINT8
    }
  }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" s: "NCHW_VECT_C" } }
}
-}
-- | 
maxPoolWithArgmax :: forall v'1 targmax t . (OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] targmax,
                                             OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                     ByteString -- ^ __padding__
                     -> Tensor v'1 t -- ^ __input__
                     -> (Tensor Build t, Tensor Build targmax)
                     -- ^ (__output__, __argmax__)
                     --
                     -- * __output__
                     --
                     -- * __argmax__
maxPoolWithArgmax :: ByteString
-> Tensor v'1 t -> (Tensor Build t, Tensor Build targmax)
maxPoolWithArgmax = OpParams
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build targmax)
forall (v'1 :: * -> *) targmax t.
(OneOf '[Int32, Int64] targmax,
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t) =>
OpParams
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build targmax)
maxPoolWithArgmax' OpParams
forall a. a -> a
id
maxPoolWithArgmax' :: forall v'1 targmax t . (OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] targmax,
                                              OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                      ByteString -- ^ __padding__
                      -> Tensor v'1 t -- ^ __input__
                      -> (Tensor Build t, Tensor Build targmax)
                      -- ^ (__output__, __argmax__)
                      --
                      -- * __output__
                      --
                      -- * __argmax__
maxPoolWithArgmax' :: OpParams
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build targmax)
maxPoolWithArgmax' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build targmax)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build targmax))
-> Build OpDef -> (Tensor Build t, Tensor Build targmax)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MaxPoolWithArgmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Targmax" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ targmax -> DataType
forall a. TensorType a => a -> DataType
tensorType (targmax
forall a. HasCallStack => a
undefined :: targmax)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "argmax" type_attr: "Targmax" }
attr {
  name: "ksize" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "strides" type: "list(int)" has_minimum: true minimum: 4
}
attr {
  name: "Targmax"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "include_batch_in_index"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
maximum :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Word.Word16,
                                       Data.Word.Word8, Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor Build t -- ^ __z__
maximum :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
maximum = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
maximum' OpParams
forall a. a -> a
id
maximum' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Word.Word16,
                                        Data.Word.Word8, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
maximum' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
maximum' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Maximum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
mean :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tidx) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build t -- ^ __output__
mean :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
mean = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
mean' OpParams
forall a. a -> a
id
mean' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t,
                                  OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] tidx) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 tidx -- ^ __reduction_indices__
         -> Tensor Build t -- ^ __output__
mean' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
mean' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Mean"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
merge :: forall v'1 t . (TensorType t) => [Tensor v'1 t] -- ^ __inputs__
         -> (Tensor Build t, Tensor Build Data.Int.Int32)
         -- ^ (__output__, __value_index__)
         --
         -- * __output__
         --
         -- * __value_index__
merge :: [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
merge = OpParams -> [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
merge' OpParams
forall a. a -> a
id
merge' :: forall v'1 t . (TensorType t) => OpParams ->
          [Tensor v'1 t] -- ^ __inputs__
          -> (Tensor Build t, Tensor Build Data.Int.Int32)
          -- ^ (__output__, __value_index__)
          --
          -- * __output__
          --
          -- * __value_index__
merge' :: OpParams -> [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
merge' op'options :: OpParams
op'options inputs :: [Tensor v'1 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs)])] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Int32))
-> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Merge"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs) :: Int64
{-
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "value_index" type: DT_INT32 }
attr { name: "T" type: "type" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
mergeSummary :: [Tensor v'1 Data.ByteString.ByteString] -- ^ __inputs__
                -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
mergeSummary :: [Tensor v'1 ByteString] -> Tensor Build ByteString
mergeSummary = OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
mergeSummary' OpParams
forall a. a -> a
id
mergeSummary' :: OpParams ->
                 [Tensor v'1 Data.ByteString.ByteString] -- ^ __inputs__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
mergeSummary' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
mergeSummary' op'options :: OpParams
op'options
              inputs :: [Tensor v'1 ByteString]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ByteString]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 ByteString] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 ByteString]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MergeSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ByteString]
inputs) :: Int64
{-
input_arg { name: "inputs" type: DT_STRING number_attr: "N" }
output_arg { name: "summary" type: DT_STRING }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
mergeV2Checkpoints :: forall v'1 v'2 m' . (MonadBuild m') =>
                      Tensor v'1 Data.ByteString.ByteString -- ^ __checkpoint_prefixes__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __destination_prefix__
                      -> m' (ControlNode)
mergeV2Checkpoints :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
mergeV2Checkpoints = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
mergeV2Checkpoints' OpParams
forall a. a -> a
id
mergeV2Checkpoints' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 Data.ByteString.ByteString -- ^ __checkpoint_prefixes__
                       -> Tensor v'2 Data.ByteString.ByteString -- ^ __destination_prefix__
                       -> m' (ControlNode)
mergeV2Checkpoints' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
mergeV2Checkpoints' op'options :: OpParams
op'options checkpoint_prefixes :: Tensor v'1 ByteString
checkpoint_prefixes
                    destination_prefix :: Tensor v'2 ByteString
destination_prefix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
checkpoint_prefixes,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
destination_prefix]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MergeV2Checkpoints"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "checkpoint_prefixes" type: DT_STRING }
input_arg { name: "destination_prefix" type: DT_STRING }
attr {
  name: "delete_old_dirs" type: "bool" default_value { b: true }
}
-}
-- | 
mfcc :: Tensor v'1 Float -- ^ __spectrogram__
        -> Tensor v'2 Data.Int.Int32 -- ^ __sample_rate__
        -> Tensor Build Float -- ^ __output__
mfcc :: Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
mfcc = OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
mfcc' OpParams
forall a. a -> a
id
mfcc' :: OpParams ->
         Tensor v'1 Float -- ^ __spectrogram__
         -> Tensor v'2 Data.Int.Int32 -- ^ __sample_rate__
         -> Tensor Build Float -- ^ __output__
mfcc' :: OpParams
-> Tensor v'1 Float -> Tensor v'2 Int32 -> Tensor Build Float
mfcc' op'options :: OpParams
op'options spectrogram :: Tensor v'1 Float
spectrogram sample_rate :: Tensor v'2 Int32
sample_rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
spectrogram,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
sample_rate]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Mfcc"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "spectrogram" type: DT_FLOAT }
input_arg { name: "sample_rate" type: DT_INT32 }
output_arg { name: "output" type: DT_FLOAT }
attr {
  name: "upper_frequency_limit"
  type: "float"
  default_value { f: 4000.0 }
}
attr {
  name: "lower_frequency_limit"
  type: "float"
  default_value { f: 20.0 }
}
attr {
  name: "filterbank_channel_count"
  type: "int"
  default_value { i: 40 }
}
attr {
  name: "dct_coefficient_count" type: "int" default_value { i: 13 }
}
-}
-- | 
min :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t,
                                OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
       Tensor v'1 t -- ^ __input__
       -> Tensor v'2 tidx -- ^ __reduction_indices__
       -> Tensor Build t -- ^ __output__
min :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
min = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
min' OpParams
forall a. a -> a
id
min' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tidx) => OpParams ->
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build t -- ^ __output__
min' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
min' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Min"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
minimum :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Word.Word16,
                                       Data.Word.Word8, Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor Build t -- ^ __z__
minimum :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
minimum = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
minimum' OpParams
forall a. a -> a
id
minimum' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Word.Word16,
                                        Data.Word.Word8, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
minimum' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
minimum' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Minimum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
mirrorPad :: forall v'1 v'2 t tpaddings . (TensorType t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tpaddings) =>
             ByteString -- ^ __mode__
             -> Tensor v'1 t -- ^ __input__
             -> Tensor v'2 tpaddings -- ^ __paddings__
             -> Tensor Build t -- ^ __output__
mirrorPad :: ByteString
-> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
mirrorPad = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
mirrorPad' OpParams
forall a. a -> a
id
mirrorPad' :: forall v'1 v'2 t tpaddings . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tpaddings) =>
              OpParams ->
              ByteString -- ^ __mode__
              -> Tensor v'1 t -- ^ __input__
              -> Tensor v'2 tpaddings -- ^ __paddings__
              -> Tensor Build t -- ^ __output__
mirrorPad' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
mirrorPad' op'options :: OpParams
op'options mode :: ByteString
mode input :: Tensor v'1 t
input paddings :: Tensor v'2 tpaddings
paddings | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tpaddings
paddings]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MirrorPad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mode
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "mode"
  type: "string"
  allowed_values { list { s: "REFLECT" s: "SYMMETRIC" } }
}
-}
-- | 
mirrorPadGrad :: forall v'1 v'2 t tpaddings . (TensorType t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tpaddings) =>
                 ByteString -- ^ __mode__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 tpaddings -- ^ __paddings__
                 -> Tensor Build t -- ^ __output__
mirrorPadGrad :: ByteString
-> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
mirrorPadGrad = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
mirrorPadGrad' OpParams
forall a. a -> a
id
mirrorPadGrad' :: forall v'1 v'2 t tpaddings . (TensorType t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tpaddings) =>
                  OpParams ->
                  ByteString -- ^ __mode__
                  -> Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 tpaddings -- ^ __paddings__
                  -> Tensor Build t -- ^ __output__
mirrorPadGrad' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor Build t
mirrorPadGrad' op'options :: OpParams
op'options mode :: ByteString
mode input :: Tensor v'1 t
input paddings :: Tensor v'2 tpaddings
paddings | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tpaddings
paddings]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MirrorPadGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mode
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "mode"
  type: "string"
  allowed_values { list { s: "REFLECT" s: "SYMMETRIC" } }
}
-}
-- | 
mlirPassthroughOp :: forall v'1 tinputs toutputs . (TensorTypes tinputs,
                                                    TensorTypes toutputs) =>
                     ByteString -- ^ __mlir_module__
                     -> TensorList (v'1) tinputs -- ^ __inputs__
                     -> TensorList (Build) toutputs -- ^ __outputs__
mlirPassthroughOp :: ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
mlirPassthroughOp = OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
forall (v'1 :: * -> *) (tinputs :: [*]) (toutputs :: [*]).
(TensorTypes tinputs, TensorTypes toutputs) =>
OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
mlirPassthroughOp' OpParams
forall a. a -> a
id
mlirPassthroughOp' :: forall v'1 tinputs toutputs . (TensorTypes tinputs,
                                                     TensorTypes toutputs) =>
                      OpParams ->
                      ByteString -- ^ __mlir_module__
                      -> TensorList (v'1) tinputs -- ^ __inputs__
                      -> TensorList (Build) toutputs -- ^ __outputs__
mlirPassthroughOp' :: OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
mlirPassthroughOp' op'options :: OpParams
op'options mlir_module :: ByteString
mlir_module inputs :: TensorList v'1 tinputs
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build toutputs
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build toutputs)
-> Build OpDef -> TensorList Build toutputs
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tinputs -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tinputs
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MlirPassthroughOp"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tinputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tinputs
forall k (t :: k). Proxy t
Proxy :: Proxy tinputs)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutputs
forall k (t :: k). Proxy t
Proxy :: Proxy toutputs)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mlir_module" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mlir_module
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "Tinputs" }
output_arg { name: "outputs" type_list_attr: "Toutputs" }
attr { name: "mlir_module" type: "string" }
attr { name: "Tinputs" type: "list(type)" has_minimum: true }
attr { name: "Toutputs" type: "list(type)" has_minimum: true }
-}
-- | 
mod :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                   Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
mod :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mod = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mod' OpParams
forall a. a -> a
id
mod' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                    Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
mod' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mod' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Mod"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
-}
-- | 
modelDataset :: [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor Build Variant -- ^ __handle__
modelDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
modelDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
modelDataset' OpParams
forall a. a -> a
id
modelDataset' :: OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor Build Variant -- ^ __handle__
modelDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
modelDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ModelDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "algorithm" type: "int" default_value { i: 0 } }
attr { name: "cpu_budget" type: "int" default_value { i: 0 } }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
mul :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word8, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
mul :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mul = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mul' OpParams
forall a. a -> a
id
mul' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word8, Double,
                                    Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
mul' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mul' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Mul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
mulNoNan :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
mulNoNan :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mulNoNan = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mulNoNan' OpParams
forall a. a -> a
id
mulNoNan' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build t -- ^ __z__
mulNoNan' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
mulNoNan' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "MulNoNan"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
multiDeviceIterator :: forall m' . (MonadBuild m') =>
                       ByteString -- ^ __container__
                       -> [DataType] -- ^ __output_types__
                       -> ByteString -- ^ __shared_name__
                       -> m' (Tensor Value ResourceHandle) -- ^ __handle__
multiDeviceIterator :: ByteString
-> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIterator = OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
multiDeviceIterator' OpParams
forall a. a -> a
id
multiDeviceIterator' :: forall m' . (MonadBuild m') => OpParams ->
                        ByteString -- ^ __container__
                        -> [DataType] -- ^ __output_types__
                        -> ByteString -- ^ __shared_name__
                        -> m' (Tensor Value ResourceHandle) -- ^ __handle__
multiDeviceIterator' :: OpParams
-> ByteString
-> [DataType]
-> ByteString
-> m' (Tensor Value ResourceHandle)
multiDeviceIterator' op'options :: OpParams
op'options container :: ByteString
container output_types :: [DataType]
output_types
                     shared_name :: ByteString
shared_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MultiDeviceIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "container" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
container
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "devices" type: "list(string)" has_minimum: true minimum: 1
}
attr { name: "shared_name" type: "string" }
attr { name: "container" type: "string" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
multiDeviceIteratorFromStringHandle :: forall v'1 m' . (MonadBuild m') =>
                                       Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                                       -> m' (Tensor Value ResourceHandle) -- ^ __multi_device_iterator__
multiDeviceIteratorFromStringHandle :: Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorFromStringHandle = OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorFromStringHandle' OpParams
forall a. a -> a
id
multiDeviceIteratorFromStringHandle' :: forall v'1 m' . (MonadBuild m') =>
                                        OpParams ->
                                        Tensor v'1 Data.ByteString.ByteString -- ^ __string_handle__
                                        -> m' (Tensor Value ResourceHandle) -- ^ __multi_device_iterator__
multiDeviceIteratorFromStringHandle' :: OpParams
-> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
multiDeviceIteratorFromStringHandle' op'options :: OpParams
op'options
                                     string_handle :: Tensor v'1 ByteString
string_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
string_handle]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MultiDeviceIteratorFromStringHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "string_handle" type: DT_STRING }
output_arg { name: "multi_device_iterator" type: DT_RESOURCE }
attr {
  name: "output_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
-}
-- | 
multiDeviceIteratorGetNextFromShard :: forall v'1 v'2 v'3 output_types
                                       m' . (MonadBuild m',
                                             TensorTypes output_types) =>
                                       Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                                       -> Tensor v'2 Data.Int.Int32 -- ^ __shard_num__
                                       -> Tensor v'3 Data.Int.Int64 -- ^ __incarnation_id__
                                       -> m' (TensorList (Value) output_types) -- ^ __components__
multiDeviceIteratorGetNextFromShard :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int64
-> m' (TensorList Value output_types)
multiDeviceIteratorGetNextFromShard = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int64
-> m' (TensorList Value output_types)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (output_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes output_types) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int64
-> m' (TensorList Value output_types)
multiDeviceIteratorGetNextFromShard' OpParams
forall a. a -> a
id
multiDeviceIteratorGetNextFromShard' :: forall v'1 v'2 v'3 output_types
                                        m' . (MonadBuild m',
                                              TensorTypes output_types) =>
                                        OpParams ->
                                        Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                                        -> Tensor v'2 Data.Int.Int32 -- ^ __shard_num__
                                        -> Tensor v'3 Data.Int.Int64 -- ^ __incarnation_id__
                                        -> m' (TensorList (Value) output_types) -- ^ __components__
multiDeviceIteratorGetNextFromShard' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Int64
-> m' (TensorList Value output_types)
multiDeviceIteratorGetNextFromShard' op'options :: OpParams
op'options multi_device_iterator :: Tensor v'1 ResourceHandle
multi_device_iterator shard_num :: Tensor v'2 Int32
shard_num
                                     incarnation_id :: Tensor v'3 Int64
incarnation_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value output_types)
 -> m' (TensorList Value output_types))
-> Build (TensorList Value output_types)
-> m' (TensorList Value output_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
multi_device_iterator,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
shard_num,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
incarnation_id]
        [Int64] -> OpDef -> Build (TensorList Value output_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MultiDeviceIteratorGetNextFromShard"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "multi_device_iterator" type: DT_RESOURCE }
input_arg { name: "shard_num" type: DT_INT32 }
input_arg { name: "incarnation_id" type: DT_INT64 }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
multiDeviceIteratorInit :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                           Tensor v'1 Variant -- ^ __dataset__
                           -> Tensor v'2 ResourceHandle -- ^ __multi_device_iterator__
                           -> Tensor v'3 Data.Int.Int64 -- ^ __max_buffer_size__
                           -> m' (Tensor Value Data.Int.Int64) -- ^ __incarnation_id__
multiDeviceIteratorInit :: Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
multiDeviceIteratorInit = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
multiDeviceIteratorInit' OpParams
forall a. a -> a
id
multiDeviceIteratorInit' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                            OpParams ->
                            Tensor v'1 Variant -- ^ __dataset__
                            -> Tensor v'2 ResourceHandle -- ^ __multi_device_iterator__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __max_buffer_size__
                            -> m' (Tensor Value Data.Int.Int64) -- ^ __incarnation_id__
multiDeviceIteratorInit' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value Int64)
multiDeviceIteratorInit' op'options :: OpParams
op'options dataset :: Tensor v'1 Variant
dataset multi_device_iterator :: Tensor v'2 ResourceHandle
multi_device_iterator
                         max_buffer_size :: Tensor v'3 Int64
max_buffer_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
multi_device_iterator,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
max_buffer_size]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MultiDeviceIteratorInit"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dataset" type: DT_VARIANT }
input_arg { name: "multi_device_iterator" type: DT_RESOURCE }
input_arg { name: "max_buffer_size" type: DT_INT64 }
output_arg { name: "incarnation_id" type: DT_INT64 }
-}
-- | 
multiDeviceIteratorToStringHandle :: forall v'1 m' . (MonadBuild m') =>
                                     Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                                     -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __string_handle__
multiDeviceIteratorToStringHandle :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
multiDeviceIteratorToStringHandle = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
multiDeviceIteratorToStringHandle' OpParams
forall a. a -> a
id
multiDeviceIteratorToStringHandle' :: forall v'1 m' . (MonadBuild m') =>
                                      OpParams ->
                                      Tensor v'1 ResourceHandle -- ^ __multi_device_iterator__
                                      -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __string_handle__
multiDeviceIteratorToStringHandle' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
multiDeviceIteratorToStringHandle' op'options :: OpParams
op'options
                                   multi_device_iterator :: Tensor v'1 ResourceHandle
multi_device_iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
multi_device_iterator]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MultiDeviceIteratorToStringHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "multi_device_iterator" type: DT_RESOURCE }
output_arg { name: "string_handle" type: DT_STRING }
-}
-- | 
multinomial :: forall v'1 v'2 t output_dtype m' . (MonadBuild m',
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double, Float] t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] output_dtype) =>
               Tensor v'1 t -- ^ __logits__
               -> Tensor v'2 Data.Int.Int32 -- ^ __num_samples__
               -> m' (Tensor Value output_dtype) -- ^ __output__
multinomial :: Tensor v'1 t -> Tensor v'2 Int32 -> m' (Tensor Value output_dtype)
multinomial = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> m' (Tensor Value output_dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) t output_dtype
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] output_dtype) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> m' (Tensor Value output_dtype)
multinomial' OpParams
forall a. a -> a
id
multinomial' :: forall v'1 v'2 t output_dtype m' . (MonadBuild m',
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word16,
                                                            Data.Word.Word32,
                                                            Data.Word.Word64,
                                                            Data.Word.Word8,
                                                            Double, Float] t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] output_dtype) =>
                OpParams ->
                Tensor v'1 t -- ^ __logits__
                -> Tensor v'2 Data.Int.Int32 -- ^ __num_samples__
                -> m' (Tensor Value output_dtype) -- ^ __output__
multinomial' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> m' (Tensor Value output_dtype)
multinomial' op'options :: OpParams
op'options logits :: Tensor v'1 t
logits num_samples :: Tensor v'2 Int32
num_samples | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value output_dtype) -> m' (Tensor Value output_dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value output_dtype)
 -> m' (Tensor Value output_dtype))
-> Build (Tensor Value output_dtype)
-> m' (Tensor Value output_dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
logits,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_samples]
        [Int64] -> OpDef -> Build (Tensor Value output_dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Multinomial"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_dtype
forall a. HasCallStack => a
undefined :: output_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "logits" type_attr: "T" }
input_arg { name: "num_samples" type: DT_INT32 }
output_arg { name: "output" type_attr: "output_dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "output_dtype"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
mutableDenseHashTable :: forall v'1 key_dtype m' . (MonadBuild m',
                                                    TensorType key_dtype) =>
                         DataType -- ^ __value_dtype__
                         -> Tensor v'1 key_dtype -- ^ __empty_key__
                         -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableDenseHashTable :: DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
mutableDenseHashTable = OpParams
-> DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
forall (v'1 :: * -> *) key_dtype (m' :: * -> *).
(MonadBuild m', TensorType key_dtype) =>
OpParams
-> DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
mutableDenseHashTable' OpParams
forall a. a -> a
id
mutableDenseHashTable' :: forall v'1 key_dtype m' . (MonadBuild m',
                                                     TensorType key_dtype) =>
                          OpParams ->
                          DataType -- ^ __value_dtype__
                          -> Tensor v'1 key_dtype -- ^ __empty_key__
                          -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableDenseHashTable' :: OpParams
-> DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
mutableDenseHashTable' op'options :: OpParams
op'options value_dtype :: DataType
value_dtype empty_key :: Tensor v'1 key_dtype
empty_key | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 key_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 key_dtype
empty_key]
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableDenseHashTable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ key_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (key_dtype
forall a. HasCallStack => a
undefined :: key_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "empty_key" type_attr: "key_dtype" }
output_arg { name: "table_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
attr {
  name: "value_shape" type: "shape" default_value { shape { } }
}
attr {
  name: "initial_num_buckets" type: "int" default_value { i: 131072 }
}
attr {
  name: "max_load_factor" type: "float" default_value { f: 0.8 }
}
-}
-- | 
mutableDenseHashTableV2 :: forall v'1 v'2 key_dtype m' . (MonadBuild m',
                                                          TensorType key_dtype) =>
                           DataType -- ^ __value_dtype__
                           -> Tensor v'1 key_dtype -- ^ __empty_key__
                           -> Tensor v'2 key_dtype -- ^ __deleted_key__
                           -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableDenseHashTableV2 :: DataType
-> Tensor v'1 key_dtype
-> Tensor v'2 key_dtype
-> m' (Tensor Value ResourceHandle)
mutableDenseHashTableV2 = OpParams
-> DataType
-> Tensor v'1 key_dtype
-> Tensor v'2 key_dtype
-> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) (v'2 :: * -> *) key_dtype (m' :: * -> *).
(MonadBuild m', TensorType key_dtype) =>
OpParams
-> DataType
-> Tensor v'1 key_dtype
-> Tensor v'2 key_dtype
-> m' (Tensor Value ResourceHandle)
mutableDenseHashTableV2' OpParams
forall a. a -> a
id
mutableDenseHashTableV2' :: forall v'1 v'2 key_dtype m' . (MonadBuild m',
                                                           TensorType key_dtype) =>
                            OpParams ->
                            DataType -- ^ __value_dtype__
                            -> Tensor v'1 key_dtype -- ^ __empty_key__
                            -> Tensor v'2 key_dtype -- ^ __deleted_key__
                            -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableDenseHashTableV2' :: OpParams
-> DataType
-> Tensor v'1 key_dtype
-> Tensor v'2 key_dtype
-> m' (Tensor Value ResourceHandle)
mutableDenseHashTableV2' op'options :: OpParams
op'options value_dtype :: DataType
value_dtype empty_key :: Tensor v'1 key_dtype
empty_key
                         deleted_key :: Tensor v'2 key_dtype
deleted_key | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 key_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 key_dtype
empty_key,
                                                             Tensor v'2 key_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 key_dtype
deleted_key]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableDenseHashTableV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ key_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (key_dtype
forall a. HasCallStack => a
undefined :: key_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "empty_key" type_attr: "key_dtype" }
input_arg { name: "deleted_key" type_attr: "key_dtype" }
output_arg { name: "table_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
attr {
  name: "value_shape" type: "shape" default_value { shape { } }
}
attr {
  name: "initial_num_buckets" type: "int" default_value { i: 131072 }
}
attr {
  name: "max_load_factor" type: "float" default_value { f: 0.8 }
}
-}
-- | 
mutableHashTable :: forall m' . (MonadBuild m') => DataType -- ^ __key_dtype__
                    -> DataType -- ^ __value_dtype__
                    -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableHashTable :: DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTable = OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTable' OpParams
forall a. a -> a
id
mutableHashTable' :: forall m' . (MonadBuild m') => OpParams ->
                     DataType -- ^ __key_dtype__
                     -> DataType -- ^ __value_dtype__
                     -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableHashTable' :: OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTable' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableHashTable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
-}
-- | 
mutableHashTableOfTensors :: forall m' . (MonadBuild m') =>
                             DataType -- ^ __key_dtype__
                             -> DataType -- ^ __value_dtype__
                             -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableHashTableOfTensors :: DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTableOfTensors = OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTableOfTensors' OpParams
forall a. a -> a
id
mutableHashTableOfTensors' :: forall m' . (MonadBuild m') => OpParams ->
                              DataType -- ^ __key_dtype__
                              -> DataType -- ^ __value_dtype__
                              -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __table_handle__
mutableHashTableOfTensors' :: OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
mutableHashTableOfTensors' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableHashTableOfTensors"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
attr {
  name: "value_shape" type: "shape" default_value { shape { } }
}
-}
-- | 
mutableHashTableOfTensorsV2 :: forall m' . (MonadBuild m') =>
                               DataType -- ^ __key_dtype__
                               -> DataType -- ^ __value_dtype__
                               -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableHashTableOfTensorsV2 :: DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableOfTensorsV2 = OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableOfTensorsV2' OpParams
forall a. a -> a
id
mutableHashTableOfTensorsV2' :: forall m' . (MonadBuild m') => OpParams ->
                                DataType -- ^ __key_dtype__
                                -> DataType -- ^ __value_dtype__
                                -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableHashTableOfTensorsV2' :: OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableOfTensorsV2' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype
                             value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableHashTableOfTensorsV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
attr {
  name: "value_shape" type: "shape" default_value { shape { } }
}
-}
-- | 
mutableHashTableV2 :: forall m' . (MonadBuild m') => DataType -- ^ __key_dtype__
                      -> DataType -- ^ __value_dtype__
                      -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableHashTableV2 :: DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableV2 = OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableV2' OpParams
forall a. a -> a
id
mutableHashTableV2' :: forall m' . (MonadBuild m') => OpParams ->
                       DataType -- ^ __key_dtype__
                       -> DataType -- ^ __value_dtype__
                       -> m' (Tensor Value ResourceHandle) -- ^ __table_handle__
mutableHashTableV2' :: OpParams
-> DataType -> DataType -> m' (Tensor Value ResourceHandle)
mutableHashTableV2' op'options :: OpParams
op'options key_dtype :: DataType
key_dtype value_dtype :: DataType
value_dtype | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutableHashTableV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
key_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "value_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
value_dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "table_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "use_node_name_sharing"
  type: "bool"
  default_value { b: false }
}
attr { name: "key_dtype" type: "type" }
attr { name: "value_dtype" type: "type" }
-}
-- | 
mutexLock :: forall v'1 m' . (MonadBuild m') =>
             Tensor v'1 ResourceHandle -- ^ __mutex__
             -> m' (Tensor Value Variant) -- ^ __mutex_lock__
mutexLock :: Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
mutexLock = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
mutexLock' OpParams
forall a. a -> a
id
mutexLock' :: forall v'1 m' . (MonadBuild m') => OpParams ->
              Tensor v'1 ResourceHandle -- ^ __mutex__
              -> m' (Tensor Value Variant) -- ^ __mutex_lock__
mutexLock' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
mutexLock' op'options :: OpParams
op'options mutex :: Tensor v'1 ResourceHandle
mutex | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
mutex]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutexLock"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "mutex" type: DT_RESOURCE }
output_arg { name: "mutex_lock" type: DT_VARIANT }
-}
-- | 
mutexV2 :: forall m' . (MonadBuild m') =>
           m' (Tensor Value ResourceHandle) -- ^ __resource__
mutexV2 :: m' (Tensor Value ResourceHandle)
mutexV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
mutexV2' OpParams
forall a. a -> a
id
mutexV2' :: forall m' . (MonadBuild m') => OpParams ->
            m' (Tensor Value ResourceHandle) -- ^ __resource__
mutexV2' :: OpParams -> m' (Tensor Value ResourceHandle)
mutexV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "MutexV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resource" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
ncclAllReduce :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Word.Word16,
                                                           Double, Float] t) =>
                 Data.Int.Int64 -- ^ __num_devices__
                 -> ByteString -- ^ __reduction__
                 -> ByteString -- ^ __shared_name__
                 -> Tensor v'1 t -- ^ __input__
                 -> m' (Tensor Value t) -- ^ __data__
ncclAllReduce :: Int64
-> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
ncclAllReduce = OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
ncclAllReduce' OpParams
forall a. a -> a
id
ncclAllReduce' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Word.Word16,
                                                            Double, Float] t) =>
                  OpParams ->
                  Data.Int.Int64 -- ^ __num_devices__
                  -> ByteString -- ^ __reduction__
                  -> ByteString -- ^ __shared_name__
                  -> Tensor v'1 t -- ^ __input__
                  -> m' (Tensor Value t) -- ^ __data__
ncclAllReduce' :: OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
ncclAllReduce' op'options :: OpParams
op'options num_devices :: Int64
num_devices reduction :: ByteString
reduction shared_name :: ByteString
shared_name
               input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NcclAllReduce"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "reduction" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
reduction
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "data" type_attr: "T" }
attr {
  name: "reduction"
  type: "string"
  allowed_values { list { s: "min" s: "max" s: "prod" s: "sum" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "num_devices" type: "int" }
attr { name: "shared_name" type: "string" }
-}
-- | 
ncclBroadcast :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Word.Word16,
                                                           Double, Float] t) =>
                 Shape -- ^ __shape__
                 -> Tensor v'1 t -- ^ __input__
                 -> m' (Tensor Value t) -- ^ __output__
ncclBroadcast :: Shape -> Tensor v'1 t -> m' (Tensor Value t)
ncclBroadcast = OpParams -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
ncclBroadcast' OpParams
forall a. a -> a
id
ncclBroadcast' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Word.Word16,
                                                            Double, Float] t) =>
                  OpParams ->
                  Shape -- ^ __shape__
                  -> Tensor v'1 t -- ^ __input__
                  -> m' (Tensor Value t) -- ^ __output__
ncclBroadcast' :: OpParams -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
ncclBroadcast' op'options :: OpParams
op'options shape :: Shape
shape input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NcclBroadcast"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "shape" type: "shape" }
-}
-- | 
ncclReduce :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Word.Word16,
                                                        Double, Float] t) =>
              ByteString -- ^ __reduction__
              -> [Tensor v'1 t] -- ^ __input__
              -> m' (Tensor Value t) -- ^ __data__
ncclReduce :: ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
ncclReduce = OpParams -> ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams -> ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
ncclReduce' OpParams
forall a. a -> a
id
ncclReduce' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Word.Word16,
                                                         Double, Float] t) =>
               OpParams ->
               ByteString -- ^ __reduction__
               -> [Tensor v'1 t] -- ^ __input__
               -> m' (Tensor Value t) -- ^ __data__
ncclReduce' :: OpParams -> ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
ncclReduce' op'options :: OpParams
op'options reduction :: ByteString
reduction
            input :: [Tensor v'1 t]
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_devices", [("input", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input)])] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NcclReduce"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "reduction" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
reduction
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_devices :: Int64
num_devices = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input) :: Int64
{-
input_arg {
  name: "input" type_attr: "T" number_attr: "num_devices"
}
output_arg { name: "data" type_attr: "T" }
attr {
  name: "reduction"
  type: "string"
  allowed_values { list { s: "min" s: "max" s: "prod" s: "sum" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "num_devices" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
ndtri :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
ndtri :: Tensor v'1 t -> Tensor Build t
ndtri = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
ndtri' OpParams
forall a. a -> a
id
ndtri' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
ndtri' :: OpParams -> Tensor v'1 t -> Tensor Build t
ndtri' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Ndtri"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
nearestNeighbors :: Tensor v'1 Float -- ^ __points__
                    -> Tensor v'2 Float -- ^ __centers__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __k__
                    -> (Tensor Build Data.Int.Int64, Tensor Build Float)
                    -- ^ (__nearest_center_indices__, __nearest_center_distances__)
                    --
                    -- * __nearest_center_indices__
                    --
                    -- * __nearest_center_distances__
nearestNeighbors :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Float)
nearestNeighbors = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Float)
nearestNeighbors' OpParams
forall a. a -> a
id
nearestNeighbors' :: OpParams ->
                     Tensor v'1 Float -- ^ __points__
                     -> Tensor v'2 Float -- ^ __centers__
                     -> Tensor v'3 Data.Int.Int64 -- ^ __k__
                     -> (Tensor Build Data.Int.Int64, Tensor Build Float)
                     -- ^ (__nearest_center_indices__, __nearest_center_distances__)
                     --
                     -- * __nearest_center_indices__
                     --
                     -- * __nearest_center_distances__
nearestNeighbors' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Float)
nearestNeighbors' op'options :: OpParams
op'options points :: Tensor v'1 Float
points centers :: Tensor v'2 Float
centers k :: Tensor v'3 Int64
k | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build Float))
-> Build OpDef -> (Tensor Build Int64, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
points,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
centers,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
k]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NearestNeighbors"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "points" type: DT_FLOAT }
input_arg { name: "centers" type: DT_FLOAT }
input_arg { name: "k" type: DT_INT64 }
output_arg { name: "nearest_center_indices" type: DT_INT64 }
output_arg { name: "nearest_center_distances" type: DT_FLOAT }
-}
-- | 
neg :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
neg :: Tensor v'1 t -> Tensor Build t
neg = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
neg' OpParams
forall a. a -> a
id
neg' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
neg' :: OpParams -> Tensor v'1 t -> Tensor Build t
neg' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Neg"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
negTrain :: forall v'3 v'4 v'5 m' . (MonadBuild m') =>
            Data.Int.Int64 -- ^ __num_negative_samples__
            -> Tensor Ref Float -- ^ __w_in__
            -> Tensor Ref Float -- ^ __w_out__
            -> Tensor v'3 Data.Int.Int32 -- ^ __examples__
            -> Tensor v'4 Data.Int.Int32 -- ^ __labels__
            -> Tensor v'5 Float -- ^ __lr__
            -> m' (ControlNode)
negTrain :: Int64
-> Tensor Ref Float
-> Tensor Ref Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> m' ControlNode
negTrain = OpParams
-> Int64
-> Tensor Ref Float
-> Tensor Ref Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> m' ControlNode
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor Ref Float
-> Tensor Ref Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> m' ControlNode
negTrain' OpParams
forall a. a -> a
id
negTrain' :: forall v'3 v'4 v'5 m' . (MonadBuild m') => OpParams ->
             Data.Int.Int64 -- ^ __num_negative_samples__
             -> Tensor Ref Float -- ^ __w_in__
             -> Tensor Ref Float -- ^ __w_out__
             -> Tensor v'3 Data.Int.Int32 -- ^ __examples__
             -> Tensor v'4 Data.Int.Int32 -- ^ __labels__
             -> Tensor v'5 Float -- ^ __lr__
             -> m' (ControlNode)
negTrain' :: OpParams
-> Int64
-> Tensor Ref Float
-> Tensor Ref Float
-> Tensor v'3 Int32
-> Tensor v'4 Int32
-> Tensor v'5 Float
-> m' ControlNode
negTrain' op'options :: OpParams
op'options num_negative_samples :: Int64
num_negative_samples w_in :: Tensor Ref Float
w_in w_out :: Tensor Ref Float
w_out examples :: Tensor v'3 Int32
examples labels :: Tensor v'4 Int32
labels
          lr :: Tensor v'5 Float
lr | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref Float
w_in,
                                                             Tensor Ref Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref Float
w_out,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
examples,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
labels,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
lr]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NegTrain"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_negative_samples" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_negative_samples
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "w_in" type: DT_FLOAT is_ref: true }
input_arg { name: "w_out" type: DT_FLOAT is_ref: true }
input_arg { name: "examples" type: DT_INT32 }
input_arg { name: "labels" type: DT_INT32 }
input_arg { name: "lr" type: DT_FLOAT }
attr { name: "vocab_count" type: "list(int)" }
attr { name: "num_negative_samples" type: "int" }
-}
-- | 
nextAfter :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
             Tensor v'1 t -- ^ __x1__
             -> Tensor v'2 t -- ^ __x2__
             -> Tensor Build t -- ^ __output__
nextAfter :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
nextAfter = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
nextAfter' OpParams
forall a. a -> a
id
nextAfter' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
              Tensor v'1 t -- ^ __x1__
              -> Tensor v'2 t -- ^ __x2__
              -> Tensor Build t -- ^ __output__
nextAfter' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
nextAfter' op'options :: OpParams
op'options x1 :: Tensor v'1 t
x1 x2 :: Tensor v'2 t
x2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x1,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NextAfter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x1" type_attr: "T" }
input_arg { name: "x2" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_DOUBLE type: DT_FLOAT } }
}
-}
-- | 
nextIteration :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __data__
                 -> Tensor Build t -- ^ __output__
nextIteration :: Tensor v'1 t -> Tensor Build t
nextIteration = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
nextIteration' OpParams
forall a. a -> a
id
nextIteration' :: forall v'1 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __data__
                  -> Tensor Build t -- ^ __output__
nextIteration' :: OpParams -> Tensor v'1 t -> Tensor Build t
nextIteration' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data']
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NextIteration"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
noOp :: forall m' . (MonadBuild m') => m' (ControlNode)
noOp :: m' ControlNode
noOp = OpParams -> m' ControlNode
forall (m' :: * -> *). MonadBuild m' => OpParams -> m' ControlNode
noOp' OpParams
forall a. a -> a
id
noOp' :: forall m' . (MonadBuild m') => OpParams ->
         m' (ControlNode)
noOp' :: OpParams -> m' ControlNode
noOp' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NoOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-

-}
-- | 
nonDeterministicInts :: forall v'1 dtype shape_dtype m' . (MonadBuild m',
                                                           TensorType dtype,
                                                           TensorType shape_dtype) =>
                        Tensor v'1 shape_dtype -- ^ __shape__
                        -> m' (Tensor Value dtype) -- ^ __output__
nonDeterministicInts :: Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
nonDeterministicInts = OpParams -> Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams -> Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
nonDeterministicInts' OpParams
forall a. a -> a
id
nonDeterministicInts' :: forall v'1 dtype shape_dtype m' . (MonadBuild m',
                                                            TensorType dtype,
                                                            TensorType shape_dtype) =>
                         OpParams ->
                         Tensor v'1 shape_dtype -- ^ __shape__
                         -> m' (Tensor Value dtype) -- ^ __output__
nonDeterministicInts' :: OpParams -> Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
nonDeterministicInts' op'options :: OpParams
op'options shape :: Tensor v'1 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "NonDeterministicInts"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_INT64 }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
nonMaxSuppression :: Tensor v'1 Float -- ^ __boxes__
                     -> Tensor v'2 Float -- ^ __scores__
                     -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                     -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppression :: Tensor v'1 Float
-> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build Int32
nonMaxSuppression = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build Int32
nonMaxSuppression' OpParams
forall a. a -> a
id
nonMaxSuppression' :: OpParams ->
                      Tensor v'1 Float -- ^ __boxes__
                      -> Tensor v'2 Float -- ^ __scores__
                      -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                      -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppression' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor Build Int32
nonMaxSuppression' op'options :: OpParams
op'options boxes :: Tensor v'1 Float
boxes scores :: Tensor v'2 Float
scores max_output_size :: Tensor v'3 Int32
max_output_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
boxes,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppression"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type: DT_FLOAT }
input_arg { name: "scores" type: DT_FLOAT }
input_arg { name: "max_output_size" type: DT_INT32 }
output_arg { name: "selected_indices" type: DT_INT32 }
attr {
  name: "iou_threshold" type: "float" default_value { f: 0.5 }
}
-}
-- | 
nonMaxSuppressionV2 :: forall v'1 v'2 v'3 v'4 t
                       t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                      OneOf '[Data.Word.Word16,
                                              Float] t_threshold) =>
                       Tensor v'1 t -- ^ __boxes__
                       -> Tensor v'2 t -- ^ __scores__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                       -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                       -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionV2 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t t_threshold.
(OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV2' OpParams
forall a. a -> a
id
nonMaxSuppressionV2' :: forall v'1 v'2 v'3 v'4 t
                        t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                       OneOf '[Data.Word.Word16,
                                               Float] t_threshold) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __boxes__
                        -> Tensor v'2 t -- ^ __scores__
                        -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                        -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                        -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV2' op'options :: OpParams
op'options boxes :: Tensor v'1 t
boxes scores :: Tensor v'2 t
scores max_output_size :: Tensor v'3 Int32
max_output_size
                     iou_threshold :: Tensor v'4 t_threshold
iou_threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
boxes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size,
                                                             Tensor v'4 t_threshold -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t_threshold
iou_threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppressionV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T_threshold" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t_threshold -> DataType
forall a. TensorType a => a -> DataType
tensorType (t_threshold
forall a. HasCallStack => a
undefined :: t_threshold)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type_attr: "T" }
input_arg { name: "scores" type_attr: "T" }
input_arg { name: "max_output_size" type: DT_INT32 }
input_arg { name: "iou_threshold" type_attr: "T_threshold" }
output_arg { name: "selected_indices" type: DT_INT32 }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "T_threshold"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
nonMaxSuppressionV3 :: forall v'1 v'2 v'3 v'4 v'5 t
                       t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                      OneOf '[Data.Word.Word16,
                                              Float] t_threshold) =>
                       Tensor v'1 t -- ^ __boxes__
                       -> Tensor v'2 t -- ^ __scores__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                       -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                       -> Tensor v'5 t_threshold -- ^ __score_threshold__
                       -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionV3 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t t_threshold.
(OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV3' OpParams
forall a. a -> a
id
nonMaxSuppressionV3' :: forall v'1 v'2 v'3 v'4 v'5 t
                        t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                       OneOf '[Data.Word.Word16,
                                               Float] t_threshold) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __boxes__
                        -> Tensor v'2 t -- ^ __scores__
                        -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                        -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                        -> Tensor v'5 t_threshold -- ^ __score_threshold__
                        -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> Tensor Build Int32
nonMaxSuppressionV3' op'options :: OpParams
op'options boxes :: Tensor v'1 t
boxes scores :: Tensor v'2 t
scores max_output_size :: Tensor v'3 Int32
max_output_size iou_threshold :: Tensor v'4 t_threshold
iou_threshold
                     score_threshold :: Tensor v'5 t_threshold
score_threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
boxes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size,
                                                             Tensor v'4 t_threshold -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t_threshold
iou_threshold,
                                                             Tensor v'5 t_threshold -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t_threshold
score_threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppressionV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T_threshold" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t_threshold -> DataType
forall a. TensorType a => a -> DataType
tensorType (t_threshold
forall a. HasCallStack => a
undefined :: t_threshold)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type_attr: "T" }
input_arg { name: "scores" type_attr: "T" }
input_arg { name: "max_output_size" type: DT_INT32 }
input_arg { name: "iou_threshold" type_attr: "T_threshold" }
input_arg { name: "score_threshold" type_attr: "T_threshold" }
output_arg { name: "selected_indices" type: DT_INT32 }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "T_threshold"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
-}
-- | 
nonMaxSuppressionV4 :: forall v'1 v'2 v'3 v'4 v'5 t
                       t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                      OneOf '[Data.Word.Word16,
                                              Float] t_threshold) =>
                       Tensor v'1 t -- ^ __boxes__
                       -> Tensor v'2 t -- ^ __scores__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                       -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                       -> Tensor v'5 t_threshold -- ^ __score_threshold__
                       -> (Tensor Build Data.Int.Int32,
                           Tensor Build Data.Int.Int32)
                       -- ^ (__selected_indices__, __valid_outputs__)
                       --
                       -- * __selected_indices__
                       --
                       -- * __valid_outputs__
nonMaxSuppressionV4 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> (Tensor Build Int32, Tensor Build Int32)
nonMaxSuppressionV4 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> (Tensor Build Int32, Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t t_threshold.
(OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> (Tensor Build Int32, Tensor Build Int32)
nonMaxSuppressionV4' OpParams
forall a. a -> a
id
nonMaxSuppressionV4' :: forall v'1 v'2 v'3 v'4 v'5 t
                        t_threshold . (OneOf '[Data.Word.Word16, Float] t,
                                       OneOf '[Data.Word.Word16,
                                               Float] t_threshold) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __boxes__
                        -> Tensor v'2 t -- ^ __scores__
                        -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                        -> Tensor v'4 t_threshold -- ^ __iou_threshold__
                        -> Tensor v'5 t_threshold -- ^ __score_threshold__
                        -> (Tensor Build Data.Int.Int32,
                            Tensor Build Data.Int.Int32)
                        -- ^ (__selected_indices__, __valid_outputs__)
                        --
                        -- * __selected_indices__
                        --
                        -- * __valid_outputs__
nonMaxSuppressionV4' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t_threshold
-> Tensor v'5 t_threshold
-> (Tensor Build Int32, Tensor Build Int32)
nonMaxSuppressionV4' op'options :: OpParams
op'options boxes :: Tensor v'1 t
boxes scores :: Tensor v'2 t
scores max_output_size :: Tensor v'3 Int32
max_output_size iou_threshold :: Tensor v'4 t_threshold
iou_threshold
                     score_threshold :: Tensor v'5 t_threshold
score_threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int32, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int32, Tensor Build Int32))
-> Build OpDef -> (Tensor Build Int32, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
boxes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size,
                                                             Tensor v'4 t_threshold -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t_threshold
iou_threshold,
                                                             Tensor v'5 t_threshold -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t_threshold
score_threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppressionV4"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T_threshold" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t_threshold -> DataType
forall a. TensorType a => a -> DataType
tensorType (t_threshold
forall a. HasCallStack => a
undefined :: t_threshold)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type_attr: "T" }
input_arg { name: "scores" type_attr: "T" }
input_arg { name: "max_output_size" type: DT_INT32 }
input_arg { name: "iou_threshold" type_attr: "T_threshold" }
input_arg { name: "score_threshold" type_attr: "T_threshold" }
output_arg { name: "selected_indices" type: DT_INT32 }
output_arg { name: "valid_outputs" type: DT_INT32 }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "T_threshold"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "pad_to_max_output_size"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
nonMaxSuppressionV5 :: forall v'1 v'2 v'3 v'4 v'5 v'6
                       t . (OneOf '[Data.Word.Word16, Float] t) =>
                       Tensor v'1 t -- ^ __boxes__
                       -> Tensor v'2 t -- ^ __scores__
                       -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                       -> Tensor v'4 t -- ^ __iou_threshold__
                       -> Tensor v'5 t -- ^ __score_threshold__
                       -> Tensor v'6 t -- ^ __soft_nms_sigma__
                       -> (Tensor Build Data.Int.Int32, Tensor Build t,
                           Tensor Build Data.Int.Int32)
                       -- ^ (__selected_indices__, __selected_scores__, __valid_outputs__)
                       --
                       -- * __selected_indices__
                       --
                       -- * __selected_scores__
                       --
                       -- * __valid_outputs__
nonMaxSuppressionV5 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
nonMaxSuppressionV5 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
OneOf '[Word16, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
nonMaxSuppressionV5' OpParams
forall a. a -> a
id
nonMaxSuppressionV5' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                        t . (OneOf '[Data.Word.Word16, Float] t) => OpParams ->
                        Tensor v'1 t -- ^ __boxes__
                        -> Tensor v'2 t -- ^ __scores__
                        -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                        -> Tensor v'4 t -- ^ __iou_threshold__
                        -> Tensor v'5 t -- ^ __score_threshold__
                        -> Tensor v'6 t -- ^ __soft_nms_sigma__
                        -> (Tensor Build Data.Int.Int32, Tensor Build t,
                            Tensor Build Data.Int.Int32)
                        -- ^ (__selected_indices__, __selected_scores__, __valid_outputs__)
                        --
                        -- * __selected_indices__
                        --
                        -- * __selected_scores__
                        --
                        -- * __valid_outputs__
nonMaxSuppressionV5' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Int32
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
nonMaxSuppressionV5' op'options :: OpParams
op'options boxes :: Tensor v'1 t
boxes scores :: Tensor v'2 t
scores max_output_size :: Tensor v'3 Int32
max_output_size iou_threshold :: Tensor v'4 t
iou_threshold
                     score_threshold :: Tensor v'5 t
score_threshold soft_nms_sigma :: Tensor v'6 t
soft_nms_sigma | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int32, Tensor Build t, Tensor Build Int32))
-> Build OpDef
-> (Tensor Build Int32, Tensor Build t, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
boxes,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
iou_threshold,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
score_threshold,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
soft_nms_sigma]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppressionV5"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "boxes" type_attr: "T" }
input_arg { name: "scores" type_attr: "T" }
input_arg { name: "max_output_size" type: DT_INT32 }
input_arg { name: "iou_threshold" type_attr: "T" }
input_arg { name: "score_threshold" type_attr: "T" }
input_arg { name: "soft_nms_sigma" type_attr: "T" }
output_arg { name: "selected_indices" type: DT_INT32 }
output_arg { name: "selected_scores" type_attr: "T" }
output_arg { name: "valid_outputs" type: DT_INT32 }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "pad_to_max_output_size"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
nonMaxSuppressionWithOverlaps :: Tensor v'1 Float -- ^ __overlaps__
                                 -> Tensor v'2 Float -- ^ __scores__
                                 -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                                 -> Tensor v'4 Float -- ^ __overlap_threshold__
                                 -> Tensor v'5 Float -- ^ __score_threshold__
                                 -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionWithOverlaps :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor Build Int32
nonMaxSuppressionWithOverlaps = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *).
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor Build Int32
nonMaxSuppressionWithOverlaps' OpParams
forall a. a -> a
id
nonMaxSuppressionWithOverlaps' :: OpParams ->
                                  Tensor v'1 Float -- ^ __overlaps__
                                  -> Tensor v'2 Float -- ^ __scores__
                                  -> Tensor v'3 Data.Int.Int32 -- ^ __max_output_size__
                                  -> Tensor v'4 Float -- ^ __overlap_threshold__
                                  -> Tensor v'5 Float -- ^ __score_threshold__
                                  -> Tensor Build Data.Int.Int32 -- ^ __selected_indices__
nonMaxSuppressionWithOverlaps' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor Build Int32
nonMaxSuppressionWithOverlaps' op'options :: OpParams
op'options overlaps :: Tensor v'1 Float
overlaps scores :: Tensor v'2 Float
scores max_output_size :: Tensor v'3 Int32
max_output_size
                               overlap_threshold :: Tensor v'4 Float
overlap_threshold
                               score_threshold :: Tensor v'5 Float
score_threshold | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
overlaps,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
scores,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
max_output_size,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
overlap_threshold,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
score_threshold]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonMaxSuppressionWithOverlaps"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "overlaps" type: DT_FLOAT }
input_arg { name: "scores" type: DT_FLOAT }
input_arg { name: "max_output_size" type: DT_INT32 }
input_arg { name: "overlap_threshold" type: DT_FLOAT }
input_arg { name: "score_threshold" type: DT_FLOAT }
output_arg { name: "selected_indices" type: DT_INT32 }
-}
-- | 
nonSerializableDataset :: [DataType] -- ^ __output_types__
                          -> Tensor v'1 Variant -- ^ __input_dataset__
                          -> Tensor Build Variant -- ^ __handle__
nonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
nonSerializableDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
nonSerializableDataset' OpParams
forall a. a -> a
id
nonSerializableDataset' :: OpParams ->
                           [DataType] -- ^ __output_types__
                           -> Tensor v'1 Variant -- ^ __input_dataset__
                           -> Tensor Build Variant -- ^ __handle__
nonSerializableDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
nonSerializableDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
                        input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NonSerializableDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
notEqual :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float), Bool,
                                        Data.ByteString.ByteString,
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build Bool -- ^ __z__
notEqual :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
notEqual = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32,
    Int64, Int8, Word16, Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
notEqual' OpParams
forall a. a -> a
id
notEqual' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float), Bool,
                                         Data.ByteString.ByteString,
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 t -- ^ __y__
             -> Tensor Build Bool -- ^ __z__
notEqual' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
notEqual' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NotEqual"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type: DT_BOOL }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
      type: DT_COMPLEX64
      type: DT_QUINT8
      type: DT_QINT8
      type: DT_QINT32
      type: DT_STRING
      type: DT_BOOL
      type: DT_COMPLEX128
    }
  }
}
attr {
  name: "incompatible_shape_error"
  type: "bool"
  default_value { b: true }
}
-}
-- | 
nthElement :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) =>
              Tensor v'1 t -- ^ __input__
              -> Tensor v'2 Data.Int.Int32 -- ^ __n__
              -> Tensor Build t -- ^ __values__
nthElement :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
nthElement = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
nthElement' OpParams
forall a. a -> a
id
nthElement' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t) => OpParams ->
               Tensor v'1 t -- ^ __input__
               -> Tensor v'2 Data.Int.Int32 -- ^ __n__
               -> Tensor Build t -- ^ __values__
nthElement' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
nthElement' op'options :: OpParams
op'options input :: Tensor v'1 t
input n :: Tensor v'2 Int32
n | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
n]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "NthElement"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "n" type: DT_INT32 }
output_arg { name: "values" type_attr: "T" }
attr { name: "reverse" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
oneHot :: forall v'1 v'2 v'3 v'4 t tI . (TensorType t, OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Word.Word8] tI) =>
          Tensor v'1 tI -- ^ __indices__
          -> Tensor v'2 Data.Int.Int32 -- ^ __depth__
          -> Tensor v'3 t -- ^ __on_value__
          -> Tensor v'4 t -- ^ __off_value__
          -> Tensor Build t -- ^ __output__
oneHot :: Tensor v'1 tI
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
oneHot = OpParams
-> Tensor v'1 tI
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tI.
(TensorType t, OneOf '[Int32, Int64, Word8] tI) =>
OpParams
-> Tensor v'1 tI
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
oneHot' OpParams
forall a. a -> a
id
oneHot' :: forall v'1 v'2 v'3 v'4 t tI . (TensorType t, OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Word.Word8] tI) =>
           OpParams ->
           Tensor v'1 tI -- ^ __indices__
           -> Tensor v'2 Data.Int.Int32 -- ^ __depth__
           -> Tensor v'3 t -- ^ __on_value__
           -> Tensor v'4 t -- ^ __off_value__
           -> Tensor Build t -- ^ __output__
oneHot' :: OpParams
-> Tensor v'1 tI
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
oneHot' op'options :: OpParams
op'options indices :: Tensor v'1 tI
indices depth :: Tensor v'2 Int32
depth on_value :: Tensor v'3 t
on_value off_value :: Tensor v'4 t
off_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tI -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tI
indices,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
depth,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
on_value,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
off_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OneHot"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "TI" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tI -> DataType
forall a. TensorType a => a -> DataType
tensorType (tI
forall a. HasCallStack => a
undefined :: tI)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type_attr: "TI" }
input_arg { name: "depth" type: DT_INT32 }
input_arg { name: "on_value" type_attr: "T" }
input_arg { name: "off_value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "axis" type: "int" default_value { i: -1 } }
attr { name: "T" type: "type" }
attr {
  name: "TI"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values {
    list { type: DT_UINT8 type: DT_INT32 type: DT_INT64 }
  }
}
-}
-- | 
onesLike :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float), Bool,
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word8, Double,
                                    Float] t) => Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
onesLike :: Tensor v'1 t -> Tensor Build t
onesLike = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Word16, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
onesLike' OpParams
forall a. a -> a
id
onesLike' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float), Bool,
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
onesLike' :: OpParams -> Tensor v'1 t -> Tensor Build t
onesLike' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OnesLike"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_BOOL
    }
  }
}
-}
-- | 
optimizeDataset :: [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __optimizations__
                   -> Tensor Build Variant -- ^ __handle__
optimizeDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
optimizeDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
optimizeDataset' OpParams
forall a. a -> a
id
optimizeDataset' :: OpParams ->
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __optimizations__
                    -> Tensor Build Variant -- ^ __handle__
optimizeDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
optimizeDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                 optimizations :: Tensor v'2 ByteString
optimizations | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
optimizations]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OptimizeDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "optimizations" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "optimization_configs"
  type: "list(string)"
  default_value { list { } }
}
-}
-- | 
optionalFromValue :: forall v'1 toutput_types . (TensorTypes toutput_types) =>
                     TensorList (v'1) toutput_types -- ^ __components__
                     -> Tensor Build Variant -- ^ __optional__
optionalFromValue :: TensorList v'1 toutput_types -> Tensor Build Variant
optionalFromValue = OpParams -> TensorList v'1 toutput_types -> Tensor Build Variant
forall (v'1 :: * -> *) (input_types :: [*]).
TensorTypes input_types =>
OpParams -> TensorList v'1 input_types -> Tensor Build Variant
optionalFromValue' OpParams
forall a. a -> a
id
optionalFromValue' :: forall v'1 toutput_types . (TensorTypes toutput_types) =>
                      OpParams ->
                      TensorList (v'1) toutput_types -- ^ __components__
                      -> Tensor Build Variant -- ^ __optional__
optionalFromValue' :: OpParams -> TensorList v'1 toutput_types -> Tensor Build Variant
optionalFromValue' op'options :: OpParams
op'options components :: TensorList v'1 toutput_types
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 toutput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 toutput_types
components]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OptionalFromValue"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutput_types
forall k (t :: k). Proxy t
Proxy :: Proxy toutput_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "components" type_list_attr: "Toutput_types" }
output_arg { name: "optional" type: DT_VARIANT }
attr {
  name: "Toutput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
optionalGetValue :: forall v'1 output_types . (TensorTypes output_types) =>
                    Tensor v'1 Variant -- ^ __optional__
                    -> TensorList (Build) output_types -- ^ __components__
optionalGetValue :: Tensor v'1 Variant -> TensorList Build output_types
optionalGetValue = OpParams -> Tensor v'1 Variant -> TensorList Build output_types
forall (v'1 :: * -> *) (output_types :: [*]).
TensorTypes output_types =>
OpParams -> Tensor v'1 Variant -> TensorList Build output_types
optionalGetValue' OpParams
forall a. a -> a
id
optionalGetValue' :: forall v'1 output_types . (TensorTypes output_types) =>
                     OpParams ->
                     Tensor v'1 Variant -- ^ __optional__
                     -> TensorList (Build) output_types -- ^ __components__
optionalGetValue' :: OpParams -> Tensor v'1 Variant -> TensorList Build output_types
optionalGetValue' op'options :: OpParams
op'options optional :: Tensor v'1 Variant
optional | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build output_types
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build output_types)
-> Build OpDef -> TensorList Build output_types
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
optional]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OptionalGetValue"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "optional" type: DT_VARIANT }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
optionalHasValue :: Tensor v'1 Variant -- ^ __optional__
                    -> Tensor Build Bool -- ^ __has_value__
optionalHasValue :: Tensor v'1 Variant -> Tensor Build Bool
optionalHasValue = OpParams -> Tensor v'1 Variant -> Tensor Build Bool
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Bool
optionalHasValue' OpParams
forall a. a -> a
id
optionalHasValue' :: OpParams ->
                     Tensor v'1 Variant -- ^ __optional__
                     -> Tensor Build Bool -- ^ __has_value__
optionalHasValue' :: OpParams -> Tensor v'1 Variant -> Tensor Build Bool
optionalHasValue' op'options :: OpParams
op'options optional :: Tensor v'1 Variant
optional | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
optional]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OptionalHasValue"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "optional" type: DT_VARIANT }
output_arg { name: "has_value" type: DT_BOOL }
-}
-- | 
optionalNone :: Tensor Build Variant -- ^ __optional__
optionalNone :: Tensor Build Variant
optionalNone = OpParams -> Tensor Build Variant
optionalNone' OpParams
forall a. a -> a
id
optionalNone' :: OpParams ->
                 Tensor Build Variant -- ^ __optional__
optionalNone' :: OpParams -> Tensor Build Variant
optionalNone' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "OptionalNone"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "optional" type: DT_VARIANT }
-}
-- | 
orderedMapClear :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
                   -> m' (ControlNode)
orderedMapClear :: [DataType] -> m' ControlNode
orderedMapClear = OpParams -> [DataType] -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' ControlNode
orderedMapClear' OpParams
forall a. a -> a
id
orderedMapClear' :: forall m' . (MonadBuild m') => OpParams ->
                    [DataType] -- ^ __dtypes__
                    -> m' (ControlNode)
orderedMapClear' :: OpParams -> [DataType] -> m' ControlNode
orderedMapClear' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapClear"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapIncompleteSize :: forall m' . (MonadBuild m') =>
                            [DataType] -- ^ __dtypes__
                            -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
orderedMapIncompleteSize :: [DataType] -> m' (Tensor Value Int32)
orderedMapIncompleteSize = OpParams -> [DataType] -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value Int32)
orderedMapIncompleteSize' OpParams
forall a. a -> a
id
orderedMapIncompleteSize' :: forall m' . (MonadBuild m') => OpParams ->
                             [DataType] -- ^ __dtypes__
                             -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
orderedMapIncompleteSize' :: OpParams -> [DataType] -> m' (Tensor Value Int32)
orderedMapIncompleteSize' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapIncompleteSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapPeek :: forall v'1 v'2 dtypes m' . (MonadBuild m',
                                              TensorTypes dtypes) =>
                  Tensor v'1 Data.Int.Int64 -- ^ __key__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                  -> m' (TensorList (Value) dtypes) -- ^ __values__
orderedMapPeek :: Tensor v'1 Int64
-> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
orderedMapPeek = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (v'2 :: * -> *) (dtypes :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
orderedMapPeek' OpParams
forall a. a -> a
id
orderedMapPeek' :: forall v'1 v'2 dtypes m' . (MonadBuild m',
                                               TensorTypes dtypes) =>
                   OpParams ->
                   Tensor v'1 Data.Int.Int64 -- ^ __key__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                   -> m' (TensorList (Value) dtypes) -- ^ __values__
orderedMapPeek' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
orderedMapPeek' op'options :: OpParams
op'options key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapPeek"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapSize :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
                  -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
orderedMapSize :: [DataType] -> m' (Tensor Value Int32)
orderedMapSize = OpParams -> [DataType] -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value Int32)
orderedMapSize' OpParams
forall a. a -> a
id
orderedMapSize' :: forall m' . (MonadBuild m') => OpParams ->
                   [DataType] -- ^ __dtypes__
                   -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
orderedMapSize' :: OpParams -> [DataType] -> m' (Tensor Value Int32)
orderedMapSize' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapStage :: forall v'1 v'2 v'3 fake_dtypes m' . (MonadBuild m',
                                                        TensorTypes fake_dtypes) =>
                   [DataType] -- ^ __dtypes__
                   -> Tensor v'1 Data.Int.Int64 -- ^ __key__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                   -> TensorList (v'3) fake_dtypes -- ^ __values__
                   -> m' (ControlNode)
orderedMapStage :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
orderedMapStage = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (fake_dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes fake_dtypes) =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
orderedMapStage' OpParams
forall a. a -> a
id
orderedMapStage' :: forall v'1 v'2 v'3 fake_dtypes m' . (MonadBuild m',
                                                         TensorTypes fake_dtypes) =>
                    OpParams ->
                    [DataType] -- ^ __dtypes__
                    -> Tensor v'1 Data.Int.Int64 -- ^ __key__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                    -> TensorList (v'3) fake_dtypes -- ^ __values__
                    -> m' (ControlNode)
orderedMapStage' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> TensorList v'3 fake_dtypes
-> m' ControlNode
orderedMapStage' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices values :: TensorList v'3 fake_dtypes
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             TensorList v'3 fake_dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 fake_dtypes
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapStage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "fake_dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy fake_dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy fake_dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy fake_dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "values" type_list_attr: "fake_dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr {
  name: "fake_dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapUnstage :: forall v'1 v'2 dtypes m' . (MonadBuild m',
                                                 TensorTypes dtypes) =>
                     Tensor v'1 Data.Int.Int64 -- ^ __key__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                     -> m' (TensorList (Value) dtypes) -- ^ __values__
orderedMapUnstage :: Tensor v'1 Int64
-> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
orderedMapUnstage = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (v'2 :: * -> *) (dtypes :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
orderedMapUnstage' OpParams
forall a. a -> a
id
orderedMapUnstage' :: forall v'1 v'2 dtypes m' . (MonadBuild m',
                                                  TensorTypes dtypes) =>
                      OpParams ->
                      Tensor v'1 Data.Int.Int64 -- ^ __key__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                      -> m' (TensorList (Value) dtypes) -- ^ __values__
orderedMapUnstage' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int32
-> m' (TensorList Value dtypes)
orderedMapUnstage' op'options :: OpParams
op'options key :: Tensor v'1 Int64
key indices :: Tensor v'2 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
key,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapUnstage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "key" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
orderedMapUnstageNoKey :: forall v'1 dtypes m' . (MonadBuild m',
                                                  TensorTypes dtypes) =>
                          Tensor v'1 Data.Int.Int32 -- ^ __indices__
                          -> m' ((Tensor Value Data.Int.Int64,
                                  TensorList (Value) dtypes))
                          -- ^ (__key__, __values__)
                          --
                          -- * __key__
                          --
                          -- * __values__
orderedMapUnstageNoKey :: Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
orderedMapUnstageNoKey = OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
orderedMapUnstageNoKey' OpParams
forall a. a -> a
id
orderedMapUnstageNoKey' :: forall v'1 dtypes m' . (MonadBuild m',
                                                   TensorTypes dtypes) =>
                           OpParams ->
                           Tensor v'1 Data.Int.Int32 -- ^ __indices__
                           -> m' ((Tensor Value Data.Int.Int64,
                                   TensorList (Value) dtypes))
                           -- ^ (__key__, __values__)
                           --
                           -- * __key__
                           --
                           -- * __values__
orderedMapUnstageNoKey' :: OpParams
-> Tensor v'1 Int32
-> m' (Tensor Value Int64, TensorList Value dtypes)
orderedMapUnstageNoKey' op'options :: OpParams
op'options indices :: Tensor v'1 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, TensorList Value dtypes)
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, TensorList Value dtypes)
 -> m' (Tensor Value Int64, TensorList Value dtypes))
-> Build (Tensor Value Int64, TensorList Value dtypes)
-> m' (Tensor Value Int64, TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
indices]
        [Int64]
-> OpDef -> Build (Tensor Value Int64, TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OrderedMapUnstageNoKey"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "key" type: DT_INT64 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
outfeedDequeue :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                  Shape -- ^ __shape__
                  -> m' (Tensor Value dtype) -- ^ __output__
outfeedDequeue :: Shape -> m' (Tensor Value dtype)
outfeedDequeue = OpParams -> Shape -> m' (Tensor Value dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Value dtype)
outfeedDequeue' OpParams
forall a. a -> a
id
outfeedDequeue' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                   OpParams ->
                   Shape -- ^ __shape__
                   -> m' (Tensor Value dtype) -- ^ __output__
outfeedDequeue' :: OpParams -> Shape -> m' (Tensor Value dtype)
outfeedDequeue' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OutfeedDequeue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
outfeedDequeueTuple :: forall dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
                       m' (TensorList (Value) dtypes) -- ^ __outputs__
outfeedDequeueTuple :: m' (TensorList Value dtypes)
outfeedDequeueTuple = OpParams -> m' (TensorList Value dtypes)
forall (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> m' (TensorList Value dtypes)
outfeedDequeueTuple' OpParams
forall a. a -> a
id
outfeedDequeueTuple' :: forall dtypes m' . (MonadBuild m',
                                            TensorTypes dtypes) => OpParams ->
                        m' (TensorList (Value) dtypes) -- ^ __outputs__
outfeedDequeueTuple' :: OpParams -> m' (TensorList Value dtypes)
outfeedDequeueTuple' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OutfeedDequeueTuple"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "outputs" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "shapes" type: "list(shape)" }
attr { name: "device_ordinal" type: "int" default_value { i: -1 } }
-}
-- | 
outfeedEnqueue :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                  Tensor v'1 dtype -- ^ __input__
                  -> m' (ControlNode)
outfeedEnqueue :: Tensor v'1 dtype -> m' ControlNode
outfeedEnqueue = OpParams -> Tensor v'1 dtype -> m' ControlNode
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 dtype -> m' ControlNode
outfeedEnqueue' OpParams
forall a. a -> a
id
outfeedEnqueue' :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                   OpParams ->
                   Tensor v'1 dtype -- ^ __input__
                   -> m' (ControlNode)
outfeedEnqueue' :: OpParams -> Tensor v'1 dtype -> m' ControlNode
outfeedEnqueue' op'options :: OpParams
op'options input :: Tensor v'1 dtype
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 dtype
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OutfeedEnqueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
outfeedEnqueueTuple :: forall v'1 dtypes m' . (MonadBuild m',
                                               TensorTypes dtypes) =>
                       TensorList (v'1) dtypes -- ^ __inputs__
                       -> m' (ControlNode)
outfeedEnqueueTuple :: TensorList v'1 dtypes -> m' ControlNode
outfeedEnqueueTuple = OpParams -> TensorList v'1 dtypes -> m' ControlNode
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> TensorList v'1 dtypes -> m' ControlNode
outfeedEnqueueTuple' OpParams
forall a. a -> a
id
outfeedEnqueueTuple' :: forall v'1 dtypes m' . (MonadBuild m',
                                                TensorTypes dtypes) =>
                        OpParams ->
                        TensorList (v'1) dtypes -- ^ __inputs__
                        -> m' (ControlNode)
outfeedEnqueueTuple' :: OpParams -> TensorList v'1 dtypes -> m' ControlNode
outfeedEnqueueTuple' op'options :: OpParams
op'options inputs :: TensorList v'1 dtypes
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 dtypes
inputs]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "OutfeedEnqueueTuple"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | 
pack :: forall v'1 t . (TensorType t) => [Tensor v'1 t] -- ^ __values__
        -> Tensor Build t -- ^ __output__
pack :: [Tensor v'1 t] -> Tensor Build t
pack = OpParams -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> [Tensor v'1 t] -> Tensor Build t
pack' OpParams
forall a. a -> a
id
pack' :: forall v'1 t . (TensorType t) => OpParams ->
         [Tensor v'1 t] -- ^ __values__
         -> Tensor Build t -- ^ __output__
pack' :: OpParams -> [Tensor v'1 t] -> Tensor Build t
pack' op'options :: OpParams
op'options values :: [Tensor v'1 t]
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("values", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Pack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values) :: Int64
{-
input_arg { name: "values" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
attr { name: "axis" type: "int" default_value { i: 0 } }
-}
-- | 
pad :: forall v'1 v'2 t tpaddings . (TensorType t, OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tpaddings) =>
       Tensor v'1 t -- ^ __input__
       -> Tensor v'2 tpaddings -- ^ __paddings__
       -> Tensor Build t -- ^ __output__
pad :: Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
pad = OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
pad' OpParams
forall a. a -> a
id
pad' :: forall v'1 v'2 t tpaddings . (TensorType t, OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tpaddings) =>
        OpParams ->
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tpaddings -- ^ __paddings__
        -> Tensor Build t -- ^ __output__
pad' :: OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
pad' op'options :: OpParams
op'options input :: Tensor v'1 t
input paddings :: Tensor v'2 tpaddings
paddings | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tpaddings
paddings]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Pad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
padV2 :: forall v'1 v'2 v'3 t tpaddings . (TensorType t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tpaddings) =>
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 tpaddings -- ^ __paddings__
         -> Tensor v'3 t -- ^ __constant_values__
         -> Tensor Build t -- ^ __output__
padV2 :: Tensor v'1 t
-> Tensor v'2 tpaddings -> Tensor v'3 t -> Tensor Build t
padV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
padV2' OpParams
forall a. a -> a
id
padV2' :: forall v'1 v'2 v'3 t tpaddings . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tpaddings) =>
          OpParams ->
          Tensor v'1 t -- ^ __input__
          -> Tensor v'2 tpaddings -- ^ __paddings__
          -> Tensor v'3 t -- ^ __constant_values__
          -> Tensor Build t -- ^ __output__
padV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
padV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input paddings :: Tensor v'2 tpaddings
paddings constant_values :: Tensor v'3 t
constant_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tpaddings
paddings,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
constant_values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PadV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
input_arg { name: "constant_values" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
paddedBatchDataset :: forall v'1 v'2 v'3 v'4
                      toutput_types . (TensorTypes toutput_types) =>
                      Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                      -> [Tensor v'3 Data.Int.Int64] -- ^ __padded_shapes__
                      -> TensorList (v'4) toutput_types -- ^ __padding_values__
                      -> Tensor Build Variant -- ^ __handle__
paddedBatchDataset :: Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor Build Variant
paddedBatchDataset = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (toutput_types :: [*]).
TensorTypes toutput_types =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor Build Variant
paddedBatchDataset' OpParams
forall a. a -> a
id
paddedBatchDataset' :: forall v'1 v'2 v'3 v'4
                       toutput_types . (TensorTypes toutput_types) =>
                       OpParams ->
                       Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                       -> [Tensor v'3 Data.Int.Int64] -- ^ __padded_shapes__
                       -> TensorList (v'4) toutput_types -- ^ __padding_values__
                       -> Tensor Build Variant -- ^ __handle__
paddedBatchDataset' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor Build Variant
paddedBatchDataset' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset batch_size :: Tensor v'2 Int64
batch_size padded_shapes :: [Tensor v'3 Int64]
padded_shapes
                    padding_values :: TensorList v'4 toutput_types
padding_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("padded_shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
padded_shapes)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
padded_shapes,
                                                             TensorList v'4 toutput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 toutput_types
padding_values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PaddedBatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutput_types
forall k (t :: k). Proxy t
Proxy :: Proxy toutput_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
padded_shapes) :: Int64
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
input_arg { name: "padded_shapes" type: DT_INT64 number_attr: "N" }
input_arg {
  name: "padding_values" type_list_attr: "Toutput_types"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "Toutput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
paddedBatchDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5
                        toutput_types . (TensorTypes toutput_types) =>
                        Tensor v'1 Variant -- ^ __input_dataset__
                        -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                        -> [Tensor v'3 Data.Int.Int64] -- ^ __padded_shapes__
                        -> TensorList (v'4) toutput_types -- ^ __padding_values__
                        -> Tensor v'5 Bool -- ^ __drop_remainder__
                        -> Tensor Build Variant -- ^ __handle__
paddedBatchDatasetV2 :: Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor v'5 Bool
-> Tensor Build Variant
paddedBatchDatasetV2 = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor v'5 Bool
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (toutput_types :: [*]).
TensorTypes toutput_types =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor v'5 Bool
-> Tensor Build Variant
paddedBatchDatasetV2' OpParams
forall a. a -> a
id
paddedBatchDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5
                         toutput_types . (TensorTypes toutput_types) =>
                         OpParams ->
                         Tensor v'1 Variant -- ^ __input_dataset__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __batch_size__
                         -> [Tensor v'3 Data.Int.Int64] -- ^ __padded_shapes__
                         -> TensorList (v'4) toutput_types -- ^ __padding_values__
                         -> Tensor v'5 Bool -- ^ __drop_remainder__
                         -> Tensor Build Variant -- ^ __handle__
paddedBatchDatasetV2' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> [Tensor v'3 Int64]
-> TensorList v'4 toutput_types
-> Tensor v'5 Bool
-> Tensor Build Variant
paddedBatchDatasetV2' op'options :: OpParams
op'options input_dataset :: Tensor v'1 Variant
input_dataset batch_size :: Tensor v'2 Int64
batch_size padded_shapes :: [Tensor v'3 Int64]
padded_shapes
                      padding_values :: TensorList v'4 toutput_types
padding_values
                      drop_remainder :: Tensor v'5 Bool
drop_remainder | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("padded_shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
padded_shapes)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_size,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
padded_shapes,
                                                             TensorList v'4 toutput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 toutput_types
padding_values,
                                                             Tensor v'5 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Bool
drop_remainder]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PaddedBatchDatasetV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutput_types
forall k (t :: k). Proxy t
Proxy :: Proxy toutput_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
padded_shapes) :: Int64
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "batch_size" type: DT_INT64 }
input_arg { name: "padded_shapes" type: DT_INT64 number_attr: "N" }
input_arg {
  name: "padding_values" type_list_attr: "Toutput_types"
}
input_arg { name: "drop_remainder" type: DT_BOOL }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "parallel_copy" type: "bool" default_value { b: false }
}
attr {
  name: "Toutput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
paddingFIFOQueue :: forall m' . (MonadBuild m') =>
                    [DataType] -- ^ __component_types__
                    -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
paddingFIFOQueue :: [DataType] -> m' (Tensor Ref ByteString)
paddingFIFOQueue = OpParams -> [DataType] -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Ref ByteString)
paddingFIFOQueue' OpParams
forall a. a -> a
id
paddingFIFOQueue' :: forall m' . (MonadBuild m') => OpParams ->
                     [DataType] -- ^ __component_types__
                     -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
paddingFIFOQueue' :: OpParams -> [DataType] -> m' (Tensor Ref ByteString)
paddingFIFOQueue' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PaddingFIFOQueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
paddingFIFOQueueV2 :: forall m' . (MonadBuild m') =>
                      [DataType] -- ^ __component_types__
                      -> m' (Tensor Value ResourceHandle) -- ^ __handle__
paddingFIFOQueueV2 :: [DataType] -> m' (Tensor Value ResourceHandle)
paddingFIFOQueueV2 = OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
paddingFIFOQueueV2' OpParams
forall a. a -> a
id
paddingFIFOQueueV2' :: forall m' . (MonadBuild m') => OpParams ->
                       [DataType] -- ^ __component_types__
                       -> m' (Tensor Value ResourceHandle) -- ^ __handle__
paddingFIFOQueueV2' :: OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
paddingFIFOQueueV2' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PaddingFIFOQueueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
parallelConcat :: forall v'1 t . (TensorType t) => Shape -- ^ __shape__
                  -> [Tensor v'1 t] -- ^ __values__
                  -> Tensor Build t -- ^ __output__
parallelConcat :: Shape -> [Tensor v'1 t] -> Tensor Build t
parallelConcat = OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
parallelConcat' OpParams
forall a. a -> a
id
parallelConcat' :: forall v'1 t . (TensorType t) => OpParams ->
                   Shape -- ^ __shape__
                   -> [Tensor v'1 t] -- ^ __values__
                   -> Tensor Build t -- ^ __output__
parallelConcat' :: OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
parallelConcat' op'options :: OpParams
op'options shape :: Shape
shape
                values :: [Tensor v'1 t]
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("values", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParallelConcat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
values) :: Int64
{-
input_arg { name: "values" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
attr { name: "shape" type: "shape" }
-}
-- | 
parallelDynamicStitch :: forall v'1 v'2 t . (TensorType t) =>
                         [Tensor v'1 Data.Int.Int32] -- ^ __indices__
                         -> [Tensor v'2 t] -- ^ __data__
                         -> Tensor Build t -- ^ __merged__
parallelDynamicStitch :: [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
parallelDynamicStitch = OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
parallelDynamicStitch' OpParams
forall a. a -> a
id
parallelDynamicStitch' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                          [Tensor v'1 Data.Int.Int32] -- ^ __indices__
                          -> [Tensor v'2 t] -- ^ __data__
                          -> Tensor Build t -- ^ __merged__
parallelDynamicStitch' :: OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
parallelDynamicStitch' op'options :: OpParams
op'options indices :: [Tensor v'1 Int32]
indices
                       data' :: [Tensor v'2 t]
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
indices),
                                                     ("data", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
data')])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int32]
indices,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
data']
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParallelDynamicStitch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT32 number_attr: "N" }
input_arg { name: "data" type_attr: "T" number_attr: "N" }
output_arg { name: "merged" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | 
parameterizedTruncatedNormal :: forall v'1 v'2 v'3 v'4 v'5 dtype t
                                m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                             Double,
                                                             Float] dtype,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] t) =>
                                Tensor v'1 t -- ^ __shape__
                                -> Tensor v'2 dtype -- ^ __means__
                                -> Tensor v'3 dtype -- ^ __stdevs__
                                -> Tensor v'4 dtype -- ^ __minvals__
                                -> Tensor v'5 dtype -- ^ __maxvals__
                                -> m' (Tensor Value dtype) -- ^ __output__
parameterizedTruncatedNormal :: Tensor v'1 t
-> Tensor v'2 dtype
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
parameterizedTruncatedNormal = OpParams
-> Tensor v'1 t
-> Tensor v'2 dtype
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) dtype t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] dtype,
 OneOf '[Int32, Int64] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 dtype
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
parameterizedTruncatedNormal' OpParams
forall a. a -> a
id
parameterizedTruncatedNormal' :: forall v'1 v'2 v'3 v'4 v'5 dtype t
                                 m' . (MonadBuild m', OneOf '[Data.Word.Word16,
                                                              Double,
                                                              Float] dtype,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] t) => OpParams ->
                                 Tensor v'1 t -- ^ __shape__
                                 -> Tensor v'2 dtype -- ^ __means__
                                 -> Tensor v'3 dtype -- ^ __stdevs__
                                 -> Tensor v'4 dtype -- ^ __minvals__
                                 -> Tensor v'5 dtype -- ^ __maxvals__
                                 -> m' (Tensor Value dtype) -- ^ __output__
parameterizedTruncatedNormal' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 dtype
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
parameterizedTruncatedNormal' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape means :: Tensor v'2 dtype
means stdevs :: Tensor v'3 dtype
stdevs minvals :: Tensor v'4 dtype
minvals
                              maxvals :: Tensor v'5 dtype
maxvals | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 dtype
means,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
stdevs,
                                                             Tensor v'4 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 dtype
minvals,
                                                             Tensor v'5 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 dtype
maxvals]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ParameterizedTruncatedNormal"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "means" type_attr: "dtype" }
input_arg { name: "stdevs" type_attr: "dtype" }
input_arg { name: "minvals" type_attr: "dtype" }
input_arg { name: "maxvals" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
parseExample :: forall v'1 v'2 v'3 v'4 v'5 sparse_types
                tdense . (OneOfs '[Data.ByteString.ByteString, Data.Int.Int64,
                                   Float] sparse_types,
                          OneOfs '[Data.ByteString.ByteString, Data.Int.Int64,
                                   Float] tdense) =>
                Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __names__
                -> [Tensor v'3 Data.ByteString.ByteString] -- ^ __sparse_keys__
                -> [Tensor v'4 Data.ByteString.ByteString] -- ^ __dense_keys__
                -> TensorList (v'5) tdense -- ^ __dense_defaults__
                -> ([Tensor Build Data.Int.Int64],
                    TensorList (Build) sparse_types,
                    [Tensor Build Data.Int.Int64], TensorList (Build) tdense)
                -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__)
                --
                -- * __sparse_indices__
                --
                -- * __sparse_values__
                --
                -- * __sparse_shapes__
                --
                -- * __dense_values__
parseExample :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> [Tensor v'3 ByteString]
-> [Tensor v'4 ByteString]
-> TensorList v'5 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseExample = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> [Tensor v'3 ByteString]
-> [Tensor v'4 ByteString]
-> TensorList v'5 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (sparse_types :: [*])
       (tdense :: [*]).
(OneOfs '[ByteString, Int64, Float] sparse_types,
 OneOfs '[ByteString, Int64, Float] tdense) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> [Tensor v'3 ByteString]
-> [Tensor v'4 ByteString]
-> TensorList v'5 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseExample' OpParams
forall a. a -> a
id
parseExample' :: forall v'1 v'2 v'3 v'4 v'5 sparse_types
                 tdense . (OneOfs '[Data.ByteString.ByteString, Data.Int.Int64,
                                    Float] sparse_types,
                           OneOfs '[Data.ByteString.ByteString, Data.Int.Int64,
                                    Float] tdense) => OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __names__
                 -> [Tensor v'3 Data.ByteString.ByteString] -- ^ __sparse_keys__
                 -> [Tensor v'4 Data.ByteString.ByteString] -- ^ __dense_keys__
                 -> TensorList (v'5) tdense -- ^ __dense_defaults__
                 -> ([Tensor Build Data.Int.Int64],
                     TensorList (Build) sparse_types,
                     [Tensor Build Data.Int.Int64], TensorList (Build) tdense)
                 -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__)
                 --
                 -- * __sparse_indices__
                 --
                 -- * __sparse_values__
                 --
                 -- * __sparse_shapes__
                 --
                 -- * __dense_values__
parseExample' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> [Tensor v'3 ByteString]
-> [Tensor v'4 ByteString]
-> TensorList v'5 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseExample' op'options :: OpParams
op'options serialized :: Tensor v'1 ByteString
serialized names :: Tensor v'2 ByteString
names sparse_keys :: [Tensor v'3 ByteString]
sparse_keys dense_keys :: [Tensor v'4 ByteString]
dense_keys
              dense_defaults :: TensorList v'5 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("Nsparse", [("sparse_keys", [Tensor v'3 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 ByteString]
sparse_keys)]),
                                              ("Ndense", [("dense_keys", [Tensor v'4 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 ByteString]
dense_keys)])] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
nsparse, Int64
nsparse] (Build OpDef
 -> ([Tensor Build Int64], TensorList Build sparse_types,
     [Tensor Build Int64], TensorList Build tdense))
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
serialized,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
names,
                                                             [Tensor v'3 ByteString] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 ByteString]
sparse_keys,
                                                             [Tensor v'4 ByteString] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 ByteString]
dense_keys,
                                                             TensorList v'5 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'5 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseExample"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Nsparse" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
nsparse
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Ndense" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
ndense
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    nsparse :: Int64
nsparse = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 ByteString]
sparse_keys) :: Int64
    ndense :: Int64
ndense = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 ByteString]
dense_keys) :: Int64
{-
input_arg { name: "serialized" type: DT_STRING }
input_arg { name: "names" type: DT_STRING }
input_arg {
  name: "sparse_keys" type: DT_STRING number_attr: "Nsparse"
}
input_arg {
  name: "dense_keys" type: DT_STRING number_attr: "Ndense"
}
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg {
  name: "sparse_indices" type: DT_INT64 number_attr: "Nsparse"
}
output_arg { name: "sparse_values" type_list_attr: "sparse_types" }
output_arg {
  name: "sparse_shapes" type: DT_INT64 number_attr: "Nsparse"
}
output_arg { name: "dense_values" type_list_attr: "Tdense" }
attr { name: "Nsparse" type: "int" has_minimum: true }
attr { name: "Ndense" type: "int" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
-}
-- | 
parseExampleDataset :: forall v'1 v'2 v'3
                       tdense . (OneOfs '[Data.ByteString.ByteString,
                                          Data.Int.Int64, Float] tdense) =>
                       [DataType] -- ^ __output_types__
                       -> [DataType] -- ^ __sparse_types__
                       -> Tensor v'1 Variant -- ^ __input_dataset__
                       -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                       -> TensorList (v'3) tdense -- ^ __dense_defaults__
                       -> Tensor Build Variant -- ^ __handle__
parseExampleDataset :: [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDataset = OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (tdense :: [*]).
OneOfs '[ByteString, Int64, Float] tdense =>
OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDataset' OpParams
forall a. a -> a
id
parseExampleDataset' :: forall v'1 v'2 v'3
                        tdense . (OneOfs '[Data.ByteString.ByteString,
                                           Data.Int.Int64, Float] tdense) =>
                        OpParams ->
                        [DataType] -- ^ __output_types__
                        -> [DataType] -- ^ __sparse_types__
                        -> Tensor v'1 Variant -- ^ __input_dataset__
                        -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                        -> TensorList (v'3) tdense -- ^ __dense_defaults__
                        -> Tensor Build Variant -- ^ __handle__
parseExampleDataset' :: OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types sparse_types :: [DataType]
sparse_types input_dataset :: Tensor v'1 Variant
input_dataset
                     num_parallel_calls :: Tensor v'2 Int64
num_parallel_calls dense_defaults :: TensorList v'3 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_parallel_calls,
                                                             TensorList v'3 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseExampleDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
sparse_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_parallel_calls" type: DT_INT64 }
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "sparse_keys" type: "list(string)" has_minimum: true }
attr { name: "dense_keys" type: "list(string)" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "sloppy" type: "bool" default_value { b: false } }
attr {
  name: "ragged_keys"
  type: "list(string)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "ragged_value_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "ragged_split_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
parseExampleDatasetV2 :: forall v'1 v'2 v'3
                         tdense . (OneOfs '[Data.ByteString.ByteString,
                                            Data.Int.Int64, Float] tdense) =>
                         [DataType] -- ^ __output_types__
                         -> [DataType] -- ^ __sparse_types__
                         -> Tensor v'1 Variant -- ^ __input_dataset__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                         -> TensorList (v'3) tdense -- ^ __dense_defaults__
                         -> Tensor Build Variant -- ^ __handle__
parseExampleDatasetV2 :: [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDatasetV2 = OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (tdense :: [*]).
OneOfs '[ByteString, Int64, Float] tdense =>
OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDatasetV2' OpParams
forall a. a -> a
id
parseExampleDatasetV2' :: forall v'1 v'2 v'3
                          tdense . (OneOfs '[Data.ByteString.ByteString,
                                             Data.Int.Int64, Float] tdense) =>
                          OpParams ->
                          [DataType] -- ^ __output_types__
                          -> [DataType] -- ^ __sparse_types__
                          -> Tensor v'1 Variant -- ^ __input_dataset__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __num_parallel_calls__
                          -> TensorList (v'3) tdense -- ^ __dense_defaults__
                          -> Tensor Build Variant -- ^ __handle__
parseExampleDatasetV2' :: OpParams
-> [DataType]
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> TensorList v'3 tdense
-> Tensor Build Variant
parseExampleDatasetV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types sparse_types :: [DataType]
sparse_types input_dataset :: Tensor v'1 Variant
input_dataset
                       num_parallel_calls :: Tensor v'2 Int64
num_parallel_calls dense_defaults :: TensorList v'3 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_parallel_calls,
                                                             TensorList v'3 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseExampleDatasetV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
sparse_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_parallel_calls" type: DT_INT64 }
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "sparse_keys" type: "list(string)" has_minimum: true }
attr { name: "dense_keys" type: "list(string)" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "deterministic" type: "string" default_value { s: "default" }
}
attr {
  name: "ragged_keys"
  type: "list(string)"
  default_value { list { } }
  has_minimum: true
}
attr {
  name: "ragged_value_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "ragged_split_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
parseExampleV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types
                  ragged_value_types
                  ragged_split_types . (OneOfs '[Data.ByteString.ByteString,
                                                 Data.Int.Int64, Float] tdense,
                                        OneOfs '[Data.ByteString.ByteString,
                                                 Data.Int.Int64,
                                                 Float] sparse_types,
                                        OneOfs '[Data.ByteString.ByteString,
                                                 Data.Int.Int64,
                                                 Float] ragged_value_types,
                                        OneOfs '[Data.Int.Int32,
                                                 Data.Int.Int64] ragged_split_types) =>
                  Data.Int.Int64 -- ^ __num_sparse__
                  -> Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __names__
                  -> Tensor v'3 Data.ByteString.ByteString -- ^ __sparse_keys__
                  -> Tensor v'4 Data.ByteString.ByteString -- ^ __dense_keys__
                  -> Tensor v'5 Data.ByteString.ByteString -- ^ __ragged_keys__
                  -> TensorList (v'6) tdense -- ^ __dense_defaults__
                  -> ([Tensor Build Data.Int.Int64],
                      TensorList (Build) sparse_types,
                      [Tensor Build Data.Int.Int64], TensorList (Build) tdense,
                      TensorList (Build) ragged_value_types,
                      TensorList (Build) ragged_split_types)
                  -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__, __ragged_values__, __ragged_row_splits__)
                  --
                  -- * __sparse_indices__
                  --
                  -- * __sparse_values__
                  --
                  -- * __sparse_shapes__
                  --
                  -- * __dense_values__
                  --
                  -- * __ragged_values__
                  --
                  -- * __ragged_row_splits__
parseExampleV2 :: Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> TensorList v'6 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
parseExampleV2 = OpParams
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> TensorList v'6 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (tdense :: [*])
       (sparse_types :: [*]) (ragged_value_types :: [*])
       (ragged_split_types :: [*]).
(OneOfs '[ByteString, Int64, Float] tdense,
 OneOfs '[ByteString, Int64, Float] sparse_types,
 OneOfs '[ByteString, Int64, Float] ragged_value_types,
 OneOfs '[Int32, Int64] ragged_split_types) =>
OpParams
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> TensorList v'6 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
parseExampleV2' OpParams
forall a. a -> a
id
parseExampleV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tdense sparse_types
                   ragged_value_types
                   ragged_split_types . (OneOfs '[Data.ByteString.ByteString,
                                                  Data.Int.Int64, Float] tdense,
                                         OneOfs '[Data.ByteString.ByteString,
                                                  Data.Int.Int64,
                                                  Float] sparse_types,
                                         OneOfs '[Data.ByteString.ByteString,
                                                  Data.Int.Int64,
                                                  Float] ragged_value_types,
                                         OneOfs '[Data.Int.Int32,
                                                  Data.Int.Int64] ragged_split_types) =>
                   OpParams ->
                   Data.Int.Int64 -- ^ __num_sparse__
                   -> Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __names__
                   -> Tensor v'3 Data.ByteString.ByteString -- ^ __sparse_keys__
                   -> Tensor v'4 Data.ByteString.ByteString -- ^ __dense_keys__
                   -> Tensor v'5 Data.ByteString.ByteString -- ^ __ragged_keys__
                   -> TensorList (v'6) tdense -- ^ __dense_defaults__
                   -> ([Tensor Build Data.Int.Int64],
                       TensorList (Build) sparse_types,
                       [Tensor Build Data.Int.Int64], TensorList (Build) tdense,
                       TensorList (Build) ragged_value_types,
                       TensorList (Build) ragged_split_types)
                   -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__, __ragged_values__, __ragged_row_splits__)
                   --
                   -- * __sparse_indices__
                   --
                   -- * __sparse_values__
                   --
                   -- * __sparse_shapes__
                   --
                   -- * __dense_values__
                   --
                   -- * __ragged_values__
                   --
                   -- * __ragged_row_splits__
parseExampleV2' :: OpParams
-> Int64
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> TensorList v'6 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
parseExampleV2' op'options :: OpParams
op'options num_sparse :: Int64
num_sparse serialized :: Tensor v'1 ByteString
serialized names :: Tensor v'2 ByteString
names sparse_keys :: Tensor v'3 ByteString
sparse_keys dense_keys :: Tensor v'4 ByteString
dense_keys
                ragged_keys :: Tensor v'5 ByteString
ragged_keys dense_defaults :: TensorList v'6 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_sparse, Int64
num_sparse] (Build OpDef
 -> ([Tensor Build Int64], TensorList Build sparse_types,
     [Tensor Build Int64], TensorList Build tdense,
     TensorList Build ragged_value_types,
     TensorList Build ragged_split_types))
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense,
    TensorList Build ragged_value_types,
    TensorList Build ragged_split_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
serialized,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
names,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
sparse_keys,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
dense_keys,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
ragged_keys,
                                                             TensorList v'6 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'6 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseExampleV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "ragged_value_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy ragged_value_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy ragged_value_types
forall k (t :: k). Proxy t
Proxy :: Proxy ragged_value_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "ragged_split_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy ragged_split_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy ragged_split_types
forall k (t :: k). Proxy t
Proxy :: Proxy ragged_split_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "serialized" type: DT_STRING }
input_arg { name: "names" type: DT_STRING }
input_arg { name: "sparse_keys" type: DT_STRING }
input_arg { name: "dense_keys" type: DT_STRING }
input_arg { name: "ragged_keys" type: DT_STRING }
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg {
  name: "sparse_indices" type: DT_INT64 number_attr: "num_sparse"
}
output_arg { name: "sparse_values" type_list_attr: "sparse_types" }
output_arg {
  name: "sparse_shapes" type: DT_INT64 number_attr: "num_sparse"
}
output_arg { name: "dense_values" type_list_attr: "Tdense" }
output_arg {
  name: "ragged_values" type_list_attr: "ragged_value_types"
}
output_arg {
  name: "ragged_row_splits" type_list_attr: "ragged_split_types"
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "num_sparse" type: "int" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "ragged_value_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "ragged_split_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
-}
-- | 
parseSingleExample :: forall v'1 v'2 sparse_types
                      tdense . (OneOfs '[Data.ByteString.ByteString,
                                         Data.Int.Int64, Float] sparse_types,
                                OneOfs '[Data.ByteString.ByteString,
                                         Data.Int.Int64, Float] tdense) =>
                      Data.Int.Int64 -- ^ __num_sparse__
                      -> Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                      -> TensorList (v'2) tdense -- ^ __dense_defaults__
                      -> ([Tensor Build Data.Int.Int64],
                          TensorList (Build) sparse_types,
                          [Tensor Build Data.Int.Int64],
                          TensorList (Build) tdense)
                      -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__)
                      --
                      -- * __sparse_indices__
                      --
                      -- * __sparse_values__
                      --
                      -- * __sparse_shapes__
                      --
                      -- * __dense_values__
parseSingleExample :: Int64
-> Tensor v'1 ByteString
-> TensorList v'2 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseSingleExample = OpParams
-> Int64
-> Tensor v'1 ByteString
-> TensorList v'2 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall (v'1 :: * -> *) (v'2 :: * -> *) (sparse_types :: [*])
       (tdense :: [*]).
(OneOfs '[ByteString, Int64, Float] sparse_types,
 OneOfs '[ByteString, Int64, Float] tdense) =>
OpParams
-> Int64
-> Tensor v'1 ByteString
-> TensorList v'2 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseSingleExample' OpParams
forall a. a -> a
id
parseSingleExample' :: forall v'1 v'2 sparse_types
                       tdense . (OneOfs '[Data.ByteString.ByteString,
                                          Data.Int.Int64, Float] sparse_types,
                                 OneOfs '[Data.ByteString.ByteString,
                                          Data.Int.Int64, Float] tdense) =>
                       OpParams ->
                       Data.Int.Int64 -- ^ __num_sparse__
                       -> Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                       -> TensorList (v'2) tdense -- ^ __dense_defaults__
                       -> ([Tensor Build Data.Int.Int64],
                           TensorList (Build) sparse_types,
                           [Tensor Build Data.Int.Int64],
                           TensorList (Build) tdense)
                       -- ^ (__sparse_indices__, __sparse_values__, __sparse_shapes__, __dense_values__)
                       --
                       -- * __sparse_indices__
                       --
                       -- * __sparse_values__
                       --
                       -- * __sparse_shapes__
                       --
                       -- * __dense_values__
parseSingleExample' :: OpParams
-> Int64
-> Tensor v'1 ByteString
-> TensorList v'2 tdense
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
parseSingleExample' op'options :: OpParams
op'options num_sparse :: Int64
num_sparse serialized :: Tensor v'1 ByteString
serialized
                    dense_defaults :: TensorList v'2 tdense
dense_defaults | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_sparse, Int64
num_sparse] (Build OpDef
 -> ([Tensor Build Int64], TensorList Build sparse_types,
     [Tensor Build Int64], TensorList Build tdense))
-> Build OpDef
-> ([Tensor Build Int64], TensorList Build sparse_types,
    [Tensor Build Int64], TensorList Build tdense)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
serialized,
                                                             TensorList v'2 tdense -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tdense
dense_defaults]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseSingleExample"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tdense" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tdense -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tdense
forall k (t :: k). Proxy t
Proxy :: Proxy tdense)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "serialized" type: DT_STRING }
input_arg { name: "dense_defaults" type_list_attr: "Tdense" }
output_arg {
  name: "sparse_indices" type: DT_INT64 number_attr: "num_sparse"
}
output_arg { name: "sparse_values" type_list_attr: "sparse_types" }
output_arg {
  name: "sparse_shapes" type: DT_INT64 number_attr: "num_sparse"
}
output_arg { name: "dense_values" type_list_attr: "Tdense" }
attr { name: "num_sparse" type: "int" has_minimum: true }
attr { name: "sparse_keys" type: "list(string)" has_minimum: true }
attr { name: "dense_keys" type: "list(string)" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr {
  name: "Tdense"
  type: "list(type)"
  has_minimum: true
  allowed_values {
    list { type: DT_FLOAT type: DT_INT64 type: DT_STRING }
  }
}
attr { name: "dense_shapes" type: "list(shape)" has_minimum: true }
-}
-- | 
parseTensor :: forall v'1 out_type . (TensorType out_type) =>
               Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
               -> Tensor Build out_type -- ^ __output__
parseTensor :: Tensor v'1 ByteString -> Tensor Build out_type
parseTensor = OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
forall (v'1 :: * -> *) out_type.
TensorType out_type =>
OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
parseTensor' OpParams
forall a. a -> a
id
parseTensor' :: forall v'1 out_type . (TensorType out_type) => OpParams ->
                Tensor v'1 Data.ByteString.ByteString -- ^ __serialized__
                -> Tensor Build out_type -- ^ __output__
parseTensor' :: OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
parseTensor' op'options :: OpParams
op'options serialized :: Tensor v'1 ByteString
serialized | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
serialized]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ParseTensor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "serialized" type: DT_STRING }
output_arg { name: "output" type_attr: "out_type" }
attr { name: "out_type" type: "type" }
-}
-- | 
placeholder :: forall dtype . (TensorType dtype) =>
               Tensor Build dtype -- ^ __output__
placeholder :: Tensor Build dtype
placeholder = OpParams -> Tensor Build dtype
forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
placeholder' OpParams
forall a. a -> a
id
placeholder' :: forall dtype . (TensorType dtype) => OpParams ->
                Tensor Build dtype -- ^ __output__
placeholder' :: OpParams -> Tensor Build dtype
placeholder' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Placeholder"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
placeholderV2 :: forall dtype . (TensorType dtype) => Shape -- ^ __shape__
                 -> Tensor Build dtype -- ^ __output__
placeholderV2 :: Shape -> Tensor Build dtype
placeholderV2 = OpParams -> Shape -> Tensor Build dtype
forall dtype.
TensorType dtype =>
OpParams -> Shape -> Tensor Build dtype
placeholderV2' OpParams
forall a. a -> a
id
placeholderV2' :: forall dtype . (TensorType dtype) => OpParams ->
                  Shape -- ^ __shape__
                  -> Tensor Build dtype -- ^ __output__
placeholderV2' :: OpParams -> Shape -> Tensor Build dtype
placeholderV2' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PlaceholderV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
-}
-- | 
placeholderWithDefault :: forall v'1 dtype . (TensorType dtype) =>
                          Shape -- ^ __shape__
                          -> Tensor v'1 dtype -- ^ __input__
                          -> Tensor Build dtype -- ^ __output__
placeholderWithDefault :: Shape -> Tensor v'1 dtype -> Tensor Build dtype
placeholderWithDefault = OpParams -> Shape -> Tensor v'1 dtype -> Tensor Build dtype
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
placeholderWithDefault' OpParams
forall a. a -> a
id
placeholderWithDefault' :: forall v'1 dtype . (TensorType dtype) => OpParams ->
                           Shape -- ^ __shape__
                           -> Tensor v'1 dtype -- ^ __input__
                           -> Tensor Build dtype -- ^ __output__
placeholderWithDefault' :: OpParams -> Shape -> Tensor v'1 dtype -> Tensor Build dtype
placeholderWithDefault' op'options :: OpParams
op'options shape :: Shape
shape input :: Tensor v'1 dtype
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 dtype
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PlaceholderWithDefault"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
-}
-- | 
polygamma :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
             Tensor v'1 t -- ^ __a__
             -> Tensor v'2 t -- ^ __x__
             -> Tensor Build t -- ^ __z__
polygamma :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
polygamma = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
polygamma' OpParams
forall a. a -> a
id
polygamma' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
              Tensor v'1 t -- ^ __a__
              -> Tensor v'2 t -- ^ __x__
              -> Tensor Build t -- ^ __z__
polygamma' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
polygamma' op'options :: OpParams
op'options a :: Tensor v'1 t
a x :: Tensor v'2 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Polygamma"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "x" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
populationCount :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64,
                                           Data.Word.Word8] t) =>
                   Tensor v'1 t -- ^ __x__
                   -> Tensor Build Data.Word.Word8 -- ^ __y__
populationCount :: Tensor v'1 t -> Tensor Build Word8
populationCount = OpParams -> Tensor v'1 t -> Tensor Build Word8
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor Build Word8
populationCount' OpParams
forall a. a -> a
id
populationCount' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64,
                                            Data.Word.Word8] t) => OpParams ->
                    Tensor v'1 t -- ^ __x__
                    -> Tensor Build Data.Word.Word8 -- ^ __y__
populationCount' :: OpParams -> Tensor v'1 t -> Tensor Build Word8
populationCount' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Word8
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Word8)
-> Build OpDef -> Tensor Build Word8
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PopulationCount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
pow :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int32,
                                   Data.Int.Int64, Data.Word.Word16, Double,
                                   Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
pow :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
pow = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
pow' OpParams
forall a. a -> a
id
pow' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int32, Data.Int.Int64,
                                    Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
pow' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
pow' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Pow"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_HALF
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
prefetchDataset :: [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                   -> Tensor Build Variant -- ^ __handle__
prefetchDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
prefetchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
prefetchDataset' OpParams
forall a. a -> a
id
prefetchDataset' :: OpParams ->
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                    -> Tensor Build Variant -- ^ __handle__
prefetchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
prefetchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                 buffer_size :: Tensor v'2 Int64
buffer_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PrefetchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "slack_period" type: "int" default_value { i: 0 } }
attr {
  name: "legacy_autotune" type: "bool" default_value { b: true }
}
-}
-- | 
prelinearize :: forall v'1 dtype . (TensorType dtype) =>
                Tensor v'1 dtype -- ^ __input__
                -> Tensor Build Variant -- ^ __output__
prelinearize :: Tensor v'1 dtype -> Tensor Build Variant
prelinearize = OpParams -> Tensor v'1 dtype -> Tensor Build Variant
forall (v'1 :: * -> *) dtype.
TensorType dtype =>
OpParams -> Tensor v'1 dtype -> Tensor Build Variant
prelinearize' OpParams
forall a. a -> a
id
prelinearize' :: forall v'1 dtype . (TensorType dtype) => OpParams ->
                 Tensor v'1 dtype -- ^ __input__
                 -> Tensor Build Variant -- ^ __output__
prelinearize' :: OpParams -> Tensor v'1 dtype -> Tensor Build Variant
prelinearize' op'options :: OpParams
op'options input :: Tensor v'1 dtype
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 dtype
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Prelinearize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "dtype" }
output_arg { name: "output" type: DT_VARIANT }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" default_value { shape { } } }
attr {
  name: "layout" type: "list(int)" default_value { list { } }
}
-}
-- | 
prelinearizeTuple :: forall v'1 dtypes . (TensorTypes dtypes) =>
                     TensorList (v'1) dtypes -- ^ __inputs__
                     -> Tensor Build Variant -- ^ __output__
prelinearizeTuple :: TensorList v'1 dtypes -> Tensor Build Variant
prelinearizeTuple = OpParams -> TensorList v'1 dtypes -> Tensor Build Variant
forall (v'1 :: * -> *) (input_types :: [*]).
TensorTypes input_types =>
OpParams -> TensorList v'1 input_types -> Tensor Build Variant
prelinearizeTuple' OpParams
forall a. a -> a
id
prelinearizeTuple' :: forall v'1 dtypes . (TensorTypes dtypes) => OpParams ->
                      TensorList (v'1) dtypes -- ^ __inputs__
                      -> Tensor Build Variant -- ^ __output__
prelinearizeTuple' :: OpParams -> TensorList v'1 dtypes -> Tensor Build Variant
prelinearizeTuple' op'options :: OpParams
op'options inputs :: TensorList v'1 dtypes
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 dtypes
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PrelinearizeTuple"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "dtypes" }
output_arg { name: "output" type: DT_VARIANT }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "shapes" type: "list(shape)" }
attr {
  name: "layouts" type: "list(int)" default_value { list { } }
}
-}
-- | 
preventGradient :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                   -> Tensor Build t -- ^ __output__
preventGradient :: Tensor v'1 t -> Tensor Build t
preventGradient = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
preventGradient' OpParams
forall a. a -> a
id
preventGradient' :: forall v'1 t . (TensorType t) => OpParams ->
                    Tensor v'1 t -- ^ __input__
                    -> Tensor Build t -- ^ __output__
preventGradient' :: OpParams -> Tensor v'1 t -> Tensor Build t
preventGradient' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PreventGradient"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "message" type: "string" default_value { s: "" } }
-}
-- | 
print :: forall v'1 v'2 t u m' . (MonadBuild m', TensorType t, TensorTypes u) =>
         Tensor v'1 t -- ^ __input__
         -> TensorList (v'2) u -- ^ __data__
         -> m' (Tensor Value t) -- ^ __output__
print :: Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
print = OpParams -> Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t (u :: [*]) (m' :: * -> *).
(MonadBuild m', TensorType t, TensorTypes u) =>
OpParams -> Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
print' OpParams
forall a. a -> a
id
print' :: forall v'1 v'2 t u m' . (MonadBuild m', TensorType t,
                                   TensorTypes u) => OpParams ->
          Tensor v'1 t -- ^ __input__
          -> TensorList (v'2) u -- ^ __data__
          -> m' (Tensor Value t) -- ^ __output__
print' :: OpParams -> Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
print' op'options :: OpParams
op'options input :: Tensor v'1 t
input data' :: TensorList v'2 u
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             TensorList v'2 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 u
data']
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Print"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy u -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy u
forall k (t :: k). Proxy t
Proxy :: Proxy u)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "data" type_list_attr: "U" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "U" type: "list(type)" has_minimum: true }
attr { name: "message" type: "string" default_value { s: "" } }
attr { name: "first_n" type: "int" default_value { i: -1 } }
attr { name: "summarize" type: "int" default_value { i: 3 } }
-}
-- | 
printV2 :: forall v'1 m' . (MonadBuild m') =>
           Tensor v'1 Data.ByteString.ByteString -- ^ __input__
           -> m' (ControlNode)
printV2 :: Tensor v'1 ByteString -> m' ControlNode
printV2 = OpParams -> Tensor v'1 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' ControlNode
printV2' OpParams
forall a. a -> a
id
printV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
            Tensor v'1 Data.ByteString.ByteString -- ^ __input__
            -> m' (ControlNode)
printV2' :: OpParams -> Tensor v'1 ByteString -> m' ControlNode
printV2' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PrintV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
attr {
  name: "output_stream" type: "string" default_value { s: "stderr" }
}
attr { name: "end" type: "string" default_value { s: "\n" } }
-}
-- | 
priorityQueue :: forall m' . (MonadBuild m') =>
                 m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
priorityQueue :: m' (Tensor Ref ByteString)
priorityQueue = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
priorityQueue' OpParams
forall a. a -> a
id
priorityQueue' :: forall m' . (MonadBuild m') => OpParams ->
                  m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
priorityQueue' :: OpParams -> m' (Tensor Ref ByteString)
priorityQueue' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PriorityQueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "component_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "shapes" type: "list(shape)" has_minimum: true }
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
priorityQueueV2 :: forall m' . (MonadBuild m') =>
                   m' (Tensor Value ResourceHandle) -- ^ __handle__
priorityQueueV2 :: m' (Tensor Value ResourceHandle)
priorityQueueV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
priorityQueueV2' OpParams
forall a. a -> a
id
priorityQueueV2' :: forall m' . (MonadBuild m') => OpParams ->
                    m' (Tensor Value ResourceHandle) -- ^ __handle__
priorityQueueV2' :: OpParams -> m' (Tensor Value ResourceHandle)
priorityQueueV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PriorityQueueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "component_types"
  type: "list(type)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "shapes" type: "list(shape)" has_minimum: true }
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
privateThreadPoolDataset :: [DataType] -- ^ __output_types__
                            -> Tensor v'1 Variant -- ^ __input_dataset__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __num_threads__
                            -> Tensor Build Variant -- ^ __handle__
privateThreadPoolDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
privateThreadPoolDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
privateThreadPoolDataset' OpParams
forall a. a -> a
id
privateThreadPoolDataset' :: OpParams ->
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __num_threads__
                             -> Tensor Build Variant -- ^ __handle__
privateThreadPoolDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
privateThreadPoolDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                          num_threads :: Tensor v'2 Int64
num_threads | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_threads]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PrivateThreadPoolDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_threads" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
prod :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tidx) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build t -- ^ __output__
prod :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
prod = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
prod' OpParams
forall a. a -> a
id
prod' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t,
                                  OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] tidx) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 tidx -- ^ __reduction_indices__
         -> Tensor Build t -- ^ __output__
prod' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
prod' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Prod"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
pyFunc :: forall v'1 tin tout m' . (MonadBuild m', TensorTypes tin,
                                    TensorTypes tout) =>
          ByteString -- ^ __token__
          -> TensorList (v'1) tin -- ^ __input__
          -> m' (TensorList (Value) tout) -- ^ __output__
pyFunc :: ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
pyFunc = OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
forall (v'1 :: * -> *) (tin :: [*]) (tout :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes tin, TensorTypes tout) =>
OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
pyFunc' OpParams
forall a. a -> a
id
pyFunc' :: forall v'1 tin tout m' . (MonadBuild m', TensorTypes tin,
                                     TensorTypes tout) => OpParams ->
           ByteString -- ^ __token__
           -> TensorList (v'1) tin -- ^ __input__
           -> m' (TensorList (Value) tout) -- ^ __output__
pyFunc' :: OpParams
-> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
pyFunc' op'options :: OpParams
op'options token :: ByteString
token input :: TensorList v'1 tin
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value tout) -> m' (TensorList Value tout)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value tout) -> m' (TensorList Value tout))
-> Build (TensorList Value tout) -> m' (TensorList Value tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tin
input]
        [Int64] -> OpDef -> Build (TensorList Value tout)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "PyFunc"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tin -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tin
forall k (t :: k). Proxy t
Proxy :: Proxy tin)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tout -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tout
forall k (t :: k). Proxy t
Proxy :: Proxy tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "token" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
token
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_list_attr: "Tin" }
output_arg { name: "output" type_list_attr: "Tout" }
attr { name: "token" type: "string" }
attr { name: "Tin" type: "list(type)" has_minimum: true }
attr { name: "Tout" type: "list(type)" has_minimum: true }
-}
-- | 
pyFuncStateless :: forall v'1 tin tout . (TensorTypes tin, TensorTypes tout) =>
                   ByteString -- ^ __token__
                   -> TensorList (v'1) tin -- ^ __input__
                   -> TensorList (Build) tout -- ^ __output__
pyFuncStateless :: ByteString -> TensorList v'1 tin -> TensorList Build tout
pyFuncStateless = OpParams
-> ByteString -> TensorList v'1 tin -> TensorList Build tout
forall (v'1 :: * -> *) (tinputs :: [*]) (toutputs :: [*]).
(TensorTypes tinputs, TensorTypes toutputs) =>
OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
pyFuncStateless' OpParams
forall a. a -> a
id
pyFuncStateless' :: forall v'1 tin tout . (TensorTypes tin, TensorTypes tout) =>
                    OpParams ->
                    ByteString -- ^ __token__
                    -> TensorList (v'1) tin -- ^ __input__
                    -> TensorList (Build) tout -- ^ __output__
pyFuncStateless' :: OpParams
-> ByteString -> TensorList v'1 tin -> TensorList Build tout
pyFuncStateless' op'options :: OpParams
op'options token :: ByteString
token input :: TensorList v'1 tin
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build tout)
-> Build OpDef -> TensorList Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tin
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "PyFuncStateless"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tin -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tin
forall k (t :: k). Proxy t
Proxy :: Proxy tin)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tout -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tout
forall k (t :: k). Proxy t
Proxy :: Proxy tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "token" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
token
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_list_attr: "Tin" }
output_arg { name: "output" type_list_attr: "Tout" }
attr { name: "token" type: "string" }
attr { name: "Tin" type: "list(type)" has_minimum: true }
attr { name: "Tout" type: "list(type)" has_minimum: true }
-}
-- | 
qr :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                              (Data.Complex.Complex Float), Data.Word.Word16,
                              Double, Float] t) => Tensor v'1 t -- ^ __input__
      -> (Tensor Build t, Tensor Build t) -- ^ (__q__, __r__)
      --
      -- * __q__
      --
      -- * __r__
qr :: Tensor v'1 t -> (Tensor Build t, Tensor Build t)
qr = OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
qr' OpParams
forall a. a -> a
id
qr' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => OpParams ->
       Tensor v'1 t -- ^ __input__
       -> (Tensor Build t, Tensor Build t) -- ^ (__q__, __r__)
       --
       -- * __q__
       --
       -- * __r__
qr' :: OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
qr' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Qr"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "q" type_attr: "T" }
output_arg { name: "r" type_attr: "T" }
attr {
  name: "full_matrices" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
quantizeAndDequantize :: forall v'1 t . (OneOf '[Data.Word.Word16, Double,
                                                 Float] t) =>
                         Tensor v'1 t -- ^ __input__
                         -> Tensor Build t -- ^ __output__
quantizeAndDequantize :: Tensor v'1 t -> Tensor Build t
quantizeAndDequantize = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
quantizeAndDequantize' OpParams
forall a. a -> a
id
quantizeAndDequantize' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double,
                                                  Float] t) => OpParams ->
                          Tensor v'1 t -- ^ __input__
                          -> Tensor Build t -- ^ __output__
quantizeAndDequantize' :: OpParams -> Tensor v'1 t -> Tensor Build t
quantizeAndDequantize' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizeAndDequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "signed_input" type: "bool" default_value { b: true }
}
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "range_given" type: "bool" default_value { b: false }
}
attr { name: "input_min" type: "float" default_value { f: 0.0 } }
attr { name: "input_max" type: "float" default_value { f: 0.0 } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
quantizeAndDequantizeV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                           Double, Float] t) =>
                           Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 t -- ^ __input_min__
                           -> Tensor v'3 t -- ^ __input_max__
                           -> Tensor Build t -- ^ __output__
quantizeAndDequantizeV2 :: Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
quantizeAndDequantizeV2 = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
quantizeAndDequantizeV2' OpParams
forall a. a -> a
id
quantizeAndDequantizeV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                            Double, Float] t) =>
                            OpParams ->
                            Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 t -- ^ __input_min__
                            -> Tensor v'3 t -- ^ __input_max__
                            -> Tensor Build t -- ^ __output__
quantizeAndDequantizeV2' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
quantizeAndDequantizeV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_min :: Tensor v'2 t
input_min
                         input_max :: Tensor v'3 t
input_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_min,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizeAndDequantizeV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_min" type_attr: "T" }
input_arg { name: "input_max" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "signed_input" type: "bool" default_value { b: true }
}
attr { name: "num_bits" type: "int" default_value { i: 8 } }
attr {
  name: "range_given" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "round_mode"
  type: "string"
  default_value { s: "HALF_TO_EVEN" }
  allowed_values { list { s: "HALF_TO_EVEN" s: "HALF_UP" } }
}
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
attr { name: "axis" type: "int" default_value { i: -1 } }
-}
-- | 
quantizeAndDequantizeV3 :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                           Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 t -- ^ __input_min__
                           -> Tensor v'3 t -- ^ __input_max__
                           -> Tensor v'4 Data.Int.Int32 -- ^ __num_bits__
                           -> Tensor Build t -- ^ __output__
quantizeAndDequantizeV3 :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor Build t
quantizeAndDequantizeV3 = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor Build t
quantizeAndDequantizeV3' OpParams
forall a. a -> a
id
quantizeAndDequantizeV3' :: forall v'1 v'2 v'3 v'4
                            t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
                            OpParams ->
                            Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 t -- ^ __input_min__
                            -> Tensor v'3 t -- ^ __input_max__
                            -> Tensor v'4 Data.Int.Int32 -- ^ __num_bits__
                            -> Tensor Build t -- ^ __output__
quantizeAndDequantizeV3' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 Int32
-> Tensor Build t
quantizeAndDequantizeV3' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_min :: Tensor v'2 t
input_min input_max :: Tensor v'3 t
input_max
                         num_bits :: Tensor v'4 Int32
num_bits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_min,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
input_max,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
num_bits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizeAndDequantizeV3"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_min" type_attr: "T" }
input_arg { name: "input_max" type_attr: "T" }
input_arg { name: "num_bits" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "signed_input" type: "bool" default_value { b: true }
}
attr { name: "range_given" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
attr { name: "axis" type: "int" default_value { i: -1 } }
-}
-- | 
quantizeDownAndShrinkRange :: forall v'1 v'2 v'3 tinput
                              out_type . (OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Word.Word16,
                                                  Data.Word.Word8] tinput,
                                          OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Word.Word16,
                                                  Data.Word.Word8] out_type) =>
                              Tensor v'1 tinput -- ^ __input__
                              -> Tensor v'2 Float -- ^ __input_min__
                              -> Tensor v'3 Float -- ^ __input_max__
                              -> (Tensor Build out_type, Tensor Build Float,
                                  Tensor Build Float)
                              -- ^ (__output__, __output_min__, __output_max__)
                              --
                              -- * __output__
                              --
                              -- * __output_min__
                              --
                              -- * __output_max__
quantizeDownAndShrinkRange :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizeDownAndShrinkRange = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tinput
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizeDownAndShrinkRange' OpParams
forall a. a -> a
id
quantizeDownAndShrinkRange' :: forall v'1 v'2 v'3 tinput
                               out_type . (OneOf '[Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Word.Word16,
                                                   Data.Word.Word8] tinput,
                                           OneOf '[Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Word.Word16,
                                                   Data.Word.Word8] out_type) =>
                               OpParams ->
                               Tensor v'1 tinput -- ^ __input__
                               -> Tensor v'2 Float -- ^ __input_min__
                               -> Tensor v'3 Float -- ^ __input_max__
                               -> (Tensor Build out_type, Tensor Build Float,
                                   Tensor Build Float)
                               -- ^ (__output__, __output_min__, __output_max__)
                               --
                               -- * __output__
                               --
                               -- * __output_min__
                               --
                               -- * __output_max__
quantizeDownAndShrinkRange' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizeDownAndShrinkRange' op'options :: OpParams
op'options input :: Tensor v'1 tinput
input input_min :: Tensor v'2 Float
input_min
                            input_max :: Tensor v'3 Float
input_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
input_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizeDownAndShrinkRange"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizeV2 :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] t) =>
              Tensor v'1 Float -- ^ __input__
              -> Tensor v'2 Float -- ^ __min_range__
              -> Tensor v'3 Float -- ^ __max_range__
              -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
              -- ^ (__output__, __output_min__, __output_max__)
              --
              -- * __output__
              --
              -- * __output_min__
              --
              -- * __output_max__
quantizeV2 :: Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizeV2 = OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int16, Int32, Word16, Word8] t =>
OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizeV2' OpParams
forall a. a -> a
id
quantizeV2' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] t) =>
               OpParams ->
               Tensor v'1 Float -- ^ __input__
               -> Tensor v'2 Float -- ^ __min_range__
               -> Tensor v'3 Float -- ^ __max_range__
               -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
               -- ^ (__output__, __output_min__, __output_max__)
               --
               -- * __output__
               --
               -- * __output_min__
               --
               -- * __output_max__
quantizeV2' :: OpParams
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizeV2' op'options :: OpParams
op'options input :: Tensor v'1 Float
input min_range :: Tensor v'2 Float
min_range max_range :: Tensor v'3 Float
max_range | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_range,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_range]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizeV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_FLOAT }
input_arg { name: "min_range" type: DT_FLOAT }
input_arg { name: "max_range" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "mode"
  type: "string"
  default_value { s: "MIN_COMBINED" }
  allowed_values {
    list { s: "MIN_COMBINED" s: "MIN_FIRST" s: "SCALED" }
  }
}
attr {
  name: "round_mode"
  type: "string"
  default_value { s: "HALF_AWAY_FROM_ZERO" }
  allowed_values {
    list { s: "HALF_AWAY_FROM_ZERO" s: "HALF_TO_EVEN" }
  }
}
attr {
  name: "narrow_range" type: "bool" default_value { b: false }
}
attr { name: "axis" type: "int" default_value { i: -1 } }
attr {
  name: "ensure_minimum_range"
  type: "float"
  default_value { f: 1.0e-2 }
}
-}
-- | 
quantizedAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16, Data.Word.Word8] t1,
                           OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16, Data.Word.Word8] t2,
                           OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16,
                                   Data.Word.Word8] toutput) =>
                Tensor v'1 t1 -- ^ __x__
                -> Tensor v'2 t2 -- ^ __y__
                -> Tensor v'3 Float -- ^ __min_x__
                -> Tensor v'4 Float -- ^ __max_x__
                -> Tensor v'5 Float -- ^ __min_y__
                -> Tensor v'6 Float -- ^ __max_y__
                -> (Tensor Build toutput, Tensor Build Float,
                    Tensor Build Float) -- ^ (__z__, __min_z__, __max_z__)
                --
                -- * __z__
                --
                -- * __min_z__
                --
                -- * __max_z__
quantizedAdd :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedAdd = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t1 t2 toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedAdd' OpParams
forall a. a -> a
id
quantizedAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                 toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16, Data.Word.Word8] t1,
                            OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16, Data.Word.Word8] t2,
                            OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16,
                                    Data.Word.Word8] toutput) => OpParams ->
                 Tensor v'1 t1 -- ^ __x__
                 -> Tensor v'2 t2 -- ^ __y__
                 -> Tensor v'3 Float -- ^ __min_x__
                 -> Tensor v'4 Float -- ^ __max_x__
                 -> Tensor v'5 Float -- ^ __min_y__
                 -> Tensor v'6 Float -- ^ __max_y__
                 -> (Tensor Build toutput, Tensor Build Float,
                     Tensor Build Float) -- ^ (__z__, __min_z__, __max_z__)
                 --
                 -- * __z__
                 --
                 -- * __min_z__
                 --
                 -- * __max_z__
quantizedAdd' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedAdd' op'options :: OpParams
op'options x :: Tensor v'1 t1
x y :: Tensor v'2 t2
y min_x :: Tensor v'3 Float
min_x max_x :: Tensor v'4 Float
max_x min_y :: Tensor v'5 Float
min_y max_y :: Tensor v'6 Float
max_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
x,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
y,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_x,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_x,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_y,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T1" }
input_arg { name: "y" type_attr: "T2" }
input_arg { name: "min_x" type: DT_FLOAT }
input_arg { name: "max_x" type: DT_FLOAT }
input_arg { name: "min_y" type: DT_FLOAT }
input_arg { name: "max_y" type: DT_FLOAT }
output_arg { name: "z" type_attr: "Toutput" }
output_arg { name: "min_z" type: DT_FLOAT }
output_arg { name: "max_z" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedAvgPool :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Word.Word16,
                                                    Data.Word.Word8] t) =>
                    ByteString -- ^ __padding__
                    -> Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 Float -- ^ __min_input__
                    -> Tensor v'3 Float -- ^ __max_input__
                    -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                    -- ^ (__output__, __min_output__, __max_output__)
                    --
                    -- * __output__
                    --
                    -- * __min_output__
                    --
                    -- * __max_output__
quantizedAvgPool :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedAvgPool = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int16, Int32, Word16, Word8] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedAvgPool' OpParams
forall a. a -> a
id
quantizedAvgPool' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] t) =>
                     OpParams ->
                     ByteString -- ^ __padding__
                     -> Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 Float -- ^ __min_input__
                     -> Tensor v'3 Float -- ^ __max_input__
                     -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                     -- ^ (__output__, __min_output__, __max_output__)
                     --
                     -- * __output__
                     --
                     -- * __min_output__
                     --
                     -- * __max_output__
quantizedAvgPool' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedAvgPool' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input min_input :: Tensor v'2 Float
min_input
                  max_input :: Tensor v'3 Float
max_input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_input,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedAvgPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "ksize" type: "list(int)" }
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
quantizedBatchNormWithGlobalNormalization :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7
                                             v'8 v'9 v'10 v'11 v'12 v'13 v'14
                                             v'15 tinput
                                             out_type . (OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] tinput,
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] out_type) =>
                                             Bool -- ^ __scale_after_normalization__
                                             -> Float -- ^ __variance_epsilon__
                                             -> Tensor v'1 tinput -- ^ __t__
                                             -> Tensor v'2 Float -- ^ __t_min__
                                             -> Tensor v'3 Float -- ^ __t_max__
                                             -> Tensor v'4 tinput -- ^ __m__
                                             -> Tensor v'5 Float -- ^ __m_min__
                                             -> Tensor v'6 Float -- ^ __m_max__
                                             -> Tensor v'7 tinput -- ^ __v__
                                             -> Tensor v'8 Float -- ^ __v_min__
                                             -> Tensor v'9 Float -- ^ __v_max__
                                             -> Tensor v'10 tinput -- ^ __beta__
                                             -> Tensor v'11 Float -- ^ __beta_min__
                                             -> Tensor v'12 Float -- ^ __beta_max__
                                             -> Tensor v'13 tinput -- ^ __gamma__
                                             -> Tensor v'14 Float -- ^ __gamma_min__
                                             -> Tensor v'15 Float -- ^ __gamma_max__
                                             -> (Tensor Build out_type,
                                                 Tensor Build Float,
                                                 Tensor Build Float)
                                             -- ^ (__result__, __result_min__, __result_max__)
                                             --
                                             -- * __result__
                                             --
                                             -- * __result_min__
                                             --
                                             -- * __result_max__
quantizedBatchNormWithGlobalNormalization :: Bool
-> Float
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 tinput
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 tinput
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tinput
-> Tensor v'11 Float
-> Tensor v'12 Float
-> Tensor v'13 tinput
-> Tensor v'14 Float
-> Tensor v'15 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedBatchNormWithGlobalNormalization = OpParams
-> Bool
-> Float
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 tinput
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 tinput
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tinput
-> Tensor v'11 Float
-> Tensor v'12 Float
-> Tensor v'13 tinput
-> Tensor v'14 Float
-> Tensor v'15 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) (v'13 :: * -> *) (v'14 :: * -> *) (v'15 :: * -> *)
       tinput out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Bool
-> Float
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 tinput
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 tinput
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tinput
-> Tensor v'11 Float
-> Tensor v'12 Float
-> Tensor v'13 tinput
-> Tensor v'14 Float
-> Tensor v'15 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedBatchNormWithGlobalNormalization' OpParams
forall a. a -> a
id
quantizedBatchNormWithGlobalNormalization' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7
                                              v'8 v'9 v'10 v'11 v'12 v'13 v'14
                                              v'15 tinput
                                              out_type . (OneOf '[Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word8] tinput,
                                                          OneOf '[Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word8] out_type) =>
                                              OpParams ->
                                              Bool -- ^ __scale_after_normalization__
                                              -> Float -- ^ __variance_epsilon__
                                              -> Tensor v'1 tinput -- ^ __t__
                                              -> Tensor v'2 Float -- ^ __t_min__
                                              -> Tensor v'3 Float -- ^ __t_max__
                                              -> Tensor v'4 tinput -- ^ __m__
                                              -> Tensor v'5 Float -- ^ __m_min__
                                              -> Tensor v'6 Float -- ^ __m_max__
                                              -> Tensor v'7 tinput -- ^ __v__
                                              -> Tensor v'8 Float -- ^ __v_min__
                                              -> Tensor v'9 Float -- ^ __v_max__
                                              -> Tensor v'10 tinput -- ^ __beta__
                                              -> Tensor v'11 Float -- ^ __beta_min__
                                              -> Tensor v'12 Float -- ^ __beta_max__
                                              -> Tensor v'13 tinput -- ^ __gamma__
                                              -> Tensor v'14 Float -- ^ __gamma_min__
                                              -> Tensor v'15 Float -- ^ __gamma_max__
                                              -> (Tensor Build out_type,
                                                  Tensor Build Float,
                                                  Tensor Build Float)
                                              -- ^ (__result__, __result_min__, __result_max__)
                                              --
                                              -- * __result__
                                              --
                                              -- * __result_min__
                                              --
                                              -- * __result_max__
quantizedBatchNormWithGlobalNormalization' :: OpParams
-> Bool
-> Float
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 tinput
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 tinput
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tinput
-> Tensor v'11 Float
-> Tensor v'12 Float
-> Tensor v'13 tinput
-> Tensor v'14 Float
-> Tensor v'15 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedBatchNormWithGlobalNormalization' op'options :: OpParams
op'options scale_after_normalization :: Bool
scale_after_normalization
                                           variance_epsilon :: Float
variance_epsilon t :: Tensor v'1 tinput
t t_min :: Tensor v'2 Float
t_min t_max :: Tensor v'3 Float
t_max m :: Tensor v'4 tinput
m
                                           m_min :: Tensor v'5 Float
m_min m_max :: Tensor v'6 Float
m_max v :: Tensor v'7 tinput
v v_min :: Tensor v'8 Float
v_min v_max :: Tensor v'9 Float
v_max beta :: Tensor v'10 tinput
beta
                                           beta_min :: Tensor v'11 Float
beta_min beta_max :: Tensor v'12 Float
beta_max gamma :: Tensor v'13 tinput
gamma gamma_min :: Tensor v'14 Float
gamma_min
                                           gamma_max :: Tensor v'15 Float
gamma_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
t,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
t_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
t_max,
                                                             Tensor v'4 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tinput
m,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
m_min,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
m_max,
                                                             Tensor v'7 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 tinput
v,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
v_min,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
v_max,
                                                             Tensor v'10 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 tinput
beta,
                                                             Tensor v'11 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 Float
beta_min,
                                                             Tensor v'12 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 Float
beta_max,
                                                             Tensor v'13 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'13 tinput
gamma,
                                                             Tensor v'14 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'14 Float
gamma_min,
                                                             Tensor v'15 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'15 Float
gamma_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedBatchNormWithGlobalNormalization"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "scale_after_normalization" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
scale_after_normalization
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "variance_epsilon" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
variance_epsilon
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "t" type_attr: "Tinput" }
input_arg { name: "t_min" type: DT_FLOAT }
input_arg { name: "t_max" type: DT_FLOAT }
input_arg { name: "m" type_attr: "Tinput" }
input_arg { name: "m_min" type: DT_FLOAT }
input_arg { name: "m_max" type: DT_FLOAT }
input_arg { name: "v" type_attr: "Tinput" }
input_arg { name: "v_min" type: DT_FLOAT }
input_arg { name: "v_max" type: DT_FLOAT }
input_arg { name: "beta" type_attr: "Tinput" }
input_arg { name: "beta_min" type: DT_FLOAT }
input_arg { name: "beta_max" type: DT_FLOAT }
input_arg { name: "gamma" type_attr: "Tinput" }
input_arg { name: "gamma_min" type: DT_FLOAT }
input_arg { name: "gamma_max" type: DT_FLOAT }
output_arg { name: "result" type_attr: "out_type" }
output_arg { name: "result_min" type: DT_FLOAT }
output_arg { name: "result_max" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "variance_epsilon" type: "float" }
attr { name: "scale_after_normalization" type: "bool" }
-}
-- | 
quantizedBiasAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                    out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16, Data.Word.Word8] t1,
                                OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16, Data.Word.Word8] t2,
                                OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16,
                                        Data.Word.Word8] out_type) =>
                    Tensor v'1 t1 -- ^ __input__
                    -> Tensor v'2 t2 -- ^ __bias__
                    -> Tensor v'3 Float -- ^ __min_input__
                    -> Tensor v'4 Float -- ^ __max_input__
                    -> Tensor v'5 Float -- ^ __min_bias__
                    -> Tensor v'6 Float -- ^ __max_bias__
                    -> (Tensor Build out_type, Tensor Build Float,
                        Tensor Build Float)
                    -- ^ (__output__, __min_out__, __max_out__)
                    --
                    -- * __output__
                    --
                    -- * __min_out__
                    --
                    -- * __max_out__
quantizedBiasAdd :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedBiasAdd = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t1 t2 toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedBiasAdd' OpParams
forall a. a -> a
id
quantizedBiasAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                     out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Word.Word16, Data.Word.Word8] t1,
                                 OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Word.Word16, Data.Word.Word8] t2,
                                 OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Word.Word16,
                                         Data.Word.Word8] out_type) =>
                     OpParams ->
                     Tensor v'1 t1 -- ^ __input__
                     -> Tensor v'2 t2 -- ^ __bias__
                     -> Tensor v'3 Float -- ^ __min_input__
                     -> Tensor v'4 Float -- ^ __max_input__
                     -> Tensor v'5 Float -- ^ __min_bias__
                     -> Tensor v'6 Float -- ^ __max_bias__
                     -> (Tensor Build out_type, Tensor Build Float,
                         Tensor Build Float)
                     -- ^ (__output__, __min_out__, __max_out__)
                     --
                     -- * __output__
                     --
                     -- * __min_out__
                     --
                     -- * __max_out__
quantizedBiasAdd' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedBiasAdd' op'options :: OpParams
op'options input :: Tensor v'1 t1
input bias :: Tensor v'2 t2
bias min_input :: Tensor v'3 Float
min_input max_input :: Tensor v'4 Float
max_input min_bias :: Tensor v'5 Float
min_bias
                  max_bias :: Tensor v'6 Float
max_bias | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
input,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
bias,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_bias,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_bias]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedBiasAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T1" }
input_arg { name: "bias" type_attr: "T2" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_bias" type: DT_FLOAT }
input_arg { name: "max_bias" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedConcat :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                   Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
                   -> [Tensor v'2 t] -- ^ __values__
                   -> [Tensor v'3 Float] -- ^ __input_mins__
                   -> [Tensor v'4 Float] -- ^ __input_maxes__
                   -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                   -- ^ (__output__, __output_min__, __output_max__)
                   --
                   -- * __output__
                   --
                   -- * __output_min__
                   --
                   -- * __output_max__
quantizedConcat :: Tensor v'1 Int32
-> [Tensor v'2 t]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedConcat = OpParams
-> Tensor v'1 Int32
-> [Tensor v'2 t]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int32
-> [Tensor v'2 t]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedConcat' OpParams
forall a. a -> a
id
quantizedConcat' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                    Tensor v'1 Data.Int.Int32 -- ^ __concat_dim__
                    -> [Tensor v'2 t] -- ^ __values__
                    -> [Tensor v'3 Float] -- ^ __input_mins__
                    -> [Tensor v'4 Float] -- ^ __input_maxes__
                    -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                    -- ^ (__output__, __output_min__, __output_max__)
                    --
                    -- * __output__
                    --
                    -- * __output_min__
                    --
                    -- * __output_max__
quantizedConcat' :: OpParams
-> Tensor v'1 Int32
-> [Tensor v'2 t]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedConcat' op'options :: OpParams
op'options concat_dim :: Tensor v'1 Int32
concat_dim values :: [Tensor v'2 t]
values input_mins :: [Tensor v'3 Float]
input_mins
                 input_maxes :: [Tensor v'4 Float]
input_maxes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("values", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
values),
                                                     ("input_mins", [Tensor v'3 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Float]
input_mins),
                                                     ("input_maxes", [Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
input_maxes)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
concat_dim,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
values,
                                                             [Tensor v'3 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Float]
input_mins,
                                                             [Tensor v'4 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Float]
input_maxes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConcat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
values) :: Int64
{-
input_arg { name: "concat_dim" type: DT_INT32 }
input_arg { name: "values" type_attr: "T" number_attr: "N" }
input_arg { name: "input_mins" type: DT_FLOAT number_attr: "N" }
input_arg { name: "input_maxes" type: DT_FLOAT number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "T" type: "type" }
-}
-- | 
quantizedConv2D :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                   out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] tinput,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] tfilter,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] out_type) =>
                   ByteString -- ^ __padding__
                   -> Tensor v'1 tinput -- ^ __input__
                   -> Tensor v'2 tfilter -- ^ __filter__
                   -> Tensor v'3 Float -- ^ __min_input__
                   -> Tensor v'4 Float -- ^ __max_input__
                   -> Tensor v'5 Float -- ^ __min_filter__
                   -> Tensor v'6 Float -- ^ __max_filter__
                   -> (Tensor Build out_type, Tensor Build Float,
                       Tensor Build Float)
                   -- ^ (__output__, __min_output__, __max_output__)
                   --
                   -- * __output__
                   --
                   -- * __min_output__
                   --
                   -- * __max_output__
quantizedConv2D :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2D = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) tinput tfilter
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2D' OpParams
forall a. a -> a
id
quantizedConv2D' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                    out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16,
                                        Data.Word.Word8] tinput,
                                OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16,
                                        Data.Word.Word8] tfilter,
                                OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Word.Word16,
                                        Data.Word.Word8] out_type) =>
                    OpParams ->
                    ByteString -- ^ __padding__
                    -> Tensor v'1 tinput -- ^ __input__
                    -> Tensor v'2 tfilter -- ^ __filter__
                    -> Tensor v'3 Float -- ^ __min_input__
                    -> Tensor v'4 Float -- ^ __max_input__
                    -> Tensor v'5 Float -- ^ __min_filter__
                    -> Tensor v'6 Float -- ^ __max_filter__
                    -> (Tensor Build out_type, Tensor Build Float,
                        Tensor Build Float)
                    -- ^ (__output__, __min_output__, __max_output__)
                    --
                    -- * __output__
                    --
                    -- * __min_output__
                    --
                    -- * __max_output__
quantizedConv2D' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input max_input :: Tensor v'4 Float
max_input min_filter :: Tensor v'5 Float
min_filter
                 max_filter :: Tensor v'6 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
quantizedConv2DAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                          out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] tinput,
                                      OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] tfilter,
                                      OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] out_type) =>
                          ByteString -- ^ __padding__
                          -> Tensor v'1 tinput -- ^ __input__
                          -> Tensor v'2 tfilter -- ^ __filter__
                          -> Tensor v'3 Float -- ^ __min_input__
                          -> Tensor v'4 Float -- ^ __max_input__
                          -> Tensor v'5 Float -- ^ __min_filter__
                          -> Tensor v'6 Float -- ^ __max_filter__
                          -> (Tensor Build out_type, Tensor Build Float,
                              Tensor Build Float)
                          -- ^ (__output__, __min_output__, __max_output__)
                          --
                          -- * __output__
                          --
                          -- * __min_output__
                          --
                          -- * __max_output__
quantizedConv2DAndRelu :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRelu = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) tinput tfilter
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRelu' OpParams
forall a. a -> a
id
quantizedConv2DAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                           out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] tinput,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] tfilter,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] out_type) =>
                           OpParams ->
                           ByteString -- ^ __padding__
                           -> Tensor v'1 tinput -- ^ __input__
                           -> Tensor v'2 tfilter -- ^ __filter__
                           -> Tensor v'3 Float -- ^ __min_input__
                           -> Tensor v'4 Float -- ^ __max_input__
                           -> Tensor v'5 Float -- ^ __min_filter__
                           -> Tensor v'6 Float -- ^ __max_filter__
                           -> (Tensor Build out_type, Tensor Build Float,
                               Tensor Build Float)
                           -- ^ (__output__, __min_output__, __max_output__)
                           --
                           -- * __output__
                           --
                           -- * __min_output__
                           --
                           -- * __max_output__
quantizedConv2DAndRelu' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRelu' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input max_input :: Tensor v'4 Float
max_input
                        min_filter :: Tensor v'5 Float
min_filter max_filter :: Tensor v'6 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DAndRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                       tinput tfilter
                                       out_type . (OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] tinput,
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] tfilter,
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] out_type) =>
                                       ByteString -- ^ __padding__
                                       -> Tensor v'1 tinput -- ^ __input__
                                       -> Tensor v'2 tfilter -- ^ __filter__
                                       -> Tensor v'3 Float -- ^ __min_input__
                                       -> Tensor v'4 Float -- ^ __max_input__
                                       -> Tensor v'5 Float -- ^ __min_filter__
                                       -> Tensor v'6 Float -- ^ __max_filter__
                                       -> Tensor v'7 Float -- ^ __min_freezed_output__
                                       -> Tensor v'8 Float -- ^ __max_freezed_output__
                                       -> (Tensor Build out_type,
                                           Tensor Build Float,
                                           Tensor Build Float)
                                       -- ^ (__output__, __min_output__, __max_output__)
                                       --
                                       -- * __output__
                                       --
                                       -- * __min_output__
                                       --
                                       -- * __max_output__
quantizedConv2DAndReluAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndReluAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                        tinput tfilter
                                        out_type . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tinput,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tfilter,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] out_type) =>
                                        OpParams ->
                                        ByteString -- ^ __padding__
                                        -> Tensor v'1 tinput -- ^ __input__
                                        -> Tensor v'2 tfilter -- ^ __filter__
                                        -> Tensor v'3 Float -- ^ __min_input__
                                        -> Tensor v'4 Float -- ^ __max_input__
                                        -> Tensor v'5 Float -- ^ __min_filter__
                                        -> Tensor v'6 Float -- ^ __max_filter__
                                        -> Tensor v'7 Float -- ^ __min_freezed_output__
                                        -> Tensor v'8 Float -- ^ __max_freezed_output__
                                        -> (Tensor Build out_type,
                                            Tensor Build Float,
                                            Tensor Build Float)
                                        -- ^ (__output__, __min_output__, __max_output__)
                                        --
                                        -- * __output__
                                        --
                                        -- * __min_output__
                                        --
                                        -- * __max_output__
quantizedConv2DAndReluAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndReluAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input
                                     max_input :: Tensor v'4 Float
max_input min_filter :: Tensor v'5 Float
min_filter max_filter :: Tensor v'6 Float
max_filter
                                     min_freezed_output :: Tensor v'7 Float
min_freezed_output
                                     max_freezed_output :: Tensor v'8 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
min_freezed_output,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput
                                tfilter out_type . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tinput,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tfilter,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] out_type) =>
                                ByteString -- ^ __padding__
                                -> Tensor v'1 tinput -- ^ __input__
                                -> Tensor v'2 tfilter -- ^ __filter__
                                -> Tensor v'3 Float -- ^ __min_input__
                                -> Tensor v'4 Float -- ^ __max_input__
                                -> Tensor v'5 Float -- ^ __min_filter__
                                -> Tensor v'6 Float -- ^ __max_filter__
                                -> Tensor v'7 Float -- ^ __min_freezed_output__
                                -> Tensor v'8 Float -- ^ __max_freezed_output__
                                -> (Tensor Build out_type, Tensor Build Float,
                                    Tensor Build Float)
                                -- ^ (__output__, __min_output__, __max_output__)
                                --
                                -- * __output__
                                --
                                -- * __min_output__
                                --
                                -- * __max_output__
quantizedConv2DAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 tinput
                                 tfilter out_type . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] tinput,
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] tfilter,
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] out_type) =>
                                 OpParams ->
                                 ByteString -- ^ __padding__
                                 -> Tensor v'1 tinput -- ^ __input__
                                 -> Tensor v'2 tfilter -- ^ __filter__
                                 -> Tensor v'3 Float -- ^ __min_input__
                                 -> Tensor v'4 Float -- ^ __max_input__
                                 -> Tensor v'5 Float -- ^ __min_filter__
                                 -> Tensor v'6 Float -- ^ __max_filter__
                                 -> Tensor v'7 Float -- ^ __min_freezed_output__
                                 -> Tensor v'8 Float -- ^ __max_freezed_output__
                                 -> (Tensor Build out_type, Tensor Build Float,
                                     Tensor Build Float)
                                 -- ^ (__output__, __min_output__, __max_output__)
                                 --
                                 -- * __output__
                                 --
                                 -- * __min_output__
                                 --
                                 -- * __max_output__
quantizedConv2DAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input
                              max_input :: Tensor v'4 Float
max_input min_filter :: Tensor v'5 Float
min_filter max_filter :: Tensor v'6 Float
max_filter min_freezed_output :: Tensor v'7 Float
min_freezed_output
                              max_freezed_output :: Tensor v'8 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
min_freezed_output,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DPerChannel :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                             out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] tinput,
                                         OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] tfilter,
                                         OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] out_type) =>
                             ByteString -- ^ __padding__
                             -> Tensor v'1 tinput -- ^ __input__
                             -> Tensor v'2 tfilter -- ^ __filter__
                             -> Tensor v'3 Float -- ^ __min_input__
                             -> Tensor v'4 Float -- ^ __max_input__
                             -> Tensor v'5 Float -- ^ __min_filter__
                             -> Tensor v'6 Float -- ^ __max_filter__
                             -> (Tensor Build out_type, Tensor Build Float,
                                 Tensor Build Float)
                             -- ^ (__output__, __min_output__, __max_output__)
                             --
                             -- * __output__
                             --
                             -- * __min_output__
                             --
                             -- * __max_output__
quantizedConv2DPerChannel :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DPerChannel = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) tinput tfilter
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DPerChannel' OpParams
forall a. a -> a
id
quantizedConv2DPerChannel' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                              out_type . (OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Word.Word16,
                                                  Data.Word.Word8] tinput,
                                          OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Word.Word16,
                                                  Data.Word.Word8] tfilter,
                                          OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Word.Word16,
                                                  Data.Word.Word8] out_type) =>
                              OpParams ->
                              ByteString -- ^ __padding__
                              -> Tensor v'1 tinput -- ^ __input__
                              -> Tensor v'2 tfilter -- ^ __filter__
                              -> Tensor v'3 Float -- ^ __min_input__
                              -> Tensor v'4 Float -- ^ __max_input__
                              -> Tensor v'5 Float -- ^ __min_filter__
                              -> Tensor v'6 Float -- ^ __max_filter__
                              -> (Tensor Build out_type, Tensor Build Float,
                                  Tensor Build Float)
                              -- ^ (__output__, __min_output__, __max_output__)
                              --
                              -- * __output__
                              --
                              -- * __min_output__
                              --
                              -- * __max_output__
quantizedConv2DPerChannel' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DPerChannel' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input max_input :: Tensor v'4 Float
max_input
                           min_filter :: Tensor v'5 Float
min_filter max_filter :: Tensor v'6 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DPerChannel"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
quantizedConv2DWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter
                           out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] tinput,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] tfilter,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] out_type) =>
                           ByteString -- ^ __padding__
                           -> Tensor v'1 tinput -- ^ __input__
                           -> Tensor v'2 tfilter -- ^ __filter__
                           -> Tensor v'3 Float -- ^ __bias__
                           -> Tensor v'4 Float -- ^ __min_input__
                           -> Tensor v'5 Float -- ^ __max_input__
                           -> Tensor v'6 Float -- ^ __min_filter__
                           -> Tensor v'7 Float -- ^ __max_filter__
                           -> (Tensor Build out_type, Tensor Build Float,
                               Tensor Build Float)
                           -- ^ (__output__, __min_output__, __max_output__)
                           --
                           -- * __output__
                           --
                           -- * __min_output__
                           --
                           -- * __max_output__
quantizedConv2DWithBias :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBias = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBias' OpParams
forall a. a -> a
id
quantizedConv2DWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput tfilter
                            out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] tinput,
                                        OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] tfilter,
                                        OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] out_type) =>
                            OpParams ->
                            ByteString -- ^ __padding__
                            -> Tensor v'1 tinput -- ^ __input__
                            -> Tensor v'2 tfilter -- ^ __filter__
                            -> Tensor v'3 Float -- ^ __bias__
                            -> Tensor v'4 Float -- ^ __min_input__
                            -> Tensor v'5 Float -- ^ __max_input__
                            -> Tensor v'6 Float -- ^ __min_filter__
                            -> Tensor v'7 Float -- ^ __max_filter__
                            -> (Tensor Build out_type, Tensor Build Float,
                                Tensor Build Float)
                            -- ^ (__output__, __min_output__, __max_output__)
                            --
                            -- * __output__
                            --
                            -- * __min_output__
                            --
                            -- * __max_output__
quantizedConv2DWithBias' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBias' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 Float
bias min_input :: Tensor v'4 Float
min_input
                         max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter max_filter :: Tensor v'7 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBias"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput
                                  tfilter out_type . (OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Word.Word16,
                                                              Data.Word.Word8] tinput,
                                                      OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Word.Word16,
                                                              Data.Word.Word8] tfilter,
                                                      OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Word.Word16,
                                                              Data.Word.Word8] out_type) =>
                                  ByteString -- ^ __padding__
                                  -> Tensor v'1 tinput -- ^ __input__
                                  -> Tensor v'2 tfilter -- ^ __filter__
                                  -> Tensor v'3 Float -- ^ __bias__
                                  -> Tensor v'4 Float -- ^ __min_input__
                                  -> Tensor v'5 Float -- ^ __max_input__
                                  -> Tensor v'6 Float -- ^ __min_filter__
                                  -> Tensor v'7 Float -- ^ __max_filter__
                                  -> (Tensor Build out_type, Tensor Build Float,
                                      Tensor Build Float)
                                  -- ^ (__output__, __min_output__, __max_output__)
                                  --
                                  -- * __output__
                                  --
                                  -- * __min_output__
                                  --
                                  -- * __max_output__
quantizedConv2DWithBiasAndRelu :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRelu = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRelu' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput
                                   tfilter out_type . (OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] tinput,
                                                       OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] tfilter,
                                                       OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] out_type) =>
                                   OpParams ->
                                   ByteString -- ^ __padding__
                                   -> Tensor v'1 tinput -- ^ __input__
                                   -> Tensor v'2 tfilter -- ^ __filter__
                                   -> Tensor v'3 Float -- ^ __bias__
                                   -> Tensor v'4 Float -- ^ __min_input__
                                   -> Tensor v'5 Float -- ^ __max_input__
                                   -> Tensor v'6 Float -- ^ __min_filter__
                                   -> Tensor v'7 Float -- ^ __max_filter__
                                   -> (Tensor Build out_type,
                                       Tensor Build Float, Tensor Build Float)
                                   -- ^ (__output__, __min_output__, __max_output__)
                                   --
                                   -- * __output__
                                   --
                                   -- * __min_output__
                                   --
                                   -- * __max_output__
quantizedConv2DWithBiasAndRelu' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRelu' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 Float
bias min_input :: Tensor v'4 Float
min_input
                                max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                max_filter :: Tensor v'7 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasAndRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6
                                               v'7 v'8 v'9 tinput tfilter tbias
                                               out_type . (OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] tinput,
                                                           OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] tfilter,
                                                           OneOf '[Data.Int.Int32,
                                                                   Float] tbias,
                                                           OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] out_type) =>
                                               ByteString -- ^ __padding__
                                               -> Tensor v'1 tinput -- ^ __input__
                                               -> Tensor v'2 tfilter -- ^ __filter__
                                               -> Tensor v'3 tbias -- ^ __bias__
                                               -> Tensor v'4 Float -- ^ __min_input__
                                               -> Tensor v'5 Float -- ^ __max_input__
                                               -> Tensor v'6 Float -- ^ __min_filter__
                                               -> Tensor v'7 Float -- ^ __max_filter__
                                               -> Tensor v'8 Float -- ^ __min_freezed_output__
                                               -> Tensor v'9 Float -- ^ __max_freezed_output__
                                               -> (Tensor Build out_type,
                                                   Tensor Build Float,
                                                   Tensor Build Float)
                                               -- ^ (__output__, __min_output__, __max_output__)
                                               --
                                               -- * __output__
                                               --
                                               -- * __min_output__
                                               --
                                               -- * __max_output__
quantizedConv2DWithBiasAndReluAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndReluAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) tinput tfilter tbias out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                                                v'7 v'8 v'9 tinput tfilter tbias
                                                out_type . (OneOf '[Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word8] tinput,
                                                            OneOf '[Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word8] tfilter,
                                                            OneOf '[Data.Int.Int32,
                                                                    Float] tbias,
                                                            OneOf '[Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word8] out_type) =>
                                                OpParams ->
                                                ByteString -- ^ __padding__
                                                -> Tensor v'1 tinput -- ^ __input__
                                                -> Tensor v'2 tfilter -- ^ __filter__
                                                -> Tensor v'3 tbias -- ^ __bias__
                                                -> Tensor v'4 Float -- ^ __min_input__
                                                -> Tensor v'5 Float -- ^ __max_input__
                                                -> Tensor v'6 Float -- ^ __min_filter__
                                                -> Tensor v'7 Float -- ^ __max_filter__
                                                -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                -> (Tensor Build out_type,
                                                    Tensor Build Float,
                                                    Tensor Build Float)
                                                -- ^ (__output__, __min_output__, __max_output__)
                                                --
                                                -- * __output__
                                                --
                                                -- * __min_output__
                                                --
                                                -- * __max_output__
quantizedConv2DWithBiasAndReluAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndReluAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter
                                             bias :: Tensor v'3 tbias
bias min_input :: Tensor v'4 Float
min_input max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                             max_filter :: Tensor v'7 Float
max_filter min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                             max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                        v'9 tinput tfilter tbias
                                        out_type . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tinput,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tfilter,
                                                    OneOf '[Data.Int.Int32,
                                                            Float] tbias,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] out_type) =>
                                        ByteString -- ^ __padding__
                                        -> Tensor v'1 tinput -- ^ __input__
                                        -> Tensor v'2 tfilter -- ^ __filter__
                                        -> Tensor v'3 tbias -- ^ __bias__
                                        -> Tensor v'4 Float -- ^ __min_input__
                                        -> Tensor v'5 Float -- ^ __max_input__
                                        -> Tensor v'6 Float -- ^ __min_filter__
                                        -> Tensor v'7 Float -- ^ __max_filter__
                                        -> Tensor v'8 Float -- ^ __min_freezed_output__
                                        -> Tensor v'9 Float -- ^ __max_freezed_output__
                                        -> (Tensor Build out_type,
                                            Tensor Build Float,
                                            Tensor Build Float)
                                        -- ^ (__output__, __min_output__, __max_output__)
                                        --
                                        -- * __output__
                                        --
                                        -- * __min_output__
                                        --
                                        -- * __max_output__
quantizedConv2DWithBiasAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) tinput tfilter tbias out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                         v'9 tinput tfilter tbias
                                         out_type . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] tinput,
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] tfilter,
                                                     OneOf '[Data.Int.Int32,
                                                             Float] tbias,
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] out_type) =>
                                         OpParams ->
                                         ByteString -- ^ __padding__
                                         -> Tensor v'1 tinput -- ^ __input__
                                         -> Tensor v'2 tfilter -- ^ __filter__
                                         -> Tensor v'3 tbias -- ^ __bias__
                                         -> Tensor v'4 Float -- ^ __min_input__
                                         -> Tensor v'5 Float -- ^ __max_input__
                                         -> Tensor v'6 Float -- ^ __min_filter__
                                         -> Tensor v'7 Float -- ^ __max_filter__
                                         -> Tensor v'8 Float -- ^ __min_freezed_output__
                                         -> Tensor v'9 Float -- ^ __max_freezed_output__
                                         -> (Tensor Build out_type,
                                             Tensor Build Float,
                                             Tensor Build Float)
                                         -- ^ (__output__, __min_output__, __max_output__)
                                         --
                                         -- * __output__
                                         --
                                         -- * __min_output__
                                         --
                                         -- * __max_output__
quantizedConv2DWithBiasAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 tbias
bias
                                      min_input :: Tensor v'4 Float
min_input max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter max_filter :: Tensor v'7 Float
max_filter
                                      min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                      max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasSignedSumAndReluAndRequantize :: forall v'1 v'2 v'3 v'4
                                                        v'5 v'6 v'7 v'8 v'9 v'10
                                                        v'11 v'12 tinput tfilter
                                                        tbias tsummand
                                                        out_type . (OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] tinput,
                                                                    OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] tfilter,
                                                                    OneOf '[Data.Int.Int32,
                                                                            Float] tbias,
                                                                    OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] tsummand,
                                                                    OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] out_type) =>
                                                        ByteString -- ^ __padding__
                                                        -> Tensor v'1 tinput -- ^ __input__
                                                        -> Tensor v'2 tfilter -- ^ __filter__
                                                        -> Tensor v'3 tbias -- ^ __bias__
                                                        -> Tensor v'4 Float -- ^ __min_input__
                                                        -> Tensor v'5 Float -- ^ __max_input__
                                                        -> Tensor v'6 Float -- ^ __min_filter__
                                                        -> Tensor v'7 Float -- ^ __max_filter__
                                                        -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                        -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                        -> Tensor v'10 tsummand -- ^ __summand__
                                                        -> Tensor v'11 Float -- ^ __min_summand__
                                                        -> Tensor v'12 Float -- ^ __max_summand__
                                                        -> (Tensor Build out_type,
                                                            Tensor Build Float,
                                                            Tensor Build Float)
                                                        -- ^ (__output__, __min_output__, __max_output__)
                                                        --
                                                        -- * __output__
                                                        --
                                                        -- * __min_output__
                                                        --
                                                        -- * __max_output__
quantizedConv2DWithBiasSignedSumAndReluAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSignedSumAndReluAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) tinput tfilter tbias tsummand out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] tsummand,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSignedSumAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasSignedSumAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4
                                                         v'5 v'6 v'7 v'8 v'9
                                                         v'10 v'11 v'12 tinput
                                                         tfilter tbias tsummand
                                                         out_type . (OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] tinput,
                                                                     OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] tfilter,
                                                                     OneOf '[Data.Int.Int32,
                                                                             Float] tbias,
                                                                     OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] tsummand,
                                                                     OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] out_type) =>
                                                         OpParams ->
                                                         ByteString -- ^ __padding__
                                                         -> Tensor v'1 tinput -- ^ __input__
                                                         -> Tensor v'2 tfilter -- ^ __filter__
                                                         -> Tensor v'3 tbias -- ^ __bias__
                                                         -> Tensor v'4 Float -- ^ __min_input__
                                                         -> Tensor v'5 Float -- ^ __max_input__
                                                         -> Tensor v'6 Float -- ^ __min_filter__
                                                         -> Tensor v'7 Float -- ^ __max_filter__
                                                         -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                         -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                         -> Tensor v'10 tsummand -- ^ __summand__
                                                         -> Tensor v'11 Float -- ^ __min_summand__
                                                         -> Tensor v'12 Float -- ^ __max_summand__
                                                         -> (Tensor Build out_type,
                                                             Tensor Build Float,
                                                             Tensor Build Float)
                                                         -- ^ (__output__, __min_output__, __max_output__)
                                                         --
                                                         -- * __output__
                                                         --
                                                         -- * __min_output__
                                                         --
                                                         -- * __max_output__
quantizedConv2DWithBiasSignedSumAndReluAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSignedSumAndReluAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input
                                                      filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 tbias
bias min_input :: Tensor v'4 Float
min_input
                                                      max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                                      max_filter :: Tensor v'7 Float
max_filter
                                                      min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                                      max_freezed_output :: Tensor v'9 Float
max_freezed_output summand :: Tensor v'10 tsummand
summand
                                                      min_summand :: Tensor v'11 Float
min_summand
                                                      max_summand :: Tensor v'12 Float
max_summand | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output,
                                                             Tensor v'10 tsummand -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 tsummand
summand,
                                                             Tensor v'11 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 Float
min_summand,
                                                             Tensor v'12 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 Float
max_summand]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasSignedSumAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsummand" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsummand -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsummand
forall a. HasCallStack => a
undefined :: tsummand)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
input_arg { name: "summand" type_attr: "Tsummand" }
input_arg { name: "min_summand" type: DT_FLOAT }
input_arg { name: "max_summand" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Tsummand"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasSumAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                     tinput tfilter
                                     out_type . (OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8] tinput,
                                                 OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8] tfilter,
                                                 OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8] out_type) =>
                                     ByteString -- ^ __padding__
                                     -> Tensor v'1 tinput -- ^ __input__
                                     -> Tensor v'2 tfilter -- ^ __filter__
                                     -> Tensor v'3 Float -- ^ __bias__
                                     -> Tensor v'4 Float -- ^ __min_input__
                                     -> Tensor v'5 Float -- ^ __max_input__
                                     -> Tensor v'6 Float -- ^ __min_filter__
                                     -> Tensor v'7 Float -- ^ __max_filter__
                                     -> Tensor v'8 Float -- ^ __summand__
                                     -> (Tensor Build out_type,
                                         Tensor Build Float, Tensor Build Float)
                                     -- ^ (__output__, __min_output__, __max_output__)
                                     --
                                     -- * __output__
                                     --
                                     -- * __min_output__
                                     --
                                     -- * __max_output__
quantizedConv2DWithBiasSumAndRelu :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndRelu = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndRelu' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasSumAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                      tinput tfilter
                                      out_type . (OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8] tinput,
                                                  OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8] tfilter,
                                                  OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8] out_type) =>
                                      OpParams ->
                                      ByteString -- ^ __padding__
                                      -> Tensor v'1 tinput -- ^ __input__
                                      -> Tensor v'2 tfilter -- ^ __filter__
                                      -> Tensor v'3 Float -- ^ __bias__
                                      -> Tensor v'4 Float -- ^ __min_input__
                                      -> Tensor v'5 Float -- ^ __max_input__
                                      -> Tensor v'6 Float -- ^ __min_filter__
                                      -> Tensor v'7 Float -- ^ __max_filter__
                                      -> Tensor v'8 Float -- ^ __summand__
                                      -> (Tensor Build out_type,
                                          Tensor Build Float,
                                          Tensor Build Float)
                                      -- ^ (__output__, __min_output__, __max_output__)
                                      --
                                      -- * __output__
                                      --
                                      -- * __min_output__
                                      --
                                      -- * __max_output__
quantizedConv2DWithBiasSumAndRelu' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndRelu' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 Float
bias
                                   min_input :: Tensor v'4 Float
min_input max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter max_filter :: Tensor v'7 Float
max_filter
                                   summand :: Tensor v'8 Float
summand | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
summand]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasSumAndRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "summand" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedConv2DWithBiasSumAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6
                                                  v'7 v'8 v'9 v'10 v'11 v'12
                                                  tinput tfilter tbias tsummand
                                                  out_type . (OneOf '[Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word8] tinput,
                                                              OneOf '[Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word8] tfilter,
                                                              OneOf '[Data.Int.Int32,
                                                                      Float] tbias,
                                                              OneOf '[Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word8] tsummand,
                                                              OneOf '[Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word8] out_type) =>
                                                  ByteString -- ^ __padding__
                                                  -> Tensor v'1 tinput -- ^ __input__
                                                  -> Tensor v'2 tfilter -- ^ __filter__
                                                  -> Tensor v'3 tbias -- ^ __bias__
                                                  -> Tensor v'4 Float -- ^ __min_input__
                                                  -> Tensor v'5 Float -- ^ __max_input__
                                                  -> Tensor v'6 Float -- ^ __min_filter__
                                                  -> Tensor v'7 Float -- ^ __max_filter__
                                                  -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                  -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                  -> Tensor v'10 tsummand -- ^ __summand__
                                                  -> Tensor v'11 Float -- ^ __min_summand__
                                                  -> Tensor v'12 Float -- ^ __max_summand__
                                                  -> (Tensor Build out_type,
                                                      Tensor Build Float,
                                                      Tensor Build Float)
                                                  -- ^ (__output__, __min_output__, __max_output__)
                                                  --
                                                  -- * __output__
                                                  --
                                                  -- * __min_output__
                                                  --
                                                  -- * __max_output__
quantizedConv2DWithBiasSumAndReluAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndReluAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *)
       (v'12 :: * -> *) tinput tfilter tbias tsummand out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] tsummand,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedConv2DWithBiasSumAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5
                                                   v'6 v'7 v'8 v'9 v'10 v'11
                                                   v'12 tinput tfilter tbias
                                                   tsummand
                                                   out_type . (OneOf '[Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word8] tinput,
                                                               OneOf '[Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word8] tfilter,
                                                               OneOf '[Data.Int.Int32,
                                                                       Float] tbias,
                                                               OneOf '[Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word8] tsummand,
                                                               OneOf '[Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word8] out_type) =>
                                                   OpParams ->
                                                   ByteString -- ^ __padding__
                                                   -> Tensor v'1 tinput -- ^ __input__
                                                   -> Tensor v'2 tfilter -- ^ __filter__
                                                   -> Tensor v'3 tbias -- ^ __bias__
                                                   -> Tensor v'4 Float -- ^ __min_input__
                                                   -> Tensor v'5 Float -- ^ __max_input__
                                                   -> Tensor v'6 Float -- ^ __min_filter__
                                                   -> Tensor v'7 Float -- ^ __max_filter__
                                                   -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                   -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                   -> Tensor v'10 tsummand -- ^ __summand__
                                                   -> Tensor v'11 Float -- ^ __min_summand__
                                                   -> Tensor v'12 Float -- ^ __max_summand__
                                                   -> (Tensor Build out_type,
                                                       Tensor Build Float,
                                                       Tensor Build Float)
                                                   -- ^ (__output__, __min_output__, __max_output__)
                                                   --
                                                   -- * __output__
                                                   --
                                                   -- * __min_output__
                                                   --
                                                   -- * __max_output__
quantizedConv2DWithBiasSumAndReluAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor v'10 tsummand
-> Tensor v'11 Float
-> Tensor v'12 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedConv2DWithBiasSumAndReluAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter
                                                bias :: Tensor v'3 tbias
bias min_input :: Tensor v'4 Float
min_input max_input :: Tensor v'5 Float
max_input
                                                min_filter :: Tensor v'6 Float
min_filter max_filter :: Tensor v'7 Float
max_filter
                                                min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                                max_freezed_output :: Tensor v'9 Float
max_freezed_output summand :: Tensor v'10 tsummand
summand
                                                min_summand :: Tensor v'11 Float
min_summand
                                                max_summand :: Tensor v'12 Float
max_summand | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output,
                                                             Tensor v'10 tsummand -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 tsummand
summand,
                                                             Tensor v'11 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 Float
min_summand,
                                                             Tensor v'12 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'12 Float
max_summand]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedConv2DWithBiasSumAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsummand" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsummand -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsummand
forall a. HasCallStack => a
undefined :: tsummand)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
input_arg { name: "summand" type_attr: "Tsummand" }
input_arg { name: "min_summand" type: DT_FLOAT }
input_arg { name: "max_summand" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Tsummand"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedDepthwiseConv2D :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                            out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] tinput,
                                        OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] tfilter,
                                        OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Word.Word16,
                                                Data.Word.Word8] out_type) =>
                            ByteString -- ^ __padding__
                            -> Tensor v'1 tinput -- ^ __input__
                            -> Tensor v'2 tfilter -- ^ __filter__
                            -> Tensor v'3 Float -- ^ __min_input__
                            -> Tensor v'4 Float -- ^ __max_input__
                            -> Tensor v'5 Float -- ^ __min_filter__
                            -> Tensor v'6 Float -- ^ __max_filter__
                            -> (Tensor Build out_type, Tensor Build Float,
                                Tensor Build Float)
                            -- ^ (__output__, __min_output__, __max_output__)
                            --
                            -- * __output__
                            --
                            -- * __min_output__
                            --
                            -- * __max_output__
quantizedDepthwiseConv2D :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2D = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) tinput tfilter
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2D' OpParams
forall a. a -> a
id
quantizedDepthwiseConv2D' :: forall v'1 v'2 v'3 v'4 v'5 v'6 tinput tfilter
                             out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] tinput,
                                         OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] tfilter,
                                         OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Word.Word16,
                                                 Data.Word.Word8] out_type) =>
                             OpParams ->
                             ByteString -- ^ __padding__
                             -> Tensor v'1 tinput -- ^ __input__
                             -> Tensor v'2 tfilter -- ^ __filter__
                             -> Tensor v'3 Float -- ^ __min_input__
                             -> Tensor v'4 Float -- ^ __max_input__
                             -> Tensor v'5 Float -- ^ __min_filter__
                             -> Tensor v'6 Float -- ^ __max_filter__
                             -> (Tensor Build out_type, Tensor Build Float,
                                 Tensor Build Float)
                             -- ^ (__output__, __min_output__, __max_output__)
                             --
                             -- * __output__
                             --
                             -- * __min_output__
                             --
                             -- * __max_output__
quantizedDepthwiseConv2D' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter min_input :: Tensor v'3 Float
min_input max_input :: Tensor v'4 Float
max_input
                          min_filter :: Tensor v'5 Float
min_filter max_filter :: Tensor v'6 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_input,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_filter,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedDepthwiseConv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
quantizedDepthwiseConv2DWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput
                                    tfilter out_type . (OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tinput,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tfilter,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] out_type) =>
                                    ByteString -- ^ __padding__
                                    -> Tensor v'1 tinput -- ^ __input__
                                    -> Tensor v'2 tfilter -- ^ __filter__
                                    -> Tensor v'3 Float -- ^ __bias__
                                    -> Tensor v'4 Float -- ^ __min_input__
                                    -> Tensor v'5 Float -- ^ __max_input__
                                    -> Tensor v'6 Float -- ^ __min_filter__
                                    -> Tensor v'7 Float -- ^ __max_filter__
                                    -> (Tensor Build out_type,
                                        Tensor Build Float, Tensor Build Float)
                                    -- ^ (__output__, __min_output__, __max_output__)
                                    --
                                    -- * __output__
                                    --
                                    -- * __min_output__
                                    --
                                    -- * __max_output__
quantizedDepthwiseConv2DWithBias :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBias = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBias' OpParams
forall a. a -> a
id
quantizedDepthwiseConv2DWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 tinput
                                     tfilter out_type . (OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] tinput,
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] tfilter,
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] out_type) =>
                                     OpParams ->
                                     ByteString -- ^ __padding__
                                     -> Tensor v'1 tinput -- ^ __input__
                                     -> Tensor v'2 tfilter -- ^ __filter__
                                     -> Tensor v'3 Float -- ^ __bias__
                                     -> Tensor v'4 Float -- ^ __min_input__
                                     -> Tensor v'5 Float -- ^ __max_input__
                                     -> Tensor v'6 Float -- ^ __min_filter__
                                     -> Tensor v'7 Float -- ^ __max_filter__
                                     -> (Tensor Build out_type,
                                         Tensor Build Float, Tensor Build Float)
                                     -- ^ (__output__, __min_output__, __max_output__)
                                     --
                                     -- * __output__
                                     --
                                     -- * __min_output__
                                     --
                                     -- * __max_output__
quantizedDepthwiseConv2DWithBias' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBias' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 Float
bias min_input :: Tensor v'4 Float
min_input
                                  max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                  max_filter :: Tensor v'7 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedDepthwiseConv2DWithBias"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
-}
-- | 
quantizedDepthwiseConv2DWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7
                                           tinput tfilter
                                           out_type . (OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] tinput,
                                                       OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] tfilter,
                                                       OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] out_type) =>
                                           ByteString -- ^ __padding__
                                           -> Tensor v'1 tinput -- ^ __input__
                                           -> Tensor v'2 tfilter -- ^ __filter__
                                           -> Tensor v'3 Float -- ^ __bias__
                                           -> Tensor v'4 Float -- ^ __min_input__
                                           -> Tensor v'5 Float -- ^ __max_input__
                                           -> Tensor v'6 Float -- ^ __min_filter__
                                           -> Tensor v'7 Float -- ^ __max_filter__
                                           -> (Tensor Build out_type,
                                               Tensor Build Float,
                                               Tensor Build Float)
                                           -- ^ (__output__, __min_output__, __max_output__)
                                           --
                                           -- * __output__
                                           --
                                           -- * __min_output__
                                           --
                                           -- * __max_output__
quantizedDepthwiseConv2DWithBiasAndRelu :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndRelu = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       tinput tfilter out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndRelu' OpParams
forall a. a -> a
id
quantizedDepthwiseConv2DWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7
                                            tinput tfilter
                                            out_type . (OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tinput,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tfilter,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] out_type) =>
                                            OpParams ->
                                            ByteString -- ^ __padding__
                                            -> Tensor v'1 tinput -- ^ __input__
                                            -> Tensor v'2 tfilter -- ^ __filter__
                                            -> Tensor v'3 Float -- ^ __bias__
                                            -> Tensor v'4 Float -- ^ __min_input__
                                            -> Tensor v'5 Float -- ^ __max_input__
                                            -> Tensor v'6 Float -- ^ __min_filter__
                                            -> Tensor v'7 Float -- ^ __max_filter__
                                            -> (Tensor Build out_type,
                                                Tensor Build Float,
                                                Tensor Build Float)
                                            -- ^ (__output__, __min_output__, __max_output__)
                                            --
                                            -- * __output__
                                            --
                                            -- * __min_output__
                                            --
                                            -- * __max_output__
quantizedDepthwiseConv2DWithBiasAndRelu' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndRelu' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 Float
bias
                                         min_input :: Tensor v'4 Float
min_input max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                         max_filter :: Tensor v'7 Float
max_filter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedDepthwiseConv2DWithBiasAndRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4
                                                        v'5 v'6 v'7 v'8 v'9
                                                        tinput tfilter tbias
                                                        out_type . (OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] tinput,
                                                                    OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] tfilter,
                                                                    OneOf '[Data.Int.Int32,
                                                                            Float] tbias,
                                                                    OneOf '[Data.Int.Int16,
                                                                            Data.Int.Int32,
                                                                            Data.Word.Word16,
                                                                            Data.Word.Word8] out_type) =>
                                                        ByteString -- ^ __padding__
                                                        -> Tensor v'1 tinput -- ^ __input__
                                                        -> Tensor v'2 tfilter -- ^ __filter__
                                                        -> Tensor v'3 tbias -- ^ __bias__
                                                        -> Tensor v'4 Float -- ^ __min_input__
                                                        -> Tensor v'5 Float -- ^ __max_input__
                                                        -> Tensor v'6 Float -- ^ __min_filter__
                                                        -> Tensor v'7 Float -- ^ __max_filter__
                                                        -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                        -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                        -> (Tensor Build out_type,
                                                            Tensor Build Float,
                                                            Tensor Build Float)
                                                        -- ^ (__output__, __min_output__, __max_output__)
                                                        --
                                                        -- * __output__
                                                        --
                                                        -- * __min_output__
                                                        --
                                                        -- * __max_output__
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize :: ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize = OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) tinput tfilter tbias out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] tfilter,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4
                                                         v'5 v'6 v'7 v'8 v'9
                                                         tinput tfilter tbias
                                                         out_type . (OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] tinput,
                                                                     OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] tfilter,
                                                                     OneOf '[Data.Int.Int32,
                                                                             Float] tbias,
                                                                     OneOf '[Data.Int.Int16,
                                                                             Data.Int.Int32,
                                                                             Data.Word.Word16,
                                                                             Data.Word.Word8] out_type) =>
                                                         OpParams ->
                                                         ByteString -- ^ __padding__
                                                         -> Tensor v'1 tinput -- ^ __input__
                                                         -> Tensor v'2 tfilter -- ^ __filter__
                                                         -> Tensor v'3 tbias -- ^ __bias__
                                                         -> Tensor v'4 Float -- ^ __min_input__
                                                         -> Tensor v'5 Float -- ^ __max_input__
                                                         -> Tensor v'6 Float -- ^ __min_filter__
                                                         -> Tensor v'7 Float -- ^ __max_filter__
                                                         -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                         -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                         -> (Tensor Build out_type,
                                                             Tensor Build Float,
                                                             Tensor Build Float)
                                                         -- ^ (__output__, __min_output__, __max_output__)
                                                         --
                                                         -- * __output__
                                                         --
                                                         -- * __min_output__
                                                         --
                                                         -- * __max_output__
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' :: OpParams
-> ByteString
-> Tensor v'1 tinput
-> Tensor v'2 tfilter
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedDepthwiseConv2DWithBiasAndReluAndRequantize' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 tinput
input
                                                      filter :: Tensor v'2 tfilter
filter bias :: Tensor v'3 tbias
bias min_input :: Tensor v'4 Float
min_input
                                                      max_input :: Tensor v'5 Float
max_input min_filter :: Tensor v'6 Float
min_filter
                                                      max_filter :: Tensor v'7 Float
max_filter
                                                      min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                                      max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 tfilter -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tfilter
filter,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_input,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_input,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_filter,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_filter,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedDepthwiseConv2DWithBiasAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tfilter" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tfilter -> DataType
forall a. TensorType a => a -> DataType
tensorType (tfilter
forall a. HasCallStack => a
undefined :: tfilter)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "filter" type_attr: "Tfilter" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
input_arg { name: "min_filter" type: DT_FLOAT }
input_arg { name: "max_filter" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tfilter"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "padding_list" type: "list(int)" default_value { list { } }
}
-}
-- | 
quantizedInstanceNorm :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8] t) =>
                         Tensor v'1 t -- ^ __x__
                         -> Tensor v'2 Float -- ^ __x_min__
                         -> Tensor v'3 Float -- ^ __x_max__
                         -> (Tensor Build t, Tensor Build Float,
                             Tensor Build Float)
                         -- ^ (__y__, __y_min__, __y_max__)
                         --
                         -- * __y__
                         --
                         -- * __y_min__
                         --
                         -- * __y_max__
quantizedInstanceNorm :: Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedInstanceNorm = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int16, Int32, Word16, Word8] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedInstanceNorm' OpParams
forall a. a -> a
id
quantizedInstanceNorm' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __x__
                          -> Tensor v'2 Float -- ^ __x_min__
                          -> Tensor v'3 Float -- ^ __x_max__
                          -> (Tensor Build t, Tensor Build Float,
                              Tensor Build Float)
                          -- ^ (__y__, __y_min__, __y_max__)
                          --
                          -- * __y__
                          --
                          -- * __y_min__
                          --
                          -- * __y_max__
quantizedInstanceNorm' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedInstanceNorm' op'options :: OpParams
op'options x :: Tensor v'1 t
x x_min :: Tensor v'2 Float
x_min x_max :: Tensor v'3 Float
x_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
x_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
x_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedInstanceNorm"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "x_min" type: DT_FLOAT }
input_arg { name: "x_max" type: DT_FLOAT }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "y_min" type: DT_FLOAT }
output_arg { name: "y_max" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "output_range_given" type: "bool" default_value { b: false }
}
attr { name: "given_y_min" type: "float" default_value { f: 0.0 } }
attr { name: "given_y_max" type: "float" default_value { f: 0.0 } }
attr {
  name: "variance_epsilon" type: "float" default_value { f: 1.0e-5 }
}
attr {
  name: "min_separation" type: "float" default_value { f: 1.0e-3 }
}
-}
-- | 
quantizedMatMul :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                   toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Word.Word16, Data.Word.Word8] t1,
                              OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Word.Word16, Data.Word.Word8] t2,
                              OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Word.Word16,
                                      Data.Word.Word8] toutput) =>
                   Tensor v'1 t1 -- ^ __a__
                   -> Tensor v'2 t2 -- ^ __b__
                   -> Tensor v'3 Float -- ^ __min_a__
                   -> Tensor v'4 Float -- ^ __max_a__
                   -> Tensor v'5 Float -- ^ __min_b__
                   -> Tensor v'6 Float -- ^ __max_b__
                   -> (Tensor Build toutput, Tensor Build Float,
                       Tensor Build Float)
                   -- ^ (__out__, __min_out__, __max_out__)
                   --
                   -- * __out__
                   --
                   -- * __min_out__
                   --
                   -- * __max_out__
quantizedMatMul :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMul = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t1 t2 toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMul' OpParams
forall a. a -> a
id
quantizedMatMul' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                    toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16, Data.Word.Word8] t1,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16, Data.Word.Word8] t2,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] toutput) => OpParams ->
                    Tensor v'1 t1 -- ^ __a__
                    -> Tensor v'2 t2 -- ^ __b__
                    -> Tensor v'3 Float -- ^ __min_a__
                    -> Tensor v'4 Float -- ^ __max_a__
                    -> Tensor v'5 Float -- ^ __min_b__
                    -> Tensor v'6 Float -- ^ __max_b__
                    -> (Tensor Build toutput, Tensor Build Float,
                        Tensor Build Float)
                    -- ^ (__out__, __min_out__, __max_out__)
                    --
                    -- * __out__
                    --
                    -- * __min_out__
                    --
                    -- * __max_out__
quantizedMatMul' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMul' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b min_a :: Tensor v'3 Float
min_a max_a :: Tensor v'4 Float
max_a min_b :: Tensor v'5 Float
min_b max_b :: Tensor v'6 Float
max_b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_a,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_b,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "Tactivation"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedMatMulWithBias :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias
                           toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] t1,
                                      OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] t2,
                                      OneOf '[Data.Int.Int32, Float] tbias,
                                      OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Word.Word16,
                                              Data.Word.Word8] toutput) =>
                           Tensor v'1 t1 -- ^ __a__
                           -> Tensor v'2 t2 -- ^ __b__
                           -> Tensor v'3 tbias -- ^ __bias__
                           -> Tensor v'4 Float -- ^ __min_a__
                           -> Tensor v'5 Float -- ^ __max_a__
                           -> Tensor v'6 Float -- ^ __min_b__
                           -> Tensor v'7 Float -- ^ __max_b__
                           -> (Tensor Build toutput, Tensor Build Float,
                               Tensor Build Float)
                           -- ^ (__out__, __min_out__, __max_out__)
                           --
                           -- * __out__
                           --
                           -- * __min_out__
                           --
                           -- * __max_out__
quantizedMatMulWithBias :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBias = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t1
       t2 tbias toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBias' OpParams
forall a. a -> a
id
quantizedMatMulWithBias' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2 tbias
                            toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] t1,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] t2,
                                       OneOf '[Data.Int.Int32, Float] tbias,
                                       OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] toutput) =>
                            OpParams ->
                            Tensor v'1 t1 -- ^ __a__
                            -> Tensor v'2 t2 -- ^ __b__
                            -> Tensor v'3 tbias -- ^ __bias__
                            -> Tensor v'4 Float -- ^ __min_a__
                            -> Tensor v'5 Float -- ^ __max_a__
                            -> Tensor v'6 Float -- ^ __min_b__
                            -> Tensor v'7 Float -- ^ __max_b__
                            -> (Tensor Build toutput, Tensor Build Float,
                                Tensor Build Float)
                            -- ^ (__out__, __min_out__, __max_out__)
                            --
                            -- * __out__
                            --
                            -- * __min_out__
                            --
                            -- * __max_out__
quantizedMatMulWithBias' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBias' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b bias :: Tensor v'3 tbias
bias min_a :: Tensor v'4 Float
min_a max_a :: Tensor v'5 Float
max_a min_b :: Tensor v'6 Float
min_b
                         max_b :: Tensor v'7 Float
max_b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_a,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_b,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMulWithBias"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "input_quant_mode"
  type: "string"
  default_value { s: "MIN_FIRST" }
  allowed_values { list { s: "MIN_FIRST" s: "SCALED" } }
}
-}
-- | 
quantizedMatMulWithBiasAndDequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                        v'9 t1 t2 tbias
                                        toutput . (OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] t1,
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] t2,
                                                   OneOf '[Data.Int.Int32,
                                                           Float] tbias,
                                                   OneOf '[Float] toutput) =>
                                        Tensor v'1 t1 -- ^ __a__
                                        -> Tensor v'2 t2 -- ^ __b__
                                        -> Tensor v'3 tbias -- ^ __bias__
                                        -> Tensor v'4 Float -- ^ __min_a__
                                        -> Tensor v'5 Float -- ^ __max_a__
                                        -> Tensor v'6 Float -- ^ __min_b__
                                        -> Tensor v'7 Float -- ^ __max_b__
                                        -> Tensor v'8 Float -- ^ __min_freezed_output__
                                        -> Tensor v'9 Float -- ^ __max_freezed_output__
                                        -> Tensor Build toutput -- ^ __out__
quantizedMatMulWithBiasAndDequantize :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor Build toutput
quantizedMatMulWithBiasAndDequantize = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor Build toutput
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t1 t2 tbias toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int32, Float] tbias, OneOf '[Float] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor Build toutput
quantizedMatMulWithBiasAndDequantize' OpParams
forall a. a -> a
id
quantizedMatMulWithBiasAndDequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                         v'9 t1 t2 tbias
                                         toutput . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] t1,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] t2,
                                                    OneOf '[Data.Int.Int32,
                                                            Float] tbias,
                                                    OneOf '[Float] toutput) =>
                                         OpParams ->
                                         Tensor v'1 t1 -- ^ __a__
                                         -> Tensor v'2 t2 -- ^ __b__
                                         -> Tensor v'3 tbias -- ^ __bias__
                                         -> Tensor v'4 Float -- ^ __min_a__
                                         -> Tensor v'5 Float -- ^ __max_a__
                                         -> Tensor v'6 Float -- ^ __min_b__
                                         -> Tensor v'7 Float -- ^ __max_b__
                                         -> Tensor v'8 Float -- ^ __min_freezed_output__
                                         -> Tensor v'9 Float -- ^ __max_freezed_output__
                                         -> Tensor Build toutput -- ^ __out__
quantizedMatMulWithBiasAndDequantize' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> Tensor Build toutput
quantizedMatMulWithBiasAndDequantize' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b bias :: Tensor v'3 tbias
bias min_a :: Tensor v'4 Float
min_a max_a :: Tensor v'5 Float
max_a min_b :: Tensor v'6 Float
min_b
                                      max_b :: Tensor v'7 Float
max_b min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                      max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build toutput
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build toutput)
-> Build OpDef -> Tensor Build toutput
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_a,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_b,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_b,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMulWithBiasAndDequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Toutput"
  type: "type"
  allowed_values { list { type: DT_FLOAT } }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "input_quant_mode"
  type: "string"
  default_value { s: "MIN_FIRST" }
  allowed_values { list { s: "MIN_FIRST" s: "SCALED" } }
}
-}
-- | 
quantizedMatMulWithBiasAndRelu :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2
                                  toutput . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] t1,
                                             OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] t2,
                                             OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] toutput) =>
                                  Tensor v'1 t1 -- ^ __a__
                                  -> Tensor v'2 t2 -- ^ __b__
                                  -> Tensor v'3 Float -- ^ __bias__
                                  -> Tensor v'4 Float -- ^ __min_a__
                                  -> Tensor v'5 Float -- ^ __max_a__
                                  -> Tensor v'6 Float -- ^ __min_b__
                                  -> Tensor v'7 Float -- ^ __max_b__
                                  -> (Tensor Build toutput, Tensor Build Float,
                                      Tensor Build Float)
                                  -- ^ (__out__, __min_out__, __max_out__)
                                  --
                                  -- * __out__
                                  --
                                  -- * __min_out__
                                  --
                                  -- * __max_out__
quantizedMatMulWithBiasAndRelu :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRelu = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t1
       t2 toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRelu' OpParams
forall a. a -> a
id
quantizedMatMulWithBiasAndRelu' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t1 t2
                                   toutput . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8] t1,
                                              OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8] t2,
                                              OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8] toutput) =>
                                   OpParams ->
                                   Tensor v'1 t1 -- ^ __a__
                                   -> Tensor v'2 t2 -- ^ __b__
                                   -> Tensor v'3 Float -- ^ __bias__
                                   -> Tensor v'4 Float -- ^ __min_a__
                                   -> Tensor v'5 Float -- ^ __max_a__
                                   -> Tensor v'6 Float -- ^ __min_b__
                                   -> Tensor v'7 Float -- ^ __max_b__
                                   -> (Tensor Build toutput, Tensor Build Float,
                                       Tensor Build Float)
                                   -- ^ (__out__, __min_out__, __max_out__)
                                   --
                                   -- * __out__
                                   --
                                   -- * __min_out__
                                   --
                                   -- * __max_out__
quantizedMatMulWithBiasAndRelu' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRelu' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b bias :: Tensor v'3 Float
bias min_a :: Tensor v'4 Float
min_a max_a :: Tensor v'5 Float
max_a min_b :: Tensor v'6 Float
min_b
                                max_b :: Tensor v'7 Float
max_b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_a,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_b,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMulWithBiasAndRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "bias" type: DT_FLOAT }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "input_quant_mode"
  type: "string"
  default_value { s: "MIN_FIRST" }
  allowed_values { list { s: "MIN_FIRST" s: "SCALED" } }
}
-}
-- | 
quantizedMatMulWithBiasAndReluAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6
                                               v'7 v'8 v'9 t1 t2 tbias
                                               toutput . (OneOf '[Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word8] t1,
                                                          OneOf '[Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word8] t2,
                                                          OneOf '[Data.Int.Int32,
                                                                  Float] tbias,
                                                          OneOf '[Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word8] toutput) =>
                                               Tensor v'1 t1 -- ^ __a__
                                               -> Tensor v'2 t2 -- ^ __b__
                                               -> Tensor v'3 tbias -- ^ __bias__
                                               -> Tensor v'4 Float -- ^ __min_a__
                                               -> Tensor v'5 Float -- ^ __max_a__
                                               -> Tensor v'6 Float -- ^ __min_b__
                                               -> Tensor v'7 Float -- ^ __max_b__
                                               -> Tensor v'8 Float -- ^ __min_freezed_output__
                                               -> Tensor v'9 Float -- ^ __max_freezed_output__
                                               -> (Tensor Build toutput,
                                                   Tensor Build Float,
                                                   Tensor Build Float)
                                               -- ^ (__out__, __min_out__, __max_out__)
                                               --
                                               -- * __out__
                                               --
                                               -- * __min_out__
                                               --
                                               -- * __max_out__
quantizedMatMulWithBiasAndReluAndRequantize :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndReluAndRequantize = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t1 t2 tbias toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndReluAndRequantize' OpParams
forall a. a -> a
id
quantizedMatMulWithBiasAndReluAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                                                v'7 v'8 v'9 t1 t2 tbias
                                                toutput . (OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] t1,
                                                           OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] t2,
                                                           OneOf '[Data.Int.Int32,
                                                                   Float] tbias,
                                                           OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word8] toutput) =>
                                                OpParams ->
                                                Tensor v'1 t1 -- ^ __a__
                                                -> Tensor v'2 t2 -- ^ __b__
                                                -> Tensor v'3 tbias -- ^ __bias__
                                                -> Tensor v'4 Float -- ^ __min_a__
                                                -> Tensor v'5 Float -- ^ __max_a__
                                                -> Tensor v'6 Float -- ^ __min_b__
                                                -> Tensor v'7 Float -- ^ __max_b__
                                                -> Tensor v'8 Float -- ^ __min_freezed_output__
                                                -> Tensor v'9 Float -- ^ __max_freezed_output__
                                                -> (Tensor Build toutput,
                                                    Tensor Build Float,
                                                    Tensor Build Float)
                                                -- ^ (__out__, __min_out__, __max_out__)
                                                --
                                                -- * __out__
                                                --
                                                -- * __min_out__
                                                --
                                                -- * __max_out__
quantizedMatMulWithBiasAndReluAndRequantize' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndReluAndRequantize' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b bias :: Tensor v'3 tbias
bias min_a :: Tensor v'4 Float
min_a max_a :: Tensor v'5 Float
max_a
                                             min_b :: Tensor v'6 Float
min_b max_b :: Tensor v'7 Float
max_b min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                             max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_a,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_b,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_b,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMulWithBiasAndReluAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "input_quant_mode"
  type: "string"
  default_value { s: "MIN_FIRST" }
  allowed_values { list { s: "MIN_FIRST" s: "SCALED" } }
}
-}
-- | 
quantizedMatMulWithBiasAndRequantize :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                        v'9 t1 t2 tbias
                                        toutput . (OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] t1,
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] t2,
                                                   OneOf '[Data.Int.Int32,
                                                           Float] tbias,
                                                   OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Word.Word16,
                                                           Data.Word.Word8] toutput) =>
                                        Tensor v'1 t1 -- ^ __a__
                                        -> Tensor v'2 t2 -- ^ __b__
                                        -> Tensor v'3 tbias -- ^ __bias__
                                        -> Tensor v'4 Float -- ^ __min_a__
                                        -> Tensor v'5 Float -- ^ __max_a__
                                        -> Tensor v'6 Float -- ^ __min_b__
                                        -> Tensor v'7 Float -- ^ __max_b__
                                        -> Tensor v'8 Float -- ^ __min_freezed_output__
                                        -> Tensor v'9 Float -- ^ __max_freezed_output__
                                        -> (Tensor Build toutput,
                                            Tensor Build Float,
                                            Tensor Build Float)
                                        -- ^ (__out__, __min_out__, __max_out__)
                                        --
                                        -- * __out__
                                        --
                                        -- * __min_out__
                                        --
                                        -- * __max_out__
quantizedMatMulWithBiasAndRequantize :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRequantize = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t1 t2 tbias toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int32, Float] tbias,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRequantize' OpParams
forall a. a -> a
id
quantizedMatMulWithBiasAndRequantize' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                         v'9 t1 t2 tbias
                                         toutput . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] t1,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] t2,
                                                    OneOf '[Data.Int.Int32,
                                                            Float] tbias,
                                                    OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] toutput) =>
                                         OpParams ->
                                         Tensor v'1 t1 -- ^ __a__
                                         -> Tensor v'2 t2 -- ^ __b__
                                         -> Tensor v'3 tbias -- ^ __bias__
                                         -> Tensor v'4 Float -- ^ __min_a__
                                         -> Tensor v'5 Float -- ^ __max_a__
                                         -> Tensor v'6 Float -- ^ __min_b__
                                         -> Tensor v'7 Float -- ^ __max_b__
                                         -> Tensor v'8 Float -- ^ __min_freezed_output__
                                         -> Tensor v'9 Float -- ^ __max_freezed_output__
                                         -> (Tensor Build toutput,
                                             Tensor Build Float,
                                             Tensor Build Float)
                                         -- ^ (__out__, __min_out__, __max_out__)
                                         --
                                         -- * __out__
                                         --
                                         -- * __min_out__
                                         --
                                         -- * __max_out__
quantizedMatMulWithBiasAndRequantize' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 tbias
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> Tensor v'7 Float
-> Tensor v'8 Float
-> Tensor v'9 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMatMulWithBiasAndRequantize' op'options :: OpParams
op'options a :: Tensor v'1 t1
a b :: Tensor v'2 t2
b bias :: Tensor v'3 tbias
bias min_a :: Tensor v'4 Float
min_a max_a :: Tensor v'5 Float
max_a min_b :: Tensor v'6 Float
min_b
                                      max_b :: Tensor v'7 Float
max_b min_freezed_output :: Tensor v'8 Float
min_freezed_output
                                      max_freezed_output :: Tensor v'9 Float
max_freezed_output | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
a,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
b,
                                                             Tensor v'3 tbias -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tbias
bias,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
min_a,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
max_a,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
min_b,
                                                             Tensor v'7 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Float
max_b,
                                                             Tensor v'8 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Float
min_freezed_output,
                                                             Tensor v'9 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Float
max_freezed_output]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMatMulWithBiasAndRequantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tbias" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tbias -> DataType
forall a. TensorType a => a -> DataType
tensorType (tbias
forall a. HasCallStack => a
undefined :: tbias)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T1" }
input_arg { name: "b" type_attr: "T2" }
input_arg { name: "bias" type_attr: "Tbias" }
input_arg { name: "min_a" type: DT_FLOAT }
input_arg { name: "max_a" type: DT_FLOAT }
input_arg { name: "min_b" type: DT_FLOAT }
input_arg { name: "max_b" type: DT_FLOAT }
input_arg { name: "min_freezed_output" type: DT_FLOAT }
input_arg { name: "max_freezed_output" type: DT_FLOAT }
output_arg { name: "out" type_attr: "Toutput" }
output_arg { name: "min_out" type: DT_FLOAT }
output_arg { name: "max_out" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Tbias"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_QINT32 } }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "input_quant_mode"
  type: "string"
  default_value { s: "MIN_FIRST" }
  allowed_values { list { s: "MIN_FIRST" s: "SCALED" } }
}
-}
-- | 
quantizedMaxPool :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Word.Word16,
                                                    Data.Word.Word8] t) =>
                    ByteString -- ^ __padding__
                    -> Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 Float -- ^ __min_input__
                    -> Tensor v'3 Float -- ^ __max_input__
                    -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                    -- ^ (__output__, __min_output__, __max_output__)
                    --
                    -- * __output__
                    --
                    -- * __min_output__
                    --
                    -- * __max_output__
quantizedMaxPool :: ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedMaxPool = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int16, Int32, Word16, Word8] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedMaxPool' OpParams
forall a. a -> a
id
quantizedMaxPool' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8] t) =>
                     OpParams ->
                     ByteString -- ^ __padding__
                     -> Tensor v'1 t -- ^ __input__
                     -> Tensor v'2 Float -- ^ __min_input__
                     -> Tensor v'3 Float -- ^ __max_input__
                     -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                     -- ^ (__output__, __min_output__, __max_output__)
                     --
                     -- * __output__
                     --
                     -- * __min_output__
                     --
                     -- * __max_output__
quantizedMaxPool' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedMaxPool' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input min_input :: Tensor v'2 Float
min_input
                  max_input :: Tensor v'3 Float
max_input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_input,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMaxPool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "min_input" type: DT_FLOAT }
input_arg { name: "max_input" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "min_output" type: DT_FLOAT }
output_arg { name: "max_output" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "ksize" type: "list(int)" }
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
-}
-- | 
quantizedMul :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16, Data.Word.Word8] t1,
                           OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16, Data.Word.Word8] t2,
                           OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16,
                                   Data.Word.Word8] toutput) =>
                Tensor v'1 t1 -- ^ __x__
                -> Tensor v'2 t2 -- ^ __y__
                -> Tensor v'3 Float -- ^ __min_x__
                -> Tensor v'4 Float -- ^ __max_x__
                -> Tensor v'5 Float -- ^ __min_y__
                -> Tensor v'6 Float -- ^ __max_y__
                -> (Tensor Build toutput, Tensor Build Float,
                    Tensor Build Float) -- ^ (__z__, __min_z__, __max_z__)
                --
                -- * __z__
                --
                -- * __min_z__
                --
                -- * __max_z__
quantizedMul :: Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMul = OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t1 t2 toutput.
(OneOf '[Int16, Int32, Word16, Word8] t1,
 OneOf '[Int16, Int32, Word16, Word8] t2,
 OneOf '[Int16, Int32, Word16, Word8] toutput) =>
OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMul' OpParams
forall a. a -> a
id
quantizedMul' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2
                 toutput . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16, Data.Word.Word8] t1,
                            OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16, Data.Word.Word8] t2,
                            OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Word.Word16,
                                    Data.Word.Word8] toutput) => OpParams ->
                 Tensor v'1 t1 -- ^ __x__
                 -> Tensor v'2 t2 -- ^ __y__
                 -> Tensor v'3 Float -- ^ __min_x__
                 -> Tensor v'4 Float -- ^ __max_x__
                 -> Tensor v'5 Float -- ^ __min_y__
                 -> Tensor v'6 Float -- ^ __max_y__
                 -> (Tensor Build toutput, Tensor Build Float,
                     Tensor Build Float) -- ^ (__z__, __min_z__, __max_z__)
                 --
                 -- * __z__
                 --
                 -- * __min_z__
                 --
                 -- * __max_z__
quantizedMul' :: OpParams
-> Tensor v'1 t1
-> Tensor v'2 t2
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> Tensor v'6 Float
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
quantizedMul' op'options :: OpParams
op'options x :: Tensor v'1 t1
x y :: Tensor v'2 t2
y min_x :: Tensor v'3 Float
min_x max_x :: Tensor v'4 Float
max_x min_y :: Tensor v'5 Float
min_y max_y :: Tensor v'6 Float
max_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build toutput, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build toutput, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t1 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t1
x,
                                                             Tensor v'2 t2 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t2
y,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_x,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_x,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
min_y,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
max_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T1" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t1 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t1
forall a. HasCallStack => a
undefined :: t1)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T2" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t2 -> DataType
forall a. TensorType a => a -> DataType
tensorType (t2
forall a. HasCallStack => a
undefined :: t2)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ toutput -> DataType
forall a. TensorType a => a -> DataType
tensorType (toutput
forall a. HasCallStack => a
undefined :: toutput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T1" }
input_arg { name: "y" type_attr: "T2" }
input_arg { name: "min_x" type: DT_FLOAT }
input_arg { name: "max_x" type: DT_FLOAT }
input_arg { name: "min_y" type: DT_FLOAT }
input_arg { name: "max_y" type: DT_FLOAT }
output_arg { name: "z" type_attr: "Toutput" }
output_arg { name: "min_z" type: DT_FLOAT }
output_arg { name: "max_z" type: DT_FLOAT }
attr {
  name: "T1"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "T2"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "Toutput"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedRelu :: forall v'1 v'2 v'3 tinput out_type . (OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] tinput,
                                                       OneOf '[Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Word.Word16,
                                                               Data.Word.Word8] out_type) =>
                 Tensor v'1 tinput -- ^ __features__
                 -> Tensor v'2 Float -- ^ __min_features__
                 -> Tensor v'3 Float -- ^ __max_features__
                 -> (Tensor Build out_type, Tensor Build Float,
                     Tensor Build Float)
                 -- ^ (__activations__, __min_activations__, __max_activations__)
                 --
                 -- * __activations__
                 --
                 -- * __min_activations__
                 --
                 -- * __max_activations__
quantizedRelu :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tinput
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu' OpParams
forall a. a -> a
id
quantizedRelu' :: forall v'1 v'2 v'3 tinput out_type . (OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tinput,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] out_type) =>
                  OpParams ->
                  Tensor v'1 tinput -- ^ __features__
                  -> Tensor v'2 Float -- ^ __min_features__
                  -> Tensor v'3 Float -- ^ __max_features__
                  -> (Tensor Build out_type, Tensor Build Float,
                      Tensor Build Float)
                  -- ^ (__activations__, __min_activations__, __max_activations__)
                  --
                  -- * __activations__
                  --
                  -- * __min_activations__
                  --
                  -- * __max_activations__
quantizedRelu' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu' op'options :: OpParams
op'options features :: Tensor v'1 tinput
features min_features :: Tensor v'2 Float
min_features
               max_features :: Tensor v'3 Float
max_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
features,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_features,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedRelu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "Tinput" }
input_arg { name: "min_features" type: DT_FLOAT }
input_arg { name: "max_features" type: DT_FLOAT }
output_arg { name: "activations" type_attr: "out_type" }
output_arg { name: "min_activations" type: DT_FLOAT }
output_arg { name: "max_activations" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedRelu6 :: forall v'1 v'2 v'3 tinput out_type . (OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] tinput,
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Word.Word16,
                                                                Data.Word.Word8] out_type) =>
                  Tensor v'1 tinput -- ^ __features__
                  -> Tensor v'2 Float -- ^ __min_features__
                  -> Tensor v'3 Float -- ^ __max_features__
                  -> (Tensor Build out_type, Tensor Build Float,
                      Tensor Build Float)
                  -- ^ (__activations__, __min_activations__, __max_activations__)
                  --
                  -- * __activations__
                  --
                  -- * __min_activations__
                  --
                  -- * __max_activations__
quantizedRelu6 :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu6 = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tinput
       out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu6' OpParams
forall a. a -> a
id
quantizedRelu6' :: forall v'1 v'2 v'3 tinput out_type . (OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] tinput,
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] out_type) =>
                   OpParams ->
                   Tensor v'1 tinput -- ^ __features__
                   -> Tensor v'2 Float -- ^ __min_features__
                   -> Tensor v'3 Float -- ^ __max_features__
                   -> (Tensor Build out_type, Tensor Build Float,
                       Tensor Build Float)
                   -- ^ (__activations__, __min_activations__, __max_activations__)
                   --
                   -- * __activations__
                   --
                   -- * __min_activations__
                   --
                   -- * __max_activations__
quantizedRelu6' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedRelu6' op'options :: OpParams
op'options features :: Tensor v'1 tinput
features min_features :: Tensor v'2 Float
min_features
                max_features :: Tensor v'3 Float
max_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
features,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
min_features,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
max_features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedRelu6"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "Tinput" }
input_arg { name: "min_features" type: DT_FLOAT }
input_arg { name: "max_features" type: DT_FLOAT }
output_arg { name: "activations" type_attr: "out_type" }
output_arg { name: "min_activations" type: DT_FLOAT }
output_arg { name: "max_activations" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedReluX :: forall v'1 v'2 v'3 v'4 tinput
                  out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Word.Word16, Data.Word.Word8] tinput,
                              OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Word.Word16,
                                      Data.Word.Word8] out_type) =>
                  Tensor v'1 tinput -- ^ __features__
                  -> Tensor v'2 Float -- ^ __max_value__
                  -> Tensor v'3 Float -- ^ __min_features__
                  -> Tensor v'4 Float -- ^ __max_features__
                  -> (Tensor Build out_type, Tensor Build Float,
                      Tensor Build Float)
                  -- ^ (__activations__, __min_activations__, __max_activations__)
                  --
                  -- * __activations__
                  --
                  -- * __min_activations__
                  --
                  -- * __max_activations__
quantizedReluX :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedReluX = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) tinput out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedReluX' OpParams
forall a. a -> a
id
quantizedReluX' :: forall v'1 v'2 v'3 v'4 tinput
                   out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] tinput,
                               OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Word.Word16,
                                       Data.Word.Word8] out_type) => OpParams ->
                   Tensor v'1 tinput -- ^ __features__
                   -> Tensor v'2 Float -- ^ __max_value__
                   -> Tensor v'3 Float -- ^ __min_features__
                   -> Tensor v'4 Float -- ^ __max_features__
                   -> (Tensor Build out_type, Tensor Build Float,
                       Tensor Build Float)
                   -- ^ (__activations__, __min_activations__, __max_activations__)
                   --
                   -- * __activations__
                   --
                   -- * __min_activations__
                   --
                   -- * __max_activations__
quantizedReluX' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
quantizedReluX' op'options :: OpParams
op'options features :: Tensor v'1 tinput
features max_value :: Tensor v'2 Float
max_value min_features :: Tensor v'3 Float
min_features
                max_features :: Tensor v'4 Float
max_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
features,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
max_value,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_features,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max_features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedReluX"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "Tinput" }
input_arg { name: "max_value" type: DT_FLOAT }
input_arg { name: "min_features" type: DT_FLOAT }
input_arg { name: "max_features" type: DT_FLOAT }
output_arg { name: "activations" type_attr: "out_type" }
output_arg { name: "min_activations" type: DT_FLOAT }
output_arg { name: "max_activations" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
quantizedReshape :: forall v'1 v'2 v'3 v'4 t tshape . (TensorType t,
                                                       OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tshape) =>
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor v'2 tshape -- ^ __shape__
                    -> Tensor v'3 Float -- ^ __input_min__
                    -> Tensor v'4 Float -- ^ __input_max__
                    -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                    -- ^ (__output__, __output_min__, __output_max__)
                    --
                    -- * __output__
                    --
                    -- * __output_min__
                    --
                    -- * __output_max__
quantizedReshape :: Tensor v'1 t
-> Tensor v'2 tshape
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedReshape = OpParams
-> Tensor v'1 t
-> Tensor v'2 tshape
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tshape.
(TensorType t, OneOf '[Int32, Int64] tshape) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tshape
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedReshape' OpParams
forall a. a -> a
id
quantizedReshape' :: forall v'1 v'2 v'3 v'4 t tshape . (TensorType t,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tshape) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __tensor__
                     -> Tensor v'2 tshape -- ^ __shape__
                     -> Tensor v'3 Float -- ^ __input_min__
                     -> Tensor v'4 Float -- ^ __input_max__
                     -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
                     -- ^ (__output__, __output_min__, __output_max__)
                     --
                     -- * __output__
                     --
                     -- * __output_min__
                     --
                     -- * __output_max__
quantizedReshape' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tshape
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedReshape' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor shape :: Tensor v'2 tshape
shape input_min :: Tensor v'3 Float
input_min
                  input_max :: Tensor v'4 Float
input_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tshape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tshape
shape,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_min,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
input_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedReshape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tshape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tshape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tshape
forall a. HasCallStack => a
undefined :: tshape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "shape" type_attr: "Tshape" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr { name: "T" type: "type" }
attr {
  name: "Tshape"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
quantizedResizeBilinear :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                               Data.Word.Word8,
                                                               Float] t) =>
                           Tensor v'1 t -- ^ __images__
                           -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                           -> Tensor v'3 Float -- ^ __min__
                           -> Tensor v'4 Float -- ^ __max__
                           -> (Tensor Build t, Tensor Build Float,
                               Tensor Build Float)
                           -- ^ (__resized_images__, __out_min__, __out_max__)
                           --
                           -- * __resized_images__
                           --
                           -- * __out_min__
                           --
                           -- * __out_max__
quantizedResizeBilinear :: Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedResizeBilinear = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Int32, Word8, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedResizeBilinear' OpParams
forall a. a -> a
id
quantizedResizeBilinear' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                                Data.Word.Word8,
                                                                Float] t) =>
                            OpParams ->
                            Tensor v'1 t -- ^ __images__
                            -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                            -> Tensor v'3 Float -- ^ __min__
                            -> Tensor v'4 Float -- ^ __max__
                            -> (Tensor Build t, Tensor Build Float,
                                Tensor Build Float)
                            -- ^ (__resized_images__, __out_min__, __out_max__)
                            --
                            -- * __resized_images__
                            --
                            -- * __out_min__
                            --
                            -- * __out_max__
quantizedResizeBilinear' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
quantizedResizeBilinear' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size min :: Tensor v'3 Float
min max :: Tensor v'4 Float
max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build t, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "QuantizedResizeBilinear"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "min" type: DT_FLOAT }
input_arg { name: "max" type: DT_FLOAT }
output_arg { name: "resized_images" type_attr: "T" }
output_arg { name: "out_min" type: DT_FLOAT }
output_arg { name: "out_max" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_QUINT8 type: DT_QINT32 type: DT_FLOAT }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
queueClose :: forall m' . (MonadBuild m') =>
              Tensor Ref Data.ByteString.ByteString -- ^ __handle__
              -> m' (ControlNode)
queueClose :: Tensor Ref ByteString -> m' ControlNode
queueClose = OpParams -> Tensor Ref ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' ControlNode
queueClose' OpParams
forall a. a -> a
id
queueClose' :: forall m' . (MonadBuild m') => OpParams ->
               Tensor Ref Data.ByteString.ByteString -- ^ __handle__
               -> m' (ControlNode)
queueClose' :: OpParams -> Tensor Ref ByteString -> m' ControlNode
queueClose' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueClose"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "cancel_pending_enqueues"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
queueCloseV2 :: forall v'1 m' . (MonadBuild m') =>
                Tensor v'1 ResourceHandle -- ^ __handle__
                -> m' (ControlNode)
queueCloseV2 :: Tensor v'1 ResourceHandle -> m' ControlNode
queueCloseV2 = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
queueCloseV2' OpParams
forall a. a -> a
id
queueCloseV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                 Tensor v'1 ResourceHandle -- ^ __handle__
                 -> m' (ControlNode)
queueCloseV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
queueCloseV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueCloseV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "cancel_pending_enqueues"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
queueDequeue :: forall component_types m' . (MonadBuild m',
                                             TensorTypes component_types) =>
                Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeue :: Tensor Ref ByteString -> m' (TensorList Value component_types)
queueDequeue = OpParams
-> Tensor Ref ByteString -> m' (TensorList Value component_types)
forall (component_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor Ref ByteString -> m' (TensorList Value component_types)
queueDequeue' OpParams
forall a. a -> a
id
queueDequeue' :: forall component_types m' . (MonadBuild m',
                                              TensorTypes component_types) =>
                 OpParams ->
                 Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                 -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeue' :: OpParams
-> Tensor Ref ByteString -> m' (TensorList Value component_types)
queueDequeue' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueDequeueMany :: forall v'2 component_types m' . (MonadBuild m',
                                                     TensorTypes component_types) =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                    -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueMany :: Tensor Ref ByteString
-> Tensor v'2 Int32 -> m' (TensorList Value component_types)
queueDequeueMany = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
forall (v'2 :: * -> *) (component_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueMany' OpParams
forall a. a -> a
id
queueDequeueMany' :: forall v'2 component_types m' . (MonadBuild m',
                                                      TensorTypes component_types) =>
                     OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                     -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueMany' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueMany' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle n :: Tensor v'2 Int32
n | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
n]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeueMany"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "n" type: DT_INT32 }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueDequeueManyV2 :: forall v'1 v'2 component_types m' . (MonadBuild m',
                                                           TensorTypes component_types) =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                      -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueManyV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32 -> m' (TensorList Value component_types)
queueDequeueManyV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
forall (v'1 :: * -> *) (v'2 :: * -> *) (component_types :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueManyV2' OpParams
forall a. a -> a
id
queueDequeueManyV2' :: forall v'1 v'2 component_types m' . (MonadBuild m',
                                                            TensorTypes component_types) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                       -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueManyV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueManyV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle n :: Tensor v'2 Int32
n | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
n]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeueManyV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "n" type: DT_INT32 }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueDequeueUpTo :: forall v'2 component_types m' . (MonadBuild m',
                                                     TensorTypes component_types) =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                    -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueUpTo :: Tensor Ref ByteString
-> Tensor v'2 Int32 -> m' (TensorList Value component_types)
queueDequeueUpTo = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
forall (v'2 :: * -> *) (component_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueUpTo' OpParams
forall a. a -> a
id
queueDequeueUpTo' :: forall v'2 component_types m' . (MonadBuild m',
                                                      TensorTypes component_types) =>
                     OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                     -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueUpTo' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueUpTo' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle n :: Tensor v'2 Int32
n | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
n]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeueUpTo"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "n" type: DT_INT32 }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueDequeueUpToV2 :: forall v'1 v'2 component_types m' . (MonadBuild m',
                                                           TensorTypes component_types) =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                      -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueUpToV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32 -> m' (TensorList Value component_types)
queueDequeueUpToV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
forall (v'1 :: * -> *) (v'2 :: * -> *) (component_types :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes component_types) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueUpToV2' OpParams
forall a. a -> a
id
queueDequeueUpToV2' :: forall v'1 v'2 component_types m' . (MonadBuild m',
                                                            TensorTypes component_types) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __n__
                       -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueUpToV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (TensorList Value component_types)
queueDequeueUpToV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle n :: Tensor v'2 Int32
n | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
n]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeueUpToV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "n" type: DT_INT32 }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueDequeueV2 :: forall v'1 component_types m' . (MonadBuild m',
                                                   TensorTypes component_types) =>
                  Tensor v'1 ResourceHandle -- ^ __handle__
                  -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueV2 :: Tensor v'1 ResourceHandle -> m' (TensorList Value component_types)
queueDequeueV2 = OpParams
-> Tensor v'1 ResourceHandle
-> m' (TensorList Value component_types)
forall (v'1 :: * -> *) (output_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes output_types) =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
queueDequeueV2' OpParams
forall a. a -> a
id
queueDequeueV2' :: forall v'1 component_types m' . (MonadBuild m',
                                                    TensorTypes component_types) =>
                   OpParams ->
                   Tensor v'1 ResourceHandle -- ^ __handle__
                   -> m' (TensorList (Value) component_types) -- ^ __components__
queueDequeueV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> m' (TensorList Value component_types)
queueDequeueV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value component_types)
 -> m' (TensorList Value component_types))
-> Build (TensorList Value component_types)
-> m' (TensorList Value component_types)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build (TensorList Value component_types)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueDequeueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy component_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy component_types
forall k (t :: k). Proxy t
Proxy :: Proxy component_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "components" type_list_attr: "component_types" }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueEnqueue :: forall v'2 tcomponents m' . (MonadBuild m',
                                             TensorTypes tcomponents) =>
                Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                -> TensorList (v'2) tcomponents -- ^ __components__
                -> m' (ControlNode)
queueEnqueue :: Tensor Ref ByteString
-> TensorList v'2 tcomponents -> m' ControlNode
queueEnqueue = OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
forall (v'2 :: * -> *) (tcomponents :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes tcomponents) =>
OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueue' OpParams
forall a. a -> a
id
queueEnqueue' :: forall v'2 tcomponents m' . (MonadBuild m',
                                              TensorTypes tcomponents) =>
                 OpParams ->
                 Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                 -> TensorList (v'2) tcomponents -- ^ __components__
                 -> m' (ControlNode)
queueEnqueue' :: OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueue' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle components :: TensorList v'2 tcomponents
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             TensorList v'2 tcomponents -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tcomponents
components]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueEnqueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomponents" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tcomponents -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tcomponents
forall k (t :: k). Proxy t
Proxy :: Proxy tcomponents)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "components" type_list_attr: "Tcomponents" }
attr {
  name: "Tcomponents" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueEnqueueMany :: forall v'2 tcomponents m' . (MonadBuild m',
                                                 TensorTypes tcomponents) =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> TensorList (v'2) tcomponents -- ^ __components__
                    -> m' (ControlNode)
queueEnqueueMany :: Tensor Ref ByteString
-> TensorList v'2 tcomponents -> m' ControlNode
queueEnqueueMany = OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
forall (v'2 :: * -> *) (tcomponents :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes tcomponents) =>
OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueMany' OpParams
forall a. a -> a
id
queueEnqueueMany' :: forall v'2 tcomponents m' . (MonadBuild m',
                                                  TensorTypes tcomponents) =>
                     OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> TensorList (v'2) tcomponents -- ^ __components__
                     -> m' (ControlNode)
queueEnqueueMany' :: OpParams
-> Tensor Ref ByteString
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueMany' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle components :: TensorList v'2 tcomponents
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             TensorList v'2 tcomponents -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tcomponents
components]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueEnqueueMany"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomponents" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tcomponents -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tcomponents
forall k (t :: k). Proxy t
Proxy :: Proxy tcomponents)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "components" type_list_attr: "Tcomponents" }
attr {
  name: "Tcomponents" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueEnqueueManyV2 :: forall v'1 v'2 tcomponents m' . (MonadBuild m',
                                                       TensorTypes tcomponents) =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> TensorList (v'2) tcomponents -- ^ __components__
                      -> m' (ControlNode)
queueEnqueueManyV2 :: Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents -> m' ControlNode
queueEnqueueManyV2 = OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (tcomponents :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes tcomponents) =>
OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueManyV2' OpParams
forall a. a -> a
id
queueEnqueueManyV2' :: forall v'1 v'2 tcomponents m' . (MonadBuild m',
                                                        TensorTypes tcomponents) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> TensorList (v'2) tcomponents -- ^ __components__
                       -> m' (ControlNode)
queueEnqueueManyV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueManyV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle components :: TensorList v'2 tcomponents
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             TensorList v'2 tcomponents -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tcomponents
components]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueEnqueueManyV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomponents" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tcomponents -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tcomponents
forall k (t :: k). Proxy t
Proxy :: Proxy tcomponents)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "components" type_list_attr: "Tcomponents" }
attr {
  name: "Tcomponents" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueEnqueueV2 :: forall v'1 v'2 tcomponents m' . (MonadBuild m',
                                                   TensorTypes tcomponents) =>
                  Tensor v'1 ResourceHandle -- ^ __handle__
                  -> TensorList (v'2) tcomponents -- ^ __components__
                  -> m' (ControlNode)
queueEnqueueV2 :: Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents -> m' ControlNode
queueEnqueueV2 = OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (tcomponents :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes tcomponents) =>
OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueV2' OpParams
forall a. a -> a
id
queueEnqueueV2' :: forall v'1 v'2 tcomponents m' . (MonadBuild m',
                                                    TensorTypes tcomponents) =>
                   OpParams ->
                   Tensor v'1 ResourceHandle -- ^ __handle__
                   -> TensorList (v'2) tcomponents -- ^ __components__
                   -> m' (ControlNode)
queueEnqueueV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> TensorList v'2 tcomponents
-> m' ControlNode
queueEnqueueV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle components :: TensorList v'2 tcomponents
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             TensorList v'2 tcomponents -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 tcomponents
components]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueEnqueueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomponents" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tcomponents -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tcomponents
forall k (t :: k). Proxy t
Proxy :: Proxy tcomponents)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "components" type_list_attr: "Tcomponents" }
attr {
  name: "Tcomponents" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "timeout_ms" type: "int" default_value { i: -1 } }
-}
-- | 
queueIsClosed :: forall m' . (MonadBuild m') =>
                 Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                 -> m' (Tensor Value Bool) -- ^ __is_closed__
queueIsClosed :: Tensor Ref ByteString -> m' (Tensor Value Bool)
queueIsClosed = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Bool)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Bool)
queueIsClosed' OpParams
forall a. a -> a
id
queueIsClosed' :: forall m' . (MonadBuild m') => OpParams ->
                  Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                  -> m' (Tensor Value Bool) -- ^ __is_closed__
queueIsClosed' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Bool)
queueIsClosed' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueIsClosed"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "is_closed" type: DT_BOOL }
-}
-- | 
queueIsClosedV2 :: forall v'1 m' . (MonadBuild m') =>
                   Tensor v'1 ResourceHandle -- ^ __handle__
                   -> m' (Tensor Value Bool) -- ^ __is_closed__
queueIsClosedV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
queueIsClosedV2 = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
queueIsClosedV2' OpParams
forall a. a -> a
id
queueIsClosedV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                    Tensor v'1 ResourceHandle -- ^ __handle__
                    -> m' (Tensor Value Bool) -- ^ __is_closed__
queueIsClosedV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
queueIsClosedV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueIsClosedV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "is_closed" type: DT_BOOL }
-}
-- | 
queueSize :: forall m' . (MonadBuild m') =>
             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
             -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
queueSize :: Tensor Ref ByteString -> m' (Tensor Value Int32)
queueSize = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
queueSize' OpParams
forall a. a -> a
id
queueSize' :: forall m' . (MonadBuild m') => OpParams ->
              Tensor Ref Data.ByteString.ByteString -- ^ __handle__
              -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
queueSize' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
queueSize' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
queueSizeV2 :: forall v'1 m' . (MonadBuild m') =>
               Tensor v'1 ResourceHandle -- ^ __handle__
               -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
queueSizeV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
queueSizeV2 = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
queueSizeV2' OpParams
forall a. a -> a
id
queueSizeV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                Tensor v'1 ResourceHandle -- ^ __handle__
                -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
queueSizeV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
queueSizeV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "QueueSizeV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
rFFT :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                         OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float)] tcomplex) =>
        Tensor v'1 treal -- ^ __input__
        -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
        -> Tensor Build tcomplex -- ^ __output__
rFFT :: Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT = OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT' OpParams
forall a. a -> a
id
rFFT' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                          OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float)] tcomplex) =>
         OpParams ->
         Tensor v'1 treal -- ^ __input__
         -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
         -> Tensor Build tcomplex -- ^ __output__
rFFT' :: OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT' op'options :: OpParams
op'options input :: Tensor v'1 treal
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 treal -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 treal
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RFFT"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Treal" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
rFFT2D :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float)] tcomplex) =>
          Tensor v'1 treal -- ^ __input__
          -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
          -> Tensor Build tcomplex -- ^ __output__
rFFT2D :: Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT2D = OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT2D' OpParams
forall a. a -> a
id
rFFT2D' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float)] tcomplex) =>
           OpParams ->
           Tensor v'1 treal -- ^ __input__
           -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
           -> Tensor Build tcomplex -- ^ __output__
rFFT2D' :: OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT2D' op'options :: OpParams
op'options input :: Tensor v'1 treal
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 treal -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 treal
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RFFT2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Treal" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
rFFT3D :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float)] tcomplex) =>
          Tensor v'1 treal -- ^ __input__
          -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
          -> Tensor Build tcomplex -- ^ __output__
rFFT3D :: Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT3D = OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
forall (v'1 :: * -> *) (v'2 :: * -> *) treal tcomplex.
(OneOf '[Double, Float] treal,
 OneOf '[Complex Double, Complex Float] tcomplex) =>
OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT3D' OpParams
forall a. a -> a
id
rFFT3D' :: forall v'1 v'2 treal tcomplex . (OneOf '[Double, Float] treal,
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float)] tcomplex) =>
           OpParams ->
           Tensor v'1 treal -- ^ __input__
           -> Tensor v'2 Data.Int.Int32 -- ^ __fft_length__
           -> Tensor Build tcomplex -- ^ __output__
rFFT3D' :: OpParams
-> Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
rFFT3D' op'options :: OpParams
op'options input :: Tensor v'1 treal
input fft_length :: Tensor v'2 Int32
fft_length | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tcomplex
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tcomplex)
-> Build OpDef -> Tensor Build tcomplex
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 treal -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 treal
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
fft_length]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RFFT3D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tcomplex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tcomplex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tcomplex
forall a. HasCallStack => a
undefined :: tcomplex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Treal" }
input_arg { name: "fft_length" type: DT_INT32 }
output_arg { name: "output" type_attr: "Tcomplex" }
attr {
  name: "Treal"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tcomplex"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
-}
-- | 
rGBToHSV :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __images__
            -> Tensor Build t -- ^ __output__
rGBToHSV :: Tensor v'1 t -> Tensor Build t
rGBToHSV = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
rGBToHSV' OpParams
forall a. a -> a
id
rGBToHSV' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __images__
             -> Tensor Build t -- ^ __output__
rGBToHSV' :: OpParams -> Tensor v'1 t -> Tensor Build t
rGBToHSV' op'options :: OpParams
op'options images :: Tensor v'1 t
images | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RGBToHSV"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
raggedBincount :: forall v'1 v'2 v'3 v'4 tidx t . (OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tidx,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Double, Float] t) =>
                  Tensor v'1 Data.Int.Int64 -- ^ __splits__
                  -> Tensor v'2 tidx -- ^ __values__
                  -> Tensor v'3 tidx -- ^ __size__
                  -> Tensor v'4 t -- ^ __weights__
                  -> Tensor Build t -- ^ __output__
raggedBincount :: Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor v'4 t
-> Tensor Build t
raggedBincount = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) tidx t.
(OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64, Double, Float] t) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor v'4 t
-> Tensor Build t
raggedBincount' OpParams
forall a. a -> a
id
raggedBincount' :: forall v'1 v'2 v'3 v'4 tidx t . (OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tidx,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Double, Float] t) =>
                   OpParams ->
                   Tensor v'1 Data.Int.Int64 -- ^ __splits__
                   -> Tensor v'2 tidx -- ^ __values__
                   -> Tensor v'3 tidx -- ^ __size__
                   -> Tensor v'4 t -- ^ __weights__
                   -> Tensor Build t -- ^ __output__
raggedBincount' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor v'4 t
-> Tensor Build t
raggedBincount' op'options :: OpParams
op'options splits :: Tensor v'1 Int64
splits values :: Tensor v'2 tidx
values size :: Tensor v'3 tidx
size weights :: Tensor v'4 t
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
splits,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
values,
                                                             Tensor v'3 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tidx
size,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedBincount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "splits" type: DT_INT64 }
input_arg { name: "values" type_attr: "Tidx" }
input_arg { name: "size" type_attr: "Tidx" }
input_arg { name: "weights" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "Tidx"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "binary_output" type: "bool" default_value { b: false }
}
-}
-- | 
raggedCountSparseOutput :: forall v'1 v'2 v'3 t
                           output_type . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64, Double,
                                                  Float] output_type) =>
                           Bool -- ^ __binary_output__
                           -> Tensor v'1 Data.Int.Int64 -- ^ __splits__
                           -> Tensor v'2 t -- ^ __values__
                           -> Tensor v'3 output_type -- ^ __weights__
                           -> (Tensor Build Data.Int.Int64,
                               Tensor Build output_type,
                               Tensor Build Data.Int.Int64)
                           -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                           --
                           -- * __output_indices__
                           --
                           -- * __output_values__
                           --
                           -- * __output_dense_shape__
raggedCountSparseOutput :: Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
raggedCountSparseOutput = OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       output_type.
(OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64, Double, Float] output_type) =>
OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
raggedCountSparseOutput' OpParams
forall a. a -> a
id
raggedCountSparseOutput' :: forall v'1 v'2 v'3 t
                            output_type . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64, Double,
                                                   Float] output_type) =>
                            OpParams ->
                            Bool -- ^ __binary_output__
                            -> Tensor v'1 Data.Int.Int64 -- ^ __splits__
                            -> Tensor v'2 t -- ^ __values__
                            -> Tensor v'3 output_type -- ^ __weights__
                            -> (Tensor Build Data.Int.Int64,
                                Tensor Build output_type,
                                Tensor Build Data.Int.Int64)
                            -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                            --
                            -- * __output_indices__
                            --
                            -- * __output_values__
                            --
                            -- * __output_dense_shape__
raggedCountSparseOutput' :: OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
raggedCountSparseOutput' op'options :: OpParams
op'options binary_output :: Bool
binary_output splits :: Tensor v'1 Int64
splits values :: Tensor v'2 t
values
                         weights :: Tensor v'3 output_type
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build output_type,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
splits,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 output_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 output_type
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedCountSparseOutput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "binary_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
binary_output
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "splits" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "weights" type_attr: "output_type" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "output_type" }
output_arg { name: "output_dense_shape" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "minlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr {
  name: "maxlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr { name: "binary_output" type: "bool" }
attr {
  name: "output_type"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
raggedCross :: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types
               ragged_splits_types sparse_values_types dense_types
               out_values_type
               out_row_splits_type . (OneOfs '[Data.ByteString.ByteString,
                                               Data.Int.Int64] ragged_values_types,
                                      OneOfs '[Data.Int.Int32,
                                               Data.Int.Int64] ragged_splits_types,
                                      OneOfs '[Data.ByteString.ByteString,
                                               Data.Int.Int64] sparse_values_types,
                                      OneOfs '[Data.ByteString.ByteString,
                                               Data.Int.Int64] dense_types,
                                      OneOf '[Data.ByteString.ByteString,
                                              Data.Int.Int64] out_values_type,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] out_row_splits_type) =>
               Data.Int.Int64 -- ^ __hash_key__
               -> Bool -- ^ __hashed_output__
               -> ByteString -- ^ __input_order__
               -> Data.Int.Int64 -- ^ __num_buckets__
               -> TensorList (v'1) ragged_values_types -- ^ __ragged_values__
               -> TensorList (v'2) ragged_splits_types -- ^ __ragged_row_splits__
               -> [Tensor v'3 Data.Int.Int64] -- ^ __sparse_indices__
               -> TensorList (v'4) sparse_values_types -- ^ __sparse_values__
               -> [Tensor v'5 Data.Int.Int64] -- ^ __sparse_shape__
               -> TensorList (v'6) dense_types -- ^ __dense_inputs__
               -> (Tensor Build out_values_type,
                   Tensor Build out_row_splits_type)
               -- ^ (__output_values__, __output_row_splits__)
               --
               -- * __output_values__
               --
               -- * __output_row_splits__
raggedCross :: Int64
-> Bool
-> ByteString
-> Int64
-> TensorList v'1 ragged_values_types
-> TensorList v'2 ragged_splits_types
-> [Tensor v'3 Int64]
-> TensorList v'4 sparse_values_types
-> [Tensor v'5 Int64]
-> TensorList v'6 dense_types
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
raggedCross = OpParams
-> Int64
-> Bool
-> ByteString
-> Int64
-> TensorList v'1 ragged_values_types
-> TensorList v'2 ragged_splits_types
-> [Tensor v'3 Int64]
-> TensorList v'4 sparse_values_types
-> [Tensor v'5 Int64]
-> TensorList v'6 dense_types
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (ragged_values_types :: [*]) (ragged_splits_types :: [*])
       (sparse_values_types :: [*]) (dense_types :: [*]) out_values_type
       out_row_splits_type.
(OneOfs '[ByteString, Int64] ragged_values_types,
 OneOfs '[Int32, Int64] ragged_splits_types,
 OneOfs '[ByteString, Int64] sparse_values_types,
 OneOfs '[ByteString, Int64] dense_types,
 OneOf '[ByteString, Int64] out_values_type,
 OneOf '[Int32, Int64] out_row_splits_type) =>
OpParams
-> Int64
-> Bool
-> ByteString
-> Int64
-> TensorList v'1 ragged_values_types
-> TensorList v'2 ragged_splits_types
-> [Tensor v'3 Int64]
-> TensorList v'4 sparse_values_types
-> [Tensor v'5 Int64]
-> TensorList v'6 dense_types
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
raggedCross' OpParams
forall a. a -> a
id
raggedCross' :: forall v'1 v'2 v'3 v'4 v'5 v'6 ragged_values_types
                ragged_splits_types sparse_values_types dense_types
                out_values_type
                out_row_splits_type . (OneOfs '[Data.ByteString.ByteString,
                                                Data.Int.Int64] ragged_values_types,
                                       OneOfs '[Data.Int.Int32,
                                                Data.Int.Int64] ragged_splits_types,
                                       OneOfs '[Data.ByteString.ByteString,
                                                Data.Int.Int64] sparse_values_types,
                                       OneOfs '[Data.ByteString.ByteString,
                                                Data.Int.Int64] dense_types,
                                       OneOf '[Data.ByteString.ByteString,
                                               Data.Int.Int64] out_values_type,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] out_row_splits_type) =>
                OpParams ->
                Data.Int.Int64 -- ^ __hash_key__
                -> Bool -- ^ __hashed_output__
                -> ByteString -- ^ __input_order__
                -> Data.Int.Int64 -- ^ __num_buckets__
                -> TensorList (v'1) ragged_values_types -- ^ __ragged_values__
                -> TensorList (v'2) ragged_splits_types -- ^ __ragged_row_splits__
                -> [Tensor v'3 Data.Int.Int64] -- ^ __sparse_indices__
                -> TensorList (v'4) sparse_values_types -- ^ __sparse_values__
                -> [Tensor v'5 Data.Int.Int64] -- ^ __sparse_shape__
                -> TensorList (v'6) dense_types -- ^ __dense_inputs__
                -> (Tensor Build out_values_type,
                    Tensor Build out_row_splits_type)
                -- ^ (__output_values__, __output_row_splits__)
                --
                -- * __output_values__
                --
                -- * __output_row_splits__
raggedCross' :: OpParams
-> Int64
-> Bool
-> ByteString
-> Int64
-> TensorList v'1 ragged_values_types
-> TensorList v'2 ragged_splits_types
-> [Tensor v'3 Int64]
-> TensorList v'4 sparse_values_types
-> [Tensor v'5 Int64]
-> TensorList v'6 dense_types
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
raggedCross' op'options :: OpParams
op'options hash_key :: Int64
hash_key hashed_output :: Bool
hashed_output input_order :: ByteString
input_order num_buckets :: Int64
num_buckets
             ragged_values :: TensorList v'1 ragged_values_types
ragged_values ragged_row_splits :: TensorList v'2 ragged_splits_types
ragged_row_splits sparse_indices :: [Tensor v'3 Int64]
sparse_indices sparse_values :: TensorList v'4 sparse_values_types
sparse_values
             sparse_shape :: [Tensor v'5 Int64]
sparse_shape
             dense_inputs :: TensorList v'6 dense_types
dense_inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("Nsparse", [("sparse_indices", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
sparse_indices),
                                                        ("sparse_shape", [Tensor v'5 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'5 Int64]
sparse_shape)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_values_type,
     Tensor Build out_row_splits_type))
-> Build OpDef
-> (Tensor Build out_values_type, Tensor Build out_row_splits_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 ragged_values_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 ragged_values_types
ragged_values,
                                                             TensorList v'2 ragged_splits_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 ragged_splits_types
ragged_row_splits,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
sparse_indices,
                                                             TensorList v'4 sparse_values_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 sparse_values_types
sparse_values,
                                                             [Tensor v'5 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'5 Int64]
sparse_shape,
                                                             TensorList v'6 dense_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'6 dense_types
dense_inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedCross"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "ragged_values_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy ragged_values_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy ragged_values_types
forall k (t :: k). Proxy t
Proxy :: Proxy ragged_values_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "ragged_splits_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy ragged_splits_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy ragged_splits_types
forall k (t :: k). Proxy t
Proxy :: Proxy ragged_splits_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_values_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_values_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_values_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_values_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dense_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dense_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dense_types
forall k (t :: k). Proxy t
Proxy :: Proxy dense_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_values_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_values_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_values_type
forall a. HasCallStack => a
undefined :: out_values_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_row_splits_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_row_splits_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_row_splits_type
forall a. HasCallStack => a
undefined :: out_row_splits_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "hash_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
hash_key
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "hashed_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
hashed_output
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_order" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
input_order
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Nsparse" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
nsparse
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    nsparse :: Int64
nsparse = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
sparse_indices) :: Int64
{-
input_arg {
  name: "ragged_values" type_list_attr: "ragged_values_types"
}
input_arg {
  name: "ragged_row_splits" type_list_attr: "ragged_splits_types"
}
input_arg {
  name: "sparse_indices" type: DT_INT64 number_attr: "Nsparse"
}
input_arg {
  name: "sparse_values" type_list_attr: "sparse_values_types"
}
input_arg {
  name: "sparse_shape" type: DT_INT64 number_attr: "Nsparse"
}
input_arg { name: "dense_inputs" type_list_attr: "dense_types" }
output_arg { name: "output_values" type_attr: "out_values_type" }
output_arg {
  name: "output_row_splits" type_attr: "out_row_splits_type"
}
attr { name: "Nsparse" type: "int" has_minimum: true }
attr { name: "input_order" type: "string" }
attr { name: "hashed_output" type: "bool" }
attr { name: "num_buckets" type: "int" has_minimum: true }
attr { name: "hash_key" type: "int" }
attr {
  name: "ragged_values_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "ragged_splits_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "sparse_values_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "dense_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "out_values_type"
  type: "type"
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "out_row_splits_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
raggedGather :: forall v'1 v'2 v'3 tvalues tindices
                tsplits . (TensorType tvalues, OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices,
                           OneOf '[Data.Int.Int32, Data.Int.Int64] tsplits) =>
                Data.Int.Int64 -- ^ __OUTPUT_RAGGED_RANK__
                -> [Tensor v'1 tsplits] -- ^ __params_nested_splits__
                -> Tensor v'2 tvalues -- ^ __params_dense_values__
                -> Tensor v'3 tindices -- ^ __indices__
                -> ([Tensor Build tsplits], Tensor Build tvalues)
                -- ^ (__output_nested_splits__, __output_dense_values__)
                --
                -- * __output_nested_splits__
                --
                -- * __output_dense_values__
raggedGather :: Int64
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor v'3 tindices
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedGather = OpParams
-> Int64
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor v'3 tindices
-> ([Tensor Build tsplits], Tensor Build tvalues)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tvalues
       tindices tsplits.
(TensorType tvalues, OneOf '[Int32, Int64] tindices,
 OneOf '[Int32, Int64] tsplits) =>
OpParams
-> Int64
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor v'3 tindices
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedGather' OpParams
forall a. a -> a
id
raggedGather' :: forall v'1 v'2 v'3 tvalues tindices
                 tsplits . (TensorType tvalues, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices,
                            OneOf '[Data.Int.Int32, Data.Int.Int64] tsplits) =>
                 OpParams ->
                 Data.Int.Int64 -- ^ __OUTPUT_RAGGED_RANK__
                 -> [Tensor v'1 tsplits] -- ^ __params_nested_splits__
                 -> Tensor v'2 tvalues -- ^ __params_dense_values__
                 -> Tensor v'3 tindices -- ^ __indices__
                 -> ([Tensor Build tsplits], Tensor Build tvalues)
                 -- ^ (__output_nested_splits__, __output_dense_values__)
                 --
                 -- * __output_nested_splits__
                 --
                 -- * __output_dense_values__
raggedGather' :: OpParams
-> Int64
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor v'3 tindices
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedGather' op'options :: OpParams
op'options oUTPUT_RAGGED_RANK :: Int64
oUTPUT_RAGGED_RANK params_nested_splits :: [Tensor v'1 tsplits]
params_nested_splits
              params_dense_values :: Tensor v'2 tvalues
params_dense_values
              indices :: Tensor v'3 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("PARAMS_RAGGED_RANK", [("params_nested_splits", [Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
params_nested_splits)])] =
    [Int64]
-> Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
oUTPUT_RAGGED_RANK] (Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues))
-> Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 tsplits] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 tsplits]
params_nested_splits,
                                                             Tensor v'2 tvalues -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tvalues
params_dense_values,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedGather"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "OUTPUT_RAGGED_RANK" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
oUTPUT_RAGGED_RANK
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "PARAMS_RAGGED_RANK" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
pARAMS_RAGGED_RANK
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    pARAMS_RAGGED_RANK :: Int64
pARAMS_RAGGED_RANK = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
params_nested_splits) :: Int64
{-
input_arg {
  name: "params_nested_splits"
  type_attr: "Tsplits"
  number_attr: "PARAMS_RAGGED_RANK"
}
input_arg { name: "params_dense_values" type_attr: "Tvalues" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg {
  name: "output_nested_splits"
  type_attr: "Tsplits"
  number_attr: "OUTPUT_RAGGED_RANK"
}
output_arg { name: "output_dense_values" type_attr: "Tvalues" }
attr { name: "Tvalues" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "PARAMS_RAGGED_RANK" type: "int" has_minimum: true minimum: 1
}
attr { name: "OUTPUT_RAGGED_RANK" type: "int" has_minimum: true }
-}
-- | 
raggedRange :: forall v'1 v'2 v'3 t tsplits . (OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Word.Word16, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tsplits) =>
               Tensor v'1 t -- ^ __starts__
               -> Tensor v'2 t -- ^ __limits__
               -> Tensor v'3 t -- ^ __deltas__
               -> (Tensor Build tsplits, Tensor Build t)
               -- ^ (__rt_nested_splits__, __rt_dense_values__)
               --
               -- * __rt_nested_splits__
               --
               -- * __rt_dense_values__
raggedRange :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> (Tensor Build tsplits, Tensor Build t)
raggedRange = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> (Tensor Build tsplits, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tsplits.
(OneOf '[Int32, Int64, Word16, Double, Float] t,
 OneOf '[Int32, Int64] tsplits) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> (Tensor Build tsplits, Tensor Build t)
raggedRange' OpParams
forall a. a -> a
id
raggedRange' :: forall v'1 v'2 v'3 t tsplits . (OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Word.Word16,
                                                        Double, Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tsplits) =>
                OpParams ->
                Tensor v'1 t -- ^ __starts__
                -> Tensor v'2 t -- ^ __limits__
                -> Tensor v'3 t -- ^ __deltas__
                -> (Tensor Build tsplits, Tensor Build t)
                -- ^ (__rt_nested_splits__, __rt_dense_values__)
                --
                -- * __rt_nested_splits__
                --
                -- * __rt_dense_values__
raggedRange' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> (Tensor Build tsplits, Tensor Build t)
raggedRange' op'options :: OpParams
op'options starts :: Tensor v'1 t
starts limits :: Tensor v'2 t
limits deltas :: Tensor v'3 t
deltas | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build tsplits, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build tsplits, Tensor Build t))
-> Build OpDef -> (Tensor Build tsplits, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
starts,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
limits,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
deltas]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedRange"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "starts" type_attr: "T" }
input_arg { name: "limits" type_attr: "T" }
input_arg { name: "deltas" type_attr: "T" }
output_arg { name: "rt_nested_splits" type_attr: "Tsplits" }
output_arg { name: "rt_dense_values" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
raggedTensorFromVariant :: forall v'1 tvalues tsplits . (TensorType tvalues,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tsplits) =>
                           Data.Int.Int64 -- ^ __input_ragged_rank__
                           -> Data.Int.Int64 -- ^ __output_ragged_rank__
                           -> Tensor v'1 Variant -- ^ __encoded_ragged__
                           -> ([Tensor Build tsplits], Tensor Build tvalues)
                           -- ^ (__output_nested_splits__, __output_dense_values__)
                           --
                           -- * __output_nested_splits__
                           --
                           -- * __output_dense_values__
raggedTensorFromVariant :: Int64
-> Int64
-> Tensor v'1 Variant
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedTensorFromVariant = OpParams
-> Int64
-> Int64
-> Tensor v'1 Variant
-> ([Tensor Build tsplits], Tensor Build tvalues)
forall (v'1 :: * -> *) tvalues tsplits.
(TensorType tvalues, OneOf '[Int32, Int64] tsplits) =>
OpParams
-> Int64
-> Int64
-> Tensor v'1 Variant
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedTensorFromVariant' OpParams
forall a. a -> a
id
raggedTensorFromVariant' :: forall v'1 tvalues tsplits . (TensorType tvalues,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] tsplits) =>
                            OpParams ->
                            Data.Int.Int64 -- ^ __input_ragged_rank__
                            -> Data.Int.Int64 -- ^ __output_ragged_rank__
                            -> Tensor v'1 Variant -- ^ __encoded_ragged__
                            -> ([Tensor Build tsplits], Tensor Build tvalues)
                            -- ^ (__output_nested_splits__, __output_dense_values__)
                            --
                            -- * __output_nested_splits__
                            --
                            -- * __output_dense_values__
raggedTensorFromVariant' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Variant
-> ([Tensor Build tsplits], Tensor Build tvalues)
raggedTensorFromVariant' op'options :: OpParams
op'options input_ragged_rank :: Int64
input_ragged_rank output_ragged_rank :: Int64
output_ragged_rank
                         encoded_ragged :: Tensor v'1 Variant
encoded_ragged | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
output_ragged_rank] (Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues))
-> Build OpDef -> ([Tensor Build tsplits], Tensor Build tvalues)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
encoded_ragged]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedTensorFromVariant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_ragged_rank" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
input_ragged_rank
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_ragged_rank" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
output_ragged_rank
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "encoded_ragged" type: DT_VARIANT }
output_arg {
  name: "output_nested_splits"
  type_attr: "Tsplits"
  number_attr: "output_ragged_rank"
}
output_arg { name: "output_dense_values" type_attr: "Tvalues" }
attr {
  name: "input_ragged_rank" type: "int" has_minimum: true minimum: -1
}
attr { name: "output_ragged_rank" type: "int" has_minimum: true }
attr { name: "Tvalues" type: "type" }
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
raggedTensorToSparse :: forall v'1 v'2 t tsplits . (TensorType t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tsplits) =>
                        [Tensor v'1 tsplits] -- ^ __rt_nested_splits__
                        -> Tensor v'2 t -- ^ __rt_dense_values__
                        -> (Tensor Build Data.Int.Int64, Tensor Build t,
                            Tensor Build Data.Int.Int64)
                        -- ^ (__sparse_indices__, __sparse_values__, __sparse_dense_shape__)
                        --
                        -- * __sparse_indices__
                        --
                        -- * __sparse_values__
                        --
                        -- * __sparse_dense_shape__
raggedTensorToSparse :: [Tensor v'1 tsplits]
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
raggedTensorToSparse = OpParams
-> [Tensor v'1 tsplits]
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) t tsplits.
(TensorType t, OneOf '[Int32, Int64] tsplits) =>
OpParams
-> [Tensor v'1 tsplits]
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
raggedTensorToSparse' OpParams
forall a. a -> a
id
raggedTensorToSparse' :: forall v'1 v'2 t tsplits . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tsplits) =>
                         OpParams ->
                         [Tensor v'1 tsplits] -- ^ __rt_nested_splits__
                         -> Tensor v'2 t -- ^ __rt_dense_values__
                         -> (Tensor Build Data.Int.Int64, Tensor Build t,
                             Tensor Build Data.Int.Int64)
                         -- ^ (__sparse_indices__, __sparse_values__, __sparse_dense_shape__)
                         --
                         -- * __sparse_indices__
                         --
                         -- * __sparse_values__
                         --
                         -- * __sparse_dense_shape__
raggedTensorToSparse' :: OpParams
-> [Tensor v'1 tsplits]
-> Tensor v'2 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
raggedTensorToSparse' op'options :: OpParams
op'options rt_nested_splits :: [Tensor v'1 tsplits]
rt_nested_splits
                      rt_dense_values :: Tensor v'2 t
rt_dense_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("RAGGED_RANK", [("rt_nested_splits", [Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
rt_nested_splits)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 tsplits] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 tsplits]
rt_nested_splits,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rt_dense_values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedTensorToSparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "RAGGED_RANK" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
rAGGED_RANK
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    rAGGED_RANK :: Int64
rAGGED_RANK = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
rt_nested_splits) :: Int64
{-
input_arg {
  name: "rt_nested_splits"
  type_attr: "Tsplits"
  number_attr: "RAGGED_RANK"
}
input_arg { name: "rt_dense_values" type_attr: "T" }
output_arg { name: "sparse_indices" type: DT_INT64 }
output_arg { name: "sparse_values" type_attr: "T" }
output_arg { name: "sparse_dense_shape" type: DT_INT64 }
attr {
  name: "RAGGED_RANK" type: "int" has_minimum: true minimum: 1
}
attr { name: "T" type: "type" }
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
raggedTensorToTensor :: forall v'1 v'2 v'3 v'4 t tindex tshape . (TensorType t,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] tindex,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] tshape) =>
                        Tensor v'1 tshape -- ^ __shape__
                        -> Tensor v'2 t -- ^ __values__
                        -> Tensor v'3 t -- ^ __default_value__
                        -> [Tensor v'4 tindex] -- ^ __row_partition_tensors__
                        -> Tensor Build t -- ^ __result__
raggedTensorToTensor :: Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> [Tensor v'4 tindex]
-> Tensor Build t
raggedTensorToTensor = OpParams
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> [Tensor v'4 tindex]
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tindex tshape.
(TensorType t, OneOf '[Int32, Int64] tindex,
 OneOf '[Int32, Int64] tshape) =>
OpParams
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> [Tensor v'4 tindex]
-> Tensor Build t
raggedTensorToTensor' OpParams
forall a. a -> a
id
raggedTensorToTensor' :: forall v'1 v'2 v'3 v'4 t tindex tshape . (TensorType t,
                                                                   OneOf '[Data.Int.Int32,
                                                                           Data.Int.Int64] tindex,
                                                                   OneOf '[Data.Int.Int32,
                                                                           Data.Int.Int64] tshape) =>
                         OpParams ->
                         Tensor v'1 tshape -- ^ __shape__
                         -> Tensor v'2 t -- ^ __values__
                         -> Tensor v'3 t -- ^ __default_value__
                         -> [Tensor v'4 tindex] -- ^ __row_partition_tensors__
                         -> Tensor Build t -- ^ __result__
raggedTensorToTensor' :: OpParams
-> Tensor v'1 tshape
-> Tensor v'2 t
-> Tensor v'3 t
-> [Tensor v'4 tindex]
-> Tensor Build t
raggedTensorToTensor' op'options :: OpParams
op'options shape :: Tensor v'1 tshape
shape values :: Tensor v'2 t
values default_value :: Tensor v'3 t
default_value
                      row_partition_tensors :: [Tensor v'4 tindex]
row_partition_tensors | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_row_partition_tensors", [("row_partition_tensors", [Tensor v'4 tindex] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 tindex]
row_partition_tensors)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tshape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tshape
shape,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
default_value,
                                                             [Tensor v'4 tindex] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 tindex]
row_partition_tensors]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedTensorToTensor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindex" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindex -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindex
forall a. HasCallStack => a
undefined :: tindex)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tshape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tshape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tshape
forall a. HasCallStack => a
undefined :: tshape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_row_partition_tensors" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_row_partition_tensors
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_row_partition_tensors :: Int64
num_row_partition_tensors = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 tindex] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 tindex]
row_partition_tensors) :: Int64
{-
input_arg { name: "shape" type_attr: "Tshape" }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "default_value" type_attr: "T" }
input_arg {
  name: "row_partition_tensors"
  type_attr: "Tindex"
  number_attr: "num_row_partition_tensors"
}
output_arg { name: "result" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindex"
  type: "type"
  allowed_values { list { type: DT_INT64 type: DT_INT32 } }
}
attr {
  name: "Tshape"
  type: "type"
  allowed_values { list { type: DT_INT64 type: DT_INT32 } }
}
attr {
  name: "num_row_partition_tensors"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr { name: "row_partition_types" type: "list(string)" }
-}
-- | 
raggedTensorToVariant :: forall v'1 v'2 tvalues tsplits . (TensorType tvalues,
                                                           OneOf '[Data.Int.Int32,
                                                                   Data.Int.Int64] tsplits) =>
                         Bool -- ^ __batched_input__
                         -> [Tensor v'1 tsplits] -- ^ __rt_nested_splits__
                         -> Tensor v'2 tvalues -- ^ __rt_dense_values__
                         -> Tensor Build Variant -- ^ __encoded_ragged__
raggedTensorToVariant :: Bool
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor Build Variant
raggedTensorToVariant = OpParams
-> Bool
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) tvalues tsplits.
(TensorType tvalues, OneOf '[Int32, Int64] tsplits) =>
OpParams
-> Bool
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor Build Variant
raggedTensorToVariant' OpParams
forall a. a -> a
id
raggedTensorToVariant' :: forall v'1 v'2 tvalues tsplits . (TensorType tvalues,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tsplits) =>
                          OpParams ->
                          Bool -- ^ __batched_input__
                          -> [Tensor v'1 tsplits] -- ^ __rt_nested_splits__
                          -> Tensor v'2 tvalues -- ^ __rt_dense_values__
                          -> Tensor Build Variant -- ^ __encoded_ragged__
raggedTensorToVariant' :: OpParams
-> Bool
-> [Tensor v'1 tsplits]
-> Tensor v'2 tvalues
-> Tensor Build Variant
raggedTensorToVariant' op'options :: OpParams
op'options batched_input :: Bool
batched_input rt_nested_splits :: [Tensor v'1 tsplits]
rt_nested_splits
                       rt_dense_values :: Tensor v'2 tvalues
rt_dense_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("RAGGED_RANK", [("rt_nested_splits", [Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
rt_nested_splits)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 tsplits] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 tsplits]
rt_nested_splits,
                                                             Tensor v'2 tvalues -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tvalues
rt_dense_values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RaggedTensorToVariant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "batched_input" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
batched_input
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "RAGGED_RANK" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
rAGGED_RANK
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    rAGGED_RANK :: Int64
rAGGED_RANK = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 tsplits] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 tsplits]
rt_nested_splits) :: Int64
{-
input_arg {
  name: "rt_nested_splits"
  type_attr: "Tsplits"
  number_attr: "RAGGED_RANK"
}
input_arg { name: "rt_dense_values" type_attr: "Tvalues" }
output_arg { name: "encoded_ragged" type: DT_VARIANT }
attr { name: "RAGGED_RANK" type: "int" has_minimum: true }
attr { name: "Tvalues" type: "type" }
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "batched_input" type: "bool" }
-}
-- | 
randomCrop :: forall v'1 v'2 t m' . (MonadBuild m', OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Int.Int64,
                                                            Data.Int.Int8,
                                                            Data.Word.Word8,
                                                            Double, Float] t) =>
              Tensor v'1 t -- ^ __image__
              -> Tensor v'2 Data.Int.Int64 -- ^ __size__
              -> m' (Tensor Value t) -- ^ __output__
randomCrop :: Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
randomCrop = OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t) =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
randomCrop' OpParams
forall a. a -> a
id
randomCrop' :: forall v'1 v'2 t m' . (MonadBuild m', OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __image__
               -> Tensor v'2 Data.Int.Int64 -- ^ __size__
               -> m' (Tensor Value t) -- ^ __output__
randomCrop' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
randomCrop' op'options :: OpParams
op'options image :: Tensor v'1 t
image size :: Tensor v'2 Int64
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
image,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
size]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomCrop"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image" type_attr: "T" }
input_arg { name: "size" type: DT_INT64 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
randomDataset :: forall v'1 v'2 m' . (MonadBuild m') =>
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Data.Int.Int64 -- ^ __seed__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                 -> m' (Tensor Value Variant) -- ^ __handle__
randomDataset :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
randomDataset = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
randomDataset' OpParams
forall a. a -> a
id
randomDataset' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                  [DataType] -- ^ __output_types__
                  -> Tensor v'1 Data.Int.Int64 -- ^ __seed__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __seed2__
                  -> m' (Tensor Value Variant) -- ^ __handle__
randomDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> m' (Tensor Value Variant)
randomDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types seed :: Tensor v'1 Int64
seed seed2 :: Tensor v'2 Int64
seed2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
seed,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
seed2]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
randomGamma :: forall v'1 v'2 s t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] s,
                                        OneOf '[Data.Word.Word16, Double,
                                                Float] t) =>
               Tensor v'1 s -- ^ __shape__
               -> Tensor v'2 t -- ^ __alpha__
               -> m' (Tensor Value t) -- ^ __output__
randomGamma :: Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
randomGamma = OpParams -> Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) s t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] s,
 OneOf '[Word16, Double, Float] t) =>
OpParams -> Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
randomGamma' OpParams
forall a. a -> a
id
randomGamma' :: forall v'1 v'2 s t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] s,
                                         OneOf '[Data.Word.Word16, Double,
                                                 Float] t) => OpParams ->
                Tensor v'1 s -- ^ __shape__
                -> Tensor v'2 t -- ^ __alpha__
                -> m' (Tensor Value t) -- ^ __output__
randomGamma' :: OpParams -> Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
randomGamma' op'options :: OpParams
op'options shape :: Tensor v'1 s
shape alpha :: Tensor v'2 t
alpha | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 s
shape,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomGamma"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "alpha" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
-}
-- | 
randomGammaGrad :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                   Tensor v'1 t -- ^ __alpha__
                   -> Tensor v'2 t -- ^ __sample__
                   -> Tensor Build t -- ^ __output__
randomGammaGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
randomGammaGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
randomGammaGrad' OpParams
forall a. a -> a
id
randomGammaGrad' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
                    Tensor v'1 t -- ^ __alpha__
                    -> Tensor v'2 t -- ^ __sample__
                    -> Tensor Build t -- ^ __output__
randomGammaGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
randomGammaGrad' op'options :: OpParams
op'options alpha :: Tensor v'1 t
alpha sample :: Tensor v'2 t
sample | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
alpha,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sample]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RandomGammaGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "sample" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
randomPoisson :: forall v'1 v'2 s dtype m' . (MonadBuild m',
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] s,
                                              OneOf '[Data.Word.Word16, Double,
                                                      Float] dtype) =>
                 Tensor v'1 s -- ^ __shape__
                 -> Tensor v'2 dtype -- ^ __rate__
                 -> m' (Tensor Value dtype) -- ^ __output__
randomPoisson :: Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
randomPoisson = OpParams
-> Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) s t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] s,
 OneOf '[Word16, Double, Float] t) =>
OpParams -> Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
randomPoisson' OpParams
forall a. a -> a
id
randomPoisson' :: forall v'1 v'2 s dtype m' . (MonadBuild m',
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] s,
                                               OneOf '[Data.Word.Word16, Double,
                                                       Float] dtype) =>
                  OpParams ->
                  Tensor v'1 s -- ^ __shape__
                  -> Tensor v'2 dtype -- ^ __rate__
                  -> m' (Tensor Value dtype) -- ^ __output__
randomPoisson' :: OpParams
-> Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
randomPoisson' op'options :: OpParams
op'options shape :: Tensor v'1 s
shape rate :: Tensor v'2 dtype
rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 s
shape,
                                                             Tensor v'2 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 dtype
rate]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomPoisson"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "rate" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
-}
-- | 
randomPoissonV2 :: forall v'1 v'2 s r dtype m' . (MonadBuild m',
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] s,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Word.Word16,
                                                          Double, Float] r,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Word.Word16,
                                                          Double,
                                                          Float] dtype) =>
                   Tensor v'1 s -- ^ __shape__
                   -> Tensor v'2 r -- ^ __rate__
                   -> m' (Tensor Value dtype) -- ^ __output__
randomPoissonV2 :: Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
randomPoissonV2 = OpParams -> Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) s r dtype (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] s,
 OneOf '[Int32, Int64, Word16, Double, Float] r,
 OneOf '[Int32, Int64, Word16, Double, Float] dtype) =>
OpParams -> Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
randomPoissonV2' OpParams
forall a. a -> a
id
randomPoissonV2' :: forall v'1 v'2 s r dtype m' . (MonadBuild m',
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] s,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Word.Word16,
                                                           Double, Float] r,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Word.Word16,
                                                           Double,
                                                           Float] dtype) =>
                    OpParams ->
                    Tensor v'1 s -- ^ __shape__
                    -> Tensor v'2 r -- ^ __rate__
                    -> m' (Tensor Value dtype) -- ^ __output__
randomPoissonV2' :: OpParams -> Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
randomPoissonV2' op'options :: OpParams
op'options shape :: Tensor v'1 s
shape rate :: Tensor v'2 r
rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 s
shape,
                                                             Tensor v'2 r -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 r
rate]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomPoissonV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "R" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ r -> DataType
forall a. TensorType a => a -> DataType
tensorType (r
forall a. HasCallStack => a
undefined :: r)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "rate" type_attr: "R" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "R"
  type: "type"
  default_value { type: DT_DOUBLE }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
randomShuffle :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                 Tensor v'1 t -- ^ __value__
                 -> m' (Tensor Value t) -- ^ __output__
randomShuffle :: Tensor v'1 t -> m' (Tensor Value t)
randomShuffle = OpParams -> Tensor v'1 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value t)
randomShuffle' OpParams
forall a. a -> a
id
randomShuffle' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __value__
                  -> m' (Tensor Value t) -- ^ __output__
randomShuffle' :: OpParams -> Tensor v'1 t -> m' (Tensor Value t)
randomShuffle' op'options :: OpParams
op'options value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomShuffle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "T" type: "type" }
-}
-- | 
randomShuffleQueue :: forall m' . (MonadBuild m') =>
                      [DataType] -- ^ __component_types__
                      -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
randomShuffleQueue :: [DataType] -> m' (Tensor Ref ByteString)
randomShuffleQueue = OpParams -> [DataType] -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Ref ByteString)
randomShuffleQueue' OpParams
forall a. a -> a
id
randomShuffleQueue' :: forall m' . (MonadBuild m') => OpParams ->
                       [DataType] -- ^ __component_types__
                       -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
randomShuffleQueue' :: OpParams -> [DataType] -> m' (Tensor Ref ByteString)
randomShuffleQueue' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomShuffleQueue"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr {
  name: "min_after_dequeue" type: "int" default_value { i: 0 }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
randomShuffleQueueV2 :: forall m' . (MonadBuild m') =>
                        [DataType] -- ^ __component_types__
                        -> m' (Tensor Value ResourceHandle) -- ^ __handle__
randomShuffleQueueV2 :: [DataType] -> m' (Tensor Value ResourceHandle)
randomShuffleQueueV2 = OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
randomShuffleQueueV2' OpParams
forall a. a -> a
id
randomShuffleQueueV2' :: forall m' . (MonadBuild m') => OpParams ->
                         [DataType] -- ^ __component_types__
                         -> m' (Tensor Value ResourceHandle) -- ^ __handle__
randomShuffleQueueV2' :: OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
randomShuffleQueueV2' op'options :: OpParams
op'options component_types :: [DataType]
component_types | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomShuffleQueueV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "component_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
component_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "component_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "shapes"
  type: "list(shape)"
  default_value { list { } }
  has_minimum: true
}
attr { name: "capacity" type: "int" default_value { i: -1 } }
attr {
  name: "min_after_dequeue" type: "int" default_value { i: 0 }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
randomStandardNormal :: forall v'1 dtype t m' . (MonadBuild m',
                                                 OneOf '[Data.Word.Word16,
                                                         Double, Float] dtype,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] t) =>
                        Tensor v'1 t -- ^ __shape__
                        -> m' (Tensor Value dtype) -- ^ __output__
randomStandardNormal :: Tensor v'1 t -> m' (Tensor Value dtype)
randomStandardNormal = OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] dtype,
 OneOf '[Int32, Int64] t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
randomStandardNormal' OpParams
forall a. a -> a
id
randomStandardNormal' :: forall v'1 dtype t m' . (MonadBuild m',
                                                  OneOf '[Data.Word.Word16,
                                                          Double, Float] dtype,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] t) =>
                         OpParams ->
                         Tensor v'1 t -- ^ __shape__
                         -> m' (Tensor Value dtype) -- ^ __output__
randomStandardNormal' :: OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
randomStandardNormal' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomStandardNormal"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
randomUniform :: forall v'1 dtype t m' . (MonadBuild m',
                                          OneOf '[Data.Word.Word16, Double,
                                                  Float] dtype,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] t) =>
                 Tensor v'1 t -- ^ __shape__
                 -> m' (Tensor Value dtype) -- ^ __output__
randomUniform :: Tensor v'1 t -> m' (Tensor Value dtype)
randomUniform = OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] dtype,
 OneOf '[Int32, Int64] t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
randomUniform' OpParams
forall a. a -> a
id
randomUniform' :: forall v'1 dtype t m' . (MonadBuild m',
                                           OneOf '[Data.Word.Word16, Double,
                                                   Float] dtype,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] t) =>
                  OpParams ->
                  Tensor v'1 t -- ^ __shape__
                  -> m' (Tensor Value dtype) -- ^ __output__
randomUniform' :: OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
randomUniform' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomUniform"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
randomUniformInt :: forall v'1 v'2 v'3 tout t m' . (MonadBuild m',
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tout,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] t) =>
                    Tensor v'1 t -- ^ __shape__
                    -> Tensor v'2 tout -- ^ __minval__
                    -> Tensor v'3 tout -- ^ __maxval__
                    -> m' (Tensor Value tout) -- ^ __output__
randomUniformInt :: Tensor v'1 t
-> Tensor v'2 tout -> Tensor v'3 tout -> m' (Tensor Value tout)
randomUniformInt = OpParams
-> Tensor v'1 t
-> Tensor v'2 tout
-> Tensor v'3 tout
-> m' (Tensor Value tout)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tout t
       (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] tout,
 OneOf '[Int32, Int64] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tout
-> Tensor v'3 tout
-> m' (Tensor Value tout)
randomUniformInt' OpParams
forall a. a -> a
id
randomUniformInt' :: forall v'1 v'2 v'3 tout t m' . (MonadBuild m',
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tout,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __shape__
                     -> Tensor v'2 tout -- ^ __minval__
                     -> Tensor v'3 tout -- ^ __maxval__
                     -> m' (Tensor Value tout) -- ^ __output__
randomUniformInt' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tout
-> Tensor v'3 tout
-> m' (Tensor Value tout)
randomUniformInt' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape minval :: Tensor v'2 tout
minval maxval :: Tensor v'3 tout
maxval | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tout) -> m' (Tensor Value tout)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tout) -> m' (Tensor Value tout))
-> Build (Tensor Value tout) -> m' (Tensor Value tout)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tout
minval,
                                                             Tensor v'3 tout -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tout
maxval]
        [Int64] -> OpDef -> Build (Tensor Value tout)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RandomUniformInt"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "minval" type_attr: "Tout" }
input_arg { name: "maxval" type_attr: "Tout" }
output_arg { name: "output" type_attr: "Tout" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "Tout"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
range :: forall v'1 v'2 v'3 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] tidx) =>
         Tensor v'1 tidx -- ^ __start__
         -> Tensor v'2 tidx -- ^ __limit__
         -> Tensor v'3 tidx -- ^ __delta__
         -> Tensor Build tidx -- ^ __output__
range :: Tensor v'1 tidx
-> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor Build tidx
range = OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor Build tidx
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tidx.
OneOf '[Int32, Int64, Word16, Double, Float] tidx =>
OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor Build tidx
range' OpParams
forall a. a -> a
id
range' :: forall v'1 v'2 v'3 tidx . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Data.Word.Word16, Double,
                                             Float] tidx) => OpParams ->
          Tensor v'1 tidx -- ^ __start__
          -> Tensor v'2 tidx -- ^ __limit__
          -> Tensor v'3 tidx -- ^ __delta__
          -> Tensor Build tidx -- ^ __output__
range' :: OpParams
-> Tensor v'1 tidx
-> Tensor v'2 tidx
-> Tensor v'3 tidx
-> Tensor Build tidx
range' op'options :: OpParams
op'options start :: Tensor v'1 tidx
start limit :: Tensor v'2 tidx
limit delta :: Tensor v'3 tidx
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tidx
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tidx)
-> Build OpDef -> Tensor Build tidx
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tidx
start,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
limit,
                                                             Tensor v'3 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tidx
delta]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Range"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "start" type_attr: "Tidx" }
input_arg { name: "limit" type_attr: "Tidx" }
input_arg { name: "delta" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "Tidx" }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
rangeDataset :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                [DataType] -- ^ __output_types__
                -> Tensor v'1 Data.Int.Int64 -- ^ __start__
                -> Tensor v'2 Data.Int.Int64 -- ^ __stop__
                -> Tensor v'3 Data.Int.Int64 -- ^ __step__
                -> m' (Tensor Value Variant) -- ^ __handle__
rangeDataset :: [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
rangeDataset = OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
rangeDataset' OpParams
forall a. a -> a
id
rangeDataset' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Data.Int.Int64 -- ^ __start__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __stop__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __step__
                 -> m' (Tensor Value Variant) -- ^ __handle__
rangeDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
rangeDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types start :: Tensor v'1 Int64
start stop :: Tensor v'2 Int64
stop step :: Tensor v'3 Int64
step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
start,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
stop,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
step]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RangeDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "start" type: DT_INT64 }
input_arg { name: "stop" type: DT_INT64 }
input_arg { name: "step" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
rank :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
        -> Tensor Build Data.Int.Int32 -- ^ __output__
rank :: Tensor v'1 t -> Tensor Build Int32
rank = OpParams -> Tensor v'1 t -> Tensor Build Int32
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build Int32
rank' OpParams
forall a. a -> a
id
rank' :: forall v'1 t . (TensorType t) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build Data.Int.Int32 -- ^ __output__
rank' :: OpParams -> Tensor v'1 t -> Tensor Build Int32
rank' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Rank"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_INT32 }
attr { name: "T" type: "type" }
-}
-- | 
readFile :: Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
            -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
readFile :: Tensor v'1 ByteString -> Tensor Build ByteString
readFile = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
readFile' OpParams
forall a. a -> a
id
readFile' :: OpParams ->
             Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
             -> Tensor Build Data.ByteString.ByteString -- ^ __contents__
readFile' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
readFile' op'options :: OpParams
op'options filename :: Tensor v'1 ByteString
filename | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filename]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReadFile"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filename" type: DT_STRING }
output_arg { name: "contents" type: DT_STRING }
-}
-- | 
readVariableOp :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                  Tensor v'1 ResourceHandle -- ^ __resource__
                  -> m' (Tensor Value dtype) -- ^ __value__
readVariableOp :: Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
readVariableOp = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
readVariableOp' OpParams
forall a. a -> a
id
readVariableOp' :: forall v'1 dtype m' . (MonadBuild m', TensorType dtype) =>
                   OpParams ->
                   Tensor v'1 ResourceHandle -- ^ __resource__
                   -> m' (Tensor Value dtype) -- ^ __value__
readVariableOp' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
readVariableOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReadVariableOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
readerNumRecordsProduced :: forall m' . (MonadBuild m') =>
                            Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                            -> m' (Tensor Value Data.Int.Int64) -- ^ __records_produced__
readerNumRecordsProduced :: Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumRecordsProduced = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumRecordsProduced' OpParams
forall a. a -> a
id
readerNumRecordsProduced' :: forall m' . (MonadBuild m') => OpParams ->
                             Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                             -> m' (Tensor Value Data.Int.Int64) -- ^ __records_produced__
readerNumRecordsProduced' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumRecordsProduced' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderNumRecordsProduced"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
output_arg { name: "records_produced" type: DT_INT64 }
-}
-- | 
readerNumRecordsProducedV2 :: forall v'1 m' . (MonadBuild m') =>
                              Tensor v'1 ResourceHandle -- ^ __reader_handle__
                              -> m' (Tensor Value Data.Int.Int64) -- ^ __records_produced__
readerNumRecordsProducedV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumRecordsProducedV2 = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumRecordsProducedV2' OpParams
forall a. a -> a
id
readerNumRecordsProducedV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __reader_handle__
                               -> m' (Tensor Value Data.Int.Int64) -- ^ __records_produced__
readerNumRecordsProducedV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumRecordsProducedV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderNumRecordsProducedV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
output_arg { name: "records_produced" type: DT_INT64 }
-}
-- | 
readerNumWorkUnitsCompleted :: forall m' . (MonadBuild m') =>
                               Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                               -> m' (Tensor Value Data.Int.Int64) -- ^ __units_completed__
readerNumWorkUnitsCompleted :: Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumWorkUnitsCompleted = OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumWorkUnitsCompleted' OpParams
forall a. a -> a
id
readerNumWorkUnitsCompleted' :: forall m' . (MonadBuild m') => OpParams ->
                                Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                                -> m' (Tensor Value Data.Int.Int64) -- ^ __units_completed__
readerNumWorkUnitsCompleted' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
readerNumWorkUnitsCompleted' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderNumWorkUnitsCompleted"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
output_arg { name: "units_completed" type: DT_INT64 }
-}
-- | 
readerNumWorkUnitsCompletedV2 :: forall v'1 m' . (MonadBuild m') =>
                                 Tensor v'1 ResourceHandle -- ^ __reader_handle__
                                 -> m' (Tensor Value Data.Int.Int64) -- ^ __units_completed__
readerNumWorkUnitsCompletedV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumWorkUnitsCompletedV2 = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumWorkUnitsCompletedV2' OpParams
forall a. a -> a
id
readerNumWorkUnitsCompletedV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                                  Tensor v'1 ResourceHandle -- ^ __reader_handle__
                                  -> m' (Tensor Value Data.Int.Int64) -- ^ __units_completed__
readerNumWorkUnitsCompletedV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
readerNumWorkUnitsCompletedV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64) -> m' (Tensor Value Int64))
-> Build (Tensor Value Int64) -> m' (Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderNumWorkUnitsCompletedV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
output_arg { name: "units_completed" type: DT_INT64 }
-}
-- | 
readerRead :: forall m' . (MonadBuild m') =>
              Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
              -> Tensor Ref Data.ByteString.ByteString -- ^ __queue_handle__
              -> m' ((Tensor Value Data.ByteString.ByteString,
                      Tensor Value Data.ByteString.ByteString))
              -- ^ (__key__, __value__)
              --
              -- * __key__
              --
              -- * __value__
readerRead :: Tensor Ref ByteString
-> Tensor Ref ByteString
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerRead = OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerRead' OpParams
forall a. a -> a
id
readerRead' :: forall m' . (MonadBuild m') => OpParams ->
               Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
               -> Tensor Ref Data.ByteString.ByteString -- ^ __queue_handle__
               -> m' ((Tensor Value Data.ByteString.ByteString,
                       Tensor Value Data.ByteString.ByteString))
               -- ^ (__key__, __value__)
               --
               -- * __key__
               --
               -- * __value__
readerRead' :: OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerRead' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle queue_handle :: Tensor Ref ByteString
queue_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString, Tensor Value ByteString)
 -> m' (Tensor Value ByteString, Tensor Value ByteString))
-> Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle,
                                                             Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
queue_handle]
        [Int64]
-> OpDef
-> Build (Tensor Value ByteString, Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderRead"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
input_arg { name: "queue_handle" type: DT_STRING is_ref: true }
output_arg { name: "key" type: DT_STRING }
output_arg { name: "value" type: DT_STRING }
-}
-- | 
readerReadUpTo :: forall v'3 m' . (MonadBuild m') =>
                  Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                  -> Tensor Ref Data.ByteString.ByteString -- ^ __queue_handle__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __num_records__
                  -> m' ((Tensor Value Data.ByteString.ByteString,
                          Tensor Value Data.ByteString.ByteString))
                  -- ^ (__keys__, __values__)
                  --
                  -- * __keys__
                  --
                  -- * __values__
readerReadUpTo :: Tensor Ref ByteString
-> Tensor Ref ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpTo = OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (v'3 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpTo' OpParams
forall a. a -> a
id
readerReadUpTo' :: forall v'3 m' . (MonadBuild m') => OpParams ->
                   Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                   -> Tensor Ref Data.ByteString.ByteString -- ^ __queue_handle__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __num_records__
                   -> m' ((Tensor Value Data.ByteString.ByteString,
                           Tensor Value Data.ByteString.ByteString))
                   -- ^ (__keys__, __values__)
                   --
                   -- * __keys__
                   --
                   -- * __values__
readerReadUpTo' :: OpParams
-> Tensor Ref ByteString
-> Tensor Ref ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpTo' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle queue_handle :: Tensor Ref ByteString
queue_handle
                num_records :: Tensor v'3 Int64
num_records | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString, Tensor Value ByteString)
 -> m' (Tensor Value ByteString, Tensor Value ByteString))
-> Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle,
                                                             Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
queue_handle,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
num_records]
        [Int64]
-> OpDef
-> Build (Tensor Value ByteString, Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderReadUpTo"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
input_arg { name: "queue_handle" type: DT_STRING is_ref: true }
input_arg { name: "num_records" type: DT_INT64 }
output_arg { name: "keys" type: DT_STRING }
output_arg { name: "values" type: DT_STRING }
-}
-- | 
readerReadUpToV2 :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                    Tensor v'1 ResourceHandle -- ^ __reader_handle__
                    -> Tensor v'2 ResourceHandle -- ^ __queue_handle__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __num_records__
                    -> m' ((Tensor Value Data.ByteString.ByteString,
                            Tensor Value Data.ByteString.ByteString))
                    -- ^ (__keys__, __values__)
                    --
                    -- * __keys__
                    --
                    -- * __values__
readerReadUpToV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpToV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpToV2' OpParams
forall a. a -> a
id
readerReadUpToV2' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                     Tensor v'1 ResourceHandle -- ^ __reader_handle__
                     -> Tensor v'2 ResourceHandle -- ^ __queue_handle__
                     -> Tensor v'3 Data.Int.Int64 -- ^ __num_records__
                     -> m' ((Tensor Value Data.ByteString.ByteString,
                             Tensor Value Data.ByteString.ByteString))
                     -- ^ (__keys__, __values__)
                     --
                     -- * __keys__
                     --
                     -- * __values__
readerReadUpToV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 Int64
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadUpToV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle queue_handle :: Tensor v'2 ResourceHandle
queue_handle
                  num_records :: Tensor v'3 Int64
num_records | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString, Tensor Value ByteString)
 -> m' (Tensor Value ByteString, Tensor Value ByteString))
-> Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
queue_handle,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
num_records]
        [Int64]
-> OpDef
-> Build (Tensor Value ByteString, Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderReadUpToV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
input_arg { name: "queue_handle" type: DT_RESOURCE }
input_arg { name: "num_records" type: DT_INT64 }
output_arg { name: "keys" type: DT_STRING }
output_arg { name: "values" type: DT_STRING }
-}
-- | 
readerReadV2 :: forall v'1 v'2 m' . (MonadBuild m') =>
                Tensor v'1 ResourceHandle -- ^ __reader_handle__
                -> Tensor v'2 ResourceHandle -- ^ __queue_handle__
                -> m' ((Tensor Value Data.ByteString.ByteString,
                        Tensor Value Data.ByteString.ByteString))
                -- ^ (__key__, __value__)
                --
                -- * __key__
                --
                -- * __value__
readerReadV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadV2' OpParams
forall a. a -> a
id
readerReadV2' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                 Tensor v'1 ResourceHandle -- ^ __reader_handle__
                 -> Tensor v'2 ResourceHandle -- ^ __queue_handle__
                 -> m' ((Tensor Value Data.ByteString.ByteString,
                         Tensor Value Data.ByteString.ByteString))
                 -- ^ (__key__, __value__)
                 --
                 -- * __key__
                 --
                 -- * __value__
readerReadV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value ByteString, Tensor Value ByteString)
readerReadV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle queue_handle :: Tensor v'2 ResourceHandle
queue_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString, Tensor Value ByteString)
 -> m' (Tensor Value ByteString, Tensor Value ByteString))
-> Build (Tensor Value ByteString, Tensor Value ByteString)
-> m' (Tensor Value ByteString, Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
queue_handle]
        [Int64]
-> OpDef
-> Build (Tensor Value ByteString, Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderReadV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
input_arg { name: "queue_handle" type: DT_RESOURCE }
output_arg { name: "key" type: DT_STRING }
output_arg { name: "value" type: DT_STRING }
-}
-- | 
readerReset :: forall m' . (MonadBuild m') =>
               Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
               -> m' (ControlNode)
readerReset :: Tensor Ref ByteString -> m' ControlNode
readerReset = OpParams -> Tensor Ref ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' ControlNode
readerReset' OpParams
forall a. a -> a
id
readerReset' :: forall m' . (MonadBuild m') => OpParams ->
                Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                -> m' (ControlNode)
readerReset' :: OpParams -> Tensor Ref ByteString -> m' ControlNode
readerReset' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderReset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
-}
-- | 
readerResetV2 :: forall v'1 m' . (MonadBuild m') =>
                 Tensor v'1 ResourceHandle -- ^ __reader_handle__
                 -> m' (ControlNode)
readerResetV2 :: Tensor v'1 ResourceHandle -> m' ControlNode
readerResetV2 = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
readerResetV2' OpParams
forall a. a -> a
id
readerResetV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                  Tensor v'1 ResourceHandle -- ^ __reader_handle__
                  -> m' (ControlNode)
readerResetV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
readerResetV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderResetV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
-}
-- | 
readerRestoreState :: forall v'2 m' . (MonadBuild m') =>
                      Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                      -> Tensor v'2 Data.ByteString.ByteString -- ^ __state__
                      -> m' (ControlNode)
readerRestoreState :: Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
readerRestoreState = OpParams
-> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
forall (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
readerRestoreState' OpParams
forall a. a -> a
id
readerRestoreState' :: forall v'2 m' . (MonadBuild m') => OpParams ->
                       Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                       -> Tensor v'2 Data.ByteString.ByteString -- ^ __state__
                       -> m' (ControlNode)
readerRestoreState' :: OpParams
-> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
readerRestoreState' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle state :: Tensor v'2 ByteString
state | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
state]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderRestoreState"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
input_arg { name: "state" type: DT_STRING }
-}
-- | 
readerRestoreStateV2 :: forall v'1 v'2 m' . (MonadBuild m') =>
                        Tensor v'1 ResourceHandle -- ^ __reader_handle__
                        -> Tensor v'2 Data.ByteString.ByteString -- ^ __state__
                        -> m' (ControlNode)
readerRestoreStateV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString -> m' ControlNode
readerRestoreStateV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
readerRestoreStateV2' OpParams
forall a. a -> a
id
readerRestoreStateV2' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __reader_handle__
                         -> Tensor v'2 Data.ByteString.ByteString -- ^ __state__
                         -> m' (ControlNode)
readerRestoreStateV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
readerRestoreStateV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle state :: Tensor v'2 ByteString
state | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
state]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderRestoreStateV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
input_arg { name: "state" type: DT_STRING }
-}
-- | 
readerSerializeState :: forall m' . (MonadBuild m') =>
                        Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                        -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __state__
readerSerializeState :: Tensor Ref ByteString -> m' (Tensor Value ByteString)
readerSerializeState = OpParams -> Tensor Ref ByteString -> m' (Tensor Value ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value ByteString)
readerSerializeState' OpParams
forall a. a -> a
id
readerSerializeState' :: forall m' . (MonadBuild m') => OpParams ->
                         Tensor Ref Data.ByteString.ByteString -- ^ __reader_handle__
                         -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __state__
readerSerializeState' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value ByteString)
readerSerializeState' op'options :: OpParams
op'options reader_handle :: Tensor Ref ByteString
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderSerializeState"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_STRING is_ref: true }
output_arg { name: "state" type: DT_STRING }
-}
-- | 
readerSerializeStateV2 :: forall v'1 m' . (MonadBuild m') =>
                          Tensor v'1 ResourceHandle -- ^ __reader_handle__
                          -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __state__
readerSerializeStateV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
readerSerializeStateV2 = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
readerSerializeStateV2' OpParams
forall a. a -> a
id
readerSerializeStateV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __reader_handle__
                           -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __state__
readerSerializeStateV2' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
readerSerializeStateV2' op'options :: OpParams
op'options reader_handle :: Tensor v'1 ResourceHandle
reader_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
reader_handle]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ReaderSerializeStateV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reader_handle" type: DT_RESOURCE }
output_arg { name: "state" type: DT_STRING }
-}
-- | 
real :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float)] t,
                             OneOf '[Double, Float] tout) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor Build tout -- ^ __output__
real :: Tensor v'1 t -> Tensor Build tout
real = OpParams -> Tensor v'1 t -> Tensor Build tout
forall (v'1 :: * -> *) t tout.
(OneOf '[Complex Double, Complex Float] t,
 OneOf '[Double, Float] tout) =>
OpParams -> Tensor v'1 t -> Tensor Build tout
real' OpParams
forall a. a -> a
id
real' :: forall v'1 t tout . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float)] t,
                              OneOf '[Double, Float] tout) => OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build tout -- ^ __output__
real' :: OpParams -> Tensor v'1 t -> Tensor Build tout
real' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tout
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tout)
-> Build OpDef -> Tensor Build tout
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Real"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tout" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tout -> DataType
forall a. TensorType a => a -> DataType
tensorType (tout
forall a. HasCallStack => a
undefined :: tout)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "Tout" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_COMPLEX64 }
  allowed_values { list { type: DT_COMPLEX64 type: DT_COMPLEX128 } }
}
attr {
  name: "Tout"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
realDiv :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word8,
                                       Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor Build t -- ^ __z__
realDiv :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
realDiv = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
realDiv' OpParams
forall a. a -> a
id
realDiv' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
realDiv' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
realDiv' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RealDiv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
rebatchDataset :: [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __num_replicas__
                  -> Tensor Build Variant -- ^ __handle__
rebatchDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
rebatchDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
rebatchDataset' OpParams
forall a. a -> a
id
rebatchDataset' :: OpParams ->
                   [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __num_replicas__
                   -> Tensor Build Variant -- ^ __handle__
rebatchDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
rebatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                num_replicas :: Tensor v'2 Int64
num_replicas | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_replicas]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RebatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_replicas" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "use_fallback" type: "bool" default_value { b: true }
}
-}
-- | 
reciprocal :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Double, Float] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
reciprocal :: Tensor v'1 t -> Tensor Build t
reciprocal = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
reciprocal' OpParams
forall a. a -> a
id
reciprocal' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Double, Float] t) =>
               OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor Build t -- ^ __y__
reciprocal' :: OpParams -> Tensor v'1 t -> Tensor Build t
reciprocal' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Reciprocal"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
reciprocalGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Word.Word16, Double,
                                              Float] t) =>
                  Tensor v'1 t -- ^ __y__
                  -> Tensor v'2 t -- ^ __dy__
                  -> Tensor Build t -- ^ __z__
reciprocalGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reciprocalGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reciprocalGrad' OpParams
forall a. a -> a
id
reciprocalGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Word.Word16, Double,
                                               Float] t) => OpParams ->
                   Tensor v'1 t -- ^ __y__
                   -> Tensor v'2 t -- ^ __dy__
                   -> Tensor Build t -- ^ __z__
reciprocalGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reciprocalGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReciprocalGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
recordInput :: forall m' . (MonadBuild m') => ByteString -- ^ __file_pattern__
               -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __records__
recordInput :: ByteString -> m' (Tensor Value ByteString)
recordInput = OpParams -> ByteString -> m' (Tensor Value ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> ByteString -> m' (Tensor Value ByteString)
recordInput' OpParams
forall a. a -> a
id
recordInput' :: forall m' . (MonadBuild m') => OpParams ->
                ByteString -- ^ __file_pattern__
                -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __records__
recordInput' :: OpParams -> ByteString -> m' (Tensor Value ByteString)
recordInput' op'options :: OpParams
op'options file_pattern :: ByteString
file_pattern | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RecordInput"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "file_pattern" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
file_pattern
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "records" type: DT_STRING }
attr { name: "file_pattern" type: "string" }
attr {
  name: "file_random_seed" type: "int" default_value { i: 301 }
}
attr {
  name: "file_shuffle_shift_ratio"
  type: "float"
  default_value { f: 0.0 }
}
attr {
  name: "file_buffer_size" type: "int" default_value { i: 10000 }
}
attr {
  name: "file_parallelism" type: "int" default_value { i: 16 }
}
attr { name: "batch_size" type: "int" default_value { i: 32 } }
attr {
  name: "compression_type" type: "string" default_value { s: "" }
}
-}
-- | 
recv :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
        ByteString -- ^ __recv_device__
        -> ByteString -- ^ __send_device__
        -> Data.Int.Int64 -- ^ __send_device_incarnation__
        -> ByteString -- ^ __tensor_name__
        -> m' (Tensor Value tensor_type) -- ^ __tensor__
recv :: ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
recv = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
forall tensor_type (m' :: * -> *).
(MonadBuild m', TensorType tensor_type) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
recv' OpParams
forall a. a -> a
id
recv' :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
         OpParams ->
         ByteString -- ^ __recv_device__
         -> ByteString -- ^ __send_device__
         -> Data.Int.Int64 -- ^ __send_device_incarnation__
         -> ByteString -- ^ __tensor_name__
         -> m' (Tensor Value tensor_type) -- ^ __tensor__
recv' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
recv' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation
      tensor_name :: ByteString
tensor_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type))
-> Build (Tensor Value tensor_type)
-> m' (Tensor Value tensor_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value tensor_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Recv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tensor_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (tensor_type
forall a. HasCallStack => a
undefined :: tensor_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "tensor" type_attr: "tensor_type" }
attr { name: "tensor_type" type: "type" }
attr { name: "tensor_name" type: "string" }
attr { name: "send_device" type: "string" }
attr { name: "send_device_incarnation" type: "int" }
attr { name: "recv_device" type: "string" }
attr {
  name: "client_terminated" type: "bool" default_value { b: false }
}
-}
-- | 
recvTPUEmbeddingActivations :: forall m' . (MonadBuild m') =>
                               ByteString -- ^ __config__
                               -> Data.Int.Int64 -- ^ __num_outputs__
                               -> m' ([Tensor Value Float]) -- ^ __outputs__
recvTPUEmbeddingActivations :: ByteString -> Int64 -> m' [Tensor Value Float]
recvTPUEmbeddingActivations = OpParams -> ByteString -> Int64 -> m' [Tensor Value Float]
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> ByteString -> Int64 -> m' [Tensor Value Float]
recvTPUEmbeddingActivations' OpParams
forall a. a -> a
id
recvTPUEmbeddingActivations' :: forall m' . (MonadBuild m') => OpParams ->
                                ByteString -- ^ __config__
                                -> Data.Int.Int64 -- ^ __num_outputs__
                                -> m' ([Tensor Value Float]) -- ^ __outputs__
recvTPUEmbeddingActivations' :: OpParams -> ByteString -> Int64 -> m' [Tensor Value Float]
recvTPUEmbeddingActivations' op'options :: OpParams
op'options config :: ByteString
config num_outputs :: Int64
num_outputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build [Tensor Value Float] -> m' [Tensor Value Float]
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build [Tensor Value Float] -> m' [Tensor Value Float])
-> Build [Tensor Value Float] -> m' [Tensor Value Float]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build [Tensor Value Float]
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [Int64
num_outputs] (OpType -> OpDef
opDef "RecvTPUEmbeddingActivations"
                               OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
config
                               OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_outputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_outputs
                               OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "outputs" type: DT_FLOAT number_attr: "num_outputs"
}
attr {
  name: "num_outputs" type: "int" has_minimum: true minimum: 1
}
attr { name: "config" type: "string" }
-}
-- | 
reduceJoin :: Tensor v'1 Data.ByteString.ByteString -- ^ __inputs__
              -> Tensor v'2 Data.Int.Int32 -- ^ __reduction_indices__
              -> Tensor Build Data.ByteString.ByteString -- ^ __output__
reduceJoin :: Tensor v'1 ByteString
-> Tensor v'2 Int32 -> Tensor Build ByteString
reduceJoin = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
reduceJoin' OpParams
forall a. a -> a
id
reduceJoin' :: OpParams ->
               Tensor v'1 Data.ByteString.ByteString -- ^ __inputs__
               -> Tensor v'2 Data.Int.Int32 -- ^ __reduction_indices__
               -> Tensor Build Data.ByteString.ByteString -- ^ __output__
reduceJoin' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
reduceJoin' op'options :: OpParams
op'options inputs :: Tensor v'1 ByteString
inputs reduction_indices :: Tensor v'2 Int32
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
inputs,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReduceJoin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_STRING }
input_arg { name: "reduction_indices" type: DT_INT32 }
output_arg { name: "output" type: DT_STRING }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr { name: "separator" type: "string" default_value { s: "" } }
-}
-- | 
refEnter :: forall t m' . (MonadBuild m', TensorType t) =>
            ByteString -- ^ __frame_name__
            -> Tensor Ref t -- ^ __data__
            -> m' (Tensor Ref t) -- ^ __output__
refEnter :: ByteString -> Tensor Ref t -> m' (Tensor Ref t)
refEnter = OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Ref t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Ref t)
refEnter' OpParams
forall a. a -> a
id
refEnter' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
             ByteString -- ^ __frame_name__
             -> Tensor Ref t -- ^ __data__
             -> m' (Tensor Ref t) -- ^ __output__
refEnter' :: OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Ref t)
refEnter' op'options :: OpParams
op'options frame_name :: ByteString
frame_name data' :: Tensor Ref t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
data']
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefEnter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "frame_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
frame_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr { name: "frame_name" type: "string" }
attr {
  name: "is_constant" type: "bool" default_value { b: false }
}
attr {
  name: "parallel_iterations" type: "int" default_value { i: 10 }
}
-}
-- | 
refExit :: forall t m' . (MonadBuild m', TensorType t) =>
           Tensor Ref t -- ^ __data__
           -> m' (Tensor Ref t) -- ^ __output__
refExit :: Tensor Ref t -> m' (Tensor Ref t)
refExit = OpParams -> Tensor Ref t -> m' (Tensor Ref t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refExit' OpParams
forall a. a -> a
id
refExit' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
            Tensor Ref t -- ^ __data__
            -> m' (Tensor Ref t) -- ^ __output__
refExit' :: OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refExit' op'options :: OpParams
op'options data' :: Tensor Ref t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
data']
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefExit"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
-}
-- | 
refIdentity :: forall t m' . (MonadBuild m', TensorType t) =>
               Tensor Ref t -- ^ __input__
               -> m' (Tensor Ref t) -- ^ __output__
refIdentity :: Tensor Ref t -> m' (Tensor Ref t)
refIdentity = OpParams -> Tensor Ref t -> m' (Tensor Ref t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refIdentity' OpParams
forall a. a -> a
id
refIdentity' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
                Tensor Ref t -- ^ __input__
                -> m' (Tensor Ref t) -- ^ __output__
refIdentity' :: OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refIdentity' op'options :: OpParams
op'options input :: Tensor Ref t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
input]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefIdentity"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
-}
-- | 
refMerge :: forall t m' . (MonadBuild m', TensorType t) =>
            [Tensor Ref t] -- ^ __inputs__
            -> m' ((Tensor Ref t, Tensor Value Data.Int.Int32))
            -- ^ (__output__, __value_index__)
            --
            -- * __output__
            --
            -- * __value_index__
refMerge :: [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
refMerge = OpParams -> [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
refMerge' OpParams
forall a. a -> a
id
refMerge' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
             [Tensor Ref t] -- ^ __inputs__
             -> m' ((Tensor Ref t, Tensor Value Data.Int.Int32))
             -- ^ (__output__, __value_index__)
             --
             -- * __output__
             --
             -- * __value_index__
refMerge' :: OpParams -> [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
refMerge' op'options :: OpParams
op'options
          inputs :: [Tensor Ref t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor Ref t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref t]
inputs)])] =
    Build (Tensor Ref t, Tensor Value Int32)
-> m' (Tensor Ref t, Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t, Tensor Value Int32)
 -> m' (Tensor Ref t, Tensor Value Int32))
-> Build (Tensor Ref t, Tensor Value Int32)
-> m' (Tensor Ref t, Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor Ref t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor Ref t]
inputs]
        [Int64] -> OpDef -> Build (Tensor Ref t, Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefMerge"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor Ref t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref t]
inputs) :: Int64
{-
input_arg {
  name: "inputs" type_attr: "T" number_attr: "N" is_ref: true
}
output_arg { name: "output" type_attr: "T" is_ref: true }
output_arg { name: "value_index" type: DT_INT32 }
attr { name: "T" type: "type" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
refNextIteration :: forall t m' . (MonadBuild m', TensorType t) =>
                    Tensor Ref t -- ^ __data__
                    -> m' (Tensor Ref t) -- ^ __output__
refNextIteration :: Tensor Ref t -> m' (Tensor Ref t)
refNextIteration = OpParams -> Tensor Ref t -> m' (Tensor Ref t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refNextIteration' OpParams
forall a. a -> a
id
refNextIteration' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
                     Tensor Ref t -- ^ __data__
                     -> m' (Tensor Ref t) -- ^ __output__
refNextIteration' :: OpParams -> Tensor Ref t -> m' (Tensor Ref t)
refNextIteration' op'options :: OpParams
op'options data' :: Tensor Ref t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
data']
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefNextIteration"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" is_ref: true }
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
-}
-- | 
refSelect :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
             Tensor v'1 Data.Int.Int32 -- ^ __index__
             -> [Tensor Ref t] -- ^ __inputs__
             -> m' (Tensor Ref t) -- ^ __output__
refSelect :: Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
refSelect = OpParams -> Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
refSelect' OpParams
forall a. a -> a
id
refSelect' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
              Tensor v'1 Data.Int.Int32 -- ^ __index__
              -> [Tensor Ref t] -- ^ __inputs__
              -> m' (Tensor Ref t) -- ^ __output__
refSelect' :: OpParams -> Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
refSelect' op'options :: OpParams
op'options index :: Tensor v'1 Int32
index
           inputs :: [Tensor Ref t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor Ref t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref t]
inputs)])] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
index,
                                                             [Tensor Ref t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor Ref t]
inputs]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefSelect"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor Ref t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref t]
inputs) :: Int64
{-
input_arg { name: "index" type: DT_INT32 }
input_arg {
  name: "inputs" type_attr: "T" number_attr: "N" is_ref: true
}
output_arg { name: "output" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | 
refSwitch :: forall v'2 t m' . (MonadBuild m', TensorType t) =>
             Tensor Ref t -- ^ __data__
             -> Tensor v'2 Bool -- ^ __pred__
             -> m' ((Tensor Ref t, Tensor Ref t))
             -- ^ (__output_false__, __output_true__)
             --
             -- * __output_false__
             --
             -- * __output_true__
refSwitch :: Tensor Ref t -> Tensor v'2 Bool -> m' (Tensor Ref t, Tensor Ref t)
refSwitch = OpParams
-> Tensor Ref t
-> Tensor v'2 Bool
-> m' (Tensor Ref t, Tensor Ref t)
forall (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 Bool
-> m' (Tensor Ref t, Tensor Ref t)
refSwitch' OpParams
forall a. a -> a
id
refSwitch' :: forall v'2 t m' . (MonadBuild m', TensorType t) => OpParams ->
              Tensor Ref t -- ^ __data__
              -> Tensor v'2 Bool -- ^ __pred__
              -> m' ((Tensor Ref t, Tensor Ref t))
              -- ^ (__output_false__, __output_true__)
              --
              -- * __output_false__
              --
              -- * __output_true__
refSwitch' :: OpParams
-> Tensor Ref t
-> Tensor v'2 Bool
-> m' (Tensor Ref t, Tensor Ref t)
refSwitch' op'options :: OpParams
op'options data' :: Tensor Ref t
data' pred :: Tensor v'2 Bool
pred | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t, Tensor Ref t)
-> m' (Tensor Ref t, Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t, Tensor Ref t)
 -> m' (Tensor Ref t, Tensor Ref t))
-> Build (Tensor Ref t, Tensor Ref t)
-> m' (Tensor Ref t, Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
data',
                                                             Tensor v'2 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Bool
pred]
        [Int64] -> OpDef -> Build (Tensor Ref t, Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RefSwitch"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" is_ref: true }
input_arg { name: "pred" type: DT_BOOL }
output_arg { name: "output_false" type_attr: "T" is_ref: true }
output_arg { name: "output_true" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
-}
-- | 
regexFullMatch :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __pattern__
                  -> Tensor Build Bool -- ^ __output__
regexFullMatch :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor Build Bool
regexFullMatch = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor Build Bool
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor Build Bool
regexFullMatch' OpParams
forall a. a -> a
id
regexFullMatch' :: OpParams ->
                   Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __pattern__
                   -> Tensor Build Bool -- ^ __output__
regexFullMatch' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor Build Bool
regexFullMatch' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input pattern :: Tensor v'2 ByteString
pattern | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
pattern]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RegexFullMatch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
input_arg { name: "pattern" type: DT_STRING }
output_arg { name: "output" type: DT_BOOL }
-}
-- | 
regexReplace :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __pattern__
                -> Tensor v'3 Data.ByteString.ByteString -- ^ __rewrite__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
regexReplace :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build ByteString
regexReplace = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build ByteString
regexReplace' OpParams
forall a. a -> a
id
regexReplace' :: OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __pattern__
                 -> Tensor v'3 Data.ByteString.ByteString -- ^ __rewrite__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __output__
regexReplace' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build ByteString
regexReplace' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input pattern :: Tensor v'2 ByteString
pattern rewrite :: Tensor v'3 ByteString
rewrite | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
pattern,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
rewrite]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RegexReplace"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
input_arg { name: "pattern" type: DT_STRING }
input_arg { name: "rewrite" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr {
  name: "replace_global" type: "bool" default_value { b: true }
}
-}
-- | 
registerDataset :: Data.Int.Int64 -- ^ __external_state_policy__
                   -> Tensor v'1 Variant -- ^ __dataset__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __address__
                   -> Tensor v'3 Data.ByteString.ByteString -- ^ __protocol__
                   -> Tensor Build Data.Int.Int64 -- ^ __dataset_id__
registerDataset :: Int64
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build Int64
registerDataset = OpParams
-> Int64
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build Int64
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Int64
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build Int64
registerDataset' OpParams
forall a. a -> a
id
registerDataset' :: OpParams ->
                    Data.Int.Int64 -- ^ __external_state_policy__
                    -> Tensor v'1 Variant -- ^ __dataset__
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __address__
                    -> Tensor v'3 Data.ByteString.ByteString -- ^ __protocol__
                    -> Tensor Build Data.Int.Int64 -- ^ __dataset_id__
registerDataset' :: OpParams
-> Int64
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> Tensor Build Int64
registerDataset' op'options :: OpParams
op'options external_state_policy :: Int64
external_state_policy dataset :: Tensor v'1 Variant
dataset address :: Tensor v'2 ByteString
address
                 protocol :: Tensor v'3 ByteString
protocol | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
address,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
protocol]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RegisterDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "external_state_policy" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
external_state_policy
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dataset" type: DT_VARIANT }
input_arg { name: "address" type: DT_STRING }
input_arg { name: "protocol" type: DT_STRING }
output_arg { name: "dataset_id" type: DT_INT64 }
attr { name: "external_state_policy" type: "int" }
-}
-- | 
relu :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                Data.Int.Int8, Data.Word.Word16,
                                Data.Word.Word32, Data.Word.Word64,
                                Data.Word.Word8, Double, Float] t) =>
        Tensor v'1 t -- ^ __features__
        -> Tensor Build t -- ^ __activations__
relu :: Tensor v'1 t -> Tensor Build t
relu = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
relu' OpParams
forall a. a -> a
id
relu' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                 Data.Int.Int8, Data.Word.Word16,
                                 Data.Word.Word32, Data.Word.Word64,
                                 Data.Word.Word8, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __features__
         -> Tensor Build t -- ^ __activations__
relu' :: OpParams -> Tensor v'1 t -> Tensor Build t
relu' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Relu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_QINT8
    }
  }
}
-}
-- | 
relu6 :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                 Data.Int.Int8, Data.Word.Word16,
                                 Data.Word.Word32, Data.Word.Word64,
                                 Data.Word.Word8, Double, Float] t) =>
         Tensor v'1 t -- ^ __features__
         -> Tensor Build t -- ^ __activations__
relu6 :: Tensor v'1 t -> Tensor Build t
relu6 = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
relu6' OpParams
forall a. a -> a
id
relu6' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                  Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t) => OpParams ->
          Tensor v'1 t -- ^ __features__
          -> Tensor Build t -- ^ __activations__
relu6' :: OpParams -> Tensor v'1 t -> Tensor Build t
relu6' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Relu6"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
relu6Grad :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) =>
             Tensor v'1 t -- ^ __gradients__
             -> Tensor v'2 t -- ^ __features__
             -> Tensor Build t -- ^ __backprops__
relu6Grad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
relu6Grad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
relu6Grad' OpParams
forall a. a -> a
id
relu6Grad' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t) => OpParams ->
              Tensor v'1 t -- ^ __gradients__
              -> Tensor v'2 t -- ^ __features__
              -> Tensor Build t -- ^ __backprops__
relu6Grad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
relu6Grad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients features :: Tensor v'2 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Relu6Grad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "features" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
reluGrad :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
            Tensor v'1 t -- ^ __gradients__
            -> Tensor v'2 t -- ^ __features__
            -> Tensor Build t -- ^ __backprops__
reluGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reluGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reluGrad' OpParams
forall a. a -> a
id
reluGrad' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
             Tensor v'1 t -- ^ __gradients__
             -> Tensor v'2 t -- ^ __features__
             -> Tensor Build t -- ^ __backprops__
reluGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
reluGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients features :: Tensor v'2 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReluGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "features" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
remoteFusedGraphExecute :: forall v'1 tinputs toutputs . (TensorTypes tinputs,
                                                          TensorTypes toutputs) =>
                           ByteString -- ^ __serialized_remote_fused_graph_execute_info__
                           -> TensorList (v'1) tinputs -- ^ __inputs__
                           -> TensorList (Build) toutputs -- ^ __outputs__
remoteFusedGraphExecute :: ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
remoteFusedGraphExecute = OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
forall (v'1 :: * -> *) (tinputs :: [*]) (toutputs :: [*]).
(TensorTypes tinputs, TensorTypes toutputs) =>
OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
remoteFusedGraphExecute' OpParams
forall a. a -> a
id
remoteFusedGraphExecute' :: forall v'1 tinputs toutputs . (TensorTypes tinputs,
                                                           TensorTypes toutputs) =>
                            OpParams ->
                            ByteString -- ^ __serialized_remote_fused_graph_execute_info__
                            -> TensorList (v'1) tinputs -- ^ __inputs__
                            -> TensorList (Build) toutputs -- ^ __outputs__
remoteFusedGraphExecute' :: OpParams
-> ByteString
-> TensorList v'1 tinputs
-> TensorList Build toutputs
remoteFusedGraphExecute' op'options :: OpParams
op'options serialized_remote_fused_graph_execute_info :: ByteString
serialized_remote_fused_graph_execute_info
                         inputs :: TensorList v'1 tinputs
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build toutputs
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build toutputs)
-> Build OpDef -> TensorList Build toutputs
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tinputs -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tinputs
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RemoteFusedGraphExecute"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tinputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tinputs
forall k (t :: k). Proxy t
Proxy :: Proxy tinputs)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutputs
forall k (t :: k). Proxy t
Proxy :: Proxy toutputs)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "serialized_remote_fused_graph_execute_info" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
serialized_remote_fused_graph_execute_info
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "Tinputs" }
output_arg { name: "outputs" type_list_attr: "Toutputs" }
attr { name: "Tinputs" type: "list(type)" has_minimum: true }
attr { name: "Toutputs" type: "list(type)" has_minimum: true }
attr {
  name: "serialized_remote_fused_graph_execute_info" type: "string"
}
-}
-- | 
repeatDataset :: [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __count__
                 -> Tensor Build Variant -- ^ __handle__
repeatDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
repeatDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
repeatDataset' OpParams
forall a. a -> a
id
repeatDataset' :: OpParams ->
                  [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __count__
                  -> Tensor Build Variant -- ^ __handle__
repeatDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
repeatDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset count :: Tensor v'2 Int64
count | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
count]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RepeatDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "count" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
requantizationRange :: forall v'1 v'2 v'3 tinput . (OneOf '[Data.Int.Int16,
                                                            Data.Int.Int32,
                                                            Data.Word.Word16,
                                                            Data.Word.Word8] tinput) =>
                       Tensor v'1 tinput -- ^ __input__
                       -> Tensor v'2 Float -- ^ __input_min__
                       -> Tensor v'3 Float -- ^ __input_max__
                       -> (Tensor Build Float, Tensor Build Float)
                       -- ^ (__output_min__, __output_max__)
                       --
                       -- * __output_min__
                       --
                       -- * __output_max__
requantizationRange :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRange = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tinput.
OneOf '[Int16, Int32, Word16, Word8] tinput =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRange' OpParams
forall a. a -> a
id
requantizationRange' :: forall v'1 v'2 v'3 tinput . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Word.Word16,
                                                             Data.Word.Word8] tinput) =>
                        OpParams ->
                        Tensor v'1 tinput -- ^ __input__
                        -> Tensor v'2 Float -- ^ __input_min__
                        -> Tensor v'3 Float -- ^ __input_max__
                        -> (Tensor Build Float, Tensor Build Float)
                        -- ^ (__output_min__, __output_max__)
                        --
                        -- * __output_min__
                        --
                        -- * __output_max__
requantizationRange' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRange' op'options :: OpParams
op'options input :: Tensor v'1 tinput
input input_min :: Tensor v'2 Float
input_min input_max :: Tensor v'3 Float
input_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Float, Tensor Build Float))
-> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
input_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RequantizationRange"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
requantizationRangePerChannel :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word8] t) =>
                                 Float -- ^ __clip_value_max__
                                 -> Tensor v'1 t -- ^ __input__
                                 -> Tensor v'2 Float -- ^ __input_min__
                                 -> Tensor v'3 Float -- ^ __input_max__
                                 -> (Tensor Build Float, Tensor Build Float)
                                 -- ^ (__output_min__, __output_max__)
                                 --
                                 -- * __output_min__
                                 --
                                 -- * __output_max__
requantizationRangePerChannel :: Float
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRangePerChannel = OpParams
-> Float
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int16, Int32, Word16, Word8] t =>
OpParams
-> Float
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRangePerChannel' OpParams
forall a. a -> a
id
requantizationRangePerChannel' :: forall v'1 v'2 v'3
                                  t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Word.Word16,
                                               Data.Word.Word8] t) =>
                                  OpParams ->
                                  Float -- ^ __clip_value_max__
                                  -> Tensor v'1 t -- ^ __input__
                                  -> Tensor v'2 Float -- ^ __input_min__
                                  -> Tensor v'3 Float -- ^ __input_max__
                                  -> (Tensor Build Float, Tensor Build Float)
                                  -- ^ (__output_min__, __output_max__)
                                  --
                                  -- * __output_min__
                                  --
                                  -- * __output_max__
requantizationRangePerChannel' :: OpParams
-> Float
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> (Tensor Build Float, Tensor Build Float)
requantizationRangePerChannel' op'options :: OpParams
op'options clip_value_max :: Float
clip_value_max input :: Tensor v'1 t
input input_min :: Tensor v'2 Float
input_min
                               input_max :: Tensor v'3 Float
input_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Float, Tensor Build Float))
-> Build OpDef -> (Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
input_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RequantizationRangePerChannel"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "clip_value_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
clip_value_max
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr { name: "clip_value_max" type: "float" }
-}
-- | 
requantize :: forall v'1 v'2 v'3 v'4 v'5 tinput
              out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                  Data.Word.Word16, Data.Word.Word8] tinput,
                          OneOf '[Data.Int.Int16, Data.Int.Int32,
                                  Data.Word.Word16,
                                  Data.Word.Word8] out_type) =>
              Tensor v'1 tinput -- ^ __input__
              -> Tensor v'2 Float -- ^ __input_min__
              -> Tensor v'3 Float -- ^ __input_max__
              -> Tensor v'4 Float -- ^ __requested_output_min__
              -> Tensor v'5 Float -- ^ __requested_output_max__
              -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
              -- ^ (__output__, __output_min__, __output_max__)
              --
              -- * __output__
              --
              -- * __output_min__
              --
              -- * __output_max__
requantize :: Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantize = OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) tinput out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantize' OpParams
forall a. a -> a
id
requantize' :: forall v'1 v'2 v'3 v'4 v'5 tinput
               out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16, Data.Word.Word8] tinput,
                           OneOf '[Data.Int.Int16, Data.Int.Int32,
                                   Data.Word.Word16,
                                   Data.Word.Word8] out_type) => OpParams ->
               Tensor v'1 tinput -- ^ __input__
               -> Tensor v'2 Float -- ^ __input_min__
               -> Tensor v'3 Float -- ^ __input_max__
               -> Tensor v'4 Float -- ^ __requested_output_min__
               -> Tensor v'5 Float -- ^ __requested_output_max__
               -> (Tensor Build out_type, Tensor Build Float,
                   Tensor Build Float)
               -- ^ (__output__, __output_min__, __output_max__)
               --
               -- * __output__
               --
               -- * __output_min__
               --
               -- * __output_max__
requantize' :: OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantize' op'options :: OpParams
op'options input :: Tensor v'1 tinput
input input_min :: Tensor v'2 Float
input_min input_max :: Tensor v'3 Float
input_max requested_output_min :: Tensor v'4 Float
requested_output_min
            requested_output_max :: Tensor v'5 Float
requested_output_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tinput -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tinput
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
input_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_max,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
requested_output_min,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
requested_output_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Requantize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinput" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tinput -> DataType
forall a. TensorType a => a -> DataType
tensorType (tinput
forall a. HasCallStack => a
undefined :: tinput)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "Tinput" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
input_arg { name: "requested_output_min" type: DT_FLOAT }
input_arg { name: "requested_output_max" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "Tinput"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
requantizePerChannel :: forall v'1 v'2 v'3 v'4 v'5 t
                        out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                            Data.Word.Word16,
                                            Data.Word.Word8] t,
                                    OneOf '[Data.Int.Int16, Data.Int.Int32,
                                            Data.Word.Word16,
                                            Data.Word.Word8] out_type) =>
                        Tensor v'1 t -- ^ __input__
                        -> Tensor v'2 Float -- ^ __input_min__
                        -> Tensor v'3 Float -- ^ __input_max__
                        -> Tensor v'4 Float -- ^ __requested_output_min__
                        -> Tensor v'5 Float -- ^ __requested_output_max__
                        -> (Tensor Build out_type, Tensor Build Float,
                            Tensor Build Float)
                        -- ^ (__output__, __output_min__, __output_max__)
                        --
                        -- * __output__
                        --
                        -- * __output_min__
                        --
                        -- * __output_max__
requantizePerChannel :: Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantizePerChannel = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) tinput out_type.
(OneOf '[Int16, Int32, Word16, Word8] tinput,
 OneOf '[Int16, Int32, Word16, Word8] out_type) =>
OpParams
-> Tensor v'1 tinput
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantizePerChannel' OpParams
forall a. a -> a
id
requantizePerChannel' :: forall v'1 v'2 v'3 v'4 v'5 t
                         out_type . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Word.Word16,
                                             Data.Word.Word8] t,
                                     OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Word.Word16,
                                             Data.Word.Word8] out_type) =>
                         OpParams ->
                         Tensor v'1 t -- ^ __input__
                         -> Tensor v'2 Float -- ^ __input_min__
                         -> Tensor v'3 Float -- ^ __input_max__
                         -> Tensor v'4 Float -- ^ __requested_output_min__
                         -> Tensor v'5 Float -- ^ __requested_output_max__
                         -> (Tensor Build out_type, Tensor Build Float,
                             Tensor Build Float)
                         -- ^ (__output__, __output_min__, __output_max__)
                         --
                         -- * __output__
                         --
                         -- * __output_min__
                         --
                         -- * __output_max__
requantizePerChannel' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor v'5 Float
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
requantizePerChannel' op'options :: OpParams
op'options input :: Tensor v'1 t
input input_min :: Tensor v'2 Float
input_min input_max :: Tensor v'3 Float
input_max requested_output_min :: Tensor v'4 Float
requested_output_min
                      requested_output_max :: Tensor v'5 Float
requested_output_max | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float))
-> Build OpDef
-> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
input_min,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
input_max,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
requested_output_min,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
requested_output_max]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RequantizePerChannel"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "input_min" type: DT_FLOAT }
input_arg { name: "input_max" type: DT_FLOAT }
input_arg { name: "requested_output_min" type: DT_FLOAT }
input_arg { name: "requested_output_max" type: DT_FLOAT }
output_arg { name: "output" type_attr: "out_type" }
output_arg { name: "output_min" type: DT_FLOAT }
output_arg { name: "output_max" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_QINT32 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_QUINT8 }
  allowed_values {
    list {
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_QINT16
      type: DT_QUINT16
    }
  }
}
-}
-- | 
reshape :: forall v'1 v'2 t tshape . (TensorType t, OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tshape) =>
           Tensor v'1 t -- ^ __tensor__
           -> Tensor v'2 tshape -- ^ __shape__
           -> Tensor Build t -- ^ __output__
reshape :: Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
reshape = OpParams -> Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
reshape' OpParams
forall a. a -> a
id
reshape' :: forall v'1 v'2 t tshape . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tshape) =>
            OpParams ->
            Tensor v'1 t -- ^ __tensor__
            -> Tensor v'2 tshape -- ^ __shape__
            -> Tensor Build t -- ^ __output__
reshape' :: OpParams -> Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
reshape' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor shape :: Tensor v'2 tshape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tshape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tshape
shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Reshape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tshape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tshape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tshape
forall a. HasCallStack => a
undefined :: tshape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "shape" type_attr: "Tshape" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tshape"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resizeArea :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word8,
                                          Double, Float] t) =>
              Tensor v'1 t -- ^ __images__
              -> Tensor v'2 Data.Int.Int32 -- ^ __size__
              -> Tensor Build Float -- ^ __resized_images__
resizeArea :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeArea = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeArea' OpParams
forall a. a -> a
id
resizeArea' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word8,
                                           Double, Float] t) => OpParams ->
               Tensor v'1 t -- ^ __images__
               -> Tensor v'2 Data.Int.Int32 -- ^ __size__
               -> Tensor Build Float -- ^ __resized_images__
resizeArea' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeArea' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeArea"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "resized_images" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_BFLOAT16
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
-}
-- | 
resizeBicubic :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word8,
                                             Double, Float] t) =>
                 Tensor v'1 t -- ^ __images__
                 -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                 -> Tensor Build Float -- ^ __resized_images__
resizeBicubic :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBicubic = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBicubic' OpParams
forall a. a -> a
id
resizeBicubic' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16, Data.Word.Word8,
                                              Double, Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __images__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                  -> Tensor Build Float -- ^ __resized_images__
resizeBicubic' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBicubic' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeBicubic"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "resized_images" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_BFLOAT16
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resizeBicubicGrad :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                     Tensor v'1 Float -- ^ __grads__
                     -> Tensor v'2 t -- ^ __original_image__
                     -> Tensor Build t -- ^ __output__
resizeBicubicGrad :: Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBicubicGrad = OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBicubicGrad' OpParams
forall a. a -> a
id
resizeBicubicGrad' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) =>
                      OpParams ->
                      Tensor v'1 Float -- ^ __grads__
                      -> Tensor v'2 t -- ^ __original_image__
                      -> Tensor Build t -- ^ __output__
resizeBicubicGrad' :: OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBicubicGrad' op'options :: OpParams
op'options grads :: Tensor v'1 Float
grads original_image :: Tensor v'2 t
original_image | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
grads,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
original_image]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeBicubicGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type: DT_FLOAT }
input_arg { name: "original_image" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resizeBilinear :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16, Data.Word.Word8,
                                              Double, Float] t) =>
                  Tensor v'1 t -- ^ __images__
                  -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                  -> Tensor Build Float -- ^ __resized_images__
resizeBilinear :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBilinear = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBilinear' OpParams
forall a. a -> a
id
resizeBilinear' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word8, Double,
                                               Float] t) => OpParams ->
                   Tensor v'1 t -- ^ __images__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                   -> Tensor Build Float -- ^ __resized_images__
resizeBilinear' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
resizeBilinear' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeBilinear"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "resized_images" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_BFLOAT16
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resizeBilinearGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                                  Float] t) =>
                      Tensor v'1 Float -- ^ __grads__
                      -> Tensor v'2 t -- ^ __original_image__
                      -> Tensor Build t -- ^ __output__
resizeBilinearGrad :: Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBilinearGrad = OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBilinearGrad' OpParams
forall a. a -> a
id
resizeBilinearGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                                   Float] t) => OpParams ->
                       Tensor v'1 Float -- ^ __grads__
                       -> Tensor v'2 t -- ^ __original_image__
                       -> Tensor Build t -- ^ __output__
resizeBilinearGrad' :: OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
resizeBilinearGrad' op'options :: OpParams
op'options grads :: Tensor v'1 Float
grads original_image :: Tensor v'2 t
original_image | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
grads,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
original_image]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeBilinearGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type: DT_FLOAT }
input_arg { name: "original_image" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT type: DT_BFLOAT16 type: DT_HALF type: DT_DOUBLE
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resizeNearestNeighbor :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                         Tensor v'1 t -- ^ __images__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                         -> Tensor Build t -- ^ __resized_images__
resizeNearestNeighbor :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighbor = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighbor' OpParams
forall a. a -> a
id
resizeNearestNeighbor' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                          Tensor v'1 t -- ^ __images__
                          -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                          -> Tensor Build t -- ^ __resized_images__
resizeNearestNeighbor' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighbor' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeNearestNeighbor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "resized_images" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_BFLOAT16
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resizeNearestNeighborGrad :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8,
                                                         Double, Float] t) =>
                             Tensor v'1 t -- ^ __grads__
                             -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                             -> Tensor Build t -- ^ __output__
resizeNearestNeighborGrad :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighborGrad = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int8, Word16, Word8, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighborGrad' OpParams
forall a. a -> a
id
resizeNearestNeighborGrad' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32,
                                                          Data.Int.Int8,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8,
                                                          Double, Float] t) =>
                              OpParams ->
                              Tensor v'1 t -- ^ __grads__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                              -> Tensor Build t -- ^ __output__
resizeNearestNeighborGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
resizeNearestNeighborGrad' op'options :: OpParams
op'options grads :: Tensor v'1 t
grads size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
grads,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ResizeNearestNeighborGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT32
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr {
  name: "align_corners" type: "bool" default_value { b: false }
}
attr {
  name: "half_pixel_centers" type: "bool" default_value { b: false }
}
-}
-- | 
resourceAccumulatorApplyGradient :: forall v'1 v'2 v'3 dtype
                                    m' . (MonadBuild m',
                                          OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] dtype) =>
                                    Tensor v'1 ResourceHandle -- ^ __handle__
                                    -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                                    -> Tensor v'3 dtype -- ^ __gradient__
                                    -> m' (ControlNode)
resourceAccumulatorApplyGradient :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
resourceAccumulatorApplyGradient = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
resourceAccumulatorApplyGradient' OpParams
forall a. a -> a
id
resourceAccumulatorApplyGradient' :: forall v'1 v'2 v'3 dtype
                                     m' . (MonadBuild m',
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float),
                                                   Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] dtype) => OpParams ->
                                     Tensor v'1 ResourceHandle -- ^ __handle__
                                     -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                                     -> Tensor v'3 dtype -- ^ __gradient__
                                     -> m' (ControlNode)
resourceAccumulatorApplyGradient' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 dtype
-> m' ControlNode
resourceAccumulatorApplyGradient' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle local_step :: Tensor v'2 Int64
local_step
                                  gradient :: Tensor v'3 dtype
gradient | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
local_step,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
gradient]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceAccumulatorApplyGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "local_step" type: DT_INT64 }
input_arg { name: "gradient" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
resourceAccumulatorNumAccumulated :: forall v'1 m' . (MonadBuild m') =>
                                     Tensor v'1 ResourceHandle -- ^ __handle__
                                     -> m' (Tensor Value Data.Int.Int32) -- ^ __num_accumulated__
resourceAccumulatorNumAccumulated :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
resourceAccumulatorNumAccumulated = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
resourceAccumulatorNumAccumulated' OpParams
forall a. a -> a
id
resourceAccumulatorNumAccumulated' :: forall v'1 m' . (MonadBuild m') =>
                                      OpParams ->
                                      Tensor v'1 ResourceHandle -- ^ __handle__
                                      -> m' (Tensor Value Data.Int.Int32) -- ^ __num_accumulated__
resourceAccumulatorNumAccumulated' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
resourceAccumulatorNumAccumulated' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceAccumulatorNumAccumulated"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "num_accumulated" type: DT_INT32 }
-}
-- | 
resourceAccumulatorSetGlobalStep :: forall v'1 v'2 m' . (MonadBuild m') =>
                                    Tensor v'1 ResourceHandle -- ^ __handle__
                                    -> Tensor v'2 Data.Int.Int64 -- ^ __new_global_step__
                                    -> m' (ControlNode)
resourceAccumulatorSetGlobalStep :: Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
resourceAccumulatorSetGlobalStep = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
resourceAccumulatorSetGlobalStep' OpParams
forall a. a -> a
id
resourceAccumulatorSetGlobalStep' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                     OpParams ->
                                     Tensor v'1 ResourceHandle -- ^ __handle__
                                     -> Tensor v'2 Data.Int.Int64 -- ^ __new_global_step__
                                     -> m' (ControlNode)
resourceAccumulatorSetGlobalStep' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
resourceAccumulatorSetGlobalStep' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle
                                  new_global_step :: Tensor v'2 Int64
new_global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
new_global_step]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceAccumulatorSetGlobalStep"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "new_global_step" type: DT_INT64 }
-}
-- | 
resourceAccumulatorTakeGradient :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype) =>
                                   Tensor v'1 ResourceHandle -- ^ __handle__
                                   -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                                   -> m' (Tensor Value dtype) -- ^ __average__
resourceAccumulatorTakeGradient :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32 -> m' (Tensor Value dtype)
resourceAccumulatorTakeGradient = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
resourceAccumulatorTakeGradient' OpParams
forall a. a -> a
id
resourceAccumulatorTakeGradient' :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype) =>
                                    OpParams ->
                                    Tensor v'1 ResourceHandle -- ^ __handle__
                                    -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                                    -> m' (Tensor Value dtype) -- ^ __average__
resourceAccumulatorTakeGradient' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> m' (Tensor Value dtype)
resourceAccumulatorTakeGradient' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle
                                 num_required :: Tensor v'2 Int32
num_required | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_required]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceAccumulatorTakeGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "num_required" type: DT_INT32 }
output_arg { name: "average" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
resourceApplyAdaMax :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                       m' . (MonadBuild m',
                             OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) =>
                       Tensor v'1 ResourceHandle -- ^ __var__
                       -> Tensor v'2 ResourceHandle -- ^ __m__
                       -> Tensor v'3 ResourceHandle -- ^ __v__
                       -> Tensor v'4 t -- ^ __beta1_power__
                       -> Tensor v'5 t -- ^ __lr__
                       -> Tensor v'6 t -- ^ __beta1__
                       -> Tensor v'7 t -- ^ __beta2__
                       -> Tensor v'8 t -- ^ __epsilon__
                       -> Tensor v'9 t -- ^ __grad__
                       -> m' (ControlNode)
resourceApplyAdaMax :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyAdaMax = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyAdaMax' OpParams
forall a. a -> a
id
resourceApplyAdaMax' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) => OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __var__
                        -> Tensor v'2 ResourceHandle -- ^ __m__
                        -> Tensor v'3 ResourceHandle -- ^ __v__
                        -> Tensor v'4 t -- ^ __beta1_power__
                        -> Tensor v'5 t -- ^ __lr__
                        -> Tensor v'6 t -- ^ __beta1__
                        -> Tensor v'7 t -- ^ __beta2__
                        -> Tensor v'8 t -- ^ __epsilon__
                        -> Tensor v'9 t -- ^ __grad__
                        -> m' (ControlNode)
resourceApplyAdaMax' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyAdaMax' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var m :: Tensor v'2 ResourceHandle
m v :: Tensor v'3 ResourceHandle
v beta1_power :: Tensor v'4 t
beta1_power lr :: Tensor v'5 t
lr beta1 :: Tensor v'6 t
beta1 beta2 :: Tensor v'7 t
beta2 epsilon :: Tensor v'8 t
epsilon
                     grad :: Tensor v'9 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
m,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta1_power,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
beta2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdaMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "m" type: DT_RESOURCE }
input_arg { name: "v" type: DT_RESOURCE }
input_arg { name: "beta1_power" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "beta1" type_attr: "T" }
input_arg { name: "beta2" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyAdadelta :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                         m' . (MonadBuild m',
                               OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) =>
                         Tensor v'1 ResourceHandle -- ^ __var__
                         -> Tensor v'2 ResourceHandle -- ^ __accum__
                         -> Tensor v'3 ResourceHandle -- ^ __accum_update__
                         -> Tensor v'4 t -- ^ __lr__
                         -> Tensor v'5 t -- ^ __rho__
                         -> Tensor v'6 t -- ^ __epsilon__
                         -> Tensor v'7 t -- ^ __grad__
                         -> m' (ControlNode)
resourceApplyAdadelta :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAdadelta = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAdadelta' OpParams
forall a. a -> a
id
resourceApplyAdadelta' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                          m' . (MonadBuild m',
                                OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
                          Tensor v'1 ResourceHandle -- ^ __var__
                          -> Tensor v'2 ResourceHandle -- ^ __accum__
                          -> Tensor v'3 ResourceHandle -- ^ __accum_update__
                          -> Tensor v'4 t -- ^ __lr__
                          -> Tensor v'5 t -- ^ __rho__
                          -> Tensor v'6 t -- ^ __epsilon__
                          -> Tensor v'7 t -- ^ __grad__
                          -> m' (ControlNode)
resourceApplyAdadelta' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAdadelta' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum accum_update :: Tensor v'3 ResourceHandle
accum_update lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho epsilon :: Tensor v'6 t
epsilon
                       grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
accum_update,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
epsilon,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdadelta"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "accum_update" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyAdagrad :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                       OneOf '[(Data.Complex.Complex Double),
                                                               (Data.Complex.Complex Float),
                                                               Data.Int.Int16,
                                                               Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Data.Int.Int8,
                                                               Data.Word.Word16,
                                                               Data.Word.Word32,
                                                               Data.Word.Word64,
                                                               Data.Word.Word8,
                                                               Double,
                                                               Float] t) =>
                        Tensor v'1 ResourceHandle -- ^ __var__
                        -> Tensor v'2 ResourceHandle -- ^ __accum__
                        -> Tensor v'3 t -- ^ __lr__
                        -> Tensor v'4 t -- ^ __grad__
                        -> m' (ControlNode)
resourceApplyAdagrad :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> m' ControlNode
resourceApplyAdagrad = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> m' ControlNode
resourceApplyAdagrad' OpParams
forall a. a -> a
id
resourceApplyAdagrad' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                        OneOf '[(Data.Complex.Complex Double),
                                                                (Data.Complex.Complex Float),
                                                                Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Int.Int8,
                                                                Data.Word.Word16,
                                                                Data.Word.Word32,
                                                                Data.Word.Word64,
                                                                Data.Word.Word8,
                                                                Double,
                                                                Float] t) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __var__
                         -> Tensor v'2 ResourceHandle -- ^ __accum__
                         -> Tensor v'3 t -- ^ __lr__
                         -> Tensor v'4 t -- ^ __grad__
                         -> m' (ControlNode)
resourceApplyAdagrad' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> m' ControlNode
resourceApplyAdagrad' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
resourceApplyAdagradDA :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                          m' . (MonadBuild m',
                                OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
                          Tensor v'1 ResourceHandle -- ^ __var__
                          -> Tensor v'2 ResourceHandle -- ^ __gradient_accumulator__
                          -> Tensor v'3 ResourceHandle -- ^ __gradient_squared_accumulator__
                          -> Tensor v'4 t -- ^ __grad__
                          -> Tensor v'5 t -- ^ __lr__
                          -> Tensor v'6 t -- ^ __l1__
                          -> Tensor v'7 t -- ^ __l2__
                          -> Tensor v'8 Data.Int.Int64 -- ^ __global_step__
                          -> m' (ControlNode)
resourceApplyAdagradDA :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' ControlNode
resourceApplyAdagradDA = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' ControlNode
resourceApplyAdagradDA' OpParams
forall a. a -> a
id
resourceApplyAdagradDA' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                           m' . (MonadBuild m',
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __var__
                           -> Tensor v'2 ResourceHandle -- ^ __gradient_accumulator__
                           -> Tensor v'3 ResourceHandle -- ^ __gradient_squared_accumulator__
                           -> Tensor v'4 t -- ^ __grad__
                           -> Tensor v'5 t -- ^ __lr__
                           -> Tensor v'6 t -- ^ __l1__
                           -> Tensor v'7 t -- ^ __l2__
                           -> Tensor v'8 Data.Int.Int64 -- ^ __global_step__
                           -> m' (ControlNode)
resourceApplyAdagradDA' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 Int64
-> m' ControlNode
resourceApplyAdagradDA' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var gradient_accumulator :: Tensor v'2 ResourceHandle
gradient_accumulator
                        gradient_squared_accumulator :: Tensor v'3 ResourceHandle
gradient_squared_accumulator grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2
                        global_step :: Tensor v'8 Int64
global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
gradient_accumulator,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
gradient_squared_accumulator,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 Int64
global_step]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdagradDA"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "gradient_accumulator" type: DT_RESOURCE }
input_arg {
  name: "gradient_squared_accumulator" type: DT_RESOURCE
}
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "global_step" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyAdagradV2 :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                             OneOf '[(Data.Complex.Complex Double),
                                                                     (Data.Complex.Complex Float),
                                                                     Data.Int.Int16,
                                                                     Data.Int.Int32,
                                                                     Data.Int.Int64,
                                                                     Data.Int.Int8,
                                                                     Data.Word.Word16,
                                                                     Data.Word.Word32,
                                                                     Data.Word.Word64,
                                                                     Data.Word.Word8,
                                                                     Double,
                                                                     Float] t) =>
                          Tensor v'1 ResourceHandle -- ^ __var__
                          -> Tensor v'2 ResourceHandle -- ^ __accum__
                          -> Tensor v'3 t -- ^ __lr__
                          -> Tensor v'4 t -- ^ __epsilon__
                          -> Tensor v'5 t -- ^ __grad__
                          -> m' (ControlNode)
resourceApplyAdagradV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyAdagradV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyAdagradV2' OpParams
forall a. a -> a
id
resourceApplyAdagradV2' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] t) =>
                           OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __var__
                           -> Tensor v'2 ResourceHandle -- ^ __accum__
                           -> Tensor v'3 t -- ^ __lr__
                           -> Tensor v'4 t -- ^ __epsilon__
                           -> Tensor v'5 t -- ^ __grad__
                           -> m' (ControlNode)
resourceApplyAdagradV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyAdagradV2' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr epsilon :: Tensor v'4 t
epsilon
                        grad :: Tensor v'5 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
epsilon,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdagradV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
resourceApplyAdam :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t
                     m' . (MonadBuild m', OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t) =>
                     Tensor v'1 ResourceHandle -- ^ __var__
                     -> Tensor v'2 ResourceHandle -- ^ __m__
                     -> Tensor v'3 ResourceHandle -- ^ __v__
                     -> Tensor v'4 t -- ^ __beta1_power__
                     -> Tensor v'5 t -- ^ __beta2_power__
                     -> Tensor v'6 t -- ^ __lr__
                     -> Tensor v'7 t -- ^ __beta1__
                     -> Tensor v'8 t -- ^ __beta2__
                     -> Tensor v'9 t -- ^ __epsilon__
                     -> Tensor v'10 t -- ^ __grad__
                     -> m' (ControlNode)
resourceApplyAdam :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceApplyAdam = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceApplyAdam' OpParams
forall a. a -> a
id
resourceApplyAdam' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t
                      m' . (MonadBuild m',
                            OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __var__
                      -> Tensor v'2 ResourceHandle -- ^ __m__
                      -> Tensor v'3 ResourceHandle -- ^ __v__
                      -> Tensor v'4 t -- ^ __beta1_power__
                      -> Tensor v'5 t -- ^ __beta2_power__
                      -> Tensor v'6 t -- ^ __lr__
                      -> Tensor v'7 t -- ^ __beta1__
                      -> Tensor v'8 t -- ^ __beta2__
                      -> Tensor v'9 t -- ^ __epsilon__
                      -> Tensor v'10 t -- ^ __grad__
                      -> m' (ControlNode)
resourceApplyAdam' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceApplyAdam' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var m :: Tensor v'2 ResourceHandle
m v :: Tensor v'3 ResourceHandle
v beta1_power :: Tensor v'4 t
beta1_power beta2_power :: Tensor v'5 t
beta2_power lr :: Tensor v'6 t
lr beta1 :: Tensor v'7 t
beta1 beta2 :: Tensor v'8 t
beta2
                   epsilon :: Tensor v'9 t
epsilon grad :: Tensor v'10 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
m,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
v,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta1_power,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
beta2_power,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
beta1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
beta2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
epsilon,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdam"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "m" type: DT_RESOURCE }
input_arg { name: "v" type: DT_RESOURCE }
input_arg { name: "beta1_power" type_attr: "T" }
input_arg { name: "beta2_power" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "beta1" type_attr: "T" }
input_arg { name: "beta2" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyAdamWithAmsgrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                                v'11 t m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __m__
                                -> Tensor v'3 ResourceHandle -- ^ __v__
                                -> Tensor v'4 ResourceHandle -- ^ __vhat__
                                -> Tensor v'5 t -- ^ __beta1_power__
                                -> Tensor v'6 t -- ^ __beta2_power__
                                -> Tensor v'7 t -- ^ __lr__
                                -> Tensor v'8 t -- ^ __beta1__
                                -> Tensor v'9 t -- ^ __beta2__
                                -> Tensor v'10 t -- ^ __epsilon__
                                -> Tensor v'11 t -- ^ __grad__
                                -> m' (ControlNode)
resourceApplyAdamWithAmsgrad :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m' ControlNode
resourceApplyAdamWithAmsgrad = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) (v'11 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m' ControlNode
resourceApplyAdamWithAmsgrad' OpParams
forall a. a -> a
id
resourceApplyAdamWithAmsgrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10
                                 v'11 t m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) => OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 ResourceHandle -- ^ __m__
                                 -> Tensor v'3 ResourceHandle -- ^ __v__
                                 -> Tensor v'4 ResourceHandle -- ^ __vhat__
                                 -> Tensor v'5 t -- ^ __beta1_power__
                                 -> Tensor v'6 t -- ^ __beta2_power__
                                 -> Tensor v'7 t -- ^ __lr__
                                 -> Tensor v'8 t -- ^ __beta1__
                                 -> Tensor v'9 t -- ^ __beta2__
                                 -> Tensor v'10 t -- ^ __epsilon__
                                 -> Tensor v'11 t -- ^ __grad__
                                 -> m' (ControlNode)
resourceApplyAdamWithAmsgrad' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> Tensor v'11 t
-> m' ControlNode
resourceApplyAdamWithAmsgrad' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var m :: Tensor v'2 ResourceHandle
m v :: Tensor v'3 ResourceHandle
v vhat :: Tensor v'4 ResourceHandle
vhat beta1_power :: Tensor v'5 t
beta1_power beta2_power :: Tensor v'6 t
beta2_power lr :: Tensor v'7 t
lr
                              beta1 :: Tensor v'8 t
beta1 beta2 :: Tensor v'9 t
beta2 epsilon :: Tensor v'10 t
epsilon grad :: Tensor v'11 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
m,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
v,
                                                             Tensor v'4 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ResourceHandle
vhat,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
beta1_power,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta2_power,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
lr,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
beta1,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
beta2,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
epsilon,
                                                             Tensor v'11 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'11 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAdamWithAmsgrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "m" type: DT_RESOURCE }
input_arg { name: "v" type: DT_RESOURCE }
input_arg { name: "vhat" type: DT_RESOURCE }
input_arg { name: "beta1_power" type_attr: "T" }
input_arg { name: "beta2_power" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "beta1" type_attr: "T" }
input_arg { name: "beta2" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyAddSign :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) =>
                        Tensor v'1 ResourceHandle -- ^ __var__
                        -> Tensor v'2 ResourceHandle -- ^ __m__
                        -> Tensor v'3 t -- ^ __lr__
                        -> Tensor v'4 t -- ^ __alpha__
                        -> Tensor v'5 t -- ^ __sign_decay__
                        -> Tensor v'6 t -- ^ __beta__
                        -> Tensor v'7 t -- ^ __grad__
                        -> m' (ControlNode)
resourceApplyAddSign :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAddSign = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAddSign' OpParams
forall a. a -> a
id
resourceApplyAddSign' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                         m' . (MonadBuild m',
                               OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __var__
                         -> Tensor v'2 ResourceHandle -- ^ __m__
                         -> Tensor v'3 t -- ^ __lr__
                         -> Tensor v'4 t -- ^ __alpha__
                         -> Tensor v'5 t -- ^ __sign_decay__
                         -> Tensor v'6 t -- ^ __beta__
                         -> Tensor v'7 t -- ^ __grad__
                         -> m' (ControlNode)
resourceApplyAddSign' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyAddSign' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var m :: Tensor v'2 ResourceHandle
m lr :: Tensor v'3 t
lr alpha :: Tensor v'4 t
alpha sign_decay :: Tensor v'5 t
sign_decay beta :: Tensor v'6 t
beta
                      grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
alpha,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
sign_decay,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyAddSign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "m" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "sign_decay" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyCenteredRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                                m' . (MonadBuild m',
                                      OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __mg__
                                -> Tensor v'3 ResourceHandle -- ^ __ms__
                                -> Tensor v'4 ResourceHandle -- ^ __mom__
                                -> Tensor v'5 t -- ^ __lr__
                                -> Tensor v'6 t -- ^ __rho__
                                -> Tensor v'7 t -- ^ __momentum__
                                -> Tensor v'8 t -- ^ __epsilon__
                                -> Tensor v'9 t -- ^ __grad__
                                -> m' (ControlNode)
resourceApplyCenteredRMSProp :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyCenteredRMSProp = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyCenteredRMSProp' OpParams
forall a. a -> a
id
resourceApplyCenteredRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                                 m' . (MonadBuild m',
                                       OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) => OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 ResourceHandle -- ^ __mg__
                                 -> Tensor v'3 ResourceHandle -- ^ __ms__
                                 -> Tensor v'4 ResourceHandle -- ^ __mom__
                                 -> Tensor v'5 t -- ^ __lr__
                                 -> Tensor v'6 t -- ^ __rho__
                                 -> Tensor v'7 t -- ^ __momentum__
                                 -> Tensor v'8 t -- ^ __epsilon__
                                 -> Tensor v'9 t -- ^ __grad__
                                 -> m' (ControlNode)
resourceApplyCenteredRMSProp' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyCenteredRMSProp' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var mg :: Tensor v'2 ResourceHandle
mg ms :: Tensor v'3 ResourceHandle
ms mom :: Tensor v'4 ResourceHandle
mom lr :: Tensor v'5 t
lr rho :: Tensor v'6 t
rho momentum :: Tensor v'7 t
momentum epsilon :: Tensor v'8 t
epsilon
                              grad :: Tensor v'9 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
mg,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
ms,
                                                             Tensor v'4 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ResourceHandle
mom,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
rho,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
momentum,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyCenteredRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "mg" type: DT_RESOURCE }
input_arg { name: "ms" type: DT_RESOURCE }
input_arg { name: "mom" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyFtrl :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                     m' . (MonadBuild m', OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t) =>
                     Tensor v'1 ResourceHandle -- ^ __var__
                     -> Tensor v'2 ResourceHandle -- ^ __accum__
                     -> Tensor v'3 ResourceHandle -- ^ __linear__
                     -> Tensor v'4 t -- ^ __grad__
                     -> Tensor v'5 t -- ^ __lr__
                     -> Tensor v'6 t -- ^ __l1__
                     -> Tensor v'7 t -- ^ __l2__
                     -> Tensor v'8 t -- ^ __lr_power__
                     -> m' (ControlNode)
resourceApplyFtrl :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyFtrl = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyFtrl' OpParams
forall a. a -> a
id
resourceApplyFtrl' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                      m' . (MonadBuild m',
                            OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __var__
                      -> Tensor v'2 ResourceHandle -- ^ __accum__
                      -> Tensor v'3 ResourceHandle -- ^ __linear__
                      -> Tensor v'4 t -- ^ __grad__
                      -> Tensor v'5 t -- ^ __lr__
                      -> Tensor v'6 t -- ^ __l1__
                      -> Tensor v'7 t -- ^ __l2__
                      -> Tensor v'8 t -- ^ __lr_power__
                      -> m' (ControlNode)
resourceApplyFtrl' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyFtrl' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum linear :: Tensor v'3 ResourceHandle
linear grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2
                   lr_power :: Tensor v'8 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
lr_power]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyFtrl"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "linear" type: DT_RESOURCE }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
resourceApplyFtrlV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                       m' . (MonadBuild m',
                             OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) =>
                       Tensor v'1 ResourceHandle -- ^ __var__
                       -> Tensor v'2 ResourceHandle -- ^ __accum__
                       -> Tensor v'3 ResourceHandle -- ^ __linear__
                       -> Tensor v'4 t -- ^ __grad__
                       -> Tensor v'5 t -- ^ __lr__
                       -> Tensor v'6 t -- ^ __l1__
                       -> Tensor v'7 t -- ^ __l2__
                       -> Tensor v'8 t -- ^ __l2_shrinkage__
                       -> Tensor v'9 t -- ^ __lr_power__
                       -> m' (ControlNode)
resourceApplyFtrlV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyFtrlV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyFtrlV2' OpParams
forall a. a -> a
id
resourceApplyFtrlV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) => OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __var__
                        -> Tensor v'2 ResourceHandle -- ^ __accum__
                        -> Tensor v'3 ResourceHandle -- ^ __linear__
                        -> Tensor v'4 t -- ^ __grad__
                        -> Tensor v'5 t -- ^ __lr__
                        -> Tensor v'6 t -- ^ __l1__
                        -> Tensor v'7 t -- ^ __l2__
                        -> Tensor v'8 t -- ^ __l2_shrinkage__
                        -> Tensor v'9 t -- ^ __lr_power__
                        -> m' (ControlNode)
resourceApplyFtrlV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceApplyFtrlV2' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum linear :: Tensor v'3 ResourceHandle
linear grad :: Tensor v'4 t
grad lr :: Tensor v'5 t
lr l1 :: Tensor v'6 t
l1 l2 :: Tensor v'7 t
l2 l2_shrinkage :: Tensor v'8 t
l2_shrinkage
                     lr_power :: Tensor v'9 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
l1,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l2,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2_shrinkage,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
lr_power]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyFtrlV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "linear" type: DT_RESOURCE }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "l2_shrinkage" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
resourceApplyGradientDescent :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                           OneOf '[(Data.Complex.Complex Double),
                                                                   (Data.Complex.Complex Float),
                                                                   Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Int.Int64,
                                                                   Data.Int.Int8,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word32,
                                                                   Data.Word.Word64,
                                                                   Data.Word.Word8,
                                                                   Double,
                                                                   Float] t) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 t -- ^ __alpha__
                                -> Tensor v'3 t -- ^ __delta__
                                -> m' (ControlNode)
resourceApplyGradientDescent :: Tensor v'1 ResourceHandle
-> Tensor v'2 t -> Tensor v'3 t -> m' ControlNode
resourceApplyGradientDescent = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> m' ControlNode
resourceApplyGradientDescent' OpParams
forall a. a -> a
id
resourceApplyGradientDescent' :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                            OneOf '[(Data.Complex.Complex Double),
                                                                    (Data.Complex.Complex Float),
                                                                    Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Int.Int64,
                                                                    Data.Int.Int8,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word32,
                                                                    Data.Word.Word64,
                                                                    Data.Word.Word8,
                                                                    Double,
                                                                    Float] t) =>
                                 OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 t -- ^ __alpha__
                                 -> Tensor v'3 t -- ^ __delta__
                                 -> m' (ControlNode)
resourceApplyGradientDescent' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> m' ControlNode
resourceApplyGradientDescent' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var alpha :: Tensor v'2 t
alpha delta :: Tensor v'3 t
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
delta]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "delta" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyKerasMomentum :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                                 OneOf '[(Data.Complex.Complex Double),
                                                                         (Data.Complex.Complex Float),
                                                                         Data.Int.Int16,
                                                                         Data.Int.Int32,
                                                                         Data.Int.Int64,
                                                                         Data.Int.Int8,
                                                                         Data.Word.Word16,
                                                                         Data.Word.Word32,
                                                                         Data.Word.Word64,
                                                                         Data.Word.Word8,
                                                                         Double,
                                                                         Float] t) =>
                              Tensor v'1 ResourceHandle -- ^ __var__
                              -> Tensor v'2 ResourceHandle -- ^ __accum__
                              -> Tensor v'3 t -- ^ __lr__
                              -> Tensor v'4 t -- ^ __grad__
                              -> Tensor v'5 t -- ^ __momentum__
                              -> m' (ControlNode)
resourceApplyKerasMomentum :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyKerasMomentum = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyKerasMomentum' OpParams
forall a. a -> a
id
resourceApplyKerasMomentum' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                                  OneOf '[(Data.Complex.Complex Double),
                                                                          (Data.Complex.Complex Float),
                                                                          Data.Int.Int16,
                                                                          Data.Int.Int32,
                                                                          Data.Int.Int64,
                                                                          Data.Int.Int8,
                                                                          Data.Word.Word16,
                                                                          Data.Word.Word32,
                                                                          Data.Word.Word64,
                                                                          Data.Word.Word8,
                                                                          Double,
                                                                          Float] t) =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __var__
                               -> Tensor v'2 ResourceHandle -- ^ __accum__
                               -> Tensor v'3 t -- ^ __lr__
                               -> Tensor v'4 t -- ^ __grad__
                               -> Tensor v'5 t -- ^ __momentum__
                               -> m' (ControlNode)
resourceApplyKerasMomentum' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyKerasMomentum' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad
                            momentum :: Tensor v'5 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
momentum]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyKerasMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyMomentum :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                            OneOf '[(Data.Complex.Complex Double),
                                                                    (Data.Complex.Complex Float),
                                                                    Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Int.Int64,
                                                                    Data.Int.Int8,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word32,
                                                                    Data.Word.Word64,
                                                                    Data.Word.Word8,
                                                                    Double,
                                                                    Float] t) =>
                         Tensor v'1 ResourceHandle -- ^ __var__
                         -> Tensor v'2 ResourceHandle -- ^ __accum__
                         -> Tensor v'3 t -- ^ __lr__
                         -> Tensor v'4 t -- ^ __grad__
                         -> Tensor v'5 t -- ^ __momentum__
                         -> m' (ControlNode)
resourceApplyMomentum :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyMomentum = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyMomentum' OpParams
forall a. a -> a
id
resourceApplyMomentum' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                             OneOf '[(Data.Complex.Complex Double),
                                                                     (Data.Complex.Complex Float),
                                                                     Data.Int.Int16,
                                                                     Data.Int.Int32,
                                                                     Data.Int.Int64,
                                                                     Data.Int.Int8,
                                                                     Data.Word.Word16,
                                                                     Data.Word.Word32,
                                                                     Data.Word.Word64,
                                                                     Data.Word.Word8,
                                                                     Double,
                                                                     Float] t) =>
                          OpParams ->
                          Tensor v'1 ResourceHandle -- ^ __var__
                          -> Tensor v'2 ResourceHandle -- ^ __accum__
                          -> Tensor v'3 t -- ^ __lr__
                          -> Tensor v'4 t -- ^ __grad__
                          -> Tensor v'5 t -- ^ __momentum__
                          -> m' (ControlNode)
resourceApplyMomentum' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyMomentum' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad
                       momentum :: Tensor v'5 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
momentum]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyPowerSign :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                          m' . (MonadBuild m',
                                OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t) =>
                          Tensor v'1 ResourceHandle -- ^ __var__
                          -> Tensor v'2 ResourceHandle -- ^ __m__
                          -> Tensor v'3 t -- ^ __lr__
                          -> Tensor v'4 t -- ^ __logbase__
                          -> Tensor v'5 t -- ^ __sign_decay__
                          -> Tensor v'6 t -- ^ __beta__
                          -> Tensor v'7 t -- ^ __grad__
                          -> m' (ControlNode)
resourceApplyPowerSign :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyPowerSign = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyPowerSign' OpParams
forall a. a -> a
id
resourceApplyPowerSign' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                           m' . (MonadBuild m',
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t) => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __var__
                           -> Tensor v'2 ResourceHandle -- ^ __m__
                           -> Tensor v'3 t -- ^ __lr__
                           -> Tensor v'4 t -- ^ __logbase__
                           -> Tensor v'5 t -- ^ __sign_decay__
                           -> Tensor v'6 t -- ^ __beta__
                           -> Tensor v'7 t -- ^ __grad__
                           -> m' (ControlNode)
resourceApplyPowerSign' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> m' ControlNode
resourceApplyPowerSign' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var m :: Tensor v'2 ResourceHandle
m lr :: Tensor v'3 t
lr logbase :: Tensor v'4 t
logbase sign_decay :: Tensor v'5 t
sign_decay beta :: Tensor v'6 t
beta
                        grad :: Tensor v'7 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
m,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
logbase,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
sign_decay,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
beta,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyPowerSign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "m" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "logbase" type_attr: "T" }
input_arg { name: "sign_decay" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyProximalAdagrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                                m' . (MonadBuild m',
                                      OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __accum__
                                -> Tensor v'3 t -- ^ __lr__
                                -> Tensor v'4 t -- ^ __l1__
                                -> Tensor v'5 t -- ^ __l2__
                                -> Tensor v'6 t -- ^ __grad__
                                -> m' (ControlNode)
resourceApplyProximalAdagrad :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' ControlNode
resourceApplyProximalAdagrad = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' ControlNode
resourceApplyProximalAdagrad' OpParams
forall a. a -> a
id
resourceApplyProximalAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                                 m' . (MonadBuild m',
                                       OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t) => OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 ResourceHandle -- ^ __accum__
                                 -> Tensor v'3 t -- ^ __lr__
                                 -> Tensor v'4 t -- ^ __l1__
                                 -> Tensor v'5 t -- ^ __l2__
                                 -> Tensor v'6 t -- ^ __grad__
                                 -> m' (ControlNode)
resourceApplyProximalAdagrad' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> m' ControlNode
resourceApplyProximalAdagrad' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr l1 :: Tensor v'4 t
l1 l2 :: Tensor v'5 t
l2
                              grad :: Tensor v'6 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l1,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
l2,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyProximalAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyProximalGradientDescent :: forall v'1 v'2 v'3 v'4 v'5 t
                                        m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t) =>
                                        Tensor v'1 ResourceHandle -- ^ __var__
                                        -> Tensor v'2 t -- ^ __alpha__
                                        -> Tensor v'3 t -- ^ __l1__
                                        -> Tensor v'4 t -- ^ __l2__
                                        -> Tensor v'5 t -- ^ __delta__
                                        -> m' (ControlNode)
resourceApplyProximalGradientDescent :: Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyProximalGradientDescent = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyProximalGradientDescent' OpParams
forall a. a -> a
id
resourceApplyProximalGradientDescent' :: forall v'1 v'2 v'3 v'4 v'5 t
                                         m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t) => OpParams ->
                                         Tensor v'1 ResourceHandle -- ^ __var__
                                         -> Tensor v'2 t -- ^ __alpha__
                                         -> Tensor v'3 t -- ^ __l1__
                                         -> Tensor v'4 t -- ^ __l2__
                                         -> Tensor v'5 t -- ^ __delta__
                                         -> m' (ControlNode)
resourceApplyProximalGradientDescent' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> m' ControlNode
resourceApplyProximalGradientDescent' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var alpha :: Tensor v'2 t
alpha l1 :: Tensor v'3 t
l1 l2 :: Tensor v'4 t
l2
                                      delta :: Tensor v'5 t
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
l1,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l2,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
delta]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyProximalGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "delta" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceApplyRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) =>
                        Tensor v'1 ResourceHandle -- ^ __var__
                        -> Tensor v'2 ResourceHandle -- ^ __ms__
                        -> Tensor v'3 ResourceHandle -- ^ __mom__
                        -> Tensor v'4 t -- ^ __lr__
                        -> Tensor v'5 t -- ^ __rho__
                        -> Tensor v'6 t -- ^ __momentum__
                        -> Tensor v'7 t -- ^ __epsilon__
                        -> Tensor v'8 t -- ^ __grad__
                        -> m' (ControlNode)
resourceApplyRMSProp :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyRMSProp = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyRMSProp' OpParams
forall a. a -> a
id
resourceApplyRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                         m' . (MonadBuild m',
                               OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __var__
                         -> Tensor v'2 ResourceHandle -- ^ __ms__
                         -> Tensor v'3 ResourceHandle -- ^ __mom__
                         -> Tensor v'4 t -- ^ __lr__
                         -> Tensor v'5 t -- ^ __rho__
                         -> Tensor v'6 t -- ^ __momentum__
                         -> Tensor v'7 t -- ^ __epsilon__
                         -> Tensor v'8 t -- ^ __grad__
                         -> m' (ControlNode)
resourceApplyRMSProp' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> m' ControlNode
resourceApplyRMSProp' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var ms :: Tensor v'2 ResourceHandle
ms mom :: Tensor v'3 ResourceHandle
mom lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho momentum :: Tensor v'6 t
momentum epsilon :: Tensor v'7 t
epsilon
                      grad :: Tensor v'8 t
grad | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
ms,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
mom,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
epsilon,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
grad]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceApplyRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "ms" type: DT_RESOURCE }
input_arg { name: "mom" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceConditionalAccumulator :: forall m' . (MonadBuild m') =>
                                  DataType -- ^ __dtype__
                                  -> Shape -- ^ __shape__
                                  -> m' (Tensor Value ResourceHandle) -- ^ __handle__
resourceConditionalAccumulator :: DataType -> Shape -> m' (Tensor Value ResourceHandle)
resourceConditionalAccumulator = OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
resourceConditionalAccumulator' OpParams
forall a. a -> a
id
resourceConditionalAccumulator' :: forall m' . (MonadBuild m') => OpParams ->
                                   DataType -- ^ __dtype__
                                   -> Shape -- ^ __shape__
                                   -> m' (Tensor Value ResourceHandle) -- ^ __handle__
resourceConditionalAccumulator' :: OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
resourceConditionalAccumulator' op'options :: OpParams
op'options dtype :: DataType
dtype shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceConditionalAccumulator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "shape" type: "shape" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "reduction_type"
  type: "string"
  default_value { s: "MEAN" }
  allowed_values { list { s: "MEAN" s: "SUM" } }
}
-}
-- | 
resourceCountUpTo :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] t) =>
                     Data.Int.Int64 -- ^ __limit__
                     -> Tensor v'1 ResourceHandle -- ^ __resource__
                     -> m' (Tensor Value t) -- ^ __output__
resourceCountUpTo :: Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
resourceCountUpTo = OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] t) =>
OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
resourceCountUpTo' OpParams
forall a. a -> a
id
resourceCountUpTo' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] t) =>
                      OpParams ->
                      Data.Int.Int64 -- ^ __limit__
                      -> Tensor v'1 ResourceHandle -- ^ __resource__
                      -> m' (Tensor Value t) -- ^ __output__
resourceCountUpTo' :: OpParams
-> Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
resourceCountUpTo' op'options :: OpParams
op'options limit :: Int64
limit resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceCountUpTo"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "limit" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
limit
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "output" type_attr: "T" }
attr { name: "limit" type: "int" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceGather :: forall v'1 v'2 dtype tindices m' . (MonadBuild m',
                                                      TensorType dtype,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                  Tensor v'1 ResourceHandle -- ^ __resource__
                  -> Tensor v'2 tindices -- ^ __indices__
                  -> m' (Tensor Value dtype) -- ^ __output__
resourceGather :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> m' (Tensor Value dtype)
resourceGather = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype tindices
       (m' :: * -> *).
(MonadBuild m', TensorType dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
resourceGather' OpParams
forall a. a -> a
id
resourceGather' :: forall v'1 v'2 dtype tindices m' . (MonadBuild m',
                                                       TensorType dtype,
                                                       OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tindices) =>
                   OpParams ->
                   Tensor v'1 ResourceHandle -- ^ __resource__
                   -> Tensor v'2 tindices -- ^ __indices__
                   -> m' (Tensor Value dtype) -- ^ __output__
resourceGather' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
resourceGather' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceGather"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "batch_dims" type: "int" default_value { i: 0 } }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr { name: "dtype" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceGatherNd :: forall v'1 v'2 dtype tindices m' . (MonadBuild m',
                                                        TensorType dtype,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tindices) =>
                    Tensor v'1 ResourceHandle -- ^ __resource__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> m' (Tensor Value dtype) -- ^ __output__
resourceGatherNd :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> m' (Tensor Value dtype)
resourceGatherNd = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype tindices
       (m' :: * -> *).
(MonadBuild m', TensorType dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
resourceGatherNd' OpParams
forall a. a -> a
id
resourceGatherNd' :: forall v'1 v'2 dtype tindices m' . (MonadBuild m',
                                                         TensorType dtype,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices) =>
                     OpParams ->
                     Tensor v'1 ResourceHandle -- ^ __resource__
                     -> Tensor v'2 tindices -- ^ __indices__
                     -> m' (Tensor Value dtype) -- ^ __output__
resourceGatherNd' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> m' (Tensor Value dtype)
resourceGatherNd' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceGatherNd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterAdd :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterAdd :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterAdd = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterAdd' OpParams
forall a. a -> a
id
resourceScatterAdd' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterAdd' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterAdd' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterAdd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterDiv :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterDiv :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterDiv = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterDiv' OpParams
forall a. a -> a
id
resourceScatterDiv' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterDiv' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterDiv' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterDiv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterMax :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterMax :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterMax = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMax' OpParams
forall a. a -> a
id
resourceScatterMax' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterMax' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMax' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterMin :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterMin :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterMin = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMin' OpParams
forall a. a -> a
id
resourceScatterMin' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterMin' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMin' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterMin"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterMul :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterMul :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterMul = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMul' OpParams
forall a. a -> a
id
resourceScatterMul' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterMul' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterMul' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterMul"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterNdAdd :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                            TensorType t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tindices) =>
                        Tensor v'1 ResourceHandle -- ^ __ref__
                        -> Tensor v'2 tindices -- ^ __indices__
                        -> Tensor v'3 t -- ^ __updates__
                        -> m' (ControlNode)
resourceScatterNdAdd :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
resourceScatterNdAdd = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdAdd' OpParams
forall a. a -> a
id
resourceScatterNdAdd' :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                             TensorType t,
                                                             OneOf '[Data.Int.Int32,
                                                                     Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __ref__
                         -> Tensor v'2 tindices -- ^ __indices__
                         -> Tensor v'3 t -- ^ __updates__
                         -> m' (ControlNode)
resourceScatterNdAdd' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdAdd' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterNdAdd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
resourceScatterNdMax :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                            TensorType t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tindices) =>
                        Tensor v'1 ResourceHandle -- ^ __ref__
                        -> Tensor v'2 tindices -- ^ __indices__
                        -> Tensor v'3 t -- ^ __updates__
                        -> m' (ControlNode)
resourceScatterNdMax :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
resourceScatterNdMax = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdMax' OpParams
forall a. a -> a
id
resourceScatterNdMax' :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                             TensorType t,
                                                             OneOf '[Data.Int.Int32,
                                                                     Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __ref__
                         -> Tensor v'2 tindices -- ^ __indices__
                         -> Tensor v'3 t -- ^ __updates__
                         -> m' (ControlNode)
resourceScatterNdMax' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdMax' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterNdMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
resourceScatterNdMin :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                            TensorType t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tindices) =>
                        Tensor v'1 ResourceHandle -- ^ __ref__
                        -> Tensor v'2 tindices -- ^ __indices__
                        -> Tensor v'3 t -- ^ __updates__
                        -> m' (ControlNode)
resourceScatterNdMin :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
resourceScatterNdMin = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdMin' OpParams
forall a. a -> a
id
resourceScatterNdMin' :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                             TensorType t,
                                                             OneOf '[Data.Int.Int32,
                                                                     Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __ref__
                         -> Tensor v'2 tindices -- ^ __indices__
                         -> Tensor v'3 t -- ^ __updates__
                         -> m' (ControlNode)
resourceScatterNdMin' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdMin' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterNdMin"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
resourceScatterNdSub :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                            TensorType t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tindices) =>
                        Tensor v'1 ResourceHandle -- ^ __ref__
                        -> Tensor v'2 tindices -- ^ __indices__
                        -> Tensor v'3 t -- ^ __updates__
                        -> m' (ControlNode)
resourceScatterNdSub :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
resourceScatterNdSub = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdSub' OpParams
forall a. a -> a
id
resourceScatterNdSub' :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                             TensorType t,
                                                             OneOf '[Data.Int.Int32,
                                                                     Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __ref__
                         -> Tensor v'2 tindices -- ^ __indices__
                         -> Tensor v'3 t -- ^ __updates__
                         -> m' (ControlNode)
resourceScatterNdSub' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdSub' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterNdSub"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
resourceScatterNdUpdate :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                               TensorType t,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                           Tensor v'1 ResourceHandle -- ^ __ref__
                           -> Tensor v'2 tindices -- ^ __indices__
                           -> Tensor v'3 t -- ^ __updates__
                           -> m' (ControlNode)
resourceScatterNdUpdate :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
resourceScatterNdUpdate = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdUpdate' OpParams
forall a. a -> a
id
resourceScatterNdUpdate' :: forall v'1 v'2 v'3 t tindices m' . (MonadBuild m',
                                                                TensorType t,
                                                                OneOf '[Data.Int.Int32,
                                                                        Data.Int.Int64] tindices) =>
                            OpParams ->
                            Tensor v'1 ResourceHandle -- ^ __ref__
                            -> Tensor v'2 tindices -- ^ __indices__
                            -> Tensor v'3 t -- ^ __updates__
                            -> m' (ControlNode)
resourceScatterNdUpdate' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterNdUpdate' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterNdUpdate"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
resourceScatterSub :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] dtype,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 tindices -- ^ __indices__
                      -> Tensor v'3 dtype -- ^ __updates__
                      -> m' (ControlNode)
resourceScatterSub :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterSub = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterSub' OpParams
forall a. a -> a
id
resourceScatterSub' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] dtype,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 dtype -- ^ __updates__
                       -> m' (ControlNode)
resourceScatterSub' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterSub' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterSub"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceScatterUpdate :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                                 TensorType dtype,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tindices) =>
                         Tensor v'1 ResourceHandle -- ^ __resource__
                         -> Tensor v'2 tindices -- ^ __indices__
                         -> Tensor v'3 dtype -- ^ __updates__
                         -> m' (ControlNode)
resourceScatterUpdate :: Tensor v'1 ResourceHandle
-> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
resourceScatterUpdate = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' ControlNode
resourceScatterUpdate' OpParams
forall a. a -> a
id
resourceScatterUpdate' :: forall v'1 v'2 v'3 dtype tindices m' . (MonadBuild m',
                                                                  TensorType dtype,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] tindices) =>
                          OpParams ->
                          Tensor v'1 ResourceHandle -- ^ __resource__
                          -> Tensor v'2 tindices -- ^ __indices__
                          -> Tensor v'3 dtype -- ^ __updates__
                          -> m' (ControlNode)
resourceScatterUpdate' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 tindices
-> Tensor v'3 dtype
-> m' ControlNode
resourceScatterUpdate' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 dtype
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
updates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceScatterUpdate"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
resourceSparseApplyAdadelta :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t tindices
                               m' . (MonadBuild m',
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                               Tensor v'1 ResourceHandle -- ^ __var__
                               -> Tensor v'2 ResourceHandle -- ^ __accum__
                               -> Tensor v'3 ResourceHandle -- ^ __accum_update__
                               -> Tensor v'4 t -- ^ __lr__
                               -> Tensor v'5 t -- ^ __rho__
                               -> Tensor v'6 t -- ^ __epsilon__
                               -> Tensor v'7 t -- ^ __grad__
                               -> Tensor v'8 tindices -- ^ __indices__
                               -> m' (ControlNode)
resourceSparseApplyAdadelta :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' ControlNode
resourceSparseApplyAdadelta = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' ControlNode
resourceSparseApplyAdadelta' OpParams
forall a. a -> a
id
resourceSparseApplyAdadelta' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t
                                tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
                                OpParams ->
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __accum__
                                -> Tensor v'3 ResourceHandle -- ^ __accum_update__
                                -> Tensor v'4 t -- ^ __lr__
                                -> Tensor v'5 t -- ^ __rho__
                                -> Tensor v'6 t -- ^ __epsilon__
                                -> Tensor v'7 t -- ^ __grad__
                                -> Tensor v'8 tindices -- ^ __indices__
                                -> m' (ControlNode)
resourceSparseApplyAdadelta' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' ControlNode
resourceSparseApplyAdadelta' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum accum_update :: Tensor v'3 ResourceHandle
accum_update lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho epsilon :: Tensor v'6 t
epsilon
                             grad :: Tensor v'7 t
grad indices :: Tensor v'8 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
accum_update,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
epsilon,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad,
                                                             Tensor v'8 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyAdadelta"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "accum_update" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyAdagrad :: forall v'1 v'2 v'3 v'4 v'5 t tindices
                              m' . (MonadBuild m',
                                    OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tindices) =>
                              Tensor v'1 ResourceHandle -- ^ __var__
                              -> Tensor v'2 ResourceHandle -- ^ __accum__
                              -> Tensor v'3 t -- ^ __lr__
                              -> Tensor v'4 t -- ^ __grad__
                              -> Tensor v'5 tindices -- ^ __indices__
                              -> m' (ControlNode)
resourceSparseApplyAdagrad :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' ControlNode
resourceSparseApplyAdagrad = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' ControlNode
resourceSparseApplyAdagrad' OpParams
forall a. a -> a
id
resourceSparseApplyAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 t tindices
                               m' . (MonadBuild m',
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __var__
                               -> Tensor v'2 ResourceHandle -- ^ __accum__
                               -> Tensor v'3 t -- ^ __lr__
                               -> Tensor v'4 t -- ^ __grad__
                               -> Tensor v'5 tindices -- ^ __indices__
                               -> m' (ControlNode)
resourceSparseApplyAdagrad' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' ControlNode
resourceSparseApplyAdagrad' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad
                            indices :: Tensor v'5 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
resourceSparseApplyAdagradDA :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                                tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __gradient_accumulator__
                                -> Tensor v'3 ResourceHandle -- ^ __gradient_squared_accumulator__
                                -> Tensor v'4 t -- ^ __grad__
                                -> Tensor v'5 tindices -- ^ __indices__
                                -> Tensor v'6 t -- ^ __lr__
                                -> Tensor v'7 t -- ^ __l1__
                                -> Tensor v'8 t -- ^ __l2__
                                -> Tensor v'9 Data.Int.Int64 -- ^ __global_step__
                                -> m' (ControlNode)
resourceSparseApplyAdagradDA :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' ControlNode
resourceSparseApplyAdagradDA = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' ControlNode
resourceSparseApplyAdagradDA' OpParams
forall a. a -> a
id
resourceSparseApplyAdagradDA' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                                 tindices m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                                 OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 ResourceHandle -- ^ __gradient_accumulator__
                                 -> Tensor v'3 ResourceHandle -- ^ __gradient_squared_accumulator__
                                 -> Tensor v'4 t -- ^ __grad__
                                 -> Tensor v'5 tindices -- ^ __indices__
                                 -> Tensor v'6 t -- ^ __lr__
                                 -> Tensor v'7 t -- ^ __l1__
                                 -> Tensor v'8 t -- ^ __l2__
                                 -> Tensor v'9 Data.Int.Int64 -- ^ __global_step__
                                 -> m' (ControlNode)
resourceSparseApplyAdagradDA' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' ControlNode
resourceSparseApplyAdagradDA' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var gradient_accumulator :: Tensor v'2 ResourceHandle
gradient_accumulator
                              gradient_squared_accumulator :: Tensor v'3 ResourceHandle
gradient_squared_accumulator grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                              global_step :: Tensor v'9 Int64
global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
gradient_accumulator,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
gradient_squared_accumulator,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Int64
global_step]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyAdagradDA"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "gradient_accumulator" type: DT_RESOURCE }
input_arg {
  name: "gradient_squared_accumulator" type: DT_RESOURCE
}
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "global_step" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyAdagradV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                                m' . (MonadBuild m',
                                      OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tindices) =>
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __accum__
                                -> Tensor v'3 t -- ^ __lr__
                                -> Tensor v'4 t -- ^ __epsilon__
                                -> Tensor v'5 t -- ^ __grad__
                                -> Tensor v'6 tindices -- ^ __indices__
                                -> m' (ControlNode)
resourceSparseApplyAdagradV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyAdagradV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyAdagradV2' OpParams
forall a. a -> a
id
resourceSparseApplyAdagradV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                                 m' . (MonadBuild m',
                                       OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices) =>
                                 OpParams ->
                                 Tensor v'1 ResourceHandle -- ^ __var__
                                 -> Tensor v'2 ResourceHandle -- ^ __accum__
                                 -> Tensor v'3 t -- ^ __lr__
                                 -> Tensor v'4 t -- ^ __epsilon__
                                 -> Tensor v'5 t -- ^ __grad__
                                 -> Tensor v'6 tindices -- ^ __indices__
                                 -> m' (ControlNode)
resourceSparseApplyAdagradV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyAdagradV2' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr epsilon :: Tensor v'4 t
epsilon grad :: Tensor v'5 t
grad
                              indices :: Tensor v'6 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
epsilon,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad,
                                                             Tensor v'6 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyAdagradV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
resourceSparseApplyCenteredRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9
                                      v'10 t tindices m' . (MonadBuild m',
                                                            OneOf '[(Data.Complex.Complex Double),
                                                                    (Data.Complex.Complex Float),
                                                                    Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Int.Int64,
                                                                    Data.Int.Int8,
                                                                    Data.Word.Word16,
                                                                    Data.Word.Word32,
                                                                    Data.Word.Word64,
                                                                    Data.Word.Word8,
                                                                    Double,
                                                                    Float] t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] tindices) =>
                                      Tensor v'1 ResourceHandle -- ^ __var__
                                      -> Tensor v'2 ResourceHandle -- ^ __mg__
                                      -> Tensor v'3 ResourceHandle -- ^ __ms__
                                      -> Tensor v'4 ResourceHandle -- ^ __mom__
                                      -> Tensor v'5 t -- ^ __lr__
                                      -> Tensor v'6 t -- ^ __rho__
                                      -> Tensor v'7 t -- ^ __momentum__
                                      -> Tensor v'8 t -- ^ __epsilon__
                                      -> Tensor v'9 t -- ^ __grad__
                                      -> Tensor v'10 tindices -- ^ __indices__
                                      -> m' (ControlNode)
resourceSparseApplyCenteredRMSProp :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' ControlNode
resourceSparseApplyCenteredRMSProp = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' ControlNode
resourceSparseApplyCenteredRMSProp' OpParams
forall a. a -> a
id
resourceSparseApplyCenteredRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8
                                       v'9 v'10 t tindices m' . (MonadBuild m',
                                                                 OneOf '[(Data.Complex.Complex Double),
                                                                         (Data.Complex.Complex Float),
                                                                         Data.Int.Int16,
                                                                         Data.Int.Int32,
                                                                         Data.Int.Int64,
                                                                         Data.Int.Int8,
                                                                         Data.Word.Word16,
                                                                         Data.Word.Word32,
                                                                         Data.Word.Word64,
                                                                         Data.Word.Word8,
                                                                         Double,
                                                                         Float] t,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tindices) =>
                                       OpParams ->
                                       Tensor v'1 ResourceHandle -- ^ __var__
                                       -> Tensor v'2 ResourceHandle -- ^ __mg__
                                       -> Tensor v'3 ResourceHandle -- ^ __ms__
                                       -> Tensor v'4 ResourceHandle -- ^ __mom__
                                       -> Tensor v'5 t -- ^ __lr__
                                       -> Tensor v'6 t -- ^ __rho__
                                       -> Tensor v'7 t -- ^ __momentum__
                                       -> Tensor v'8 t -- ^ __epsilon__
                                       -> Tensor v'9 t -- ^ __grad__
                                       -> Tensor v'10 tindices -- ^ __indices__
                                       -> m' (ControlNode)
resourceSparseApplyCenteredRMSProp' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 ResourceHandle
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' ControlNode
resourceSparseApplyCenteredRMSProp' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var mg :: Tensor v'2 ResourceHandle
mg ms :: Tensor v'3 ResourceHandle
ms mom :: Tensor v'4 ResourceHandle
mom lr :: Tensor v'5 t
lr rho :: Tensor v'6 t
rho momentum :: Tensor v'7 t
momentum
                                    epsilon :: Tensor v'8 t
epsilon grad :: Tensor v'9 t
grad indices :: Tensor v'10 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
mg,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
ms,
                                                             Tensor v'4 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ResourceHandle
mom,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
rho,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
momentum,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad,
                                                             Tensor v'10 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyCenteredRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "mg" type: DT_RESOURCE }
input_arg { name: "ms" type: DT_RESOURCE }
input_arg { name: "mom" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyFtrl :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                           m' . (MonadBuild m',
                                 OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tindices) =>
                           Tensor v'1 ResourceHandle -- ^ __var__
                           -> Tensor v'2 ResourceHandle -- ^ __accum__
                           -> Tensor v'3 ResourceHandle -- ^ __linear__
                           -> Tensor v'4 t -- ^ __grad__
                           -> Tensor v'5 tindices -- ^ __indices__
                           -> Tensor v'6 t -- ^ __lr__
                           -> Tensor v'7 t -- ^ __l1__
                           -> Tensor v'8 t -- ^ __l2__
                           -> Tensor v'9 t -- ^ __lr_power__
                           -> m' (ControlNode)
resourceSparseApplyFtrl :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceSparseApplyFtrl = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceSparseApplyFtrl' OpParams
forall a. a -> a
id
resourceSparseApplyFtrl' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                            tindices m' . (MonadBuild m',
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float),
                                                   Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tindices) =>
                            OpParams ->
                            Tensor v'1 ResourceHandle -- ^ __var__
                            -> Tensor v'2 ResourceHandle -- ^ __accum__
                            -> Tensor v'3 ResourceHandle -- ^ __linear__
                            -> Tensor v'4 t -- ^ __grad__
                            -> Tensor v'5 tindices -- ^ __indices__
                            -> Tensor v'6 t -- ^ __lr__
                            -> Tensor v'7 t -- ^ __l1__
                            -> Tensor v'8 t -- ^ __l2__
                            -> Tensor v'9 t -- ^ __lr_power__
                            -> m' (ControlNode)
resourceSparseApplyFtrl' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' ControlNode
resourceSparseApplyFtrl' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum linear :: Tensor v'3 ResourceHandle
linear grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                         lr_power :: Tensor v'9 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
lr_power]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyFtrl"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "linear" type: DT_RESOURCE }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
resourceSparseApplyFtrlV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t
                             tindices m' . (MonadBuild m',
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float),
                                                    Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
                             Tensor v'1 ResourceHandle -- ^ __var__
                             -> Tensor v'2 ResourceHandle -- ^ __accum__
                             -> Tensor v'3 ResourceHandle -- ^ __linear__
                             -> Tensor v'4 t -- ^ __grad__
                             -> Tensor v'5 tindices -- ^ __indices__
                             -> Tensor v'6 t -- ^ __lr__
                             -> Tensor v'7 t -- ^ __l1__
                             -> Tensor v'8 t -- ^ __l2__
                             -> Tensor v'9 t -- ^ __l2_shrinkage__
                             -> Tensor v'10 t -- ^ __lr_power__
                             -> m' (ControlNode)
resourceSparseApplyFtrlV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceSparseApplyFtrlV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceSparseApplyFtrlV2' OpParams
forall a. a -> a
id
resourceSparseApplyFtrlV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 v'10 t
                              tindices m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tindices) =>
                              OpParams ->
                              Tensor v'1 ResourceHandle -- ^ __var__
                              -> Tensor v'2 ResourceHandle -- ^ __accum__
                              -> Tensor v'3 ResourceHandle -- ^ __linear__
                              -> Tensor v'4 t -- ^ __grad__
                              -> Tensor v'5 tindices -- ^ __indices__
                              -> Tensor v'6 t -- ^ __lr__
                              -> Tensor v'7 t -- ^ __l1__
                              -> Tensor v'8 t -- ^ __l2__
                              -> Tensor v'9 t -- ^ __l2_shrinkage__
                              -> Tensor v'10 t -- ^ __lr_power__
                              -> m' (ControlNode)
resourceSparseApplyFtrlV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' ControlNode
resourceSparseApplyFtrlV2' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum linear :: Tensor v'3 ResourceHandle
linear grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                           l2_shrinkage :: Tensor v'9 t
l2_shrinkage lr_power :: Tensor v'10 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
l2_shrinkage,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
lr_power]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyFtrlV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "linear" type: DT_RESOURCE }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "l2_shrinkage" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
resourceSparseApplyKerasMomentum :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                                    m' . (MonadBuild m',
                                          OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tindices) =>
                                    Tensor v'1 ResourceHandle -- ^ __var__
                                    -> Tensor v'2 ResourceHandle -- ^ __accum__
                                    -> Tensor v'3 t -- ^ __lr__
                                    -> Tensor v'4 t -- ^ __grad__
                                    -> Tensor v'5 tindices -- ^ __indices__
                                    -> Tensor v'6 t -- ^ __momentum__
                                    -> m' (ControlNode)
resourceSparseApplyKerasMomentum :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyKerasMomentum = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyKerasMomentum' OpParams
forall a. a -> a
id
resourceSparseApplyKerasMomentum' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                                     m' . (MonadBuild m',
                                           OneOf '[(Data.Complex.Complex Double),
                                                   (Data.Complex.Complex Float),
                                                   Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tindices) =>
                                     OpParams ->
                                     Tensor v'1 ResourceHandle -- ^ __var__
                                     -> Tensor v'2 ResourceHandle -- ^ __accum__
                                     -> Tensor v'3 t -- ^ __lr__
                                     -> Tensor v'4 t -- ^ __grad__
                                     -> Tensor v'5 tindices -- ^ __indices__
                                     -> Tensor v'6 t -- ^ __momentum__
                                     -> m' (ControlNode)
resourceSparseApplyKerasMomentum' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyKerasMomentum' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices
                                  momentum :: Tensor v'6 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyKerasMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "momentum" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyMomentum :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                               m' . (MonadBuild m',
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                               Tensor v'1 ResourceHandle -- ^ __var__
                               -> Tensor v'2 ResourceHandle -- ^ __accum__
                               -> Tensor v'3 t -- ^ __lr__
                               -> Tensor v'4 t -- ^ __grad__
                               -> Tensor v'5 tindices -- ^ __indices__
                               -> Tensor v'6 t -- ^ __momentum__
                               -> m' (ControlNode)
resourceSparseApplyMomentum :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyMomentum = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyMomentum' OpParams
forall a. a -> a
id
resourceSparseApplyMomentum' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t tindices
                                m' . (MonadBuild m',
                                      OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tindices) =>
                                OpParams ->
                                Tensor v'1 ResourceHandle -- ^ __var__
                                -> Tensor v'2 ResourceHandle -- ^ __accum__
                                -> Tensor v'3 t -- ^ __lr__
                                -> Tensor v'4 t -- ^ __grad__
                                -> Tensor v'5 tindices -- ^ __indices__
                                -> Tensor v'6 t -- ^ __momentum__
                                -> m' (ControlNode)
resourceSparseApplyMomentum' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' ControlNode
resourceSparseApplyMomentum' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices
                             momentum :: Tensor v'6 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "momentum" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyProximalAdagrad :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                                      tindices m' . (MonadBuild m',
                                                     OneOf '[(Data.Complex.Complex Double),
                                                             (Data.Complex.Complex Float),
                                                             Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double, Float] t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                                      Tensor v'1 ResourceHandle -- ^ __var__
                                      -> Tensor v'2 ResourceHandle -- ^ __accum__
                                      -> Tensor v'3 t -- ^ __lr__
                                      -> Tensor v'4 t -- ^ __l1__
                                      -> Tensor v'5 t -- ^ __l2__
                                      -> Tensor v'6 t -- ^ __grad__
                                      -> Tensor v'7 tindices -- ^ __indices__
                                      -> m' (ControlNode)
resourceSparseApplyProximalAdagrad :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' ControlNode
resourceSparseApplyProximalAdagrad = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' ControlNode
resourceSparseApplyProximalAdagrad' OpParams
forall a. a -> a
id
resourceSparseApplyProximalAdagrad' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
                                       tindices m' . (MonadBuild m',
                                                      OneOf '[(Data.Complex.Complex Double),
                                                              (Data.Complex.Complex Float),
                                                              Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double, Float] t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                                       OpParams ->
                                       Tensor v'1 ResourceHandle -- ^ __var__
                                       -> Tensor v'2 ResourceHandle -- ^ __accum__
                                       -> Tensor v'3 t -- ^ __lr__
                                       -> Tensor v'4 t -- ^ __l1__
                                       -> Tensor v'5 t -- ^ __l2__
                                       -> Tensor v'6 t -- ^ __grad__
                                       -> Tensor v'7 tindices -- ^ __indices__
                                       -> m' (ControlNode)
resourceSparseApplyProximalAdagrad' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' ControlNode
resourceSparseApplyProximalAdagrad' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var accum :: Tensor v'2 ResourceHandle
accum lr :: Tensor v'3 t
lr l1 :: Tensor v'4 t
l1 l2 :: Tensor v'5 t
l2 grad :: Tensor v'6 t
grad
                                    indices :: Tensor v'7 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l1,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
l2,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
grad,
                                                             Tensor v'7 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyProximalAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "accum" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyProximalGradientDescent :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                                              tindices m' . (MonadBuild m',
                                                             OneOf '[(Data.Complex.Complex Double),
                                                                     (Data.Complex.Complex Float),
                                                                     Data.Int.Int16,
                                                                     Data.Int.Int32,
                                                                     Data.Int.Int64,
                                                                     Data.Int.Int8,
                                                                     Data.Word.Word16,
                                                                     Data.Word.Word32,
                                                                     Data.Word.Word64,
                                                                     Data.Word.Word8,
                                                                     Double,
                                                                     Float] t,
                                                             OneOf '[Data.Int.Int32,
                                                                     Data.Int.Int64] tindices) =>
                                              Tensor v'1 ResourceHandle -- ^ __var__
                                              -> Tensor v'2 t -- ^ __alpha__
                                              -> Tensor v'3 t -- ^ __l1__
                                              -> Tensor v'4 t -- ^ __l2__
                                              -> Tensor v'5 t -- ^ __grad__
                                              -> Tensor v'6 tindices -- ^ __indices__
                                              -> m' (ControlNode)
resourceSparseApplyProximalGradientDescent :: Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyProximalGradientDescent = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyProximalGradientDescent' OpParams
forall a. a -> a
id
resourceSparseApplyProximalGradientDescent' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                                               tindices m' . (MonadBuild m',
                                                              OneOf '[(Data.Complex.Complex Double),
                                                                      (Data.Complex.Complex Float),
                                                                      Data.Int.Int16,
                                                                      Data.Int.Int32,
                                                                      Data.Int.Int64,
                                                                      Data.Int.Int8,
                                                                      Data.Word.Word16,
                                                                      Data.Word.Word32,
                                                                      Data.Word.Word64,
                                                                      Data.Word.Word8,
                                                                      Double,
                                                                      Float] t,
                                                              OneOf '[Data.Int.Int32,
                                                                      Data.Int.Int64] tindices) =>
                                               OpParams ->
                                               Tensor v'1 ResourceHandle -- ^ __var__
                                               -> Tensor v'2 t -- ^ __alpha__
                                               -> Tensor v'3 t -- ^ __l1__
                                               -> Tensor v'4 t -- ^ __l2__
                                               -> Tensor v'5 t -- ^ __grad__
                                               -> Tensor v'6 tindices -- ^ __indices__
                                               -> m' (ControlNode)
resourceSparseApplyProximalGradientDescent' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' ControlNode
resourceSparseApplyProximalGradientDescent' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var alpha :: Tensor v'2 t
alpha l1 :: Tensor v'3 t
l1 l2 :: Tensor v'4 t
l2 grad :: Tensor v'5 t
grad
                                            indices :: Tensor v'6 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
l1,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l2,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad,
                                                             Tensor v'6 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyProximalGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceSparseApplyRMSProp :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                              tindices m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tindices) =>
                              Tensor v'1 ResourceHandle -- ^ __var__
                              -> Tensor v'2 ResourceHandle -- ^ __ms__
                              -> Tensor v'3 ResourceHandle -- ^ __mom__
                              -> Tensor v'4 t -- ^ __lr__
                              -> Tensor v'5 t -- ^ __rho__
                              -> Tensor v'6 t -- ^ __momentum__
                              -> Tensor v'7 t -- ^ __epsilon__
                              -> Tensor v'8 t -- ^ __grad__
                              -> Tensor v'9 tindices -- ^ __indices__
                              -> m' (ControlNode)
resourceSparseApplyRMSProp :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' ControlNode
resourceSparseApplyRMSProp = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' ControlNode
resourceSparseApplyRMSProp' OpParams
forall a. a -> a
id
resourceSparseApplyRMSProp' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t
                               tindices m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __var__
                               -> Tensor v'2 ResourceHandle -- ^ __ms__
                               -> Tensor v'3 ResourceHandle -- ^ __mom__
                               -> Tensor v'4 t -- ^ __lr__
                               -> Tensor v'5 t -- ^ __rho__
                               -> Tensor v'6 t -- ^ __momentum__
                               -> Tensor v'7 t -- ^ __epsilon__
                               -> Tensor v'8 t -- ^ __grad__
                               -> Tensor v'9 tindices -- ^ __indices__
                               -> m' (ControlNode)
resourceSparseApplyRMSProp' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ResourceHandle
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' ControlNode
resourceSparseApplyRMSProp' op'options :: OpParams
op'options var :: Tensor v'1 ResourceHandle
var ms :: Tensor v'2 ResourceHandle
ms mom :: Tensor v'3 ResourceHandle
mom lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho momentum :: Tensor v'6 t
momentum epsilon :: Tensor v'7 t
epsilon grad :: Tensor v'8 t
grad
                            indices :: Tensor v'9 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
var,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
ms,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
mom,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
epsilon,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
grad,
                                                             Tensor v'9 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 tindices
indices]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceSparseApplyRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type: DT_RESOURCE }
input_arg { name: "ms" type: DT_RESOURCE }
input_arg { name: "mom" type: DT_RESOURCE }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
resourceStridedSliceAssign :: forall v'1 v'2 v'3 v'4 v'5 t index
                              m' . (MonadBuild m', TensorType t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] index) =>
                              Tensor v'1 ResourceHandle -- ^ __ref__
                              -> Tensor v'2 index -- ^ __begin__
                              -> Tensor v'3 index -- ^ __end__
                              -> Tensor v'4 index -- ^ __strides__
                              -> Tensor v'5 t -- ^ __value__
                              -> m' (ControlNode)
resourceStridedSliceAssign :: Tensor v'1 ResourceHandle
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' ControlNode
resourceStridedSliceAssign = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t index (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' ControlNode
resourceStridedSliceAssign' OpParams
forall a. a -> a
id
resourceStridedSliceAssign' :: forall v'1 v'2 v'3 v'4 v'5 t index
                               m' . (MonadBuild m', TensorType t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] index) =>
                               OpParams ->
                               Tensor v'1 ResourceHandle -- ^ __ref__
                               -> Tensor v'2 index -- ^ __begin__
                               -> Tensor v'3 index -- ^ __end__
                               -> Tensor v'4 index -- ^ __strides__
                               -> Tensor v'5 t -- ^ __value__
                               -> m' (ControlNode)
resourceStridedSliceAssign' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' ControlNode
resourceStridedSliceAssign' op'options :: OpParams
op'options ref :: Tensor v'1 ResourceHandle
ref begin :: Tensor v'2 index
begin end :: Tensor v'3 index
end strides :: Tensor v'4 index
strides
                            value :: Tensor v'5 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
ref,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
end,
                                                             Tensor v'4 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 index
strides,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
value]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ResourceStridedSliceAssign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type: DT_RESOURCE }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "end" type_attr: "Index" }
input_arg { name: "strides" type_attr: "Index" }
input_arg { name: "value" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "begin_mask" type: "int" default_value { i: 0 } }
attr { name: "end_mask" type: "int" default_value { i: 0 } }
attr { name: "ellipsis_mask" type: "int" default_value { i: 0 } }
attr { name: "new_axis_mask" type: "int" default_value { i: 0 } }
attr {
  name: "shrink_axis_mask" type: "int" default_value { i: 0 }
}
-}
-- | 
restore :: forall v'1 v'2 dt m' . (MonadBuild m', TensorType dt) =>
           Tensor v'1 Data.ByteString.ByteString -- ^ __file_pattern__
           -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_name__
           -> m' (Tensor Value dt) -- ^ __tensor__
restore :: Tensor v'1 ByteString
-> Tensor v'2 ByteString -> m' (Tensor Value dt)
restore = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> m' (Tensor Value dt)
forall (v'1 :: * -> *) (v'2 :: * -> *) dt (m' :: * -> *).
(MonadBuild m', TensorType dt) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> m' (Tensor Value dt)
restore' OpParams
forall a. a -> a
id
restore' :: forall v'1 v'2 dt m' . (MonadBuild m', TensorType dt) => OpParams ->
            Tensor v'1 Data.ByteString.ByteString -- ^ __file_pattern__
            -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_name__
            -> m' (Tensor Value dt) -- ^ __tensor__
restore' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> m' (Tensor Value dt)
restore' op'options :: OpParams
op'options file_pattern :: Tensor v'1 ByteString
file_pattern tensor_name :: Tensor v'2 ByteString
tensor_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dt) -> m' (Tensor Value dt)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dt) -> m' (Tensor Value dt))
-> Build (Tensor Value dt) -> m' (Tensor Value dt)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
file_pattern,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_name]
        [Int64] -> OpDef -> Build (Tensor Value dt)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Restore"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dt" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dt -> DataType
forall a. TensorType a => a -> DataType
tensorType (dt
forall a. HasCallStack => a
undefined :: dt)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "file_pattern" type: DT_STRING }
input_arg { name: "tensor_name" type: DT_STRING }
output_arg { name: "tensor" type_attr: "dt" }
attr { name: "dt" type: "type" }
attr {
  name: "preferred_shard" type: "int" default_value { i: -1 }
}
-}
-- | 
restoreSlice :: forall v'1 v'2 v'3 dt m' . (MonadBuild m', TensorType dt) =>
                Tensor v'1 Data.ByteString.ByteString -- ^ __file_pattern__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_name__
                -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slice__
                -> m' (Tensor Value dt) -- ^ __tensor__
restoreSlice :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value dt)
restoreSlice = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value dt)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dt
       (m' :: * -> *).
(MonadBuild m', TensorType dt) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value dt)
restoreSlice' OpParams
forall a. a -> a
id
restoreSlice' :: forall v'1 v'2 v'3 dt m' . (MonadBuild m', TensorType dt) =>
                 OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __file_pattern__
                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_name__
                 -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slice__
                 -> m' (Tensor Value dt) -- ^ __tensor__
restoreSlice' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value dt)
restoreSlice' op'options :: OpParams
op'options file_pattern :: Tensor v'1 ByteString
file_pattern tensor_name :: Tensor v'2 ByteString
tensor_name
              shape_and_slice :: Tensor v'3 ByteString
shape_and_slice | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dt) -> m' (Tensor Value dt)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dt) -> m' (Tensor Value dt))
-> Build (Tensor Value dt) -> m' (Tensor Value dt)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
file_pattern,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_name,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
shape_and_slice]
        [Int64] -> OpDef -> Build (Tensor Value dt)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RestoreSlice"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dt" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dt -> DataType
forall a. TensorType a => a -> DataType
tensorType (dt
forall a. HasCallStack => a
undefined :: dt)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "file_pattern" type: DT_STRING }
input_arg { name: "tensor_name" type: DT_STRING }
input_arg { name: "shape_and_slice" type: DT_STRING }
output_arg { name: "tensor" type_attr: "dt" }
attr { name: "dt" type: "type" }
attr {
  name: "preferred_shard" type: "int" default_value { i: -1 }
}
-}
-- | 
restoreV2 :: forall v'1 v'2 v'3 dtypes m' . (MonadBuild m',
                                             TensorTypes dtypes) =>
             Tensor v'1 Data.ByteString.ByteString -- ^ __prefix__
             -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
             -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slices__
             -> m' (TensorList (Value) dtypes) -- ^ __tensors__
restoreV2 :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (TensorList Value dtypes)
restoreV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (TensorList Value dtypes)
restoreV2' OpParams
forall a. a -> a
id
restoreV2' :: forall v'1 v'2 v'3 dtypes m' . (MonadBuild m',
                                              TensorTypes dtypes) => OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __prefix__
              -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
              -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slices__
              -> m' (TensorList (Value) dtypes) -- ^ __tensors__
restoreV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (TensorList Value dtypes)
restoreV2' op'options :: OpParams
op'options prefix :: Tensor v'1 ByteString
prefix tensor_names :: Tensor v'2 ByteString
tensor_names shape_and_slices :: Tensor v'3 ByteString
shape_and_slices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
prefix,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_names,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
shape_and_slices]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RestoreV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "prefix" type: DT_STRING }
input_arg { name: "tensor_names" type: DT_STRING }
input_arg { name: "shape_and_slices" type: DT_STRING }
output_arg { name: "tensors" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | 
retrieveTPUEmbeddingADAMParameters :: forall m' . (MonadBuild m') =>
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> m' ((Tensor Value Float,
                                              Tensor Value Float,
                                              Tensor Value Float))
                                      -- ^ (__parameters__, __momenta__, __velocities__)
                                      --
                                      -- * __parameters__
                                      --
                                      -- * __momenta__
                                      --
                                      -- * __velocities__
retrieveTPUEmbeddingADAMParameters :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParameters = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingADAMParameters' :: forall m' . (MonadBuild m') =>
                                       OpParams ->
                                       Data.Int.Int64 -- ^ __num_shards__
                                       -> Data.Int.Int64 -- ^ __shard_id__
                                       -> m' ((Tensor Value Float,
                                               Tensor Value Float,
                                               Tensor Value Float))
                                       -- ^ (__parameters__, __momenta__, __velocities__)
                                       --
                                       -- * __parameters__
                                       --
                                       -- * __momenta__
                                       --
                                       -- * __velocities__
retrieveTPUEmbeddingADAMParameters' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingADAMParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                    shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingADAMParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "momenta" type: DT_FLOAT }
output_arg { name: "velocities" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingADAMParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> m' ((Tensor Value Float,
                                                            Tensor Value Float,
                                                            Tensor Value Float,
                                                            Tensor Value Float))
                                                    -- ^ (__parameters__, __momenta__, __velocities__, __gradient_accumulators__)
                                                    --
                                                    -- * __parameters__
                                                    --
                                                    -- * __momenta__
                                                    --
                                                    -- * __velocities__
                                                    --
                                                    -- * __gradient_accumulators__
retrieveTPUEmbeddingADAMParametersGradAccumDebug :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingADAMParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingADAMParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingADAMParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                     OpParams ->
                                                     Data.Int.Int64 -- ^ __num_shards__
                                                     -> Data.Int.Int64 -- ^ __shard_id__
                                                     -> m' ((Tensor Value Float,
                                                             Tensor Value Float,
                                                             Tensor Value Float,
                                                             Tensor Value Float))
                                                     -- ^ (__parameters__, __momenta__, __velocities__, __gradient_accumulators__)
                                                     --
                                                     -- * __parameters__
                                                     --
                                                     -- * __momenta__
                                                     --
                                                     -- * __velocities__
                                                     --
                                                     -- * __gradient_accumulators__
retrieveTPUEmbeddingADAMParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingADAMParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                  shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingADAMParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "momenta" type: DT_FLOAT }
output_arg { name: "velocities" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingAdadeltaParameters :: forall m' . (MonadBuild m') =>
                                          Data.Int.Int64 -- ^ __num_shards__
                                          -> Data.Int.Int64 -- ^ __shard_id__
                                          -> m' ((Tensor Value Float,
                                                  Tensor Value Float,
                                                  Tensor Value Float))
                                          -- ^ (__parameters__, __accumulators__, __updates__)
                                          --
                                          -- * __parameters__
                                          --
                                          -- * __accumulators__
                                          --
                                          -- * __updates__
retrieveTPUEmbeddingAdadeltaParameters :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParameters = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingAdadeltaParameters' :: forall m' . (MonadBuild m') =>
                                           OpParams ->
                                           Data.Int.Int64 -- ^ __num_shards__
                                           -> Data.Int.Int64 -- ^ __shard_id__
                                           -> m' ((Tensor Value Float,
                                                   Tensor Value Float,
                                                   Tensor Value Float))
                                           -- ^ (__parameters__, __accumulators__, __updates__)
                                           --
                                           -- * __parameters__
                                           --
                                           -- * __accumulators__
                                           --
                                           -- * __updates__
retrieveTPUEmbeddingAdadeltaParameters' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                        shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingAdadeltaParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "updates" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> m' ((Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float))
                                                        -- ^ (__parameters__, __accumulators__, __updates__, __gradient_accumulators__)
                                                        --
                                                        -- * __parameters__
                                                        --
                                                        -- * __accumulators__
                                                        --
                                                        -- * __updates__
                                                        --
                                                        -- * __gradient_accumulators__
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                         OpParams ->
                                                         Data.Int.Int64 -- ^ __num_shards__
                                                         -> Data.Int.Int64 -- ^ __shard_id__
                                                         -> m' ((Tensor Value Float,
                                                                 Tensor Value Float,
                                                                 Tensor Value Float,
                                                                 Tensor Value Float))
                                                         -- ^ (__parameters__, __accumulators__, __updates__, __gradient_accumulators__)
                                                         --
                                                         -- * __parameters__
                                                         --
                                                         -- * __accumulators__
                                                         --
                                                         -- * __updates__
                                                         --
                                                         -- * __gradient_accumulators__
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                      shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingAdadeltaParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "updates" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingAdagradParameters :: forall m' . (MonadBuild m') =>
                                         Data.Int.Int64 -- ^ __num_shards__
                                         -> Data.Int.Int64 -- ^ __shard_id__
                                         -> m' ((Tensor Value Float,
                                                 Tensor Value Float))
                                         -- ^ (__parameters__, __accumulators__)
                                         --
                                         -- * __parameters__
                                         --
                                         -- * __accumulators__
retrieveTPUEmbeddingAdagradParameters :: Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParameters = OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingAdagradParameters' :: forall m' . (MonadBuild m') =>
                                          OpParams ->
                                          Data.Int.Int64 -- ^ __num_shards__
                                          -> Data.Int.Int64 -- ^ __shard_id__
                                          -> m' ((Tensor Value Float,
                                                  Tensor Value Float))
                                          -- ^ (__parameters__, __accumulators__)
                                          --
                                          -- * __parameters__
                                          --
                                          -- * __accumulators__
retrieveTPUEmbeddingAdagradParameters' :: OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                       shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float))
-> Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingAdagradParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingAdagradParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                       Data.Int.Int64 -- ^ __num_shards__
                                                       -> Data.Int.Int64 -- ^ __shard_id__
                                                       -> m' ((Tensor Value Float,
                                                               Tensor Value Float,
                                                               Tensor Value Float))
                                                       -- ^ (__parameters__, __accumulators__, __gradient_accumulators__)
                                                       --
                                                       -- * __parameters__
                                                       --
                                                       -- * __accumulators__
                                                       --
                                                       -- * __gradient_accumulators__
retrieveTPUEmbeddingAdagradParametersGradAccumDebug :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                        OpParams ->
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> m' ((Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float))
                                                        -- ^ (__parameters__, __accumulators__, __gradient_accumulators__)
                                                        --
                                                        -- * __parameters__
                                                        --
                                                        -- * __accumulators__
                                                        --
                                                        -- * __gradient_accumulators__
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                     shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingAdagradParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingCenteredRMSPropParameters :: forall m' . (MonadBuild m') =>
                                                 Data.Int.Int64 -- ^ __num_shards__
                                                 -> Data.Int.Int64 -- ^ __shard_id__
                                                 -> m' ((Tensor Value Float,
                                                         Tensor Value Float,
                                                         Tensor Value Float,
                                                         Tensor Value Float))
                                                 -- ^ (__parameters__, __ms__, __mom__, __mg__)
                                                 --
                                                 -- * __parameters__
                                                 --
                                                 -- * __ms__
                                                 --
                                                 -- * __mom__
                                                 --
                                                 -- * __mg__
retrieveTPUEmbeddingCenteredRMSPropParameters :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingCenteredRMSPropParameters = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingCenteredRMSPropParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingCenteredRMSPropParameters' :: forall m' . (MonadBuild m') =>
                                                  OpParams ->
                                                  Data.Int.Int64 -- ^ __num_shards__
                                                  -> Data.Int.Int64 -- ^ __shard_id__
                                                  -> m' ((Tensor Value Float,
                                                          Tensor Value Float,
                                                          Tensor Value Float,
                                                          Tensor Value Float))
                                                  -- ^ (__parameters__, __ms__, __mom__, __mg__)
                                                  --
                                                  -- * __parameters__
                                                  --
                                                  -- * __ms__
                                                  --
                                                  -- * __mom__
                                                  --
                                                  -- * __mg__
retrieveTPUEmbeddingCenteredRMSPropParameters' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingCenteredRMSPropParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                               shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingCenteredRMSPropParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "ms" type: DT_FLOAT }
output_arg { name: "mom" type: DT_FLOAT }
output_arg { name: "mg" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingFTRLParameters :: forall m' . (MonadBuild m') =>
                                      Data.Int.Int64 -- ^ __num_shards__
                                      -> Data.Int.Int64 -- ^ __shard_id__
                                      -> m' ((Tensor Value Float,
                                              Tensor Value Float,
                                              Tensor Value Float))
                                      -- ^ (__parameters__, __accumulators__, __linears__)
                                      --
                                      -- * __parameters__
                                      --
                                      -- * __accumulators__
                                      --
                                      -- * __linears__
retrieveTPUEmbeddingFTRLParameters :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParameters = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingFTRLParameters' :: forall m' . (MonadBuild m') =>
                                       OpParams ->
                                       Data.Int.Int64 -- ^ __num_shards__
                                       -> Data.Int.Int64 -- ^ __shard_id__
                                       -> m' ((Tensor Value Float,
                                               Tensor Value Float,
                                               Tensor Value Float))
                                       -- ^ (__parameters__, __accumulators__, __linears__)
                                       --
                                       -- * __parameters__
                                       --
                                       -- * __accumulators__
                                       --
                                       -- * __linears__
retrieveTPUEmbeddingFTRLParameters' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingFTRLParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                    shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingFTRLParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "linears" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingFTRLParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                    Data.Int.Int64 -- ^ __num_shards__
                                                    -> Data.Int.Int64 -- ^ __shard_id__
                                                    -> m' ((Tensor Value Float,
                                                            Tensor Value Float,
                                                            Tensor Value Float,
                                                            Tensor Value Float))
                                                    -- ^ (__parameters__, __accumulators__, __linears__, __gradient_accumulators__)
                                                    --
                                                    -- * __parameters__
                                                    --
                                                    -- * __accumulators__
                                                    --
                                                    -- * __linears__
                                                    --
                                                    -- * __gradient_accumulators__
retrieveTPUEmbeddingFTRLParametersGradAccumDebug :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingFTRLParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                     OpParams ->
                                                     Data.Int.Int64 -- ^ __num_shards__
                                                     -> Data.Int.Int64 -- ^ __shard_id__
                                                     -> m' ((Tensor Value Float,
                                                             Tensor Value Float,
                                                             Tensor Value Float,
                                                             Tensor Value Float))
                                                     -- ^ (__parameters__, __accumulators__, __linears__, __gradient_accumulators__)
                                                     --
                                                     -- * __parameters__
                                                     --
                                                     -- * __accumulators__
                                                     --
                                                     -- * __linears__
                                                     --
                                                     -- * __gradient_accumulators__
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                  shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingFTRLParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "linears" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingMDLAdagradLightParameters :: forall m' . (MonadBuild m') =>
                                                 Data.Int.Int64 -- ^ __num_shards__
                                                 -> Data.Int.Int64 -- ^ __shard_id__
                                                 -> m' ((Tensor Value Float,
                                                         Tensor Value Float,
                                                         Tensor Value Float,
                                                         Tensor Value Float))
                                                 -- ^ (__parameters__, __accumulators__, __weights__, __benefits__)
                                                 --
                                                 -- * __parameters__
                                                 --
                                                 -- * __accumulators__
                                                 --
                                                 -- * __weights__
                                                 --
                                                 -- * __benefits__
retrieveTPUEmbeddingMDLAdagradLightParameters :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingMDLAdagradLightParameters = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingMDLAdagradLightParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingMDLAdagradLightParameters' :: forall m' . (MonadBuild m') =>
                                                  OpParams ->
                                                  Data.Int.Int64 -- ^ __num_shards__
                                                  -> Data.Int.Int64 -- ^ __shard_id__
                                                  -> m' ((Tensor Value Float,
                                                          Tensor Value Float,
                                                          Tensor Value Float,
                                                          Tensor Value Float))
                                                  -- ^ (__parameters__, __accumulators__, __weights__, __benefits__)
                                                  --
                                                  -- * __parameters__
                                                  --
                                                  -- * __accumulators__
                                                  --
                                                  -- * __weights__
                                                  --
                                                  -- * __benefits__
retrieveTPUEmbeddingMDLAdagradLightParameters' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingMDLAdagradLightParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                               shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingMDLAdagradLightParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "weights" type: DT_FLOAT }
output_arg { name: "benefits" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingMomentumParameters :: forall m' . (MonadBuild m') =>
                                          Data.Int.Int64 -- ^ __num_shards__
                                          -> Data.Int.Int64 -- ^ __shard_id__
                                          -> m' ((Tensor Value Float,
                                                  Tensor Value Float))
                                          -- ^ (__parameters__, __momenta__)
                                          --
                                          -- * __parameters__
                                          --
                                          -- * __momenta__
retrieveTPUEmbeddingMomentumParameters :: Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParameters = OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingMomentumParameters' :: forall m' . (MonadBuild m') =>
                                           OpParams ->
                                           Data.Int.Int64 -- ^ __num_shards__
                                           -> Data.Int.Int64 -- ^ __shard_id__
                                           -> m' ((Tensor Value Float,
                                                   Tensor Value Float))
                                           -- ^ (__parameters__, __momenta__)
                                           --
                                           -- * __parameters__
                                           --
                                           -- * __momenta__
retrieveTPUEmbeddingMomentumParameters' :: OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                        shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float))
-> Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingMomentumParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "momenta" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingMomentumParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> m' ((Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float))
                                                        -- ^ (__parameters__, __momenta__, __gradient_accumulators__)
                                                        --
                                                        -- * __parameters__
                                                        --
                                                        -- * __momenta__
                                                        --
                                                        -- * __gradient_accumulators__
retrieveTPUEmbeddingMomentumParametersGradAccumDebug :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                         OpParams ->
                                                         Data.Int.Int64 -- ^ __num_shards__
                                                         -> Data.Int.Int64 -- ^ __shard_id__
                                                         -> m' ((Tensor Value Float,
                                                                 Tensor Value Float,
                                                                 Tensor Value Float))
                                                         -- ^ (__parameters__, __momenta__, __gradient_accumulators__)
                                                         --
                                                         -- * __parameters__
                                                         --
                                                         -- * __momenta__
                                                         --
                                                         -- * __gradient_accumulators__
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                      shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingMomentumParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "momenta" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingProximalAdagradParameters :: forall m' . (MonadBuild m') =>
                                                 Data.Int.Int64 -- ^ __num_shards__
                                                 -> Data.Int.Int64 -- ^ __shard_id__
                                                 -> m' ((Tensor Value Float,
                                                         Tensor Value Float))
                                                 -- ^ (__parameters__, __accumulators__)
                                                 --
                                                 -- * __parameters__
                                                 --
                                                 -- * __accumulators__
retrieveTPUEmbeddingProximalAdagradParameters :: Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParameters = OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingProximalAdagradParameters' :: forall m' . (MonadBuild m') =>
                                                  OpParams ->
                                                  Data.Int.Int64 -- ^ __num_shards__
                                                  -> Data.Int.Int64 -- ^ __shard_id__
                                                  -> m' ((Tensor Value Float,
                                                          Tensor Value Float))
                                                  -- ^ (__parameters__, __accumulators__)
                                                  --
                                                  -- * __parameters__
                                                  --
                                                  -- * __accumulators__
retrieveTPUEmbeddingProximalAdagradParameters' :: OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                               shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float))
-> Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingProximalAdagradParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                               Data.Int.Int64 -- ^ __num_shards__
                                                               -> Data.Int.Int64 -- ^ __shard_id__
                                                               -> m' ((Tensor Value Float,
                                                                       Tensor Value Float,
                                                                       Tensor Value Float))
                                                               -- ^ (__parameters__, __accumulators__, __gradient_accumulators__)
                                                               --
                                                               -- * __parameters__
                                                               --
                                                               -- * __accumulators__
                                                               --
                                                               -- * __gradient_accumulators__
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                                OpParams ->
                                                                Data.Int.Int64 -- ^ __num_shards__
                                                                -> Data.Int.Int64 -- ^ __shard_id__
                                                                -> m' ((Tensor Value Float,
                                                                        Tensor Value Float,
                                                                        Tensor Value Float))
                                                                -- ^ (__parameters__, __accumulators__, __gradient_accumulators__)
                                                                --
                                                                -- * __parameters__
                                                                --
                                                                -- * __accumulators__
                                                                --
                                                                -- * __gradient_accumulators__
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' op'options :: OpParams
op'options
                                                             num_shards :: Int64
num_shards
                                                             shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "accumulators" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingProximalYogiParameters :: forall m' . (MonadBuild m') =>
                                              Data.Int.Int64 -- ^ __num_shards__
                                              -> Data.Int.Int64 -- ^ __shard_id__
                                              -> m' ((Tensor Value Float,
                                                      Tensor Value Float,
                                                      Tensor Value Float))
                                              -- ^ (__parameters__, __v__, __m__)
                                              --
                                              -- * __parameters__
                                              --
                                              -- * __v__
                                              --
                                              -- * __m__
retrieveTPUEmbeddingProximalYogiParameters :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParameters = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingProximalYogiParameters' :: forall m' . (MonadBuild m') =>
                                               OpParams ->
                                               Data.Int.Int64 -- ^ __num_shards__
                                               -> Data.Int.Int64 -- ^ __shard_id__
                                               -> m' ((Tensor Value Float,
                                                       Tensor Value Float,
                                                       Tensor Value Float))
                                               -- ^ (__parameters__, __v__, __m__)
                                               --
                                               -- * __parameters__
                                               --
                                               -- * __v__
                                               --
                                               -- * __m__
retrieveTPUEmbeddingProximalYogiParameters' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                            shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingProximalYogiParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "v" type: DT_FLOAT }
output_arg { name: "m" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                            Data.Int.Int64 -- ^ __num_shards__
                                                            -> Data.Int.Int64 -- ^ __shard_id__
                                                            -> m' ((Tensor Value Float,
                                                                    Tensor Value Float,
                                                                    Tensor Value Float,
                                                                    Tensor Value Float))
                                                            -- ^ (__parameters__, __v__, __m__, __gradient_accumulators__)
                                                            --
                                                            -- * __parameters__
                                                            --
                                                            -- * __v__
                                                            --
                                                            -- * __m__
                                                            --
                                                            -- * __gradient_accumulators__
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                             OpParams ->
                                                             Data.Int.Int64 -- ^ __num_shards__
                                                             -> Data.Int.Int64 -- ^ __shard_id__
                                                             -> m' ((Tensor Value Float,
                                                                     Tensor Value Float,
                                                                     Tensor Value Float,
                                                                     Tensor Value Float))
                                                             -- ^ (__parameters__, __v__, __m__, __gradient_accumulators__)
                                                             --
                                                             -- * __parameters__
                                                             --
                                                             -- * __v__
                                                             --
                                                             -- * __m__
                                                             --
                                                             -- * __gradient_accumulators__
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                          shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingProximalYogiParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "v" type: DT_FLOAT }
output_arg { name: "m" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingRMSPropParameters :: forall m' . (MonadBuild m') =>
                                         Data.Int.Int64 -- ^ __num_shards__
                                         -> Data.Int.Int64 -- ^ __shard_id__
                                         -> m' ((Tensor Value Float,
                                                 Tensor Value Float,
                                                 Tensor Value Float))
                                         -- ^ (__parameters__, __ms__, __mom__)
                                         --
                                         -- * __parameters__
                                         --
                                         -- * __ms__
                                         --
                                         -- * __mom__
retrieveTPUEmbeddingRMSPropParameters :: Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParameters = OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingRMSPropParameters' :: forall m' . (MonadBuild m') =>
                                          OpParams ->
                                          Data.Int.Int64 -- ^ __num_shards__
                                          -> Data.Int.Int64 -- ^ __shard_id__
                                          -> m' ((Tensor Value Float,
                                                  Tensor Value Float,
                                                  Tensor Value Float))
                                          -- ^ (__parameters__, __ms__, __mom__)
                                          --
                                          -- * __parameters__
                                          --
                                          -- * __ms__
                                          --
                                          -- * __mom__
retrieveTPUEmbeddingRMSPropParameters' :: OpParams
-> Int64
-> Int64
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingRMSPropParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                       shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingRMSPropParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "ms" type: DT_FLOAT }
output_arg { name: "mom" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                       Data.Int.Int64 -- ^ __num_shards__
                                                       -> Data.Int.Int64 -- ^ __shard_id__
                                                       -> m' ((Tensor Value Float,
                                                               Tensor Value Float,
                                                               Tensor Value Float,
                                                               Tensor Value Float))
                                                       -- ^ (__parameters__, __ms__, __mom__, __gradient_accumulators__)
                                                       --
                                                       -- * __parameters__
                                                       --
                                                       -- * __ms__
                                                       --
                                                       -- * __mom__
                                                       --
                                                       -- * __gradient_accumulators__
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug :: Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug = OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                        OpParams ->
                                                        Data.Int.Int64 -- ^ __num_shards__
                                                        -> Data.Int.Int64 -- ^ __shard_id__
                                                        -> m' ((Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float,
                                                                Tensor Value Float))
                                                        -- ^ (__parameters__, __ms__, __mom__, __gradient_accumulators__)
                                                        --
                                                        -- * __parameters__
                                                        --
                                                        -- * __ms__
                                                        --
                                                        -- * __mom__
                                                        --
                                                        -- * __gradient_accumulators__
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' :: OpParams
-> Int64
-> Int64
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                     shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value Float, Tensor Value Float, Tensor Value Float,
   Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value Float, Tensor Value Float, Tensor Value Float,
    Tensor Value Float)
 -> m'
      (Tensor Value Float, Tensor Value Float, Tensor Value Float,
       Tensor Value Float))
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
-> m'
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value Float, Tensor Value Float, Tensor Value Float,
      Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingRMSPropParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "ms" type: DT_FLOAT }
output_arg { name: "mom" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingStochasticGradientDescentParameters :: forall m' . (MonadBuild m') =>
                                                           Data.Int.Int64 -- ^ __num_shards__
                                                           -> Data.Int.Int64 -- ^ __shard_id__
                                                           -> m' (Tensor Value Float) -- ^ __parameters__
retrieveTPUEmbeddingStochasticGradientDescentParameters :: Int64 -> Int64 -> m' (Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParameters = OpParams -> Int64 -> Int64 -> m' (Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> Int64 -> m' (Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParameters' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingStochasticGradientDescentParameters' :: forall m' . (MonadBuild m') =>
                                                            OpParams ->
                                                            Data.Int.Int64 -- ^ __num_shards__
                                                            -> Data.Int.Int64 -- ^ __shard_id__
                                                            -> m' (Tensor Value Float) -- ^ __parameters__
retrieveTPUEmbeddingStochasticGradientDescentParameters' :: OpParams -> Int64 -> Int64 -> m' (Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParameters' op'options :: OpParams
op'options num_shards :: Int64
num_shards
                                                         shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingStochasticGradientDescentParameters"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall m' . (MonadBuild m') =>
                                                                         Data.Int.Int64 -- ^ __num_shards__
                                                                         -> Data.Int.Int64 -- ^ __shard_id__
                                                                         -> m' ((Tensor Value Float,
                                                                                 Tensor Value Float))
                                                                         -- ^ (__parameters__, __gradient_accumulators__)
                                                                         --
                                                                         -- * __parameters__
                                                                         --
                                                                         -- * __gradient_accumulators__
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug = OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' OpParams
forall a. a -> a
id
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall m' . (MonadBuild m') =>
                                                                          OpParams ->
                                                                          Data.Int.Int64 -- ^ __num_shards__
                                                                          -> Data.Int.Int64 -- ^ __shard_id__
                                                                          -> m' ((Tensor Value Float,
                                                                                  Tensor Value Float))
                                                                          -- ^ (__parameters__, __gradient_accumulators__)
                                                                          --
                                                                          -- * __parameters__
                                                                          --
                                                                          -- * __gradient_accumulators__
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: OpParams
-> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' op'options :: OpParams
op'options
                                                                       num_shards :: Int64
num_shards
                                                                       shard_id :: Int64
shard_id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Float, Tensor Value Float))
-> Build (Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RetrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_shards" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_shards
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shard_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
shard_id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "parameters" type: DT_FLOAT }
output_arg { name: "gradient_accumulators" type: DT_FLOAT }
attr { name: "table_id" type: "int" default_value { i: -1 } }
attr { name: "table_name" type: "string" default_value { s: "" } }
attr { name: "num_shards" type: "int" }
attr { name: "shard_id" type: "int" }
attr { name: "config" type: "string" default_value { s: "" } }
-}
-- | 
reverse :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float), Bool,
                                       Data.ByteString.ByteString,
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word8,
                                       Double, Float] t) =>
           Tensor v'1 t -- ^ __tensor__
           -> Tensor v'2 Bool -- ^ __dims__
           -> Tensor Build t -- ^ __output__
reverse :: Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
reverse = OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32,
    Int64, Int8, Word16, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
reverse' OpParams
forall a. a -> a
id
reverse' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float), Bool,
                                        Data.ByteString.ByteString,
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word8,
                                        Double, Float] t) => OpParams ->
            Tensor v'1 t -- ^ __tensor__
            -> Tensor v'2 Bool -- ^ __dims__
            -> Tensor Build t -- ^ __output__
reverse' :: OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
reverse' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor dims :: Tensor v'2 Bool
dims | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Bool
dims]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Reverse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "dims" type: DT_BOOL }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_BOOL
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_STRING
    }
  }
}
-}
-- | 
reverseSequence :: forall v'1 v'2 t tlen . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tlen) =>
                   Data.Int.Int64 -- ^ __seq_dim__
                   -> Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 tlen -- ^ __seq_lengths__
                   -> Tensor Build t -- ^ __output__
reverseSequence :: Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
reverseSequence = OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
reverseSequence' OpParams
forall a. a -> a
id
reverseSequence' :: forall v'1 v'2 t tlen . (TensorType t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tlen) =>
                    OpParams ->
                    Data.Int.Int64 -- ^ __seq_dim__
                    -> Tensor v'1 t -- ^ __input__
                    -> Tensor v'2 tlen -- ^ __seq_lengths__
                    -> Tensor Build t -- ^ __output__
reverseSequence' :: OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
reverseSequence' op'options :: OpParams
op'options seq_dim :: Int64
seq_dim input :: Tensor v'1 t
input seq_lengths :: Tensor v'2 tlen
seq_lengths | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tlen -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tlen
seq_lengths]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReverseSequence"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tlen" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tlen -> DataType
forall a. TensorType a => a -> DataType
tensorType (tlen
forall a. HasCallStack => a
undefined :: tlen)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "seq_dim" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
seq_dim
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "seq_lengths" type_attr: "Tlen" }
output_arg { name: "output" type_attr: "T" }
attr { name: "seq_dim" type: "int" }
attr { name: "batch_dim" type: "int" default_value { i: 0 } }
attr { name: "T" type: "type" }
attr {
  name: "Tlen"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
reverseV2 :: forall v'1 v'2 tidx t . (OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tidx,
                                      OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Bool, Data.ByteString.ByteString,
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16, Data.Word.Word8,
                                              Double, Float] t) =>
             Tensor v'1 t -- ^ __tensor__
             -> Tensor v'2 tidx -- ^ __axis__
             -> Tensor Build t -- ^ __output__
reverseV2 :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
reverseV2 = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) tidx t.
(OneOf '[Int32, Int64] tidx,
 OneOf
   '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32,
     Int64, Int8, Word16, Word8, Double, Float]
   t) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
reverseV2' OpParams
forall a. a -> a
id
reverseV2' :: forall v'1 v'2 tidx t . (OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tidx,
                                       OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Bool, Data.ByteString.ByteString,
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word8, Double,
                                               Float] t) => OpParams ->
              Tensor v'1 t -- ^ __tensor__
              -> Tensor v'2 tidx -- ^ __axis__
              -> Tensor Build t -- ^ __output__
reverseV2' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
reverseV2' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor axis :: Tensor v'2 tidx
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ReverseV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "axis" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_BOOL
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_STRING
    }
  }
}
-}
-- | 
rightShift :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64,
                                          Data.Word.Word8] t) =>
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 t -- ^ __y__
              -> Tensor Build t -- ^ __z__
rightShift :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rightShift = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rightShift' OpParams
forall a. a -> a
id
rightShift' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64,
                                           Data.Word.Word8] t) => OpParams ->
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __z__
rightShift' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rightShift' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RightShift"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
rint :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
rint :: Tensor v'1 t -> Tensor Build t
rint = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
rint' OpParams
forall a. a -> a
id
rint' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
rint' :: OpParams -> Tensor v'1 t -> Tensor Build t
rint' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Rint"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
rngSkip :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
           Tensor v'1 ResourceHandle -- ^ __resource__
           -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
           -> Tensor v'3 Data.Int.Int64 -- ^ __delta__
           -> m' (ControlNode)
rngSkip :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' ControlNode
rngSkip = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' ControlNode
rngSkip' OpParams
forall a. a -> a
id
rngSkip' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
            Tensor v'1 ResourceHandle -- ^ __resource__
            -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
            -> Tensor v'3 Data.Int.Int64 -- ^ __delta__
            -> m' (ControlNode)
rngSkip' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> m' ControlNode
rngSkip' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm delta :: Tensor v'3 Int64
delta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
delta]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "RngSkip"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "delta" type: DT_INT64 }
-}
-- | 
roll :: forall v'1 v'2 v'3 t tshift taxis . (TensorType t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tshift,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] taxis) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tshift -- ^ __shift__
        -> Tensor v'3 taxis -- ^ __axis__
        -> Tensor Build t -- ^ __output__
roll :: Tensor v'1 t
-> Tensor v'2 tshift -> Tensor v'3 taxis -> Tensor Build t
roll = OpParams
-> Tensor v'1 t
-> Tensor v'2 tshift
-> Tensor v'3 taxis
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       tblock_shape tcrops.
(TensorType t, OneOf '[Int32, Int64] tblock_shape,
 OneOf '[Int32, Int64] tcrops) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
roll' OpParams
forall a. a -> a
id
roll' :: forall v'1 v'2 v'3 t tshift taxis . (TensorType t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tshift,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] taxis) =>
         OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 tshift -- ^ __shift__
         -> Tensor v'3 taxis -- ^ __axis__
         -> Tensor Build t -- ^ __output__
roll' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tshift
-> Tensor v'3 taxis
-> Tensor Build t
roll' op'options :: OpParams
op'options input :: Tensor v'1 t
input shift :: Tensor v'2 tshift
shift axis :: Tensor v'3 taxis
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tshift -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tshift
shift,
                                                             Tensor v'3 taxis -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 taxis
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Roll"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tshift" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tshift -> DataType
forall a. TensorType a => a -> DataType
tensorType (tshift
forall a. HasCallStack => a
undefined :: tshift)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Taxis" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ taxis -> DataType
forall a. TensorType a => a -> DataType
tensorType (taxis
forall a. HasCallStack => a
undefined :: taxis)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "shift" type_attr: "Tshift" }
input_arg { name: "axis" type_attr: "Taxis" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tshift"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Taxis"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
round :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
round :: Tensor v'1 t -> Tensor Build t
round = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
round' OpParams
forall a. a -> a
id
round' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
round' :: OpParams -> Tensor v'1 t -> Tensor Build t
round' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Round"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
rpc :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
       Tensor v'1 Data.ByteString.ByteString -- ^ __address__
       -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
       -> Tensor v'3 Data.ByteString.ByteString -- ^ __request__
       -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __response__
rpc :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value ByteString)
rpc = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value ByteString)
rpc' OpParams
forall a. a -> a
id
rpc' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
        Tensor v'1 Data.ByteString.ByteString -- ^ __address__
        -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
        -> Tensor v'3 Data.ByteString.ByteString -- ^ __request__
        -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __response__
rpc' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value ByteString)
rpc' op'options :: OpParams
op'options address :: Tensor v'1 ByteString
address method :: Tensor v'2 ByteString
method request :: Tensor v'3 ByteString
request | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
address,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
method,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
request]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Rpc"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "address" type: DT_STRING }
input_arg { name: "method" type: DT_STRING }
input_arg { name: "request" type: DT_STRING }
output_arg { name: "response" type: DT_STRING }
attr { name: "protocol" type: "string" default_value { s: "" } }
attr { name: "fail_fast" type: "bool" default_value { b: true } }
attr { name: "timeout_in_ms" type: "int" default_value { i: 0 } }
-}
-- | 
rsqrt :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
rsqrt :: Tensor v'1 t -> Tensor Build t
rsqrt = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
rsqrt' OpParams
forall a. a -> a
id
rsqrt' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float),
                                  Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
rsqrt' :: OpParams -> Tensor v'1 t -> Tensor Build t
rsqrt' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Rsqrt"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
rsqrtGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
             Tensor v'1 t -- ^ __y__
             -> Tensor v'2 t -- ^ __dy__
             -> Tensor Build t -- ^ __z__
rsqrtGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rsqrtGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rsqrtGrad' OpParams
forall a. a -> a
id
rsqrtGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Word.Word16, Double, Float] t) =>
              OpParams ->
              Tensor v'1 t -- ^ __y__
              -> Tensor v'2 t -- ^ __dy__
              -> Tensor Build t -- ^ __z__
rsqrtGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
rsqrtGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "RsqrtGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sampleDistortedBoundingBox :: forall v'1 v'2 t m' . (MonadBuild m',
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word8] t) =>
                              Tensor v'1 t -- ^ __image_size__
                              -> Tensor v'2 Float -- ^ __bounding_boxes__
                              -> m' ((Tensor Value t, Tensor Value t,
                                      Tensor Value Float))
                              -- ^ (__begin__, __size__, __bboxes__)
                              --
                              -- * __begin__
                              --
                              -- * __size__
                              --
                              -- * __bboxes__
sampleDistortedBoundingBox :: Tensor v'1 t
-> Tensor v'2 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBox = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBox' OpParams
forall a. a -> a
id
sampleDistortedBoundingBox' :: forall v'1 v'2 t m' . (MonadBuild m',
                                                      OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word8] t) =>
                               OpParams ->
                               Tensor v'1 t -- ^ __image_size__
                               -> Tensor v'2 Float -- ^ __bounding_boxes__
                               -> m' ((Tensor Value t, Tensor Value t,
                                       Tensor Value Float))
                               -- ^ (__begin__, __size__, __bboxes__)
                               --
                               -- * __begin__
                               --
                               -- * __size__
                               --
                               -- * __bboxes__
sampleDistortedBoundingBox' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBox' op'options :: OpParams
op'options image_size :: Tensor v'1 t
image_size
                            bounding_boxes :: Tensor v'2 Float
bounding_boxes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t, Tensor Value t, Tensor Value Float)
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t, Tensor Value t, Tensor Value Float)
 -> m' (Tensor Value t, Tensor Value t, Tensor Value Float))
-> Build (Tensor Value t, Tensor Value t, Tensor Value Float)
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
image_size,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
bounding_boxes]
        [Int64]
-> OpDef
-> Build (Tensor Value t, Tensor Value t, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SampleDistortedBoundingBox"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image_size" type_attr: "T" }
input_arg { name: "bounding_boxes" type: DT_FLOAT }
output_arg { name: "begin" type_attr: "T" }
output_arg { name: "size" type_attr: "T" }
output_arg { name: "bboxes" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "min_object_covered" type: "float" default_value { f: 0.1 }
}
attr {
  name: "aspect_ratio_range"
  type: "list(float)"
  default_value { list { f: 0.75 f: 1.33 } }
}
attr {
  name: "area_range"
  type: "list(float)"
  default_value { list { f: 5.0e-2 f: 1.0 } }
}
attr { name: "max_attempts" type: "int" default_value { i: 100 } }
attr {
  name: "use_image_if_no_bounding_boxes"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
sampleDistortedBoundingBoxV2 :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                           OneOf '[Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Int.Int64,
                                                                   Data.Int.Int8,
                                                                   Data.Word.Word8] t) =>
                                Tensor v'1 t -- ^ __image_size__
                                -> Tensor v'2 Float -- ^ __bounding_boxes__
                                -> Tensor v'3 Float -- ^ __min_object_covered__
                                -> m' ((Tensor Value t, Tensor Value t,
                                        Tensor Value Float))
                                -- ^ (__begin__, __size__, __bboxes__)
                                --
                                -- * __begin__
                                --
                                -- * __size__
                                --
                                -- * __bboxes__
sampleDistortedBoundingBoxV2 :: Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBoxV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBoxV2' OpParams
forall a. a -> a
id
sampleDistortedBoundingBoxV2' :: forall v'1 v'2 v'3 t m' . (MonadBuild m',
                                                            OneOf '[Data.Int.Int16,
                                                                    Data.Int.Int32,
                                                                    Data.Int.Int64,
                                                                    Data.Int.Int8,
                                                                    Data.Word.Word8] t) =>
                                 OpParams ->
                                 Tensor v'1 t -- ^ __image_size__
                                 -> Tensor v'2 Float -- ^ __bounding_boxes__
                                 -> Tensor v'3 Float -- ^ __min_object_covered__
                                 -> m' ((Tensor Value t, Tensor Value t,
                                         Tensor Value Float))
                                 -- ^ (__begin__, __size__, __bboxes__)
                                 --
                                 -- * __begin__
                                 --
                                 -- * __size__
                                 --
                                 -- * __bboxes__
sampleDistortedBoundingBoxV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Float
-> Tensor v'3 Float
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
sampleDistortedBoundingBoxV2' op'options :: OpParams
op'options image_size :: Tensor v'1 t
image_size bounding_boxes :: Tensor v'2 Float
bounding_boxes
                              min_object_covered :: Tensor v'3 Float
min_object_covered | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t, Tensor Value t, Tensor Value Float)
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t, Tensor Value t, Tensor Value Float)
 -> m' (Tensor Value t, Tensor Value t, Tensor Value Float))
-> Build (Tensor Value t, Tensor Value t, Tensor Value Float)
-> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
image_size,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
bounding_boxes,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
min_object_covered]
        [Int64]
-> OpDef
-> Build (Tensor Value t, Tensor Value t, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SampleDistortedBoundingBoxV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "image_size" type_attr: "T" }
input_arg { name: "bounding_boxes" type: DT_FLOAT }
input_arg { name: "min_object_covered" type: DT_FLOAT }
output_arg { name: "begin" type_attr: "T" }
output_arg { name: "size" type_attr: "T" }
output_arg { name: "bboxes" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_UINT8
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "aspect_ratio_range"
  type: "list(float)"
  default_value { list { f: 0.75 f: 1.33 } }
}
attr {
  name: "area_range"
  type: "list(float)"
  default_value { list { f: 5.0e-2 f: 1.0 } }
}
attr { name: "max_attempts" type: "int" default_value { i: 100 } }
attr {
  name: "use_image_if_no_bounding_boxes"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
samplingDataset :: [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Float -- ^ __rate__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                   -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                   -> Tensor Build Variant -- ^ __handle__
samplingDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
samplingDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
samplingDataset' OpParams
forall a. a -> a
id
samplingDataset' :: OpParams ->
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Float -- ^ __rate__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                    -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                    -> Tensor Build Variant -- ^ __handle__
samplingDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Float
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
samplingDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset rate :: Tensor v'2 Float
rate seed :: Tensor v'3 Int64
seed
                 seed2 :: Tensor v'4 Int64
seed2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
rate,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
seed2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SamplingDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "rate" type: DT_FLOAT }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
save :: forall v'1 v'2 v'3 t m' . (MonadBuild m', TensorTypes t) =>
        Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
        -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
        -> TensorList (v'3) t -- ^ __data__
        -> m' (ControlNode)
save :: Tensor v'1 ByteString
-> Tensor v'2 ByteString -> TensorList v'3 t -> m' ControlNode
save = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> TensorList v'3 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) (t :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes t) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> TensorList v'3 t
-> m' ControlNode
save' OpParams
forall a. a -> a
id
save' :: forall v'1 v'2 v'3 t m' . (MonadBuild m', TensorTypes t) => OpParams ->
         Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
         -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
         -> TensorList (v'3) t -- ^ __data__
         -> m' (ControlNode)
save' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> TensorList v'3 t
-> m' ControlNode
save' op'options :: OpParams
op'options filename :: Tensor v'1 ByteString
filename tensor_names :: Tensor v'2 ByteString
tensor_names data' :: TensorList v'3 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filename,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_names,
                                                             TensorList v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'3 t
data']
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Save"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "tensor_names" type: DT_STRING }
input_arg { name: "data" type_list_attr: "T" }
attr { name: "T" type: "list(type)" has_minimum: true minimum: 1 }
-}
-- | 
saveSlices :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m', TensorTypes t) =>
              Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
              -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
              -> Tensor v'3 Data.ByteString.ByteString -- ^ __shapes_and_slices__
              -> TensorList (v'4) t -- ^ __data__
              -> m' (ControlNode)
saveSlices :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 t
-> m' ControlNode
saveSlices = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (t :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes t) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 t
-> m' ControlNode
saveSlices' OpParams
forall a. a -> a
id
saveSlices' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m', TensorTypes t) =>
               OpParams ->
               Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
               -> Tensor v'3 Data.ByteString.ByteString -- ^ __shapes_and_slices__
               -> TensorList (v'4) t -- ^ __data__
               -> m' (ControlNode)
saveSlices' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 t
-> m' ControlNode
saveSlices' op'options :: OpParams
op'options filename :: Tensor v'1 ByteString
filename tensor_names :: Tensor v'2 ByteString
tensor_names shapes_and_slices :: Tensor v'3 ByteString
shapes_and_slices
            data' :: TensorList v'4 t
data' | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filename,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_names,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
shapes_and_slices,
                                                             TensorList v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 t
data']
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SaveSlices"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "tensor_names" type: DT_STRING }
input_arg { name: "shapes_and_slices" type: DT_STRING }
input_arg { name: "data" type_list_attr: "T" }
attr { name: "T" type: "list(type)" has_minimum: true minimum: 1 }
-}
-- | 
saveV2 :: forall v'1 v'2 v'3 v'4 dtypes m' . (MonadBuild m',
                                              TensorTypes dtypes) =>
          Tensor v'1 Data.ByteString.ByteString -- ^ __prefix__
          -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
          -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slices__
          -> TensorList (v'4) dtypes -- ^ __tensors__
          -> m' (ControlNode)
saveV2 :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 dtypes
-> m' ControlNode
saveV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 dtypes
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (t :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes t) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 t
-> m' ControlNode
saveV2' OpParams
forall a. a -> a
id
saveV2' :: forall v'1 v'2 v'3 v'4 dtypes m' . (MonadBuild m',
                                               TensorTypes dtypes) =>
           OpParams ->
           Tensor v'1 Data.ByteString.ByteString -- ^ __prefix__
           -> Tensor v'2 Data.ByteString.ByteString -- ^ __tensor_names__
           -> Tensor v'3 Data.ByteString.ByteString -- ^ __shape_and_slices__
           -> TensorList (v'4) dtypes -- ^ __tensors__
           -> m' (ControlNode)
saveV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> TensorList v'4 dtypes
-> m' ControlNode
saveV2' op'options :: OpParams
op'options prefix :: Tensor v'1 ByteString
prefix tensor_names :: Tensor v'2 ByteString
tensor_names shape_and_slices :: Tensor v'3 ByteString
shape_and_slices
        tensors :: TensorList v'4 dtypes
tensors | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
prefix,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tensor_names,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
shape_and_slices,
                                                             TensorList v'4 dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 dtypes
tensors]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SaveV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "prefix" type: DT_STRING }
input_arg { name: "tensor_names" type: DT_STRING }
input_arg { name: "shape_and_slices" type: DT_STRING }
input_arg { name: "tensors" type_list_attr: "dtypes" }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | 
scalarSummary :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t) =>
                 Tensor v'1 Data.ByteString.ByteString -- ^ __tags__
                 -> Tensor v'2 t -- ^ __values__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
scalarSummary :: Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
scalarSummary = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
scalarSummary' OpParams
forall a. a -> a
id
scalarSummary' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t) => OpParams ->
                  Tensor v'1 Data.ByteString.ByteString -- ^ __tags__
                  -> Tensor v'2 t -- ^ __values__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
scalarSummary' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
scalarSummary' op'options :: OpParams
op'options tags :: Tensor v'1 ByteString
tags values :: Tensor v'2 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tags,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ScalarSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tags" type: DT_STRING }
input_arg { name: "values" type_attr: "T" }
output_arg { name: "summary" type: DT_STRING }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
scaleAndTranslate :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word8,
                                                         Double, Float] t) =>
                     Tensor v'1 t -- ^ __images__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                     -> Tensor v'3 Float -- ^ __scale__
                     -> Tensor v'4 Float -- ^ __translation__
                     -> Tensor Build Float -- ^ __resized_images__
scaleAndTranslate :: Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
scaleAndTranslate = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
scaleAndTranslate' OpParams
forall a. a -> a
id
scaleAndTranslate' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Int.Int8,
                                                          Data.Word.Word16,
                                                          Data.Word.Word8,
                                                          Double, Float] t) =>
                      OpParams ->
                      Tensor v'1 t -- ^ __images__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                      -> Tensor v'3 Float -- ^ __scale__
                      -> Tensor v'4 Float -- ^ __translation__
                      -> Tensor Build Float -- ^ __resized_images__
scaleAndTranslate' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build Float
scaleAndTranslate' op'options :: OpParams
op'options images :: Tensor v'1 t
images size :: Tensor v'2 Int32
size scale :: Tensor v'3 Float
scale translation :: Tensor v'4 Float
translation | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
images,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
scale,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
translation]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ScaleAndTranslate"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "images" type_attr: "T" }
input_arg { name: "size" type: DT_INT32 }
input_arg { name: "scale" type: DT_FLOAT }
input_arg { name: "translation" type: DT_FLOAT }
output_arg { name: "resized_images" type: DT_FLOAT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_UINT8
      type: DT_INT16
      type: DT_UINT16
      type: DT_INT32
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
attr {
  name: "kernel_type" type: "string" default_value { s: "lanczos3" }
}
attr { name: "antialias" type: "bool" default_value { b: true } }
-}
-- | 
scaleAndTranslateGrad :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Float] t) =>
                         Tensor v'1 t -- ^ __grads__
                         -> Tensor v'2 t -- ^ __original_image__
                         -> Tensor v'3 Float -- ^ __scale__
                         -> Tensor v'4 Float -- ^ __translation__
                         -> Tensor Build t -- ^ __output__
scaleAndTranslateGrad :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build t
scaleAndTranslateGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build t
scaleAndTranslateGrad' OpParams
forall a. a -> a
id
scaleAndTranslateGrad' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Float] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __grads__
                          -> Tensor v'2 t -- ^ __original_image__
                          -> Tensor v'3 Float -- ^ __scale__
                          -> Tensor v'4 Float -- ^ __translation__
                          -> Tensor Build t -- ^ __output__
scaleAndTranslateGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Float
-> Tensor v'4 Float
-> Tensor Build t
scaleAndTranslateGrad' op'options :: OpParams
op'options grads :: Tensor v'1 t
grads original_image :: Tensor v'2 t
original_image scale :: Tensor v'3 Float
scale
                       translation :: Tensor v'4 Float
translation | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
grads,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
original_image,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
scale,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
translation]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ScaleAndTranslateGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grads" type_attr: "T" }
input_arg { name: "original_image" type_attr: "T" }
input_arg { name: "scale" type: DT_FLOAT }
input_arg { name: "translation" type: DT_FLOAT }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr {
  name: "kernel_type" type: "string" default_value { s: "lanczos3" }
}
attr { name: "antialias" type: "bool" default_value { b: true } }
-}
-- | 
scatterAdd :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterAdd :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterAdd = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterAdd' OpParams
forall a. a -> a
id
scatterAdd' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterAdd' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterAdd' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterAdd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterDiv :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterDiv :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterDiv = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterDiv' OpParams
forall a. a -> a
id
scatterDiv' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterDiv' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterDiv' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterDiv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterMax :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Word.Word16, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMax :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterMax = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMax' OpParams
forall a. a -> a
id
scatterMax' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Word.Word16, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMax' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMax' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterMin :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Word.Word16, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMin :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterMin = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMin' OpParams
forall a. a -> a
id
scatterMin' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Word.Word16, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMin' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMin' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterMin"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_BFLOAT16
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterMul :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMul :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterMul = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMul' OpParams
forall a. a -> a
id
scatterMul' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterMul' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterMul' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterMul"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterNd :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
             Tensor v'1 tindices -- ^ __indices__
             -> Tensor v'2 t -- ^ __updates__
             -> Tensor v'3 tindices -- ^ __shape__
             -> Tensor Build t -- ^ __output__
scatterNd :: Tensor v'1 tindices
-> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
scatterNd = OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices.
(TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
scatterNd' OpParams
forall a. a -> a
id
scatterNd' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
              OpParams ->
              Tensor v'1 tindices -- ^ __indices__
              -> Tensor v'2 t -- ^ __updates__
              -> Tensor v'3 tindices -- ^ __shape__
              -> Tensor Build t -- ^ __output__
scatterNd' :: OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
scatterNd' op'options :: OpParams
op'options indices :: Tensor v'1 tindices
indices updates :: Tensor v'2 t
updates shape :: Tensor v'3 tindices
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tindices
indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
updates,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ScatterNd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
input_arg { name: "shape" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
scatterNdAdd :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                Tensor Ref t -- ^ __ref__
                -> Tensor v'2 tindices -- ^ __indices__
                -> Tensor v'3 t -- ^ __updates__
                -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdAdd :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterNdAdd = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdAdd' OpParams
forall a. a -> a
id
scatterNdAdd' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindices) =>
                 OpParams ->
                 Tensor Ref t -- ^ __ref__
                 -> Tensor v'2 tindices -- ^ __indices__
                 -> Tensor v'3 t -- ^ __updates__
                 -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdAdd' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdAdd' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterNdAdd"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterNdMax :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                Tensor Ref t -- ^ __ref__
                -> Tensor v'2 tindices -- ^ __indices__
                -> Tensor v'3 t -- ^ __updates__
                -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdMax :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterNdMax = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdMax' OpParams
forall a. a -> a
id
scatterNdMax' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindices) =>
                 OpParams ->
                 Tensor Ref t -- ^ __ref__
                 -> Tensor v'2 tindices -- ^ __indices__
                 -> Tensor v'3 t -- ^ __updates__
                 -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdMax' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdMax' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterNdMax"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterNdMin :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                Tensor Ref t -- ^ __ref__
                -> Tensor v'2 tindices -- ^ __indices__
                -> Tensor v'3 t -- ^ __updates__
                -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdMin :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterNdMin = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdMin' OpParams
forall a. a -> a
id
scatterNdMin' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindices) =>
                 OpParams ->
                 Tensor Ref t -- ^ __ref__
                 -> Tensor v'2 tindices -- ^ __indices__
                 -> Tensor v'3 t -- ^ __updates__
                 -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdMin' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdMin' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterNdMin"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterNdNonAliasingAdd :: forall v'1 v'2 v'3 t
                           tindices . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Bool, Data.Int.Int16,
                                               Data.Int.Int32, Data.Int.Int64,
                                               Data.Int.Int8, Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices) =>
                           Tensor v'1 t -- ^ __input__
                           -> Tensor v'2 tindices -- ^ __indices__
                           -> Tensor v'3 t -- ^ __updates__
                           -> Tensor Build t -- ^ __output__
scatterNdNonAliasingAdd :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
scatterNdNonAliasingAdd = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices.
(OneOf
   '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
     Word16, Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
scatterNdNonAliasingAdd' OpParams
forall a. a -> a
id
scatterNdNonAliasingAdd' :: forall v'1 v'2 v'3 t
                            tindices . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Bool, Data.Int.Int16,
                                                Data.Int.Int32, Data.Int.Int64,
                                                Data.Int.Int8, Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tindices) =>
                            OpParams ->
                            Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 tindices -- ^ __indices__
                            -> Tensor v'3 t -- ^ __updates__
                            -> Tensor Build t -- ^ __output__
scatterNdNonAliasingAdd' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
scatterNdNonAliasingAdd' op'options :: OpParams
op'options input :: Tensor v'1 t
input indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ScatterNdNonAliasingAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_BOOL
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
scatterNdSub :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                OneOf '[(Data.Complex.Complex Double),
                                                        (Data.Complex.Complex Float),
                                                        Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t,
                                                OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                Tensor Ref t -- ^ __ref__
                -> Tensor v'2 tindices -- ^ __indices__
                -> Tensor v'3 t -- ^ __updates__
                -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdSub :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterNdSub = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdSub' OpParams
forall a. a -> a
id
scatterNdSub' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                                 OneOf '[(Data.Complex.Complex Double),
                                                         (Data.Complex.Complex Float),
                                                         Data.Int.Int16,
                                                         Data.Int.Int32,
                                                         Data.Int.Int64,
                                                         Data.Int.Int8,
                                                         Data.Word.Word16,
                                                         Data.Word.Word32,
                                                         Data.Word.Word64,
                                                         Data.Word.Word8,
                                                         Double, Float] t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindices) =>
                 OpParams ->
                 Tensor Ref t -- ^ __ref__
                 -> Tensor v'2 tindices -- ^ __indices__
                 -> Tensor v'3 t -- ^ __updates__
                 -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdSub' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdSub' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterNdSub"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterNdUpdate :: forall v'2 v'3 t tindices m' . (MonadBuild m', TensorType t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tindices) =>
                   Tensor Ref t -- ^ __ref__
                   -> Tensor v'2 tindices -- ^ __indices__
                   -> Tensor v'3 t -- ^ __updates__
                   -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdUpdate :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterNdUpdate = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdUpdate' OpParams
forall a. a -> a
id
scatterNdUpdate' :: forall v'2 v'3 t tindices m' . (MonadBuild m', TensorType t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tindices) =>
                    OpParams ->
                    Tensor Ref t -- ^ __ref__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> Tensor v'3 t -- ^ __updates__
                    -> m' (Tensor Ref t) -- ^ __output_ref__
scatterNdUpdate' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterNdUpdate' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterNdUpdate"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
scatterSub :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                              OneOf '[(Data.Complex.Complex Double),
                                                      (Data.Complex.Complex Float),
                                                      Data.Int.Int16,
                                                      Data.Int.Int32,
                                                      Data.Int.Int64,
                                                      Data.Int.Int8,
                                                      Data.Word.Word16,
                                                      Data.Word.Word32,
                                                      Data.Word.Word64,
                                                      Data.Word.Word8, Double,
                                                      Float] t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
              Tensor Ref t -- ^ __ref__
              -> Tensor v'2 tindices -- ^ __indices__
              -> Tensor v'3 t -- ^ __updates__
              -> m' (Tensor Ref t) -- ^ __output_ref__
scatterSub :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterSub = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterSub' OpParams
forall a. a -> a
id
scatterSub' :: forall v'2 v'3 t tindices m' . (MonadBuild m',
                                               OneOf '[(Data.Complex.Complex Double),
                                                       (Data.Complex.Complex Float),
                                                       Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor Ref t -- ^ __ref__
               -> Tensor v'2 tindices -- ^ __indices__
               -> Tensor v'3 t -- ^ __updates__
               -> m' (Tensor Ref t) -- ^ __output_ref__
scatterSub' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterSub' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterSub"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
scatterUpdate :: forall v'2 v'3 t tindices m' . (MonadBuild m', TensorType t,
                                                 OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] tindices) =>
                 Tensor Ref t -- ^ __ref__
                 -> Tensor v'2 tindices -- ^ __indices__
                 -> Tensor v'3 t -- ^ __updates__
                 -> m' (Tensor Ref t) -- ^ __output_ref__
scatterUpdate :: Tensor Ref t
-> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
scatterUpdate = OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterUpdate' OpParams
forall a. a -> a
id
scatterUpdate' :: forall v'2 v'3 t tindices m' . (MonadBuild m', TensorType t,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tindices) =>
                  OpParams ->
                  Tensor Ref t -- ^ __ref__
                  -> Tensor v'2 tindices -- ^ __indices__
                  -> Tensor v'3 t -- ^ __updates__
                  -> m' (Tensor Ref t) -- ^ __output_ref__
scatterUpdate' :: OpParams
-> Tensor Ref t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> m' (Tensor Ref t)
scatterUpdate' op'options :: OpParams
op'options ref :: Tensor Ref t
ref indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ScatterUpdate"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "use_locking" type: "bool" default_value { b: true } }
-}
-- | 
sdcaFprint :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
              -> Tensor Build Data.Int.Int64 -- ^ __output__
sdcaFprint :: Tensor v'1 ByteString -> Tensor Build Int64
sdcaFprint = OpParams -> Tensor v'1 ByteString -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Int64
sdcaFprint' OpParams
forall a. a -> a
id
sdcaFprint' :: OpParams ->
               Tensor v'1 Data.ByteString.ByteString -- ^ __input__
               -> Tensor Build Data.Int.Int64 -- ^ __output__
sdcaFprint' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int64
sdcaFprint' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SdcaFprint"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_INT64 }
-}
-- | 
sdcaOptimizer :: Float -- ^ __l1__
                 -> Float -- ^ __l2__
                 -> ByteString -- ^ __loss_type__
                 -> Data.Int.Int64 -- ^ __num_inner_iterations__
                 -> Data.Int.Int64 -- ^ __num_loss_partitions__
                 -> [Tensor v'1 Data.Int.Int64] -- ^ __sparse_example_indices__
                 -> [Tensor v'2 Data.Int.Int64] -- ^ __sparse_feature_indices__
                 -> [Tensor v'3 Float] -- ^ __sparse_feature_values__
                 -> [Tensor v'4 Float] -- ^ __dense_features__
                 -> Tensor v'5 Float -- ^ __example_weights__
                 -> Tensor v'6 Float -- ^ __example_labels__
                 -> [Tensor v'7 Data.Int.Int64] -- ^ __sparse_indices__
                 -> [Tensor v'8 Float] -- ^ __sparse_weights__
                 -> [Tensor v'9 Float] -- ^ __dense_weights__
                 -> Tensor v'10 Float -- ^ __example_state_data__
                 -> (Tensor Build Float, [Tensor Build Float],
                     [Tensor Build Float])
                 -- ^ (__out_example_state_data__, __out_delta_sparse_weights__, __out_delta_dense_weights__)
                 --
                 -- * __out_example_state_data__
                 --
                 -- * __out_delta_sparse_weights__
                 --
                 -- * __out_delta_dense_weights__
sdcaOptimizer :: Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizer = OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *).
OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizer' OpParams
forall a. a -> a
id
sdcaOptimizer' :: OpParams ->
                  Float -- ^ __l1__
                  -> Float -- ^ __l2__
                  -> ByteString -- ^ __loss_type__
                  -> Data.Int.Int64 -- ^ __num_inner_iterations__
                  -> Data.Int.Int64 -- ^ __num_loss_partitions__
                  -> [Tensor v'1 Data.Int.Int64] -- ^ __sparse_example_indices__
                  -> [Tensor v'2 Data.Int.Int64] -- ^ __sparse_feature_indices__
                  -> [Tensor v'3 Float] -- ^ __sparse_feature_values__
                  -> [Tensor v'4 Float] -- ^ __dense_features__
                  -> Tensor v'5 Float -- ^ __example_weights__
                  -> Tensor v'6 Float -- ^ __example_labels__
                  -> [Tensor v'7 Data.Int.Int64] -- ^ __sparse_indices__
                  -> [Tensor v'8 Float] -- ^ __sparse_weights__
                  -> [Tensor v'9 Float] -- ^ __dense_weights__
                  -> Tensor v'10 Float -- ^ __example_state_data__
                  -> (Tensor Build Float, [Tensor Build Float],
                      [Tensor Build Float])
                  -- ^ (__out_example_state_data__, __out_delta_sparse_weights__, __out_delta_dense_weights__)
                  --
                  -- * __out_example_state_data__
                  --
                  -- * __out_delta_sparse_weights__
                  --
                  -- * __out_delta_dense_weights__
sdcaOptimizer' :: OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizer' op'options :: OpParams
op'options l1 :: Float
l1 l2 :: Float
l2 loss_type :: ByteString
loss_type num_inner_iterations :: Int64
num_inner_iterations
               num_loss_partitions :: Int64
num_loss_partitions sparse_example_indices :: [Tensor v'1 Int64]
sparse_example_indices sparse_feature_indices :: [Tensor v'2 Int64]
sparse_feature_indices
               sparse_feature_values :: [Tensor v'3 Float]
sparse_feature_values dense_features :: [Tensor v'4 Float]
dense_features example_weights :: Tensor v'5 Float
example_weights
               example_labels :: Tensor v'6 Float
example_labels sparse_indices :: [Tensor v'7 Int64]
sparse_indices sparse_weights :: [Tensor v'8 Float]
sparse_weights dense_weights :: [Tensor v'9 Float]
dense_weights
               example_state_data :: Tensor v'10 Float
example_state_data | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_sparse_features", [("sparse_example_indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
sparse_example_indices),
                                                                            ("sparse_feature_indices", [Tensor v'2 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int64]
sparse_feature_indices),
                                                                            ("sparse_indices", [Tensor v'7 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'7 Int64]
sparse_indices),
                                                                            ("sparse_weights", [Tensor v'8 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'8 Float]
sparse_weights)]),
                                                   ("num_sparse_features_with_values", [("sparse_feature_values", [Tensor v'3 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Float]
sparse_feature_values)]),
                                                   ("num_dense_features", [("dense_features", [Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
dense_features),
                                                                           ("dense_weights", [Tensor v'9 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'9 Float]
dense_weights)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_sparse_features, Int64
num_dense_features] (Build OpDef
 -> (Tensor Build Float, [Tensor Build Float],
     [Tensor Build Float]))
-> Build OpDef
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
sparse_example_indices,
                                                             [Tensor v'2 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int64]
sparse_feature_indices,
                                                             [Tensor v'3 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Float]
sparse_feature_values,
                                                             [Tensor v'4 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Float]
dense_features,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
example_weights,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
example_labels,
                                                             [Tensor v'7 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'7 Int64]
sparse_indices,
                                                             [Tensor v'8 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'8 Float]
sparse_weights,
                                                             [Tensor v'9 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'9 Float]
dense_weights,
                                                             Tensor v'10 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 Float
example_state_data]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SdcaOptimizer"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l1" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l1
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l2" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l2
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "loss_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
loss_type
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_inner_iterations" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_inner_iterations
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_loss_partitions" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_loss_partitions
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse_features_with_values" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse_features_with_values
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_dense_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_dense_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_sparse_features :: Int64
num_sparse_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
sparse_example_indices) :: Int64
    num_sparse_features_with_values :: Int64
num_sparse_features_with_values = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Float]
sparse_feature_values) :: Int64
    num_dense_features :: Int64
num_dense_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
dense_features) :: Int64
{-
input_arg {
  name: "sparse_example_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_feature_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_feature_values"
  type: DT_FLOAT
  number_attr: "num_sparse_features_with_values"
}
input_arg {
  name: "dense_features"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
input_arg { name: "example_weights" type: DT_FLOAT }
input_arg { name: "example_labels" type: DT_FLOAT }
input_arg {
  name: "sparse_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_weights"
  type: DT_FLOAT
  number_attr: "num_sparse_features"
}
input_arg {
  name: "dense_weights"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
input_arg { name: "example_state_data" type: DT_FLOAT }
output_arg { name: "out_example_state_data" type: DT_FLOAT }
output_arg {
  name: "out_delta_sparse_weights"
  type: DT_FLOAT
  number_attr: "num_sparse_features"
}
output_arg {
  name: "out_delta_dense_weights"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
attr {
  name: "loss_type"
  type: "string"
  allowed_values {
    list {
      s: "logistic_loss"
      s: "squared_loss"
      s: "hinge_loss"
      s: "smooth_hinge_loss"
      s: "poisson_loss"
    }
  }
}
attr { name: "adaptative" type: "bool" default_value { b: false } }
attr { name: "num_sparse_features" type: "int" has_minimum: true }
attr {
  name: "num_sparse_features_with_values"
  type: "int"
  has_minimum: true
}
attr { name: "num_dense_features" type: "int" has_minimum: true }
attr { name: "l1" type: "float" }
attr { name: "l2" type: "float" }
attr {
  name: "num_loss_partitions"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr {
  name: "num_inner_iterations"
  type: "int"
  has_minimum: true
  minimum: 1
}
-}
-- | 
sdcaOptimizerV2 :: Float -- ^ __l1__
                   -> Float -- ^ __l2__
                   -> ByteString -- ^ __loss_type__
                   -> Data.Int.Int64 -- ^ __num_inner_iterations__
                   -> Data.Int.Int64 -- ^ __num_loss_partitions__
                   -> [Tensor v'1 Data.Int.Int64] -- ^ __sparse_example_indices__
                   -> [Tensor v'2 Data.Int.Int64] -- ^ __sparse_feature_indices__
                   -> [Tensor v'3 Float] -- ^ __sparse_feature_values__
                   -> [Tensor v'4 Float] -- ^ __dense_features__
                   -> Tensor v'5 Float -- ^ __example_weights__
                   -> Tensor v'6 Float -- ^ __example_labels__
                   -> [Tensor v'7 Data.Int.Int64] -- ^ __sparse_indices__
                   -> [Tensor v'8 Float] -- ^ __sparse_weights__
                   -> [Tensor v'9 Float] -- ^ __dense_weights__
                   -> Tensor v'10 Float -- ^ __example_state_data__
                   -> (Tensor Build Float, [Tensor Build Float],
                       [Tensor Build Float])
                   -- ^ (__out_example_state_data__, __out_delta_sparse_weights__, __out_delta_dense_weights__)
                   --
                   -- * __out_example_state_data__
                   --
                   -- * __out_delta_sparse_weights__
                   --
                   -- * __out_delta_dense_weights__
sdcaOptimizerV2 :: Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizerV2 = OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *).
OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizerV2' OpParams
forall a. a -> a
id
sdcaOptimizerV2' :: OpParams ->
                    Float -- ^ __l1__
                    -> Float -- ^ __l2__
                    -> ByteString -- ^ __loss_type__
                    -> Data.Int.Int64 -- ^ __num_inner_iterations__
                    -> Data.Int.Int64 -- ^ __num_loss_partitions__
                    -> [Tensor v'1 Data.Int.Int64] -- ^ __sparse_example_indices__
                    -> [Tensor v'2 Data.Int.Int64] -- ^ __sparse_feature_indices__
                    -> [Tensor v'3 Float] -- ^ __sparse_feature_values__
                    -> [Tensor v'4 Float] -- ^ __dense_features__
                    -> Tensor v'5 Float -- ^ __example_weights__
                    -> Tensor v'6 Float -- ^ __example_labels__
                    -> [Tensor v'7 Data.Int.Int64] -- ^ __sparse_indices__
                    -> [Tensor v'8 Float] -- ^ __sparse_weights__
                    -> [Tensor v'9 Float] -- ^ __dense_weights__
                    -> Tensor v'10 Float -- ^ __example_state_data__
                    -> (Tensor Build Float, [Tensor Build Float],
                        [Tensor Build Float])
                    -- ^ (__out_example_state_data__, __out_delta_sparse_weights__, __out_delta_dense_weights__)
                    --
                    -- * __out_example_state_data__
                    --
                    -- * __out_delta_sparse_weights__
                    --
                    -- * __out_delta_dense_weights__
sdcaOptimizerV2' :: OpParams
-> Float
-> Float
-> ByteString
-> Int64
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 Int64]
-> [Tensor v'3 Float]
-> [Tensor v'4 Float]
-> Tensor v'5 Float
-> Tensor v'6 Float
-> [Tensor v'7 Int64]
-> [Tensor v'8 Float]
-> [Tensor v'9 Float]
-> Tensor v'10 Float
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
sdcaOptimizerV2' op'options :: OpParams
op'options l1 :: Float
l1 l2 :: Float
l2 loss_type :: ByteString
loss_type num_inner_iterations :: Int64
num_inner_iterations
                 num_loss_partitions :: Int64
num_loss_partitions sparse_example_indices :: [Tensor v'1 Int64]
sparse_example_indices
                 sparse_feature_indices :: [Tensor v'2 Int64]
sparse_feature_indices sparse_feature_values :: [Tensor v'3 Float]
sparse_feature_values dense_features :: [Tensor v'4 Float]
dense_features
                 example_weights :: Tensor v'5 Float
example_weights example_labels :: Tensor v'6 Float
example_labels sparse_indices :: [Tensor v'7 Int64]
sparse_indices sparse_weights :: [Tensor v'8 Float]
sparse_weights
                 dense_weights :: [Tensor v'9 Float]
dense_weights
                 example_state_data :: Tensor v'10 Float
example_state_data | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_sparse_features", [("sparse_example_indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
sparse_example_indices),
                                                                              ("sparse_feature_indices", [Tensor v'2 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Int64]
sparse_feature_indices),
                                                                              ("sparse_indices", [Tensor v'7 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'7 Int64]
sparse_indices),
                                                                              ("sparse_weights", [Tensor v'8 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'8 Float]
sparse_weights)]),
                                                     ("num_sparse_features_with_values", [("sparse_feature_values", [Tensor v'3 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Float]
sparse_feature_values)]),
                                                     ("num_dense_features", [("dense_features", [Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
dense_features),
                                                                             ("dense_weights", [Tensor v'9 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'9 Float]
dense_weights)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_sparse_features, Int64
num_dense_features] (Build OpDef
 -> (Tensor Build Float, [Tensor Build Float],
     [Tensor Build Float]))
-> Build OpDef
-> (Tensor Build Float, [Tensor Build Float], [Tensor Build Float])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
sparse_example_indices,
                                                             [Tensor v'2 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Int64]
sparse_feature_indices,
                                                             [Tensor v'3 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Float]
sparse_feature_values,
                                                             [Tensor v'4 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'4 Float]
dense_features,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
example_weights,
                                                             Tensor v'6 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Float
example_labels,
                                                             [Tensor v'7 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'7 Int64]
sparse_indices,
                                                             [Tensor v'8 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'8 Float]
sparse_weights,
                                                             [Tensor v'9 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'9 Float]
dense_weights,
                                                             Tensor v'10 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 Float
example_state_data]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SdcaOptimizerV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l1" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l1
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l2" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l2
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "loss_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
loss_type
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_inner_iterations" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_inner_iterations
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_loss_partitions" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_loss_partitions
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sparse_features_with_values" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sparse_features_with_values
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_dense_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_dense_features
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_sparse_features :: Int64
num_sparse_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
sparse_example_indices) :: Int64
    num_sparse_features_with_values :: Int64
num_sparse_features_with_values = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Float]
sparse_feature_values) :: Int64
    num_dense_features :: Int64
num_dense_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'4 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'4 Float]
dense_features) :: Int64
{-
input_arg {
  name: "sparse_example_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_feature_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_feature_values"
  type: DT_FLOAT
  number_attr: "num_sparse_features_with_values"
}
input_arg {
  name: "dense_features"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
input_arg { name: "example_weights" type: DT_FLOAT }
input_arg { name: "example_labels" type: DT_FLOAT }
input_arg {
  name: "sparse_indices"
  type: DT_INT64
  number_attr: "num_sparse_features"
}
input_arg {
  name: "sparse_weights"
  type: DT_FLOAT
  number_attr: "num_sparse_features"
}
input_arg {
  name: "dense_weights"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
input_arg { name: "example_state_data" type: DT_FLOAT }
output_arg { name: "out_example_state_data" type: DT_FLOAT }
output_arg {
  name: "out_delta_sparse_weights"
  type: DT_FLOAT
  number_attr: "num_sparse_features"
}
output_arg {
  name: "out_delta_dense_weights"
  type: DT_FLOAT
  number_attr: "num_dense_features"
}
attr {
  name: "loss_type"
  type: "string"
  allowed_values {
    list {
      s: "logistic_loss"
      s: "squared_loss"
      s: "hinge_loss"
      s: "smooth_hinge_loss"
      s: "poisson_loss"
    }
  }
}
attr { name: "adaptive" type: "bool" default_value { b: false } }
attr { name: "num_sparse_features" type: "int" has_minimum: true }
attr {
  name: "num_sparse_features_with_values"
  type: "int"
  has_minimum: true
}
attr { name: "num_dense_features" type: "int" has_minimum: true }
attr { name: "l1" type: "float" }
attr { name: "l2" type: "float" }
attr {
  name: "num_loss_partitions"
  type: "int"
  has_minimum: true
  minimum: 1
}
attr {
  name: "num_inner_iterations"
  type: "int"
  has_minimum: true
  minimum: 1
}
-}
-- | 
sdcaShrinkL1 :: forall m' . (MonadBuild m') => Float -- ^ __l1__
                -> Float -- ^ __l2__
                -> [Tensor Ref Float] -- ^ __weights__
                -> m' (ControlNode)
sdcaShrinkL1 :: Float -> Float -> [Tensor Ref Float] -> m' ControlNode
sdcaShrinkL1 = OpParams -> Float -> Float -> [Tensor Ref Float] -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Float -> Float -> [Tensor Ref Float] -> m' ControlNode
sdcaShrinkL1' OpParams
forall a. a -> a
id
sdcaShrinkL1' :: forall m' . (MonadBuild m') => OpParams ->
                 Float -- ^ __l1__
                 -> Float -- ^ __l2__
                 -> [Tensor Ref Float] -- ^ __weights__
                 -> m' (ControlNode)
sdcaShrinkL1' :: OpParams -> Float -> Float -> [Tensor Ref Float] -> m' ControlNode
sdcaShrinkL1' op'options :: OpParams
op'options l1 :: Float
l1 l2 :: Float
l2
              weights :: [Tensor Ref Float]
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_features", [("weights", [Tensor Ref Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref Float]
weights)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor Ref Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor Ref Float]
weights]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SdcaShrinkL1"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l1" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l1
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "l2" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
l2
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_features" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_features
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_features :: Int64
num_features = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor Ref Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor Ref Float]
weights) :: Int64
{-
input_arg {
  name: "weights"
  type: DT_FLOAT
  number_attr: "num_features"
  is_ref: true
}
attr { name: "num_features" type: "int" has_minimum: true }
attr { name: "l1" type: "float" }
attr { name: "l2" type: "float" }
-}
-- | 
segmentMax :: forall v'1 v'2 t tindices . (OneOf '[Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tindices) =>
              Tensor v'1 t -- ^ __data__
              -> Tensor v'2 tindices -- ^ __segment_ids__
              -> Tensor Build t -- ^ __output__
segmentMax :: Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMax = OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tindices.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMax' OpParams
forall a. a -> a
id
segmentMax' :: forall v'1 v'2 t tindices . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor v'1 t -- ^ __data__
               -> Tensor v'2 tindices -- ^ __segment_ids__
               -> Tensor Build t -- ^ __output__
segmentMax' :: OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMax' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SegmentMax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
segmentMean :: forall v'1 v'2 t
               tindices . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word32, Data.Word.Word64,
                                   Data.Word.Word8, Double, Float] t,
                           OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
               Tensor v'1 t -- ^ __data__
               -> Tensor v'2 tindices -- ^ __segment_ids__
               -> Tensor Build t -- ^ __output__
segmentMean :: Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMean = OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
segmentMean' OpParams
forall a. a -> a
id
segmentMean' :: forall v'1 v'2 t
                tindices . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t, OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
                OpParams ->
                Tensor v'1 t -- ^ __data__
                -> Tensor v'2 tindices -- ^ __segment_ids__
                -> Tensor Build t -- ^ __output__
segmentMean' :: OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMean' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SegmentMean"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
segmentMin :: forall v'1 v'2 t tindices . (OneOf '[Data.Int.Int16,
                                                   Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Int.Int8,
                                                   Data.Word.Word16,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64,
                                                   Data.Word.Word8, Double,
                                                   Float] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tindices) =>
              Tensor v'1 t -- ^ __data__
              -> Tensor v'2 tindices -- ^ __segment_ids__
              -> Tensor Build t -- ^ __output__
segmentMin :: Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMin = OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tindices.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMin' OpParams
forall a. a -> a
id
segmentMin' :: forall v'1 v'2 t tindices . (OneOf '[Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor v'1 t -- ^ __data__
               -> Tensor v'2 tindices -- ^ __segment_ids__
               -> Tensor Build t -- ^ __output__
segmentMin' :: OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentMin' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SegmentMin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
segmentProd :: forall v'1 v'2 t
               tindices . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word32, Data.Word.Word64,
                                   Data.Word.Word8, Double, Float] t,
                           OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
               Tensor v'1 t -- ^ __data__
               -> Tensor v'2 tindices -- ^ __segment_ids__
               -> Tensor Build t -- ^ __output__
segmentProd :: Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentProd = OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
segmentProd' OpParams
forall a. a -> a
id
segmentProd' :: forall v'1 v'2 t
                tindices . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t, OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
                OpParams ->
                Tensor v'1 t -- ^ __data__
                -> Tensor v'2 tindices -- ^ __segment_ids__
                -> Tensor Build t -- ^ __output__
segmentProd' :: OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentProd' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SegmentProd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
segmentSum :: forall v'1 v'2 t
              tindices . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t, OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
              Tensor v'1 t -- ^ __data__
              -> Tensor v'2 tindices -- ^ __segment_ids__
              -> Tensor Build t -- ^ __output__
segmentSum :: Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentSum = OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
segmentSum' OpParams
forall a. a -> a
id
segmentSum' :: forall v'1 v'2 t
               tindices . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word32, Data.Word.Word64,
                                   Data.Word.Word8, Double, Float] t,
                           OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
               OpParams ->
               Tensor v'1 t -- ^ __data__
               -> Tensor v'2 tindices -- ^ __segment_ids__
               -> Tensor Build t -- ^ __output__
segmentSum' :: OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
segmentSum' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SegmentSum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
select :: forall v'1 v'2 v'3 t . (TensorType t) =>
          Tensor v'1 Bool -- ^ __condition__
          -> Tensor v'2 t -- ^ __t__
          -> Tensor v'3 t -- ^ __e__
          -> Tensor Build t -- ^ __output__
select :: Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
select = OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
select' OpParams
forall a. a -> a
id
select' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
           Tensor v'1 Bool -- ^ __condition__
           -> Tensor v'2 t -- ^ __t__
           -> Tensor v'3 t -- ^ __e__
           -> Tensor Build t -- ^ __output__
select' :: OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
select' op'options :: OpParams
op'options condition :: Tensor v'1 Bool
condition t :: Tensor v'2 t
t e :: Tensor v'3 t
e | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
condition,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
t,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
e]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Select"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "condition" type: DT_BOOL }
input_arg { name: "t" type_attr: "T" }
input_arg { name: "e" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
selectV2 :: forall v'1 v'2 v'3 t . (TensorType t) =>
            Tensor v'1 Bool -- ^ __condition__
            -> Tensor v'2 t -- ^ __t__
            -> Tensor v'3 t -- ^ __e__
            -> Tensor Build t -- ^ __output__
selectV2 :: Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
selectV2 = OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
selectV2' OpParams
forall a. a -> a
id
selectV2' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
             Tensor v'1 Bool -- ^ __condition__
             -> Tensor v'2 t -- ^ __t__
             -> Tensor v'3 t -- ^ __e__
             -> Tensor Build t -- ^ __output__
selectV2' :: OpParams
-> Tensor v'1 Bool
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build t
selectV2' op'options :: OpParams
op'options condition :: Tensor v'1 Bool
condition t :: Tensor v'2 t
t e :: Tensor v'3 t
e | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Bool
condition,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
t,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
e]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SelectV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "condition" type: DT_BOOL }
input_arg { name: "t" type_attr: "T" }
input_arg { name: "e" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
selfAdjointEig :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
                  Tensor v'1 t -- ^ __input__
                  -> Tensor Build t -- ^ __output__
selfAdjointEig :: Tensor v'1 t -> Tensor Build t
selfAdjointEig = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
selfAdjointEig' OpParams
forall a. a -> a
id
selfAdjointEig' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double,
                                           Float] t) => OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> Tensor Build t -- ^ __output__
selfAdjointEig' :: OpParams -> Tensor v'1 t -> Tensor Build t
selfAdjointEig' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SelfAdjointEig"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_DOUBLE type: DT_FLOAT type: DT_HALF }
  }
}
-}
-- | 
selfAdjointEigV2 :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Word.Word16, Double,
                                            Float] t) =>
                    Tensor v'1 t -- ^ __input__
                    -> (Tensor Build t, Tensor Build t) -- ^ (__e__, __v__)
                    --
                    -- * __e__
                    --
                    -- * __v__
selfAdjointEigV2 :: Tensor v'1 t -> (Tensor Build t, Tensor Build t)
selfAdjointEigV2 = OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
selfAdjointEigV2' OpParams
forall a. a -> a
id
selfAdjointEigV2' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> (Tensor Build t, Tensor Build t) -- ^ (__e__, __v__)
                     --
                     -- * __e__
                     --
                     -- * __v__
selfAdjointEigV2' :: OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
selfAdjointEigV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SelfAdjointEigV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "e" type_attr: "T" }
output_arg { name: "v" type_attr: "T" }
attr { name: "compute_v" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
selu :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
        Tensor v'1 t -- ^ __features__
        -> Tensor Build t -- ^ __activations__
selu :: Tensor v'1 t -> Tensor Build t
selu = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
selu' OpParams
forall a. a -> a
id
selu' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
         OpParams ->
         Tensor v'1 t -- ^ __features__
         -> Tensor Build t -- ^ __activations__
selu' :: OpParams -> Tensor v'1 t -> Tensor Build t
selu' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Selu"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
seluGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __gradients__
            -> Tensor v'2 t -- ^ __outputs__
            -> Tensor Build t -- ^ __backprops__
seluGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
seluGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
seluGrad' OpParams
forall a. a -> a
id
seluGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __gradients__
             -> Tensor v'2 t -- ^ __outputs__
             -> Tensor Build t -- ^ __backprops__
seluGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
seluGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients outputs :: Tensor v'2 t
outputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
outputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SeluGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "outputs" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
send :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
        ByteString -- ^ __recv_device__
        -> ByteString -- ^ __send_device__
        -> Data.Int.Int64 -- ^ __send_device_incarnation__
        -> ByteString -- ^ __tensor_name__
        -> Tensor v'1 t -- ^ __tensor__
        -> m' (ControlNode)
send :: ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
send = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
send' OpParams
forall a. a -> a
id
send' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
         ByteString -- ^ __recv_device__
         -> ByteString -- ^ __send_device__
         -> Data.Int.Int64 -- ^ __send_device_incarnation__
         -> ByteString -- ^ __tensor_name__
         -> Tensor v'1 t -- ^ __tensor__
         -> m' (ControlNode)
send' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
send' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation tensor_name :: ByteString
tensor_name
      tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Send"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "tensor_name" type: "string" }
attr { name: "send_device" type: "string" }
attr { name: "send_device_incarnation" type: "int" }
attr { name: "recv_device" type: "string" }
attr {
  name: "client_terminated" type: "bool" default_value { b: false }
}
-}
-- | 
sendTPUEmbeddingGradients :: forall v'1 v'2 m' . (MonadBuild m') =>
                             ByteString -- ^ __config__
                             -> [Tensor v'1 Float] -- ^ __inputs__
                             -> [Tensor v'2 Float] -- ^ __learning_rates__
                             -> m' (ControlNode)
sendTPUEmbeddingGradients :: ByteString
-> [Tensor v'1 Float] -> [Tensor v'2 Float] -> m' ControlNode
sendTPUEmbeddingGradients = OpParams
-> ByteString
-> [Tensor v'1 Float]
-> [Tensor v'2 Float]
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> [Tensor v'1 Float]
-> [Tensor v'2 Float]
-> m' ControlNode
sendTPUEmbeddingGradients' OpParams
forall a. a -> a
id
sendTPUEmbeddingGradients' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                              ByteString -- ^ __config__
                              -> [Tensor v'1 Float] -- ^ __inputs__
                              -> [Tensor v'2 Float] -- ^ __learning_rates__
                              -> m' (ControlNode)
sendTPUEmbeddingGradients' :: OpParams
-> ByteString
-> [Tensor v'1 Float]
-> [Tensor v'2 Float]
-> m' ControlNode
sendTPUEmbeddingGradients' op'options :: OpParams
op'options config :: ByteString
config inputs :: [Tensor v'1 Float]
inputs
                           learning_rates :: [Tensor v'2 Float]
learning_rates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
inputs)]),
                                                           ("NN", [("learning_rates", [Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
learning_rates)])] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Float]
inputs,
                                                             [Tensor v'2 Float] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 Float]
learning_rates]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SendTPUEmbeddingGradients"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
config
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "NN" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
nN
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Float]
inputs) :: Int64
    nN :: Int64
nN = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 Float] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 Float]
learning_rates) :: Int64
{-
input_arg { name: "inputs" type: DT_FLOAT number_attr: "N" }
input_arg {
  name: "learning_rates" type: DT_FLOAT number_attr: "NN"
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "NN" type: "int" default_value { i: 0 } has_minimum: true
}
attr { name: "config" type: "string" }
-}
-- | 
serializeIterator :: forall v'1 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __resource_handle__
                     -> m' (Tensor Value Variant) -- ^ __serialized__
serializeIterator :: Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
serializeIterator = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
serializeIterator' OpParams
forall a. a -> a
id
serializeIterator' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __resource_handle__
                      -> m' (Tensor Value Variant) -- ^ __serialized__
serializeIterator' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
serializeIterator' op'options :: OpParams
op'options resource_handle :: Tensor v'1 ResourceHandle
resource_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource_handle]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SerializeIterator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource_handle" type: DT_RESOURCE }
output_arg { name: "serialized" type: DT_VARIANT }
attr {
  name: "external_state_policy" type: "int" default_value { i: 0 }
}
-}
-- | 
serializeManySparse :: forall v'1 v'2 v'3 t out_type . (TensorType t,
                                                        OneOf '[Data.ByteString.ByteString,
                                                                Variant] out_type) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                       -> Tensor v'2 t -- ^ __sparse_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                       -> Tensor Build out_type -- ^ __serialized_sparse__
serializeManySparse :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
serializeManySparse = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t out_type.
(TensorType t, OneOf '[ByteString, Variant] out_type) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
serializeManySparse' OpParams
forall a. a -> a
id
serializeManySparse' :: forall v'1 v'2 v'3 t out_type . (TensorType t,
                                                         OneOf '[Data.ByteString.ByteString,
                                                                 Variant] out_type) =>
                        OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                        -> Tensor v'2 t -- ^ __sparse_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                        -> Tensor Build out_type -- ^ __serialized_sparse__
serializeManySparse' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
serializeManySparse' op'options :: OpParams
op'options sparse_indices :: Tensor v'1 Int64
sparse_indices sparse_values :: Tensor v'2 t
sparse_values
                     sparse_shape :: Tensor v'3 Int64
sparse_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sparse_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sparse_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sparse_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SerializeManySparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_indices" type: DT_INT64 }
input_arg { name: "sparse_values" type_attr: "T" }
input_arg { name: "sparse_shape" type: DT_INT64 }
output_arg { name: "serialized_sparse" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_STRING }
  allowed_values { list { type: DT_STRING type: DT_VARIANT } }
}
-}
-- | 
serializeSparse :: forall v'1 v'2 v'3 t out_type . (TensorType t,
                                                    OneOf '[Data.ByteString.ByteString,
                                                            Variant] out_type) =>
                   Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                   -> Tensor v'2 t -- ^ __sparse_values__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                   -> Tensor Build out_type -- ^ __serialized_sparse__
serializeSparse :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
serializeSparse = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t out_type.
(TensorType t, OneOf '[ByteString, Variant] out_type) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
serializeSparse' OpParams
forall a. a -> a
id
serializeSparse' :: forall v'1 v'2 v'3 t out_type . (TensorType t,
                                                     OneOf '[Data.ByteString.ByteString,
                                                             Variant] out_type) =>
                    OpParams ->
                    Tensor v'1 Data.Int.Int64 -- ^ __sparse_indices__
                    -> Tensor v'2 t -- ^ __sparse_values__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __sparse_shape__
                    -> Tensor Build out_type -- ^ __serialized_sparse__
serializeSparse' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build out_type
serializeSparse' op'options :: OpParams
op'options sparse_indices :: Tensor v'1 Int64
sparse_indices sparse_values :: Tensor v'2 t
sparse_values
                 sparse_shape :: Tensor v'3 Int64
sparse_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sparse_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sparse_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sparse_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SerializeSparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_indices" type: DT_INT64 }
input_arg { name: "sparse_values" type_attr: "T" }
input_arg { name: "sparse_shape" type: DT_INT64 }
output_arg { name: "serialized_sparse" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_STRING }
  allowed_values { list { type: DT_STRING type: DT_VARIANT } }
}
-}
-- | 
serializeTensor :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __tensor__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __serialized__
serializeTensor :: Tensor v'1 t -> Tensor Build ByteString
serializeTensor = OpParams -> Tensor v'1 t -> Tensor Build ByteString
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build ByteString
serializeTensor' OpParams
forall a. a -> a
id
serializeTensor' :: forall v'1 t . (TensorType t) => OpParams ->
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __serialized__
serializeTensor' :: OpParams -> Tensor v'1 t -> Tensor Build ByteString
serializeTensor' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SerializeTensor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
output_arg { name: "serialized" type: DT_STRING }
attr { name: "T" type: "type" }
-}
-- | 
setSize :: forall v'1 v'2 v'3 t . (OneOf '[Data.ByteString.ByteString,
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16,
                                           Data.Word.Word8] t) =>
           Tensor v'1 Data.Int.Int64 -- ^ __set_indices__
           -> Tensor v'2 t -- ^ __set_values__
           -> Tensor v'3 Data.Int.Int64 -- ^ __set_shape__
           -> Tensor Build Data.Int.Int32 -- ^ __size__
setSize :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Int32
setSize = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Int32
setSize' OpParams
forall a. a -> a
id
setSize' :: forall v'1 v'2 v'3 t . (OneOf '[Data.ByteString.ByteString,
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16,
                                            Data.Word.Word8] t) => OpParams ->
            Tensor v'1 Data.Int.Int64 -- ^ __set_indices__
            -> Tensor v'2 t -- ^ __set_values__
            -> Tensor v'3 Data.Int.Int64 -- ^ __set_shape__
            -> Tensor Build Data.Int.Int32 -- ^ __size__
setSize' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Int32
setSize' op'options :: OpParams
op'options set_indices :: Tensor v'1 Int64
set_indices set_values :: Tensor v'2 t
set_values set_shape :: Tensor v'3 Int64
set_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
set_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
set_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
set_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SetSize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "set_indices" type: DT_INT64 }
input_arg { name: "set_values" type_attr: "T" }
input_arg { name: "set_shape" type: DT_INT64 }
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_STRING
    }
  }
}
-}
-- | 
setStatsAggregatorDataset :: forall v'1 v'2 v'3 v'4 m' . (MonadBuild m') =>
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 ResourceHandle -- ^ __stats_aggregator__
                             -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                             -> Tensor v'4 Data.ByteString.ByteString -- ^ __counter_prefix__
                             -> m' (Tensor Value Variant) -- ^ __handle__
setStatsAggregatorDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
setStatsAggregatorDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
setStatsAggregatorDataset' OpParams
forall a. a -> a
id
setStatsAggregatorDataset' :: forall v'1 v'2 v'3 v'4 m' . (MonadBuild m') =>
                              OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor v'2 ResourceHandle -- ^ __stats_aggregator__
                              -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                              -> Tensor v'4 Data.ByteString.ByteString -- ^ __counter_prefix__
                              -> m' (Tensor Value Variant) -- ^ __handle__
setStatsAggregatorDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> Tensor v'3 ByteString
-> Tensor v'4 ByteString
-> m' (Tensor Value Variant)
setStatsAggregatorDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                           stats_aggregator :: Tensor v'2 ResourceHandle
stats_aggregator tag :: Tensor v'3 ByteString
tag
                           counter_prefix :: Tensor v'4 ByteString
counter_prefix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
stats_aggregator,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
counter_prefix]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SetStatsAggregatorDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "stats_aggregator" type: DT_RESOURCE }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "counter_prefix" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shape :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] out_type) =>
         Tensor v'1 t -- ^ __input__
         -> Tensor Build out_type -- ^ __output__
shape :: Tensor v'1 t -> Tensor Build out_type
shape = OpParams -> Tensor v'1 t -> Tensor Build out_type
forall (v'1 :: * -> *) t out_type.
(TensorType t, OneOf '[Int32, Int64] out_type) =>
OpParams -> Tensor v'1 t -> Tensor Build out_type
shape' OpParams
forall a. a -> a
id
shape' :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] out_type) =>
          OpParams ->
          Tensor v'1 t -- ^ __input__
          -> Tensor Build out_type -- ^ __output__
shape' :: OpParams -> Tensor v'1 t -> Tensor Build out_type
shape' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Shape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
shapeN :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] out_type) =>
          [Tensor v'1 t] -- ^ __input__
          -> [Tensor Build out_type] -- ^ __output__
shapeN :: [Tensor v'1 t] -> [Tensor Build out_type]
shapeN = OpParams -> [Tensor v'1 t] -> [Tensor Build out_type]
forall (v'1 :: * -> *) t out_type.
(TensorType t, OneOf '[Int32, Int64] out_type) =>
OpParams -> [Tensor v'1 t] -> [Tensor Build out_type]
shapeN' OpParams
forall a. a -> a
id
shapeN' :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] out_type) =>
           OpParams ->
           [Tensor v'1 t] -- ^ __input__
           -> [Tensor Build out_type] -- ^ __output__
shapeN' :: OpParams -> [Tensor v'1 t] -> [Tensor Build out_type]
shapeN' op'options :: OpParams
op'options input :: [Tensor v'1 t]
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("input", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input)])] =
    [Int64] -> Build OpDef -> [Tensor Build out_type]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
n] (Build OpDef -> [Tensor Build out_type])
-> Build OpDef -> [Tensor Build out_type]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShapeN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input) :: Int64
{-
input_arg { name: "input" type_attr: "T" number_attr: "N" }
output_arg {
  name: "output" type_attr: "out_type" number_attr: "N"
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
shardDataset :: [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.Int.Int64 -- ^ __num_shards__
                -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                -> Tensor Build Variant -- ^ __handle__
shardDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
shardDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
shardDataset' OpParams
forall a. a -> a
id
shardDataset' :: OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __num_shards__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __index__
                 -> Tensor Build Variant -- ^ __handle__
shardDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build Variant
shardDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset num_shards :: Tensor v'2 Int64
num_shards
              index :: Tensor v'3 Int64
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
num_shards,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
index]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShardDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "num_shards" type: DT_INT64 }
input_arg { name: "index" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "require_non_empty" type: "bool" default_value { b: false }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shardedFilename :: Tensor v'1 Data.ByteString.ByteString -- ^ __basename__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __shard__
                   -> Tensor v'3 Data.Int.Int32 -- ^ __num_shards__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __filename__
shardedFilename :: Tensor v'1 ByteString
-> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build ByteString
shardedFilename = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build ByteString
shardedFilename' OpParams
forall a. a -> a
id
shardedFilename' :: OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __basename__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __shard__
                    -> Tensor v'3 Data.Int.Int32 -- ^ __num_shards__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __filename__
shardedFilename' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build ByteString
shardedFilename' op'options :: OpParams
op'options basename :: Tensor v'1 ByteString
basename shard :: Tensor v'2 Int32
shard num_shards :: Tensor v'3 Int32
num_shards | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
basename,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
shard,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
num_shards]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShardedFilename"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "basename" type: DT_STRING }
input_arg { name: "shard" type: DT_INT32 }
input_arg { name: "num_shards" type: DT_INT32 }
output_arg { name: "filename" type: DT_STRING }
-}
-- | 
shardedFilespec :: Tensor v'1 Data.ByteString.ByteString -- ^ __basename__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __num_shards__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __filename__
shardedFilespec :: Tensor v'1 ByteString
-> Tensor v'2 Int32 -> Tensor Build ByteString
shardedFilespec = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
shardedFilespec' OpParams
forall a. a -> a
id
shardedFilespec' :: OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __basename__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __num_shards__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __filename__
shardedFilespec' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor Build ByteString
shardedFilespec' op'options :: OpParams
op'options basename :: Tensor v'1 ByteString
basename num_shards :: Tensor v'2 Int32
num_shards | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
basename,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_shards]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShardedFilespec"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "basename" type: DT_STRING }
input_arg { name: "num_shards" type: DT_INT32 }
output_arg { name: "filename" type: DT_STRING }
-}
-- | 
shuffleAndRepeatDataset :: [DataType] -- ^ __output_types__
                           -> Tensor v'1 Variant -- ^ __input_dataset__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                           -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                           -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                           -> Tensor v'5 Data.Int.Int64 -- ^ __count__
                           -> Tensor Build Variant -- ^ __handle__
shuffleAndRepeatDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build Variant
shuffleAndRepeatDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build Variant
shuffleAndRepeatDataset' OpParams
forall a. a -> a
id
shuffleAndRepeatDataset' :: OpParams ->
                            [DataType] -- ^ __output_types__
                            -> Tensor v'1 Variant -- ^ __input_dataset__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                            -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                            -> Tensor v'5 Data.Int.Int64 -- ^ __count__
                            -> Tensor Build Variant -- ^ __handle__
shuffleAndRepeatDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor Build Variant
shuffleAndRepeatDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset buffer_size :: Tensor v'2 Int64
buffer_size seed :: Tensor v'3 Int64
seed
                         seed2 :: Tensor v'4 Int64
seed2 count :: Tensor v'5 Int64
count | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
seed2,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
count]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShuffleAndRepeatDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
input_arg { name: "count" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "reshuffle_each_iteration"
  type: "bool"
  default_value { b: true }
}
-}
-- | 
shuffleAndRepeatDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6
                             m' . (MonadBuild m') =>
                             [DataType] -- ^ __output_types__
                             -> Tensor v'1 Variant -- ^ __input_dataset__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                             -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                             -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                             -> Tensor v'5 Data.Int.Int64 -- ^ __count__
                             -> Tensor v'6 ResourceHandle -- ^ __seed_generator__
                             -> m' (Tensor Value Variant) -- ^ __handle__
shuffleAndRepeatDatasetV2 :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ResourceHandle
-> m' (Tensor Value Variant)
shuffleAndRepeatDatasetV2 = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ResourceHandle
-> m' (Tensor Value Variant)
shuffleAndRepeatDatasetV2' OpParams
forall a. a -> a
id
shuffleAndRepeatDatasetV2' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                              m' . (MonadBuild m') => OpParams ->
                              [DataType] -- ^ __output_types__
                              -> Tensor v'1 Variant -- ^ __input_dataset__
                              -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                              -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                              -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                              -> Tensor v'5 Data.Int.Int64 -- ^ __count__
                              -> Tensor v'6 ResourceHandle -- ^ __seed_generator__
                              -> m' (Tensor Value Variant) -- ^ __handle__
shuffleAndRepeatDatasetV2' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> Tensor v'6 ResourceHandle
-> m' (Tensor Value Variant)
shuffleAndRepeatDatasetV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset buffer_size :: Tensor v'2 Int64
buffer_size
                           seed :: Tensor v'3 Int64
seed seed2 :: Tensor v'4 Int64
seed2 count :: Tensor v'5 Int64
count seed_generator :: Tensor v'6 ResourceHandle
seed_generator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
seed2,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
count,
                                                             Tensor v'6 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 ResourceHandle
seed_generator]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ShuffleAndRepeatDatasetV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
input_arg { name: "count" type: DT_INT64 }
input_arg { name: "seed_generator" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "reshuffle_each_iteration"
  type: "bool"
  default_value { b: true }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shuffleDataset :: [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                  -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                  -> Tensor Build Variant -- ^ __handle__
shuffleDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
shuffleDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
shuffleDataset' OpParams
forall a. a -> a
id
shuffleDataset' :: OpParams ->
                   [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                   -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                   -> Tensor Build Variant -- ^ __handle__
shuffleDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
shuffleDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset buffer_size :: Tensor v'2 Int64
buffer_size seed :: Tensor v'3 Int64
seed
                seed2 :: Tensor v'4 Int64
seed2 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
seed2]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ShuffleDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "reshuffle_each_iteration"
  type: "bool"
  default_value { b: true }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shuffleDatasetV2 :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                    -> Tensor v'3 ResourceHandle -- ^ __seed_generator__
                    -> m' (Tensor Value Variant) -- ^ __handle__
shuffleDatasetV2 :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV2 = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV2' OpParams
forall a. a -> a
id
shuffleDatasetV2' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                     [DataType] -- ^ __output_types__
                     -> Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                     -> Tensor v'3 ResourceHandle -- ^ __seed_generator__
                     -> m' (Tensor Value Variant) -- ^ __handle__
shuffleDatasetV2' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV2' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset buffer_size :: Tensor v'2 Int64
buffer_size
                  seed_generator :: Tensor v'3 ResourceHandle
seed_generator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size,
                                                             Tensor v'3 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ResourceHandle
seed_generator]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ShuffleDatasetV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "seed_generator" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shuffleDatasetV3 :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                    -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                    -> Tensor v'5 ResourceHandle -- ^ __seed_generator__
                    -> m' (Tensor Value Variant) -- ^ __handle__
shuffleDatasetV3 :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV3 = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV3' OpParams
forall a. a -> a
id
shuffleDatasetV3' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                     OpParams ->
                     [DataType] -- ^ __output_types__
                     -> Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __buffer_size__
                     -> Tensor v'3 Data.Int.Int64 -- ^ __seed__
                     -> Tensor v'4 Data.Int.Int64 -- ^ __seed2__
                     -> Tensor v'5 ResourceHandle -- ^ __seed_generator__
                     -> m' (Tensor Value Variant) -- ^ __handle__
shuffleDatasetV3' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 ResourceHandle
-> m' (Tensor Value Variant)
shuffleDatasetV3' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset buffer_size :: Tensor v'2 Int64
buffer_size seed :: Tensor v'3 Int64
seed seed2 :: Tensor v'4 Int64
seed2
                  seed_generator :: Tensor v'5 ResourceHandle
seed_generator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
buffer_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
seed,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
seed2,
                                                             Tensor v'5 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ResourceHandle
seed_generator]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ShuffleDatasetV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "buffer_size" type: DT_INT64 }
input_arg { name: "seed" type: DT_INT64 }
input_arg { name: "seed2" type: DT_INT64 }
input_arg { name: "seed_generator" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "reshuffle_each_iteration"
  type: "bool"
  default_value { b: true }
}
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
shutdownDistributedTPU :: forall m' . (MonadBuild m') => m' (ControlNode)
shutdownDistributedTPU :: m' ControlNode
shutdownDistributedTPU = OpParams -> m' ControlNode
forall (m' :: * -> *). MonadBuild m' => OpParams -> m' ControlNode
shutdownDistributedTPU' OpParams
forall a. a -> a
id
shutdownDistributedTPU' :: forall m' . (MonadBuild m') => OpParams ->
                           m' (ControlNode)
shutdownDistributedTPU' :: OpParams -> m' ControlNode
shutdownDistributedTPU' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ShutdownDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-

-}
-- | 
sigmoid :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float),
                                   Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
sigmoid :: Tensor v'1 t -> Tensor Build t
sigmoid = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
sigmoid' OpParams
forall a. a -> a
id
sigmoid' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor Build t -- ^ __y__
sigmoid' :: OpParams -> Tensor v'1 t -> Tensor Build t
sigmoid' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sigmoid"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sigmoidGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Word.Word16, Double,
                                           Float] t) => Tensor v'1 t -- ^ __y__
               -> Tensor v'2 t -- ^ __dy__
               -> Tensor Build t -- ^ __z__
sigmoidGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sigmoidGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sigmoidGrad' OpParams
forall a. a -> a
id
sigmoidGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Word.Word16, Double,
                                            Float] t) => OpParams ->
                Tensor v'1 t -- ^ __y__
                -> Tensor v'2 t -- ^ __dy__
                -> Tensor Build t -- ^ __z__
sigmoidGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sigmoidGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SigmoidGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sign :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int32,
                                Data.Int.Int64, Data.Word.Word16, Double,
                                Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
sign :: Tensor v'1 t -> Tensor Build t
sign = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
sign' OpParams
forall a. a -> a
id
sign' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int32,
                                 Data.Int.Int64, Data.Word.Word16, Double,
                                 Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
sign' :: OpParams -> Tensor v'1 t -> Tensor Build t
sign' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sign"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sin :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
sin :: Tensor v'1 t -> Tensor Build t
sin = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
sin' OpParams
forall a. a -> a
id
sin' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
sin' :: OpParams -> Tensor v'1 t -> Tensor Build t
sin' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sinh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
sinh :: Tensor v'1 t -> Tensor Build t
sinh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
sinh' OpParams
forall a. a -> a
id
sinh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
sinh' :: OpParams -> Tensor v'1 t -> Tensor Build t
sinh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sinh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
size :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] out_type) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor Build out_type -- ^ __output__
size :: Tensor v'1 t -> Tensor Build out_type
size = OpParams -> Tensor v'1 t -> Tensor Build out_type
forall (v'1 :: * -> *) t out_type.
(TensorType t, OneOf '[Int32, Int64] out_type) =>
OpParams -> Tensor v'1 t -> Tensor Build out_type
size' OpParams
forall a. a -> a
id
size' :: forall v'1 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] out_type) =>
         OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor Build out_type -- ^ __output__
size' :: OpParams -> Tensor v'1 t -> Tensor Build out_type
size' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Size"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
skipDataset :: [DataType] -- ^ __output_types__
               -> Tensor v'1 Variant -- ^ __input_dataset__
               -> Tensor v'2 Data.Int.Int64 -- ^ __count__
               -> Tensor Build Variant -- ^ __handle__
skipDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
skipDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
skipDataset' OpParams
forall a. a -> a
id
skipDataset' :: OpParams ->
                [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.Int.Int64 -- ^ __count__
                -> Tensor Build Variant -- ^ __handle__
skipDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
skipDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset count :: Tensor v'2 Int64
count | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
count]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SkipDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "count" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
skipgram :: forall m' . (MonadBuild m') => Data.Int.Int64 -- ^ __batch_size__
            -> ByteString -- ^ __filename__
            -> m' ((Tensor Value Data.ByteString.ByteString,
                    Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int64,
                    Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int64,
                    Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int32))
            -- ^ (__vocab_word__, __vocab_freq__, __words_per_epoch__, __current_epoch__, __total_words_processed__, __examples__, __labels__)
            --
            -- * __vocab_word__
            --
            -- * __vocab_freq__
            --
            -- * __words_per_epoch__
            --
            -- * __current_epoch__
            --
            -- * __total_words_processed__
            --
            -- * __examples__
            --
            -- * __labels__
skipgram :: Int64
-> ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
skipgram = OpParams
-> Int64
-> ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
skipgram' OpParams
forall a. a -> a
id
skipgram' :: forall m' . (MonadBuild m') => OpParams ->
             Data.Int.Int64 -- ^ __batch_size__
             -> ByteString -- ^ __filename__
             -> m' ((Tensor Value Data.ByteString.ByteString,
                     Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int64,
                     Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int64,
                     Tensor Value Data.Int.Int32, Tensor Value Data.Int.Int32))
             -- ^ (__vocab_word__, __vocab_freq__, __words_per_epoch__, __current_epoch__, __total_words_processed__, __examples__, __labels__)
             --
             -- * __vocab_word__
             --
             -- * __vocab_freq__
             --
             -- * __words_per_epoch__
             --
             -- * __current_epoch__
             --
             -- * __total_words_processed__
             --
             -- * __examples__
             --
             -- * __labels__
skipgram' :: OpParams
-> Int64
-> ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
skipgram' op'options :: OpParams
op'options batch_size :: Int64
batch_size filename :: ByteString
filename | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
   Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
   Tensor Value Int32)
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
    Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
    Tensor Value Int32)
 -> m'
      (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
       Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
       Tensor Value Int32))
-> Build
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
-> m'
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64]
-> OpDef
-> Build
     (Tensor Value ByteString, Tensor Value Int32, Tensor Value Int64,
      Tensor Value Int32, Tensor Value Int64, Tensor Value Int32,
      Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Skipgram"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "batch_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
batch_size
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "filename" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
filename
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "vocab_word" type: DT_STRING }
output_arg { name: "vocab_freq" type: DT_INT32 }
output_arg { name: "words_per_epoch" type: DT_INT64 }
output_arg { name: "current_epoch" type: DT_INT32 }
output_arg { name: "total_words_processed" type: DT_INT64 }
output_arg { name: "examples" type: DT_INT32 }
output_arg { name: "labels" type: DT_INT32 }
attr { name: "filename" type: "string" }
attr { name: "batch_size" type: "int" }
attr { name: "window_size" type: "int" default_value { i: 5 } }
attr { name: "min_count" type: "int" default_value { i: 5 } }
attr {
  name: "subsample" type: "float" default_value { f: 1.0e-3 }
}
-}
-- | 
sleepDataset :: [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.Int.Int64 -- ^ __sleep_microseconds__
                -> Tensor Build Variant -- ^ __handle__
sleepDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
sleepDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
sleepDataset' OpParams
forall a. a -> a
id
sleepDataset' :: OpParams ->
                 [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __sleep_microseconds__
                 -> Tensor Build Variant -- ^ __handle__
sleepDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
sleepDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
              sleep_microseconds :: Tensor v'2 Int64
sleep_microseconds | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
sleep_microseconds]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SleepDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "sleep_microseconds" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
slice :: forall v'1 v'2 v'3 t index . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] index) =>
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 index -- ^ __begin__
         -> Tensor v'3 index -- ^ __size__
         -> Tensor Build t -- ^ __output__
slice :: Tensor v'1 t
-> Tensor v'2 index -> Tensor v'3 index -> Tensor Build t
slice = OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindex.
(TensorType t, OneOf '[Int32, Int64] tindex) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindex
-> Tensor v'3 tindex
-> Tensor Build t
slice' OpParams
forall a. a -> a
id
slice' :: forall v'1 v'2 v'3 t index . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] index) =>
          OpParams ->
          Tensor v'1 t -- ^ __input__
          -> Tensor v'2 index -- ^ __begin__
          -> Tensor v'3 index -- ^ __size__
          -> Tensor Build t -- ^ __output__
slice' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor Build t
slice' op'options :: OpParams
op'options input :: Tensor v'1 t
input begin :: Tensor v'2 index
begin size :: Tensor v'3 index
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Slice"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "size" type_attr: "Index" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
slidingWindowDataset :: [DataType] -- ^ __output_types__
                        -> Tensor v'1 Variant -- ^ __input_dataset__
                        -> Tensor v'2 Data.Int.Int64 -- ^ __window_size__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __window_shift__
                        -> Tensor v'4 Data.Int.Int64 -- ^ __window_stride__
                        -> Tensor Build Variant -- ^ __handle__
slidingWindowDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
slidingWindowDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
slidingWindowDataset' OpParams
forall a. a -> a
id
slidingWindowDataset' :: OpParams ->
                         [DataType] -- ^ __output_types__
                         -> Tensor v'1 Variant -- ^ __input_dataset__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __window_size__
                         -> Tensor v'3 Data.Int.Int64 -- ^ __window_shift__
                         -> Tensor v'4 Data.Int.Int64 -- ^ __window_stride__
                         -> Tensor Build Variant -- ^ __handle__
slidingWindowDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build Variant
slidingWindowDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset window_size :: Tensor v'2 Int64
window_size
                      window_shift :: Tensor v'3 Int64
window_shift window_stride :: Tensor v'4 Int64
window_stride | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
window_size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
window_shift,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
window_stride]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SlidingWindowDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "window_size" type: DT_INT64 }
input_arg { name: "window_shift" type: DT_INT64 }
input_arg { name: "window_stride" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
snapshot :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
snapshot :: Tensor v'1 t -> Tensor Build t
snapshot = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
snapshot' OpParams
forall a. a -> a
id
snapshot' :: forall v'1 t . (TensorType t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor Build t -- ^ __output__
snapshot' :: OpParams -> Tensor v'1 t -> Tensor Build t
snapshot' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Snapshot"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
snapshotDataset :: [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __path__
                   -> Tensor Build Variant -- ^ __handle__
snapshotDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
snapshotDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
snapshotDataset' OpParams
forall a. a -> a
id
snapshotDataset' :: OpParams ->
                    [DataType] -- ^ __output_types__
                    -> Tensor v'1 Variant -- ^ __input_dataset__
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __path__
                    -> Tensor Build Variant -- ^ __handle__
snapshotDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ByteString
-> Tensor Build Variant
snapshotDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset path :: Tensor v'2 ByteString
path | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
path]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SnapshotDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "path" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "compression" type: "string" default_value { s: "" } }
attr {
  name: "reader_path_prefix" type: "string" default_value { s: "" }
}
attr {
  name: "writer_path_prefix" type: "string" default_value { s: "" }
}
attr {
  name: "shard_size_bytes"
  type: "int"
  default_value { i: 10737418240 }
}
attr {
  name: "pending_snapshot_expiry_seconds"
  type: "int"
  default_value { i: 86400 }
}
attr {
  name: "num_reader_threads" type: "int" default_value { i: 1 }
}
attr {
  name: "reader_buffer_size" type: "int" default_value { i: 1 }
}
attr {
  name: "num_writer_threads" type: "int" default_value { i: 1 }
}
attr {
  name: "writer_buffer_size" type: "int" default_value { i: 1 }
}
attr {
  name: "shuffle_on_read" type: "bool" default_value { b: false }
}
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr { name: "mode" type: "string" default_value { s: "auto" } }
attr {
  name: "snapshot_name" type: "string" default_value { s: "" }
}
-}
-- | 
sobolSample :: forall v'1 v'2 v'3 dtype . (OneOf '[Double, Float] dtype) =>
               Tensor v'1 Data.Int.Int32 -- ^ __dim__
               -> Tensor v'2 Data.Int.Int32 -- ^ __num_results__
               -> Tensor v'3 Data.Int.Int32 -- ^ __skip__
               -> Tensor Build dtype -- ^ __samples__
sobolSample :: Tensor v'1 Int32
-> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build dtype
sobolSample = OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype.
OneOf '[Double, Float] dtype =>
OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build dtype
sobolSample' OpParams
forall a. a -> a
id
sobolSample' :: forall v'1 v'2 v'3 dtype . (OneOf '[Double, Float] dtype) =>
                OpParams ->
                Tensor v'1 Data.Int.Int32 -- ^ __dim__
                -> Tensor v'2 Data.Int.Int32 -- ^ __num_results__
                -> Tensor v'3 Data.Int.Int32 -- ^ __skip__
                -> Tensor Build dtype -- ^ __samples__
sobolSample' :: OpParams
-> Tensor v'1 Int32
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build dtype
sobolSample' op'options :: OpParams
op'options dim :: Tensor v'1 Int32
dim num_results :: Tensor v'2 Int32
num_results skip :: Tensor v'3 Int32
skip | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
dim,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_results,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
skip]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SobolSample"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dim" type: DT_INT32 }
input_arg { name: "num_results" type: DT_INT32 }
input_arg { name: "skip" type: DT_INT32 }
output_arg { name: "samples" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
softmax :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __logits__
           -> Tensor Build t -- ^ __softmax__
softmax :: Tensor v'1 t -> Tensor Build t
softmax = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
softmax' OpParams
forall a. a -> a
id
softmax' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __logits__
            -> Tensor Build t -- ^ __softmax__
softmax' :: OpParams -> Tensor v'1 t -> Tensor Build t
softmax' op'options :: OpParams
op'options logits :: Tensor v'1 t
logits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
logits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Softmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "logits" type_attr: "T" }
output_arg { name: "softmax" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
softmaxCrossEntropyWithLogits :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16,
                                                             Double,
                                                             Float] t) =>
                                 Tensor v'1 t -- ^ __features__
                                 -> Tensor v'2 t -- ^ __labels__
                                 -> (Tensor Build t, Tensor Build t)
                                 -- ^ (__loss__, __backprop__)
                                 --
                                 -- * __loss__
                                 --
                                 -- * __backprop__
softmaxCrossEntropyWithLogits :: Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
softmaxCrossEntropyWithLogits = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
softmaxCrossEntropyWithLogits' OpParams
forall a. a -> a
id
softmaxCrossEntropyWithLogits' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                                  OpParams ->
                                  Tensor v'1 t -- ^ __features__
                                  -> Tensor v'2 t -- ^ __labels__
                                  -> (Tensor Build t, Tensor Build t)
                                  -- ^ (__loss__, __backprop__)
                                  --
                                  -- * __loss__
                                  --
                                  -- * __backprop__
softmaxCrossEntropyWithLogits' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
softmaxCrossEntropyWithLogits' op'options :: OpParams
op'options features :: Tensor v'1 t
features labels :: Tensor v'2 t
labels | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
labels]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SoftmaxCrossEntropyWithLogits"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
input_arg { name: "labels" type_attr: "T" }
output_arg { name: "loss" type_attr: "T" }
output_arg { name: "backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
softplus :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __features__
            -> Tensor Build t -- ^ __activations__
softplus :: Tensor v'1 t -> Tensor Build t
softplus = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
softplus' OpParams
forall a. a -> a
id
softplus' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __features__
             -> Tensor Build t -- ^ __activations__
softplus' :: OpParams -> Tensor v'1 t -> Tensor Build t
softplus' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Softplus"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
softplusGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) =>
                Tensor v'1 t -- ^ __gradients__
                -> Tensor v'2 t -- ^ __features__
                -> Tensor Build t -- ^ __backprops__
softplusGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softplusGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softplusGrad' OpParams
forall a. a -> a
id
softplusGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                 Tensor v'1 t -- ^ __gradients__
                 -> Tensor v'2 t -- ^ __features__
                 -> Tensor Build t -- ^ __backprops__
softplusGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softplusGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients features :: Tensor v'2 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SoftplusGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "features" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
softsign :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __features__
            -> Tensor Build t -- ^ __activations__
softsign :: Tensor v'1 t -> Tensor Build t
softsign = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
softsign' OpParams
forall a. a -> a
id
softsign' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __features__
             -> Tensor Build t -- ^ __activations__
softsign' :: OpParams -> Tensor v'1 t -> Tensor Build t
softsign' op'options :: OpParams
op'options features :: Tensor v'1 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Softsign"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
output_arg { name: "activations" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
softsignGrad :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                            Float] t) =>
                Tensor v'1 t -- ^ __gradients__
                -> Tensor v'2 t -- ^ __features__
                -> Tensor Build t -- ^ __backprops__
softsignGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softsignGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softsignGrad' OpParams
forall a. a -> a
id
softsignGrad' :: forall v'1 v'2 t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                 Tensor v'1 t -- ^ __gradients__
                 -> Tensor v'2 t -- ^ __features__
                 -> Tensor Build t -- ^ __backprops__
softsignGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
softsignGrad' op'options :: OpParams
op'options gradients :: Tensor v'1 t
gradients features :: Tensor v'2 t
features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
gradients,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
features]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SoftsignGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "gradients" type_attr: "T" }
input_arg { name: "features" type_attr: "T" }
output_arg { name: "backprops" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
spaceToBatch :: forall v'1 v'2 t tpaddings . (TensorType t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tpaddings) =>
                Data.Int.Int64 -- ^ __block_size__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor v'2 tpaddings -- ^ __paddings__
                -> Tensor Build t -- ^ __output__
spaceToBatch :: Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
spaceToBatch = OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
spaceToBatch' OpParams
forall a. a -> a
id
spaceToBatch' :: forall v'1 v'2 t tpaddings . (TensorType t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tpaddings) =>
                 OpParams ->
                 Data.Int.Int64 -- ^ __block_size__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 tpaddings -- ^ __paddings__
                 -> Tensor Build t -- ^ __output__
spaceToBatch' :: OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
spaceToBatch' op'options :: OpParams
op'options block_size :: Int64
block_size input :: Tensor v'1 t
input paddings :: Tensor v'2 tpaddings
paddings | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tpaddings
paddings]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SpaceToBatch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "block_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
block_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "block_size" type: "int" has_minimum: true minimum: 2
}
-}
-- | 
spaceToBatchND :: forall v'1 v'2 v'3 t tblock_shape tpaddings . (TensorType t,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tblock_shape,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tpaddings) =>
                  Tensor v'1 t -- ^ __input__
                  -> Tensor v'2 tblock_shape -- ^ __block_shape__
                  -> Tensor v'3 tpaddings -- ^ __paddings__
                  -> Tensor Build t -- ^ __output__
spaceToBatchND :: Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tpaddings
-> Tensor Build t
spaceToBatchND = OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tpaddings
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t
       tblock_shape tcrops.
(TensorType t, OneOf '[Int32, Int64] tblock_shape,
 OneOf '[Int32, Int64] tcrops) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tcrops
-> Tensor Build t
spaceToBatchND' OpParams
forall a. a -> a
id
spaceToBatchND' :: forall v'1 v'2 v'3 t tblock_shape tpaddings . (TensorType t,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] tblock_shape,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] tpaddings) =>
                   OpParams ->
                   Tensor v'1 t -- ^ __input__
                   -> Tensor v'2 tblock_shape -- ^ __block_shape__
                   -> Tensor v'3 tpaddings -- ^ __paddings__
                   -> Tensor Build t -- ^ __output__
spaceToBatchND' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tblock_shape
-> Tensor v'3 tpaddings
-> Tensor Build t
spaceToBatchND' op'options :: OpParams
op'options input :: Tensor v'1 t
input block_shape :: Tensor v'2 tblock_shape
block_shape paddings :: Tensor v'3 tpaddings
paddings | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tblock_shape -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tblock_shape
block_shape,
                                                             Tensor v'3 tpaddings -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tpaddings
paddings]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SpaceToBatchND"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tblock_shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tblock_shape -> DataType
forall a. TensorType a => a -> DataType
tensorType (tblock_shape
forall a. HasCallStack => a
undefined :: tblock_shape)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tpaddings" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tpaddings -> DataType
forall a. TensorType a => a -> DataType
tensorType (tpaddings
forall a. HasCallStack => a
undefined :: tpaddings)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "block_shape" type_attr: "Tblock_shape" }
input_arg { name: "paddings" type_attr: "Tpaddings" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tblock_shape"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tpaddings"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
spaceToDepth :: forall v'1 t . (TensorType t) =>
                Data.Int.Int64 -- ^ __block_size__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor Build t -- ^ __output__
spaceToDepth :: Int64 -> Tensor v'1 t -> Tensor Build t
spaceToDepth = OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
spaceToDepth' OpParams
forall a. a -> a
id
spaceToDepth' :: forall v'1 t . (TensorType t) => OpParams ->
                 Data.Int.Int64 -- ^ __block_size__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
spaceToDepth' :: OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
spaceToDepth' op'options :: OpParams
op'options block_size :: Int64
block_size input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SpaceToDepth"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "block_size" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
block_size
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "block_size" type: "int" has_minimum: true minimum: 2
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" s: "NCHW_VECT_C" } }
}
-}
-- | 
sparseAccumulatorApplyGradient :: forall v'2 v'3 v'4 v'5 dtype
                                  m' . (MonadBuild m',
                                        OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] dtype) =>
                                  Bool -- ^ __has_known_shape__
                                  -> Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                                  -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                                  -> Tensor v'3 Data.Int.Int64 -- ^ __gradient_indices__
                                  -> Tensor v'4 dtype -- ^ __gradient_values__
                                  -> Tensor v'5 Data.Int.Int64 -- ^ __gradient_shape__
                                  -> m' (ControlNode)
sparseAccumulatorApplyGradient :: Bool
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 dtype
-> Tensor v'5 Int64
-> m' ControlNode
sparseAccumulatorApplyGradient = OpParams
-> Bool
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 dtype
-> Tensor v'5 Int64
-> m' ControlNode
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *)
       (v'5 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Bool
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 dtype
-> Tensor v'5 Int64
-> m' ControlNode
sparseAccumulatorApplyGradient' OpParams
forall a. a -> a
id
sparseAccumulatorApplyGradient' :: forall v'2 v'3 v'4 v'5 dtype
                                   m' . (MonadBuild m',
                                         OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] dtype) => OpParams ->
                                   Bool -- ^ __has_known_shape__
                                   -> Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                                   -> Tensor v'2 Data.Int.Int64 -- ^ __local_step__
                                   -> Tensor v'3 Data.Int.Int64 -- ^ __gradient_indices__
                                   -> Tensor v'4 dtype -- ^ __gradient_values__
                                   -> Tensor v'5 Data.Int.Int64 -- ^ __gradient_shape__
                                   -> m' (ControlNode)
sparseAccumulatorApplyGradient' :: OpParams
-> Bool
-> Tensor Ref ByteString
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 dtype
-> Tensor v'5 Int64
-> m' ControlNode
sparseAccumulatorApplyGradient' op'options :: OpParams
op'options has_known_shape :: Bool
has_known_shape handle :: Tensor Ref ByteString
handle local_step :: Tensor v'2 Int64
local_step
                                gradient_indices :: Tensor v'3 Int64
gradient_indices gradient_values :: Tensor v'4 dtype
gradient_values
                                gradient_shape :: Tensor v'5 Int64
gradient_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
local_step,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
gradient_indices,
                                                             Tensor v'4 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 dtype
gradient_values,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
gradient_shape]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseAccumulatorApplyGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "has_known_shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
has_known_shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "local_step" type: DT_INT64 }
input_arg { name: "gradient_indices" type: DT_INT64 }
input_arg { name: "gradient_values" type_attr: "dtype" }
input_arg { name: "gradient_shape" type: DT_INT64 }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "has_known_shape" type: "bool" }
-}
-- | 
sparseAccumulatorTakeGradient :: forall v'2 dtype m' . (MonadBuild m',
                                                        OneOf '[(Data.Complex.Complex Double),
                                                                (Data.Complex.Complex Float),
                                                                Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Int.Int8,
                                                                Data.Word.Word16,
                                                                Data.Word.Word32,
                                                                Data.Word.Word64,
                                                                Data.Word.Word8,
                                                                Double,
                                                                Float] dtype) =>
                                 Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                                 -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                                 -> m' ((Tensor Value Data.Int.Int64,
                                         Tensor Value dtype,
                                         Tensor Value Data.Int.Int64))
                                 -- ^ (__indices__, __values__, __shape__)
                                 --
                                 -- * __indices__
                                 --
                                 -- * __values__
                                 --
                                 -- * __shape__
sparseAccumulatorTakeGradient :: Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
sparseAccumulatorTakeGradient = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
sparseAccumulatorTakeGradient' OpParams
forall a. a -> a
id
sparseAccumulatorTakeGradient' :: forall v'2 dtype m' . (MonadBuild m',
                                                         OneOf '[(Data.Complex.Complex Double),
                                                                 (Data.Complex.Complex Float),
                                                                 Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Int.Int8,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word32,
                                                                 Data.Word.Word64,
                                                                 Data.Word.Word8,
                                                                 Double,
                                                                 Float] dtype) =>
                                  OpParams ->
                                  Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                                  -> Tensor v'2 Data.Int.Int32 -- ^ __num_required__
                                  -> m' ((Tensor Value Data.Int.Int64,
                                          Tensor Value dtype,
                                          Tensor Value Data.Int.Int64))
                                  -- ^ (__indices__, __values__, __shape__)
                                  --
                                  -- * __indices__
                                  --
                                  -- * __values__
                                  --
                                  -- * __shape__
sparseAccumulatorTakeGradient' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
sparseAccumulatorTakeGradient' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle
                               num_required :: Tensor v'2 Int32
num_required | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64))
-> Build
     (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_required]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseAccumulatorTakeGradient"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "num_required" type: DT_INT32 }
output_arg { name: "indices" type: DT_INT64 }
output_arg { name: "values" type_attr: "dtype" }
output_arg { name: "shape" type: DT_INT64 }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
             treal . (OneOf '[(Data.Complex.Complex Double),
                              (Data.Complex.Complex Float), Data.Int.Int16,
                              Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                              Data.Word.Word16, Data.Word.Word32,
                              Data.Word.Word64, Data.Word.Word8, Double,
                              Float] t, OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] treal) =>
             Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
             -> Tensor v'2 t -- ^ __a_values__
             -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
             -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
             -> Tensor v'5 t -- ^ __b_values__
             -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
             -> Tensor v'7 treal -- ^ __thresh__
             -> (Tensor Build Data.Int.Int64, Tensor Build t,
                 Tensor Build Data.Int.Int64)
             -- ^ (__sum_indices__, __sum_values__, __sum_shape__)
             --
             -- * __sum_indices__
             --
             -- * __sum_values__
             --
             -- * __sum_shape__
sparseAdd :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor v'7 treal
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseAdd = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor v'7 treal
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       treal.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   treal) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor v'7 treal
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseAdd' OpParams
forall a. a -> a
id
sparseAdd' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
              treal . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Data.Word.Word32,
                               Data.Word.Word64, Data.Word.Word8, Double,
                               Float] t, OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] treal) => OpParams ->
              Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
              -> Tensor v'2 t -- ^ __a_values__
              -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
              -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
              -> Tensor v'5 t -- ^ __b_values__
              -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
              -> Tensor v'7 treal -- ^ __thresh__
              -> (Tensor Build Data.Int.Int64, Tensor Build t,
                  Tensor Build Data.Int.Int64)
              -- ^ (__sum_indices__, __sum_values__, __sum_shape__)
              --
              -- * __sum_indices__
              --
              -- * __sum_values__
              --
              -- * __sum_shape__
sparseAdd' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> Tensor v'7 treal
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseAdd' op'options :: OpParams
op'options a_indices :: Tensor v'1 Int64
a_indices a_values :: Tensor v'2 t
a_values a_shape :: Tensor v'3 Int64
a_shape b_indices :: Tensor v'4 Int64
b_indices b_values :: Tensor v'5 t
b_values b_shape :: Tensor v'6 Int64
b_shape
           thresh :: Tensor v'7 treal
thresh | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
a_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
a_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
a_shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
b_indices,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
b_values,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
b_shape,
                                                             Tensor v'7 treal -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 treal
thresh]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Treal" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ treal -> DataType
forall a. TensorType a => a -> DataType
tensorType (treal
forall a. HasCallStack => a
undefined :: treal)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a_indices" type: DT_INT64 }
input_arg { name: "a_values" type_attr: "T" }
input_arg { name: "a_shape" type: DT_INT64 }
input_arg { name: "b_indices" type: DT_INT64 }
input_arg { name: "b_values" type_attr: "T" }
input_arg { name: "b_shape" type: DT_INT64 }
input_arg { name: "thresh" type_attr: "Treal" }
output_arg { name: "sum_indices" type: DT_INT64 }
output_arg { name: "sum_values" type_attr: "T" }
output_arg { name: "sum_shape" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Treal"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseAddGrad :: forall v'1 v'2 v'3 v'4
                 t . (OneOf '[(Data.Complex.Complex Double),
                              (Data.Complex.Complex Float), Data.Int.Int16,
                              Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                              Data.Word.Word16, Data.Word.Word32,
                              Data.Word.Word64, Data.Word.Word8, Double,
                              Float] t) =>
                 Tensor v'1 t -- ^ __backprop_val_grad__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __a_indices__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __b_indices__
                 -> Tensor v'4 Data.Int.Int64 -- ^ __sum_indices__
                 -> (Tensor Build t, Tensor Build t)
                 -- ^ (__a_val_grad__, __b_val_grad__)
                 --
                 -- * __a_val_grad__
                 --
                 -- * __b_val_grad__
sparseAddGrad :: Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> (Tensor Build t, Tensor Build t)
sparseAddGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> (Tensor Build t, Tensor Build t)
sparseAddGrad' OpParams
forall a. a -> a
id
sparseAddGrad' :: forall v'1 v'2 v'3 v'4
                  t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Data.Word.Word32,
                               Data.Word.Word64, Data.Word.Word8, Double,
                               Float] t) => OpParams ->
                  Tensor v'1 t -- ^ __backprop_val_grad__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __a_indices__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __b_indices__
                  -> Tensor v'4 Data.Int.Int64 -- ^ __sum_indices__
                  -> (Tensor Build t, Tensor Build t)
                  -- ^ (__a_val_grad__, __b_val_grad__)
                  --
                  -- * __a_val_grad__
                  --
                  -- * __b_val_grad__
sparseAddGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> (Tensor Build t, Tensor Build t)
sparseAddGrad' op'options :: OpParams
op'options backprop_val_grad :: Tensor v'1 t
backprop_val_grad a_indices :: Tensor v'2 Int64
a_indices b_indices :: Tensor v'3 Int64
b_indices
               sum_indices :: Tensor v'4 Int64
sum_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
backprop_val_grad,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
a_indices,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
b_indices,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
sum_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseAddGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "backprop_val_grad" type_attr: "T" }
input_arg { name: "a_indices" type: DT_INT64 }
input_arg { name: "b_indices" type: DT_INT64 }
input_arg { name: "sum_indices" type: DT_INT64 }
output_arg { name: "a_val_grad" type_attr: "T" }
output_arg { name: "b_val_grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseApplyAdadelta :: forall v'4 v'5 v'6 v'7 v'8 t tindices
                       m' . (MonadBuild m',
                             OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t, OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
                       Tensor Ref t -- ^ __var__
                       -> Tensor Ref t -- ^ __accum__
                       -> Tensor Ref t -- ^ __accum_update__
                       -> Tensor v'4 t -- ^ __lr__
                       -> Tensor v'5 t -- ^ __rho__
                       -> Tensor v'6 t -- ^ __epsilon__
                       -> Tensor v'7 t -- ^ __grad__
                       -> Tensor v'8 tindices -- ^ __indices__
                       -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdadelta :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' (Tensor Ref t)
sparseApplyAdadelta = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' (Tensor Ref t)
sparseApplyAdadelta' OpParams
forall a. a -> a
id
sparseApplyAdadelta' :: forall v'4 v'5 v'6 v'7 v'8 t tindices
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                        OpParams ->
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __accum__
                        -> Tensor Ref t -- ^ __accum_update__
                        -> Tensor v'4 t -- ^ __lr__
                        -> Tensor v'5 t -- ^ __rho__
                        -> Tensor v'6 t -- ^ __epsilon__
                        -> Tensor v'7 t -- ^ __grad__
                        -> Tensor v'8 tindices -- ^ __indices__
                        -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdadelta' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 tindices
-> m' (Tensor Ref t)
sparseApplyAdadelta' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum accum_update :: Tensor Ref t
accum_update lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho epsilon :: Tensor v'6 t
epsilon grad :: Tensor v'7 t
grad
                     indices :: Tensor v'8 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum_update,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
epsilon,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
grad,
                                                             Tensor v'8 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyAdadelta"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "accum_update" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyAdagrad :: forall v'3 v'4 v'5 t tindices m' . (MonadBuild m',
                                                          OneOf '[(Data.Complex.Complex Double),
                                                                  (Data.Complex.Complex Float),
                                                                  Data.Int.Int16,
                                                                  Data.Int.Int32,
                                                                  Data.Int.Int64,
                                                                  Data.Int.Int8,
                                                                  Data.Word.Word16,
                                                                  Data.Word.Word32,
                                                                  Data.Word.Word64,
                                                                  Data.Word.Word8,
                                                                  Double,
                                                                  Float] t,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] tindices) =>
                      Tensor Ref t -- ^ __var__
                      -> Tensor Ref t -- ^ __accum__
                      -> Tensor v'3 t -- ^ __lr__
                      -> Tensor v'4 t -- ^ __grad__
                      -> Tensor v'5 tindices -- ^ __indices__
                      -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagrad :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' (Tensor Ref t)
sparseApplyAdagrad = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' (Tensor Ref t)
sparseApplyAdagrad' OpParams
forall a. a -> a
id
sparseApplyAdagrad' :: forall v'3 v'4 v'5 t tindices m' . (MonadBuild m',
                                                           OneOf '[(Data.Complex.Complex Double),
                                                                   (Data.Complex.Complex Float),
                                                                   Data.Int.Int16,
                                                                   Data.Int.Int32,
                                                                   Data.Int.Int64,
                                                                   Data.Int.Int8,
                                                                   Data.Word.Word16,
                                                                   Data.Word.Word32,
                                                                   Data.Word.Word64,
                                                                   Data.Word.Word8,
                                                                   Double,
                                                                   Float] t,
                                                           OneOf '[Data.Int.Int32,
                                                                   Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor Ref t -- ^ __var__
                       -> Tensor Ref t -- ^ __accum__
                       -> Tensor v'3 t -- ^ __lr__
                       -> Tensor v'4 t -- ^ __grad__
                       -> Tensor v'5 tindices -- ^ __indices__
                       -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagrad' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> m' (Tensor Ref t)
sparseApplyAdagrad' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
sparseApplyAdagradDA :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                        m' . (MonadBuild m',
                              OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] tindices) =>
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __gradient_accumulator__
                        -> Tensor Ref t -- ^ __gradient_squared_accumulator__
                        -> Tensor v'4 t -- ^ __grad__
                        -> Tensor v'5 tindices -- ^ __indices__
                        -> Tensor v'6 t -- ^ __lr__
                        -> Tensor v'7 t -- ^ __l1__
                        -> Tensor v'8 t -- ^ __l2__
                        -> Tensor v'9 Data.Int.Int64 -- ^ __global_step__
                        -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagradDA :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' (Tensor Ref t)
sparseApplyAdagradDA = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' (Tensor Ref t)
sparseApplyAdagradDA' OpParams
forall a. a -> a
id
sparseApplyAdagradDA' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                         m' . (MonadBuild m',
                               OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor Ref t -- ^ __var__
                         -> Tensor Ref t -- ^ __gradient_accumulator__
                         -> Tensor Ref t -- ^ __gradient_squared_accumulator__
                         -> Tensor v'4 t -- ^ __grad__
                         -> Tensor v'5 tindices -- ^ __indices__
                         -> Tensor v'6 t -- ^ __lr__
                         -> Tensor v'7 t -- ^ __l1__
                         -> Tensor v'8 t -- ^ __l2__
                         -> Tensor v'9 Data.Int.Int64 -- ^ __global_step__
                         -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagradDA' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 Int64
-> m' (Tensor Ref t)
sparseApplyAdagradDA' op'options :: OpParams
op'options var :: Tensor Ref t
var gradient_accumulator :: Tensor Ref t
gradient_accumulator
                      gradient_squared_accumulator :: Tensor Ref t
gradient_squared_accumulator grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                      global_step :: Tensor v'9 Int64
global_step | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
gradient_accumulator,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
gradient_squared_accumulator,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 Int64
global_step]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyAdagradDA"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg {
  name: "gradient_accumulator" type_attr: "T" is_ref: true
}
input_arg {
  name: "gradient_squared_accumulator" type_attr: "T" is_ref: true
}
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "global_step" type: DT_INT64 }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyAdagradV2 :: forall v'3 v'4 v'5 v'6 t tindices m' . (MonadBuild m',
                                                                OneOf '[(Data.Complex.Complex Double),
                                                                        (Data.Complex.Complex Float),
                                                                        Data.Int.Int16,
                                                                        Data.Int.Int32,
                                                                        Data.Int.Int64,
                                                                        Data.Int.Int8,
                                                                        Data.Word.Word16,
                                                                        Data.Word.Word32,
                                                                        Data.Word.Word64,
                                                                        Data.Word.Word8,
                                                                        Double,
                                                                        Float] t,
                                                                OneOf '[Data.Int.Int32,
                                                                        Data.Int.Int64] tindices) =>
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __accum__
                        -> Tensor v'3 t -- ^ __lr__
                        -> Tensor v'4 t -- ^ __epsilon__
                        -> Tensor v'5 t -- ^ __grad__
                        -> Tensor v'6 tindices -- ^ __indices__
                        -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagradV2 :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyAdagradV2 = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyAdagradV2' OpParams
forall a. a -> a
id
sparseApplyAdagradV2' :: forall v'3 v'4 v'5 v'6 t tindices m' . (MonadBuild m',
                                                                 OneOf '[(Data.Complex.Complex Double),
                                                                         (Data.Complex.Complex Float),
                                                                         Data.Int.Int16,
                                                                         Data.Int.Int32,
                                                                         Data.Int.Int64,
                                                                         Data.Int.Int8,
                                                                         Data.Word.Word16,
                                                                         Data.Word.Word32,
                                                                         Data.Word.Word64,
                                                                         Data.Word.Word8,
                                                                         Double,
                                                                         Float] t,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor Ref t -- ^ __var__
                         -> Tensor Ref t -- ^ __accum__
                         -> Tensor v'3 t -- ^ __lr__
                         -> Tensor v'4 t -- ^ __epsilon__
                         -> Tensor v'5 t -- ^ __grad__
                         -> Tensor v'6 tindices -- ^ __indices__
                         -> m' (Tensor Ref t) -- ^ __out__
sparseApplyAdagradV2' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyAdagradV2' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr epsilon :: Tensor v'4 t
epsilon grad :: Tensor v'5 t
grad
                      indices :: Tensor v'6 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
epsilon,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad,
                                                             Tensor v'6 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyAdagradV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "update_slots" type: "bool" default_value { b: true }
}
-}
-- | 
sparseApplyCenteredRMSProp :: forall v'5 v'6 v'7 v'8 v'9 v'10 t tindices
                              m' . (MonadBuild m',
                                    OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tindices) =>
                              Tensor Ref t -- ^ __var__
                              -> Tensor Ref t -- ^ __mg__
                              -> Tensor Ref t -- ^ __ms__
                              -> Tensor Ref t -- ^ __mom__
                              -> Tensor v'5 t -- ^ __lr__
                              -> Tensor v'6 t -- ^ __rho__
                              -> Tensor v'7 t -- ^ __momentum__
                              -> Tensor v'8 t -- ^ __epsilon__
                              -> Tensor v'9 t -- ^ __grad__
                              -> Tensor v'10 tindices -- ^ __indices__
                              -> m' (Tensor Ref t) -- ^ __out__
sparseApplyCenteredRMSProp :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' (Tensor Ref t)
sparseApplyCenteredRMSProp = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' (Tensor Ref t)
forall (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' (Tensor Ref t)
sparseApplyCenteredRMSProp' OpParams
forall a. a -> a
id
sparseApplyCenteredRMSProp' :: forall v'5 v'6 v'7 v'8 v'9 v'10 t tindices
                               m' . (MonadBuild m',
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                               OpParams ->
                               Tensor Ref t -- ^ __var__
                               -> Tensor Ref t -- ^ __mg__
                               -> Tensor Ref t -- ^ __ms__
                               -> Tensor Ref t -- ^ __mom__
                               -> Tensor v'5 t -- ^ __lr__
                               -> Tensor v'6 t -- ^ __rho__
                               -> Tensor v'7 t -- ^ __momentum__
                               -> Tensor v'8 t -- ^ __epsilon__
                               -> Tensor v'9 t -- ^ __grad__
                               -> Tensor v'10 tindices -- ^ __indices__
                               -> m' (Tensor Ref t) -- ^ __out__
sparseApplyCenteredRMSProp' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 tindices
-> m' (Tensor Ref t)
sparseApplyCenteredRMSProp' op'options :: OpParams
op'options var :: Tensor Ref t
var mg :: Tensor Ref t
mg ms :: Tensor Ref t
ms mom :: Tensor Ref t
mom lr :: Tensor v'5 t
lr rho :: Tensor v'6 t
rho momentum :: Tensor v'7 t
momentum epsilon :: Tensor v'8 t
epsilon
                            grad :: Tensor v'9 t
grad indices :: Tensor v'10 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mg,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ms,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mom,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
lr,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
rho,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
momentum,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
epsilon,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
grad,
                                                             Tensor v'10 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyCenteredRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "mg" type_attr: "T" is_ref: true }
input_arg { name: "ms" type_attr: "T" is_ref: true }
input_arg { name: "mom" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyFtrl :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                   m' . (MonadBuild m', OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t,
                         OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
                   Tensor Ref t -- ^ __var__
                   -> Tensor Ref t -- ^ __accum__
                   -> Tensor Ref t -- ^ __linear__
                   -> Tensor v'4 t -- ^ __grad__
                   -> Tensor v'5 tindices -- ^ __indices__
                   -> Tensor v'6 t -- ^ __lr__
                   -> Tensor v'7 t -- ^ __l1__
                   -> Tensor v'8 t -- ^ __l2__
                   -> Tensor v'9 t -- ^ __lr_power__
                   -> m' (Tensor Ref t) -- ^ __out__
sparseApplyFtrl :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
sparseApplyFtrl = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
sparseApplyFtrl' OpParams
forall a. a -> a
id
sparseApplyFtrl' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                    m' . (MonadBuild m', OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t,
                          OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
                    OpParams ->
                    Tensor Ref t -- ^ __var__
                    -> Tensor Ref t -- ^ __accum__
                    -> Tensor Ref t -- ^ __linear__
                    -> Tensor v'4 t -- ^ __grad__
                    -> Tensor v'5 tindices -- ^ __indices__
                    -> Tensor v'6 t -- ^ __lr__
                    -> Tensor v'7 t -- ^ __l1__
                    -> Tensor v'8 t -- ^ __l2__
                    -> Tensor v'9 t -- ^ __lr_power__
                    -> m' (Tensor Ref t) -- ^ __out__
sparseApplyFtrl' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> m' (Tensor Ref t)
sparseApplyFtrl' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum linear :: Tensor Ref t
linear grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                 lr_power :: Tensor v'9 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
lr_power]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyFtrl"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "linear" type_attr: "T" is_ref: true }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
sparseApplyFtrlV2 :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices
                     m' . (MonadBuild m', OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] t,
                           OneOf '[Data.Int.Int32, Data.Int.Int64] tindices) =>
                     Tensor Ref t -- ^ __var__
                     -> Tensor Ref t -- ^ __accum__
                     -> Tensor Ref t -- ^ __linear__
                     -> Tensor v'4 t -- ^ __grad__
                     -> Tensor v'5 tindices -- ^ __indices__
                     -> Tensor v'6 t -- ^ __lr__
                     -> Tensor v'7 t -- ^ __l1__
                     -> Tensor v'8 t -- ^ __l2__
                     -> Tensor v'9 t -- ^ __l2_shrinkage__
                     -> Tensor v'10 t -- ^ __lr_power__
                     -> m' (Tensor Ref t) -- ^ __out__
sparseApplyFtrlV2 :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
sparseApplyFtrlV2 = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) (v'10 :: * -> *) t
       tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
sparseApplyFtrlV2' OpParams
forall a. a -> a
id
sparseApplyFtrlV2' :: forall v'4 v'5 v'6 v'7 v'8 v'9 v'10 t tindices
                      m' . (MonadBuild m',
                            OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t, OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
                      OpParams ->
                      Tensor Ref t -- ^ __var__
                      -> Tensor Ref t -- ^ __accum__
                      -> Tensor Ref t -- ^ __linear__
                      -> Tensor v'4 t -- ^ __grad__
                      -> Tensor v'5 tindices -- ^ __indices__
                      -> Tensor v'6 t -- ^ __lr__
                      -> Tensor v'7 t -- ^ __l1__
                      -> Tensor v'8 t -- ^ __l2__
                      -> Tensor v'9 t -- ^ __l2_shrinkage__
                      -> Tensor v'10 t -- ^ __lr_power__
                      -> m' (Tensor Ref t) -- ^ __out__
sparseApplyFtrlV2' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 t
-> Tensor v'10 t
-> m' (Tensor Ref t)
sparseApplyFtrlV2' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum linear :: Tensor Ref t
linear grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices lr :: Tensor v'6 t
lr l1 :: Tensor v'7 t
l1 l2 :: Tensor v'8 t
l2
                   l2_shrinkage :: Tensor v'9 t
l2_shrinkage lr_power :: Tensor v'10 t
lr_power | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
linear,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
lr,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
l1,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
l2,
                                                             Tensor v'9 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 t
l2_shrinkage,
                                                             Tensor v'10 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'10 t
lr_power]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyFtrlV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "linear" type_attr: "T" is_ref: true }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "l2_shrinkage" type_attr: "T" }
input_arg { name: "lr_power" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "multiply_linear_by_lr"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
sparseApplyMomentum :: forall v'3 v'4 v'5 v'6 t tindices m' . (MonadBuild m',
                                                               OneOf '[(Data.Complex.Complex Double),
                                                                       (Data.Complex.Complex Float),
                                                                       Data.Int.Int16,
                                                                       Data.Int.Int32,
                                                                       Data.Int.Int64,
                                                                       Data.Int.Int8,
                                                                       Data.Word.Word16,
                                                                       Data.Word.Word32,
                                                                       Data.Word.Word64,
                                                                       Data.Word.Word8,
                                                                       Double,
                                                                       Float] t,
                                                               OneOf '[Data.Int.Int32,
                                                                       Data.Int.Int64] tindices) =>
                       Tensor Ref t -- ^ __var__
                       -> Tensor Ref t -- ^ __accum__
                       -> Tensor v'3 t -- ^ __lr__
                       -> Tensor v'4 t -- ^ __grad__
                       -> Tensor v'5 tindices -- ^ __indices__
                       -> Tensor v'6 t -- ^ __momentum__
                       -> m' (Tensor Ref t) -- ^ __out__
sparseApplyMomentum :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' (Tensor Ref t)
sparseApplyMomentum = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' (Tensor Ref t)
sparseApplyMomentum' OpParams
forall a. a -> a
id
sparseApplyMomentum' :: forall v'3 v'4 v'5 v'6 t tindices m' . (MonadBuild m',
                                                                OneOf '[(Data.Complex.Complex Double),
                                                                        (Data.Complex.Complex Float),
                                                                        Data.Int.Int16,
                                                                        Data.Int.Int32,
                                                                        Data.Int.Int64,
                                                                        Data.Int.Int8,
                                                                        Data.Word.Word16,
                                                                        Data.Word.Word32,
                                                                        Data.Word.Word64,
                                                                        Data.Word.Word8,
                                                                        Double,
                                                                        Float] t,
                                                                OneOf '[Data.Int.Int32,
                                                                        Data.Int.Int64] tindices) =>
                        OpParams ->
                        Tensor Ref t -- ^ __var__
                        -> Tensor Ref t -- ^ __accum__
                        -> Tensor v'3 t -- ^ __lr__
                        -> Tensor v'4 t -- ^ __grad__
                        -> Tensor v'5 tindices -- ^ __indices__
                        -> Tensor v'6 t -- ^ __momentum__
                        -> m' (Tensor Ref t) -- ^ __out__
sparseApplyMomentum' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 tindices
-> Tensor v'6 t
-> m' (Tensor Ref t)
sparseApplyMomentum' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr grad :: Tensor v'4 t
grad indices :: Tensor v'5 tindices
indices
                     momentum :: Tensor v'6 t
momentum | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
grad,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
indices,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyMomentum"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "momentum" type_attr: "T" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
attr {
  name: "use_nesterov" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyProximalAdagrad :: forall v'3 v'4 v'5 v'6 v'7 t tindices
                              m' . (MonadBuild m',
                                    OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tindices) =>
                              Tensor Ref t -- ^ __var__
                              -> Tensor Ref t -- ^ __accum__
                              -> Tensor v'3 t -- ^ __lr__
                              -> Tensor v'4 t -- ^ __l1__
                              -> Tensor v'5 t -- ^ __l2__
                              -> Tensor v'6 t -- ^ __grad__
                              -> Tensor v'7 tindices -- ^ __indices__
                              -> m' (Tensor Ref t) -- ^ __out__
sparseApplyProximalAdagrad :: Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' (Tensor Ref t)
sparseApplyProximalAdagrad = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' (Tensor Ref t)
forall (v'3 :: * -> *) (v'4 :: * -> *) (v'5 :: * -> *)
       (v'6 :: * -> *) (v'7 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' (Tensor Ref t)
sparseApplyProximalAdagrad' OpParams
forall a. a -> a
id
sparseApplyProximalAdagrad' :: forall v'3 v'4 v'5 v'6 v'7 t tindices
                               m' . (MonadBuild m',
                                     OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                               OpParams ->
                               Tensor Ref t -- ^ __var__
                               -> Tensor Ref t -- ^ __accum__
                               -> Tensor v'3 t -- ^ __lr__
                               -> Tensor v'4 t -- ^ __l1__
                               -> Tensor v'5 t -- ^ __l2__
                               -> Tensor v'6 t -- ^ __grad__
                               -> Tensor v'7 tindices -- ^ __indices__
                               -> m' (Tensor Ref t) -- ^ __out__
sparseApplyProximalAdagrad' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 tindices
-> m' (Tensor Ref t)
sparseApplyProximalAdagrad' op'options :: OpParams
op'options var :: Tensor Ref t
var accum :: Tensor Ref t
accum lr :: Tensor v'3 t
lr l1 :: Tensor v'4 t
l1 l2 :: Tensor v'5 t
l2 grad :: Tensor v'6 t
grad
                            indices :: Tensor v'7 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
accum,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
lr,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l1,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
l2,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
grad,
                                                             Tensor v'7 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyProximalAdagrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "accum" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyProximalGradientDescent :: forall v'2 v'3 v'4 v'5 v'6 t tindices
                                      m' . (MonadBuild m',
                                            OneOf '[(Data.Complex.Complex Double),
                                                    (Data.Complex.Complex Float),
                                                    Data.Int.Int16,
                                                    Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Int.Int8,
                                                    Data.Word.Word16,
                                                    Data.Word.Word32,
                                                    Data.Word.Word64,
                                                    Data.Word.Word8, Double,
                                                    Float] t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
                                      Tensor Ref t -- ^ __var__
                                      -> Tensor v'2 t -- ^ __alpha__
                                      -> Tensor v'3 t -- ^ __l1__
                                      -> Tensor v'4 t -- ^ __l2__
                                      -> Tensor v'5 t -- ^ __grad__
                                      -> Tensor v'6 tindices -- ^ __indices__
                                      -> m' (Tensor Ref t) -- ^ __out__
sparseApplyProximalGradientDescent :: Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyProximalGradientDescent = OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *)
       (v'5 :: * -> *) (v'6 :: * -> *) t tindices (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyProximalGradientDescent' OpParams
forall a. a -> a
id
sparseApplyProximalGradientDescent' :: forall v'2 v'3 v'4 v'5 v'6 t tindices
                                       m' . (MonadBuild m',
                                             OneOf '[(Data.Complex.Complex Double),
                                                     (Data.Complex.Complex Float),
                                                     Data.Int.Int16,
                                                     Data.Int.Int32,
                                                     Data.Int.Int64,
                                                     Data.Int.Int8,
                                                     Data.Word.Word16,
                                                     Data.Word.Word32,
                                                     Data.Word.Word64,
                                                     Data.Word.Word8, Double,
                                                     Float] t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] tindices) =>
                                       OpParams ->
                                       Tensor Ref t -- ^ __var__
                                       -> Tensor v'2 t -- ^ __alpha__
                                       -> Tensor v'3 t -- ^ __l1__
                                       -> Tensor v'4 t -- ^ __l2__
                                       -> Tensor v'5 t -- ^ __grad__
                                       -> Tensor v'6 tindices -- ^ __indices__
                                       -> m' (Tensor Ref t) -- ^ __out__
sparseApplyProximalGradientDescent' :: OpParams
-> Tensor Ref t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 tindices
-> m' (Tensor Ref t)
sparseApplyProximalGradientDescent' op'options :: OpParams
op'options var :: Tensor Ref t
var alpha :: Tensor v'2 t
alpha l1 :: Tensor v'3 t
l1 l2 :: Tensor v'4 t
l2 grad :: Tensor v'5 t
grad
                                    indices :: Tensor v'6 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
alpha,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
l1,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
l2,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
grad,
                                                             Tensor v'6 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyProximalGradientDescent"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "l1" type_attr: "T" }
input_arg { name: "l2" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseApplyRMSProp :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                      m' . (MonadBuild m',
                            OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t, OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] tindices) =>
                      Tensor Ref t -- ^ __var__
                      -> Tensor Ref t -- ^ __ms__
                      -> Tensor Ref t -- ^ __mom__
                      -> Tensor v'4 t -- ^ __lr__
                      -> Tensor v'5 t -- ^ __rho__
                      -> Tensor v'6 t -- ^ __momentum__
                      -> Tensor v'7 t -- ^ __epsilon__
                      -> Tensor v'8 t -- ^ __grad__
                      -> Tensor v'9 tindices -- ^ __indices__
                      -> m' (Tensor Ref t) -- ^ __out__
sparseApplyRMSProp :: Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' (Tensor Ref t)
sparseApplyRMSProp = OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' (Tensor Ref t)
forall (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *)
       (v'7 :: * -> *) (v'8 :: * -> *) (v'9 :: * -> *) t tindices
       (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' (Tensor Ref t)
sparseApplyRMSProp' OpParams
forall a. a -> a
id
sparseApplyRMSProp' :: forall v'4 v'5 v'6 v'7 v'8 v'9 t tindices
                       m' . (MonadBuild m',
                             OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t, OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor Ref t -- ^ __var__
                       -> Tensor Ref t -- ^ __ms__
                       -> Tensor Ref t -- ^ __mom__
                       -> Tensor v'4 t -- ^ __lr__
                       -> Tensor v'5 t -- ^ __rho__
                       -> Tensor v'6 t -- ^ __momentum__
                       -> Tensor v'7 t -- ^ __epsilon__
                       -> Tensor v'8 t -- ^ __grad__
                       -> Tensor v'9 tindices -- ^ __indices__
                       -> m' (Tensor Ref t) -- ^ __out__
sparseApplyRMSProp' :: OpParams
-> Tensor Ref t
-> Tensor Ref t
-> Tensor Ref t
-> Tensor v'4 t
-> Tensor v'5 t
-> Tensor v'6 t
-> Tensor v'7 t
-> Tensor v'8 t
-> Tensor v'9 tindices
-> m' (Tensor Ref t)
sparseApplyRMSProp' op'options :: OpParams
op'options var :: Tensor Ref t
var ms :: Tensor Ref t
ms mom :: Tensor Ref t
mom lr :: Tensor v'4 t
lr rho :: Tensor v'5 t
rho momentum :: Tensor v'6 t
momentum epsilon :: Tensor v'7 t
epsilon grad :: Tensor v'8 t
grad
                    indices :: Tensor v'9 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
var,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ms,
                                                             Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
mom,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
lr,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
rho,
                                                             Tensor v'6 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 t
momentum,
                                                             Tensor v'7 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 t
epsilon,
                                                             Tensor v'8 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'8 t
grad,
                                                             Tensor v'9 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'9 tindices
indices]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseApplyRMSProp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "var" type_attr: "T" is_ref: true }
input_arg { name: "ms" type_attr: "T" is_ref: true }
input_arg { name: "mom" type_attr: "T" is_ref: true }
input_arg { name: "lr" type_attr: "T" }
input_arg { name: "rho" type_attr: "T" }
input_arg { name: "momentum" type_attr: "T" }
input_arg { name: "epsilon" type_attr: "T" }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
output_arg { name: "out" type_attr: "T" is_ref: true }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "use_locking" type: "bool" default_value { b: false }
}
-}
-- | 
sparseBincount :: forall v'1 v'2 v'3 v'4 v'5 tidx t . (OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tidx,
                                                       OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64,
                                                               Double,
                                                               Float] t) =>
                  Tensor v'1 Data.Int.Int64 -- ^ __indices__
                  -> Tensor v'2 tidx -- ^ __values__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                  -> Tensor v'4 tidx -- ^ __size__
                  -> Tensor v'5 t -- ^ __weights__
                  -> Tensor Build t -- ^ __output__
sparseBincount :: Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 Int64
-> Tensor v'4 tidx
-> Tensor v'5 t
-> Tensor Build t
sparseBincount = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 Int64
-> Tensor v'4 tidx
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) tidx t.
(OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64, Double, Float] t) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 Int64
-> Tensor v'4 tidx
-> Tensor v'5 t
-> Tensor Build t
sparseBincount' OpParams
forall a. a -> a
id
sparseBincount' :: forall v'1 v'2 v'3 v'4 v'5 tidx t . (OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tidx,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Double,
                                                                Float] t) =>
                   OpParams ->
                   Tensor v'1 Data.Int.Int64 -- ^ __indices__
                   -> Tensor v'2 tidx -- ^ __values__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                   -> Tensor v'4 tidx -- ^ __size__
                   -> Tensor v'5 t -- ^ __weights__
                   -> Tensor Build t -- ^ __output__
sparseBincount' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tidx
-> Tensor v'3 Int64
-> Tensor v'4 tidx
-> Tensor v'5 t
-> Tensor Build t
sparseBincount' op'options :: OpParams
op'options indices :: Tensor v'1 Int64
indices values :: Tensor v'2 tidx
values dense_shape :: Tensor v'3 Int64
dense_shape size :: Tensor v'4 tidx
size
                weights :: Tensor v'5 t
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
dense_shape,
                                                             Tensor v'4 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tidx
size,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseBincount"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "Tidx" }
input_arg { name: "dense_shape" type: DT_INT64 }
input_arg { name: "size" type_attr: "Tidx" }
input_arg { name: "weights" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "Tidx"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "binary_output" type: "bool" default_value { b: false }
}
-}
-- | 
sparseConcat :: forall v'1 v'2 v'3 t . (TensorType t) =>
                Data.Int.Int64 -- ^ __concat_dim__
                -> [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                -> [Tensor v'2 t] -- ^ __values__
                -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                -> (Tensor Build Data.Int.Int64, Tensor Build t,
                    Tensor Build Data.Int.Int64)
                -- ^ (__output_indices__, __output_values__, __output_shape__)
                --
                -- * __output_indices__
                --
                -- * __output_values__
                --
                -- * __output_shape__
sparseConcat :: Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 t]
-> [Tensor v'3 Int64]
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseConcat = OpParams
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 t]
-> [Tensor v'3 Int64]
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 t]
-> [Tensor v'3 Int64]
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseConcat' OpParams
forall a. a -> a
id
sparseConcat' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                 Data.Int.Int64 -- ^ __concat_dim__
                 -> [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                 -> [Tensor v'2 t] -- ^ __values__
                 -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                 -> (Tensor Build Data.Int.Int64, Tensor Build t,
                     Tensor Build Data.Int.Int64)
                 -- ^ (__output_indices__, __output_values__, __output_shape__)
                 --
                 -- * __output_indices__
                 --
                 -- * __output_values__
                 --
                 -- * __output_shape__
sparseConcat' :: OpParams
-> Int64
-> [Tensor v'1 Int64]
-> [Tensor v'2 t]
-> [Tensor v'3 Int64]
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseConcat' op'options :: OpParams
op'options concat_dim :: Int64
concat_dim indices :: [Tensor v'1 Int64]
indices values :: [Tensor v'2 t]
values
              shapes :: [Tensor v'3 Int64]
shapes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices),
                                             ("values", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
values),
                                             ("shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
shapes)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
indices,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
values,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
shapes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseConcat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "concat_dim" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
concat_dim
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT64 number_attr: "N" }
input_arg { name: "values" type_attr: "T" number_attr: "N" }
input_arg { name: "shapes" type: DT_INT64 number_attr: "N" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "concat_dim" type: "int" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "T" type: "type" }
-}
-- | 
sparseConditionalAccumulator :: forall m' . (MonadBuild m') =>
                                DataType -- ^ __dtype__
                                -> Shape -- ^ __shape__
                                -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
sparseConditionalAccumulator :: DataType -> Shape -> m' (Tensor Ref ByteString)
sparseConditionalAccumulator = OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
sparseConditionalAccumulator' OpParams
forall a. a -> a
id
sparseConditionalAccumulator' :: forall m' . (MonadBuild m') => OpParams ->
                                 DataType -- ^ __dtype__
                                 -> Shape -- ^ __shape__
                                 -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
sparseConditionalAccumulator' :: OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
sparseConditionalAccumulator' op'options :: OpParams
op'options dtype :: DataType
dtype shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseConditionalAccumulator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "shape" type: "shape" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "reduction_type"
  type: "string"
  default_value { s: "MEAN" }
  allowed_values { list { s: "MEAN" s: "SUM" } }
}
-}
-- | 
sparseCountSparseOutput :: forall v'1 v'2 v'3 v'4 t
                           output_type . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64, Double,
                                                  Float] output_type) =>
                           Bool -- ^ __binary_output__
                           -> Tensor v'1 Data.Int.Int64 -- ^ __indices__
                           -> Tensor v'2 t -- ^ __values__
                           -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                           -> Tensor v'4 output_type -- ^ __weights__
                           -> (Tensor Build Data.Int.Int64,
                               Tensor Build output_type,
                               Tensor Build Data.Int.Int64)
                           -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                           --
                           -- * __output_indices__
                           --
                           -- * __output_values__
                           --
                           -- * __output_dense_shape__
sparseCountSparseOutput :: Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
sparseCountSparseOutput = OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t output_type.
(OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64, Double, Float] output_type) =>
OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
sparseCountSparseOutput' OpParams
forall a. a -> a
id
sparseCountSparseOutput' :: forall v'1 v'2 v'3 v'4 t
                            output_type . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64, Double,
                                                   Float] output_type) =>
                            OpParams ->
                            Bool -- ^ __binary_output__
                            -> Tensor v'1 Data.Int.Int64 -- ^ __indices__
                            -> Tensor v'2 t -- ^ __values__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                            -> Tensor v'4 output_type -- ^ __weights__
                            -> (Tensor Build Data.Int.Int64,
                                Tensor Build output_type,
                                Tensor Build Data.Int.Int64)
                            -- ^ (__output_indices__, __output_values__, __output_dense_shape__)
                            --
                            -- * __output_indices__
                            --
                            -- * __output_values__
                            --
                            -- * __output_dense_shape__
sparseCountSparseOutput' :: OpParams
-> Bool
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 output_type
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
sparseCountSparseOutput' op'options :: OpParams
op'options binary_output :: Bool
binary_output indices :: Tensor v'1 Int64
indices values :: Tensor v'2 t
values dense_shape :: Tensor v'3 Int64
dense_shape
                         weights :: Tensor v'4 output_type
weights | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build output_type,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build output_type,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
dense_shape,
                                                             Tensor v'4 output_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 output_type
weights]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseCountSparseOutput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_type
forall a. HasCallStack => a
undefined :: output_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "binary_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
binary_output
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "dense_shape" type: DT_INT64 }
input_arg { name: "weights" type_attr: "output_type" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "output_type" }
output_arg { name: "output_dense_shape" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "minlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr {
  name: "maxlength"
  type: "int"
  default_value { i: -1 }
  has_minimum: true
  minimum: -1
}
attr { name: "binary_output" type: "bool" }
attr {
  name: "output_type"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
sparseCross :: forall v'1 v'2 v'3 v'4 sparse_types dense_types
               out_type . (OneOfs '[Data.ByteString.ByteString,
                                    Data.Int.Int64] sparse_types,
                           OneOfs '[Data.ByteString.ByteString,
                                    Data.Int.Int64] dense_types,
                           OneOf '[Data.ByteString.ByteString,
                                   Data.Int.Int64] out_type) =>
               Data.Int.Int64 -- ^ __hash_key__
               -> Bool -- ^ __hashed_output__
               -> DataType -- ^ __internal_type__
               -> Data.Int.Int64 -- ^ __num_buckets__
               -> [Tensor v'1 Data.Int.Int64] -- ^ __indices__
               -> TensorList (v'2) sparse_types -- ^ __values__
               -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
               -> TensorList (v'4) dense_types -- ^ __dense_inputs__
               -> (Tensor Build Data.Int.Int64, Tensor Build out_type,
                   Tensor Build Data.Int.Int64)
               -- ^ (__output_indices__, __output_values__, __output_shape__)
               --
               -- * __output_indices__
               --
               -- * __output_values__
               --
               -- * __output_shape__
sparseCross :: Int64
-> Bool
-> DataType
-> Int64
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
sparseCross = OpParams
-> Int64
-> Bool
-> DataType
-> Int64
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (sparse_types :: [*]) (dense_types :: [*])
       out_type.
(OneOfs '[ByteString, Int64] sparse_types,
 OneOfs '[ByteString, Int64] dense_types,
 OneOf '[ByteString, Int64] out_type) =>
OpParams
-> Int64
-> Bool
-> DataType
-> Int64
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
sparseCross' OpParams
forall a. a -> a
id
sparseCross' :: forall v'1 v'2 v'3 v'4 sparse_types dense_types
                out_type . (OneOfs '[Data.ByteString.ByteString,
                                     Data.Int.Int64] sparse_types,
                            OneOfs '[Data.ByteString.ByteString,
                                     Data.Int.Int64] dense_types,
                            OneOf '[Data.ByteString.ByteString,
                                    Data.Int.Int64] out_type) => OpParams ->
                Data.Int.Int64 -- ^ __hash_key__
                -> Bool -- ^ __hashed_output__
                -> DataType -- ^ __internal_type__
                -> Data.Int.Int64 -- ^ __num_buckets__
                -> [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                -> TensorList (v'2) sparse_types -- ^ __values__
                -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                -> TensorList (v'4) dense_types -- ^ __dense_inputs__
                -> (Tensor Build Data.Int.Int64, Tensor Build out_type,
                    Tensor Build Data.Int.Int64)
                -- ^ (__output_indices__, __output_values__, __output_shape__)
                --
                -- * __output_indices__
                --
                -- * __output_values__
                --
                -- * __output_shape__
sparseCross' :: OpParams
-> Int64
-> Bool
-> DataType
-> Int64
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
sparseCross' op'options :: OpParams
op'options hash_key :: Int64
hash_key hashed_output :: Bool
hashed_output internal_type :: DataType
internal_type num_buckets :: Int64
num_buckets indices :: [Tensor v'1 Int64]
indices
             values :: TensorList v'2 sparse_types
values shapes :: [Tensor v'3 Int64]
shapes
             dense_inputs :: TensorList v'4 dense_types
dense_inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices),
                                                  ("shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
shapes)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build out_type, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
indices,
                                                             TensorList v'2 sparse_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 sparse_types
values,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
shapes,
                                                             TensorList v'4 dense_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 dense_types
dense_inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseCross"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dense_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dense_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dense_types
forall k (t :: k). Proxy t
Proxy :: Proxy dense_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "hash_key" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
hash_key
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "hashed_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
hashed_output
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "internal_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
internal_type
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT64 number_attr: "N" }
input_arg { name: "values" type_list_attr: "sparse_types" }
input_arg { name: "shapes" type: DT_INT64 number_attr: "N" }
input_arg { name: "dense_inputs" type_list_attr: "dense_types" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "out_type" }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "N" type: "int" has_minimum: true }
attr { name: "hashed_output" type: "bool" }
attr { name: "num_buckets" type: "int" has_minimum: true }
attr { name: "hash_key" type: "int" }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "dense_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "out_type"
  type: "type"
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "internal_type"
  type: "type"
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
-}
-- | 
sparseCrossHashed :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types
                     dense_types . (OneOfs '[Data.ByteString.ByteString,
                                             Data.Int.Int64] sparse_types,
                                    OneOfs '[Data.ByteString.ByteString,
                                             Data.Int.Int64] dense_types) =>
                     [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                     -> TensorList (v'2) sparse_types -- ^ __values__
                     -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                     -> TensorList (v'4) dense_types -- ^ __dense_inputs__
                     -> Tensor v'5 Data.Int.Int64 -- ^ __num_buckets__
                     -> Tensor v'6 Bool -- ^ __strong_hash__
                     -> Tensor v'7 Data.Int.Int64 -- ^ __salt__
                     -> (Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64,
                         Tensor Build Data.Int.Int64)
                     -- ^ (__output_indices__, __output_values__, __output_shape__)
                     --
                     -- * __output_indices__
                     --
                     -- * __output_values__
                     --
                     -- * __output_shape__
sparseCrossHashed :: [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 Int64
-> Tensor v'6 Bool
-> Tensor v'7 Int64
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
sparseCrossHashed = OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 Int64
-> Tensor v'6 Bool
-> Tensor v'7 Int64
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *)
       (sparse_types :: [*]) (dense_types :: [*]).
(OneOfs '[ByteString, Int64] sparse_types,
 OneOfs '[ByteString, Int64] dense_types) =>
OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 Int64
-> Tensor v'6 Bool
-> Tensor v'7 Int64
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
sparseCrossHashed' OpParams
forall a. a -> a
id
sparseCrossHashed' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 sparse_types
                      dense_types . (OneOfs '[Data.ByteString.ByteString,
                                              Data.Int.Int64] sparse_types,
                                     OneOfs '[Data.ByteString.ByteString,
                                              Data.Int.Int64] dense_types) =>
                      OpParams ->
                      [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                      -> TensorList (v'2) sparse_types -- ^ __values__
                      -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                      -> TensorList (v'4) dense_types -- ^ __dense_inputs__
                      -> Tensor v'5 Data.Int.Int64 -- ^ __num_buckets__
                      -> Tensor v'6 Bool -- ^ __strong_hash__
                      -> Tensor v'7 Data.Int.Int64 -- ^ __salt__
                      -> (Tensor Build Data.Int.Int64,
                          Tensor Build Data.Int.Int64,
                          Tensor Build Data.Int.Int64)
                      -- ^ (__output_indices__, __output_values__, __output_shape__)
                      --
                      -- * __output_indices__
                      --
                      -- * __output_values__
                      --
                      -- * __output_shape__
sparseCrossHashed' :: OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 Int64
-> Tensor v'6 Bool
-> Tensor v'7 Int64
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
sparseCrossHashed' op'options :: OpParams
op'options indices :: [Tensor v'1 Int64]
indices values :: TensorList v'2 sparse_types
values shapes :: [Tensor v'3 Int64]
shapes dense_inputs :: TensorList v'4 dense_types
dense_inputs num_buckets :: Tensor v'5 Int64
num_buckets
                   strong_hash :: Tensor v'6 Bool
strong_hash
                   salt :: Tensor v'7 Int64
salt | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices),
                                                ("shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
shapes)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
indices,
                                                             TensorList v'2 sparse_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 sparse_types
values,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
shapes,
                                                             TensorList v'4 dense_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 dense_types
dense_inputs,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
num_buckets,
                                                             Tensor v'6 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Bool
strong_hash,
                                                             Tensor v'7 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 Int64
salt]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseCrossHashed"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dense_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dense_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dense_types
forall k (t :: k). Proxy t
Proxy :: Proxy dense_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT64 number_attr: "N" }
input_arg { name: "values" type_list_attr: "sparse_types" }
input_arg { name: "shapes" type: DT_INT64 number_attr: "N" }
input_arg { name: "dense_inputs" type_list_attr: "dense_types" }
input_arg { name: "num_buckets" type: DT_INT64 }
input_arg { name: "strong_hash" type: DT_BOOL }
input_arg { name: "salt" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type: DT_INT64 }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "N" type: "int" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "dense_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
-}
-- | 
sparseCrossV2 :: forall v'1 v'2 v'3 v'4 v'5 sparse_types
                 dense_types . (OneOfs '[Data.ByteString.ByteString,
                                         Data.Int.Int64] sparse_types,
                                OneOfs '[Data.ByteString.ByteString,
                                         Data.Int.Int64] dense_types) =>
                 [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                 -> TensorList (v'2) sparse_types -- ^ __values__
                 -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                 -> TensorList (v'4) dense_types -- ^ __dense_inputs__
                 -> Tensor v'5 Data.ByteString.ByteString -- ^ __sep__
                 -> (Tensor Build Data.Int.Int64,
                     Tensor Build Data.ByteString.ByteString,
                     Tensor Build Data.Int.Int64)
                 -- ^ (__output_indices__, __output_values__, __output_shape__)
                 --
                 -- * __output_indices__
                 --
                 -- * __output_values__
                 --
                 -- * __output_shape__
sparseCrossV2 :: [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
sparseCrossV2 = OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (sparse_types :: [*])
       (dense_types :: [*]).
(OneOfs '[ByteString, Int64] sparse_types,
 OneOfs '[ByteString, Int64] dense_types) =>
OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
sparseCrossV2' OpParams
forall a. a -> a
id
sparseCrossV2' :: forall v'1 v'2 v'3 v'4 v'5 sparse_types
                  dense_types . (OneOfs '[Data.ByteString.ByteString,
                                          Data.Int.Int64] sparse_types,
                                 OneOfs '[Data.ByteString.ByteString,
                                          Data.Int.Int64] dense_types) =>
                  OpParams ->
                  [Tensor v'1 Data.Int.Int64] -- ^ __indices__
                  -> TensorList (v'2) sparse_types -- ^ __values__
                  -> [Tensor v'3 Data.Int.Int64] -- ^ __shapes__
                  -> TensorList (v'4) dense_types -- ^ __dense_inputs__
                  -> Tensor v'5 Data.ByteString.ByteString -- ^ __sep__
                  -> (Tensor Build Data.Int.Int64,
                      Tensor Build Data.ByteString.ByteString,
                      Tensor Build Data.Int.Int64)
                  -- ^ (__output_indices__, __output_values__, __output_shape__)
                  --
                  -- * __output_indices__
                  --
                  -- * __output_values__
                  --
                  -- * __output_shape__
sparseCrossV2' :: OpParams
-> [Tensor v'1 Int64]
-> TensorList v'2 sparse_types
-> [Tensor v'3 Int64]
-> TensorList v'4 dense_types
-> Tensor v'5 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
sparseCrossV2' op'options :: OpParams
op'options indices :: [Tensor v'1 Int64]
indices values :: TensorList v'2 sparse_types
values shapes :: [Tensor v'3 Int64]
shapes dense_inputs :: TensorList v'4 dense_types
dense_inputs
               sep :: Tensor v'5 ByteString
sep | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("indices", [Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices),
                                           ("shapes", [Tensor v'3 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 Int64]
shapes)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build ByteString,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int64]
indices,
                                                             TensorList v'2 sparse_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'2 sparse_types
values,
                                                             [Tensor v'3 Int64] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 Int64]
shapes,
                                                             TensorList v'4 dense_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'4 dense_types
dense_inputs,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
sep]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseCrossV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sparse_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy sparse_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy sparse_types
forall k (t :: k). Proxy t
Proxy :: Proxy sparse_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dense_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dense_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dense_types
forall k (t :: k). Proxy t
Proxy :: Proxy dense_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int64] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int64]
indices) :: Int64
{-
input_arg { name: "indices" type: DT_INT64 number_attr: "N" }
input_arg { name: "values" type_list_attr: "sparse_types" }
input_arg { name: "shapes" type: DT_INT64 number_attr: "N" }
input_arg { name: "dense_inputs" type_list_attr: "dense_types" }
input_arg { name: "sep" type: DT_STRING }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type: DT_STRING }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "N" type: "int" has_minimum: true }
attr {
  name: "sparse_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
attr {
  name: "dense_types"
  type: "list(type)"
  has_minimum: true
  allowed_values { list { type: DT_INT64 type: DT_STRING } }
}
-}
-- | 
sparseDenseCwiseAdd :: forall v'1 v'2 v'3 v'4
                       t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                       -> Tensor v'2 t -- ^ __sp_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                       -> Tensor v'4 t -- ^ __dense__
                       -> Tensor Build t -- ^ __output__
sparseDenseCwiseAdd :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseAdd = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseAdd' OpParams
forall a. a -> a
id
sparseDenseCwiseAdd' :: forall v'1 v'2 v'3 v'4
                        t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                        -> Tensor v'2 t -- ^ __sp_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                        -> Tensor v'4 t -- ^ __dense__
                        -> Tensor Build t -- ^ __output__
sparseDenseCwiseAdd' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseAdd' op'options :: OpParams
op'options sp_indices :: Tensor v'1 Int64
sp_indices sp_values :: Tensor v'2 t
sp_values sp_shape :: Tensor v'3 Int64
sp_shape
                     dense :: Tensor v'4 t
dense | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sp_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sp_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sp_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
dense]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseDenseCwiseAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sp_indices" type: DT_INT64 }
input_arg { name: "sp_values" type_attr: "T" }
input_arg { name: "sp_shape" type: DT_INT64 }
input_arg { name: "dense" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseDenseCwiseDiv :: forall v'1 v'2 v'3 v'4
                       t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                       -> Tensor v'2 t -- ^ __sp_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                       -> Tensor v'4 t -- ^ __dense__
                       -> Tensor Build t -- ^ __output__
sparseDenseCwiseDiv :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseDiv = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseDiv' OpParams
forall a. a -> a
id
sparseDenseCwiseDiv' :: forall v'1 v'2 v'3 v'4
                        t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                        -> Tensor v'2 t -- ^ __sp_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                        -> Tensor v'4 t -- ^ __dense__
                        -> Tensor Build t -- ^ __output__
sparseDenseCwiseDiv' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseDiv' op'options :: OpParams
op'options sp_indices :: Tensor v'1 Int64
sp_indices sp_values :: Tensor v'2 t
sp_values sp_shape :: Tensor v'3 Int64
sp_shape
                     dense :: Tensor v'4 t
dense | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sp_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sp_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sp_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
dense]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseDenseCwiseDiv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sp_indices" type: DT_INT64 }
input_arg { name: "sp_values" type_attr: "T" }
input_arg { name: "sp_shape" type: DT_INT64 }
input_arg { name: "dense" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseDenseCwiseMul :: forall v'1 v'2 v'3 v'4
                       t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                       -> Tensor v'2 t -- ^ __sp_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                       -> Tensor v'4 t -- ^ __dense__
                       -> Tensor Build t -- ^ __output__
sparseDenseCwiseMul :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseMul = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseMul' OpParams
forall a. a -> a
id
sparseDenseCwiseMul' :: forall v'1 v'2 v'3 v'4
                        t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                        -> Tensor v'2 t -- ^ __sp_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                        -> Tensor v'4 t -- ^ __dense__
                        -> Tensor Build t -- ^ __output__
sparseDenseCwiseMul' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseDenseCwiseMul' op'options :: OpParams
op'options sp_indices :: Tensor v'1 Int64
sp_indices sp_values :: Tensor v'2 t
sp_values sp_shape :: Tensor v'3 Int64
sp_shape
                     dense :: Tensor v'4 t
dense | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sp_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sp_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sp_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
dense]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseDenseCwiseMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sp_indices" type: DT_INT64 }
input_arg { name: "sp_values" type_attr: "T" }
input_arg { name: "sp_shape" type: DT_INT64 }
input_arg { name: "dense" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseFillEmptyRows :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __indices__
                       -> Tensor v'2 t -- ^ __values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                       -> Tensor v'4 t -- ^ __default_value__
                       -> (Tensor Build Data.Int.Int64, Tensor Build t,
                           Tensor Build Bool, Tensor Build Data.Int.Int64)
                       -- ^ (__output_indices__, __output_values__, __empty_row_indicator__, __reverse_index_map__)
                       --
                       -- * __output_indices__
                       --
                       -- * __output_values__
                       --
                       -- * __empty_row_indicator__
                       --
                       -- * __reverse_index_map__
sparseFillEmptyRows :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
sparseFillEmptyRows = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
sparseFillEmptyRows' OpParams
forall a. a -> a
id
sparseFillEmptyRows' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __indices__
                        -> Tensor v'2 t -- ^ __values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                        -> Tensor v'4 t -- ^ __default_value__
                        -> (Tensor Build Data.Int.Int64, Tensor Build t,
                            Tensor Build Bool, Tensor Build Data.Int.Int64)
                        -- ^ (__output_indices__, __output_values__, __empty_row_indicator__, __reverse_index_map__)
                        --
                        -- * __output_indices__
                        --
                        -- * __output_values__
                        --
                        -- * __empty_row_indicator__
                        --
                        -- * __reverse_index_map__
sparseFillEmptyRows' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
sparseFillEmptyRows' op'options :: OpParams
op'options indices :: Tensor v'1 Int64
indices values :: Tensor v'2 t
values dense_shape :: Tensor v'3 Int64
dense_shape
                     default_value :: Tensor v'4 t
default_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Bool,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
dense_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
default_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseFillEmptyRows"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "dense_shape" type: DT_INT64 }
input_arg { name: "default_value" type_attr: "T" }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
output_arg { name: "empty_row_indicator" type: DT_BOOL }
output_arg { name: "reverse_index_map" type: DT_INT64 }
attr { name: "T" type: "type" }
-}
-- | 
sparseFillEmptyRowsGrad :: forall v'1 v'2 t . (TensorType t) =>
                           Tensor v'1 Data.Int.Int64 -- ^ __reverse_index_map__
                           -> Tensor v'2 t -- ^ __grad_values__
                           -> (Tensor Build t, Tensor Build t)
                           -- ^ (__d_values__, __d_default_value__)
                           --
                           -- * __d_values__
                           --
                           -- * __d_default_value__
sparseFillEmptyRowsGrad :: Tensor v'1 Int64
-> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
sparseFillEmptyRowsGrad = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build t)
sparseFillEmptyRowsGrad' OpParams
forall a. a -> a
id
sparseFillEmptyRowsGrad' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                            Tensor v'1 Data.Int.Int64 -- ^ __reverse_index_map__
                            -> Tensor v'2 t -- ^ __grad_values__
                            -> (Tensor Build t, Tensor Build t)
                            -- ^ (__d_values__, __d_default_value__)
                            --
                            -- * __d_values__
                            --
                            -- * __d_default_value__
sparseFillEmptyRowsGrad' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> (Tensor Build t, Tensor Build t)
sparseFillEmptyRowsGrad' op'options :: OpParams
op'options reverse_index_map :: Tensor v'1 Int64
reverse_index_map
                         grad_values :: Tensor v'2 t
grad_values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
reverse_index_map,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
grad_values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseFillEmptyRowsGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "reverse_index_map" type: DT_INT64 }
input_arg { name: "grad_values" type_attr: "T" }
output_arg { name: "d_values" type_attr: "T" }
output_arg { name: "d_default_value" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
sparseMatMul :: forall v'1 v'2 ta tb . (OneOf '[Data.Word.Word16, Float] ta,
                                        OneOf '[Data.Word.Word16, Float] tb) =>
                Tensor v'1 ta -- ^ __a__
                -> Tensor v'2 tb -- ^ __b__
                -> Tensor Build Float -- ^ __product__
sparseMatMul :: Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
sparseMatMul = OpParams -> Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) ta tb.
(OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb) =>
OpParams -> Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
sparseMatMul' OpParams
forall a. a -> a
id
sparseMatMul' :: forall v'1 v'2 ta tb . (OneOf '[Data.Word.Word16, Float] ta,
                                         OneOf '[Data.Word.Word16, Float] tb) =>
                 OpParams ->
                 Tensor v'1 ta -- ^ __a__
                 -> Tensor v'2 tb -- ^ __b__
                 -> Tensor Build Float -- ^ __product__
sparseMatMul' :: OpParams -> Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
sparseMatMul' op'options :: OpParams
op'options a :: Tensor v'1 ta
a b :: Tensor v'2 tb
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ta -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ta
a,
                                                             Tensor v'2 tb -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tb
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Ta" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ta -> DataType
forall a. TensorType a => a -> DataType
tensorType (ta
forall a. HasCallStack => a
undefined :: ta)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tb" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tb -> DataType
forall a. TensorType a => a -> DataType
tensorType (tb
forall a. HasCallStack => a
undefined :: tb)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "Ta" }
input_arg { name: "b" type_attr: "Tb" }
output_arg { name: "product" type: DT_FLOAT }
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "a_is_sparse" type: "bool" default_value { b: false }
}
attr {
  name: "b_is_sparse" type: "bool" default_value { b: false }
}
attr {
  name: "Ta"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_BFLOAT16 } }
}
attr {
  name: "Tb"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values { list { type: DT_FLOAT type: DT_BFLOAT16 } }
}
-}
-- | 
sparseMatrixAdd :: forall v'1 v'2 v'3 v'4
                   t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Double,
                                Float] t) => Tensor v'1 Variant -- ^ __a__
                   -> Tensor v'2 Variant -- ^ __b__
                   -> Tensor v'3 t -- ^ __alpha__
                   -> Tensor v'4 t -- ^ __beta__
                   -> Tensor Build Variant -- ^ __c__
sparseMatrixAdd :: Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build Variant
sparseMatrixAdd = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build Variant
sparseMatrixAdd' OpParams
forall a. a -> a
id
sparseMatrixAdd' :: forall v'1 v'2 v'3 v'4
                    t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Double,
                                 Float] t) => OpParams ->
                    Tensor v'1 Variant -- ^ __a__
                    -> Tensor v'2 Variant -- ^ __b__
                    -> Tensor v'3 t -- ^ __alpha__
                    -> Tensor v'4 t -- ^ __beta__
                    -> Tensor Build Variant -- ^ __c__
sparseMatrixAdd' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build Variant
sparseMatrixAdd' op'options :: OpParams
op'options a :: Tensor v'1 Variant
a b :: Tensor v'2 Variant
b alpha :: Tensor v'3 t
alpha beta :: Tensor v'4 t
beta | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
a,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
b,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
alpha,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
beta]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type: DT_VARIANT }
input_arg { name: "b" type: DT_VARIANT }
input_arg { name: "alpha" type_attr: "T" }
input_arg { name: "beta" type_attr: "T" }
output_arg { name: "c" type: DT_VARIANT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sparseMatrixMatMul :: forall v'1 v'2 t . (TensorType t) =>
                      Tensor v'1 Variant -- ^ __a__
                      -> Tensor v'2 t -- ^ __b__
                      -> Tensor Build t -- ^ __output__
sparseMatrixMatMul :: Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
sparseMatrixMatMul = OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
sparseMatrixMatMul' OpParams
forall a. a -> a
id
sparseMatrixMatMul' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                       Tensor v'1 Variant -- ^ __a__
                       -> Tensor v'2 t -- ^ __b__
                       -> Tensor Build t -- ^ __output__
sparseMatrixMatMul' :: OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
sparseMatrixMatMul' op'options :: OpParams
op'options a :: Tensor v'1 Variant
a b :: Tensor v'2 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type: DT_VARIANT }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr { name: "adjoint_a" type: "bool" default_value { b: false } }
attr { name: "adjoint_b" type: "bool" default_value { b: false } }
attr {
  name: "transpose_output" type: "bool" default_value { b: false }
}
attr {
  name: "conjugate_output" type: "bool" default_value { b: false }
}
-}
-- | 
sparseMatrixMul :: forall v'1 v'2 t . (TensorType t) =>
                   Tensor v'1 Variant -- ^ __a__
                   -> Tensor v'2 t -- ^ __b__
                   -> Tensor Build Variant -- ^ __output__
sparseMatrixMul :: Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
sparseMatrixMul = OpParams
-> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
sparseMatrixMul' OpParams
forall a. a -> a
id
sparseMatrixMul' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                    Tensor v'1 Variant -- ^ __a__
                    -> Tensor v'2 t -- ^ __b__
                    -> Tensor Build Variant -- ^ __output__
sparseMatrixMul' :: OpParams
-> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
sparseMatrixMul' op'options :: OpParams
op'options a :: Tensor v'1 Variant
a b :: Tensor v'2 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type: DT_VARIANT }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "output" type: DT_VARIANT }
attr { name: "T" type: "type" }
-}
-- | 
sparseMatrixNNZ :: Tensor v'1 Variant -- ^ __sparse_matrix__
                   -> Tensor Build Data.Int.Int32 -- ^ __nnz__
sparseMatrixNNZ :: Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixNNZ = OpParams -> Tensor v'1 Variant -> Tensor Build Int32
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixNNZ' OpParams
forall a. a -> a
id
sparseMatrixNNZ' :: OpParams ->
                    Tensor v'1 Variant -- ^ __sparse_matrix__
                    -> Tensor Build Data.Int.Int32 -- ^ __nnz__
sparseMatrixNNZ' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixNNZ' op'options :: OpParams
op'options sparse_matrix :: Tensor v'1 Variant
sparse_matrix | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
sparse_matrix]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixNNZ"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_matrix" type: DT_VARIANT }
output_arg { name: "nnz" type: DT_INT32 }
-}
-- | 
sparseMatrixOrderingAMD :: Tensor v'1 Variant -- ^ __input__
                           -> Tensor Build Data.Int.Int32 -- ^ __output__
sparseMatrixOrderingAMD :: Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixOrderingAMD = OpParams -> Tensor v'1 Variant -> Tensor Build Int32
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixOrderingAMD' OpParams
forall a. a -> a
id
sparseMatrixOrderingAMD' :: OpParams ->
                            Tensor v'1 Variant -- ^ __input__
                            -> Tensor Build Data.Int.Int32 -- ^ __output__
sparseMatrixOrderingAMD' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
sparseMatrixOrderingAMD' op'options :: OpParams
op'options input :: Tensor v'1 Variant
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixOrderingAMD"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_VARIANT }
output_arg { name: "output" type: DT_INT32 }
-}
-- | 
sparseMatrixSoftmax :: DataType -- ^ __type__
                       -> Tensor v'1 Variant -- ^ __logits__
                       -> Tensor Build Variant -- ^ __softmax__
sparseMatrixSoftmax :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixSoftmax = OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixSoftmax' OpParams
forall a. a -> a
id
sparseMatrixSoftmax' :: OpParams ->
                        DataType -- ^ __type__
                        -> Tensor v'1 Variant -- ^ __logits__
                        -> Tensor Build Variant -- ^ __softmax__
sparseMatrixSoftmax' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixSoftmax' op'options :: OpParams
op'options type' :: DataType
type' logits :: Tensor v'1 Variant
logits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
logits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixSoftmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "logits" type: DT_VARIANT }
output_arg { name: "softmax" type: DT_VARIANT }
attr {
  name: "type"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
sparseMatrixSoftmaxGrad :: DataType -- ^ __type__
                           -> Tensor v'1 Variant -- ^ __softmax__
                           -> Tensor v'2 Variant -- ^ __grad_softmax__
                           -> Tensor Build Variant -- ^ __gradient__
sparseMatrixSoftmaxGrad :: DataType
-> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixSoftmaxGrad = OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
sparseMatrixSoftmaxGrad' OpParams
forall a. a -> a
id
sparseMatrixSoftmaxGrad' :: OpParams ->
                            DataType -- ^ __type__
                            -> Tensor v'1 Variant -- ^ __softmax__
                            -> Tensor v'2 Variant -- ^ __grad_softmax__
                            -> Tensor Build Variant -- ^ __gradient__
sparseMatrixSoftmaxGrad' :: OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
sparseMatrixSoftmaxGrad' op'options :: OpParams
op'options type' :: DataType
type' softmax :: Tensor v'1 Variant
softmax
                         grad_softmax :: Tensor v'2 Variant
grad_softmax | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
softmax,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
grad_softmax]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixSoftmaxGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "softmax" type: DT_VARIANT }
input_arg { name: "grad_softmax" type: DT_VARIANT }
output_arg { name: "gradient" type: DT_VARIANT }
attr {
  name: "type"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
sparseMatrixSparseCholesky :: DataType -- ^ __type__
                              -> Tensor v'1 Variant -- ^ __input__
                              -> Tensor v'2 Data.Int.Int32 -- ^ __permutation__
                              -> Tensor Build Variant -- ^ __output__
sparseMatrixSparseCholesky :: DataType
-> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
sparseMatrixSparseCholesky = OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build Variant
sparseMatrixSparseCholesky' OpParams
forall a. a -> a
id
sparseMatrixSparseCholesky' :: OpParams ->
                               DataType -- ^ __type__
                               -> Tensor v'1 Variant -- ^ __input__
                               -> Tensor v'2 Data.Int.Int32 -- ^ __permutation__
                               -> Tensor Build Variant -- ^ __output__
sparseMatrixSparseCholesky' :: OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build Variant
sparseMatrixSparseCholesky' op'options :: OpParams
op'options type' :: DataType
type' input :: Tensor v'1 Variant
input
                            permutation :: Tensor v'2 Int32
permutation | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
permutation]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixSparseCholesky"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_VARIANT }
input_arg { name: "permutation" type: DT_INT32 }
output_arg { name: "output" type: DT_VARIANT }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sparseMatrixSparseMatMul :: DataType -- ^ __type__
                            -> Tensor v'1 Variant -- ^ __a__
                            -> Tensor v'2 Variant -- ^ __b__
                            -> Tensor Build Variant -- ^ __c__
sparseMatrixSparseMatMul :: DataType
-> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
sparseMatrixSparseMatMul = OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
sparseMatrixSparseMatMul' OpParams
forall a. a -> a
id
sparseMatrixSparseMatMul' :: OpParams ->
                             DataType -- ^ __type__
                             -> Tensor v'1 Variant -- ^ __a__
                             -> Tensor v'2 Variant -- ^ __b__
                             -> Tensor Build Variant -- ^ __c__
sparseMatrixSparseMatMul' :: OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
sparseMatrixSparseMatMul' op'options :: OpParams
op'options type' :: DataType
type' a :: Tensor v'1 Variant
a b :: Tensor v'2 Variant
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
a,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixSparseMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type: DT_VARIANT }
input_arg { name: "b" type: DT_VARIANT }
output_arg { name: "c" type: DT_VARIANT }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr { name: "adjoint_a" type: "bool" default_value { b: false } }
attr { name: "adjoint_b" type: "bool" default_value { b: false } }
-}
-- | 
sparseMatrixTranspose :: DataType -- ^ __type__
                         -> Tensor v'1 Variant -- ^ __input__
                         -> Tensor Build Variant -- ^ __output__
sparseMatrixTranspose :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixTranspose = OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixTranspose' OpParams
forall a. a -> a
id
sparseMatrixTranspose' :: OpParams ->
                          DataType -- ^ __type__
                          -> Tensor v'1 Variant -- ^ __input__
                          -> Tensor Build Variant -- ^ __output__
sparseMatrixTranspose' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
sparseMatrixTranspose' op'options :: OpParams
op'options type' :: DataType
type' input :: Tensor v'1 Variant
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixTranspose"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_VARIANT }
output_arg { name: "output" type: DT_VARIANT }
attr { name: "conjugate" type: "bool" default_value { b: false } }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sparseMatrixZeros :: DataType -- ^ __type__
                     -> Tensor v'1 Data.Int.Int64 -- ^ __dense_shape__
                     -> Tensor Build Variant -- ^ __sparse_matrix__
sparseMatrixZeros :: DataType -> Tensor v'1 Int64 -> Tensor Build Variant
sparseMatrixZeros = OpParams -> DataType -> Tensor v'1 Int64 -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> DataType -> Tensor v'1 Int64 -> Tensor Build Variant
sparseMatrixZeros' OpParams
forall a. a -> a
id
sparseMatrixZeros' :: OpParams ->
                      DataType -- ^ __type__
                      -> Tensor v'1 Data.Int.Int64 -- ^ __dense_shape__
                      -> Tensor Build Variant -- ^ __sparse_matrix__
sparseMatrixZeros' :: OpParams -> DataType -> Tensor v'1 Int64 -> Tensor Build Variant
sparseMatrixZeros' op'options :: OpParams
op'options type' :: DataType
type' dense_shape :: Tensor v'1 Int64
dense_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
dense_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseMatrixZeros"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
type'
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "dense_shape" type: DT_INT64 }
output_arg { name: "sparse_matrix" type: DT_VARIANT }
attr {
  name: "type"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sparseReduceMax :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                       Data.Int.Int32,
                                                       Data.Int.Int64,
                                                       Data.Int.Int8,
                                                       Data.Word.Word16,
                                                       Data.Word.Word32,
                                                       Data.Word.Word64,
                                                       Data.Word.Word8, Double,
                                                       Float] t) =>
                   Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                   -> Tensor v'2 t -- ^ __input_values__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                   -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                   -> Tensor Build t -- ^ __output__
sparseReduceMax :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceMax = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceMax' OpParams
forall a. a -> a
id
sparseReduceMax' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                        Data.Int.Int32,
                                                        Data.Int.Int64,
                                                        Data.Int.Int8,
                                                        Data.Word.Word16,
                                                        Data.Word.Word32,
                                                        Data.Word.Word64,
                                                        Data.Word.Word8, Double,
                                                        Float] t) => OpParams ->
                    Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                    -> Tensor v'2 t -- ^ __input_values__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                    -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                    -> Tensor Build t -- ^ __output__
sparseReduceMax' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceMax' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_values :: Tensor v'2 t
input_values input_shape :: Tensor v'3 Int64
input_shape
                 reduction_axes :: Tensor v'4 Int32
reduction_axes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_shape,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
reduction_axes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReduceMax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_values" type_attr: "T" }
input_arg { name: "input_shape" type: DT_INT64 }
input_arg { name: "reduction_axes" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseReduceMaxSparse :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                         Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                         -> Tensor v'2 t -- ^ __input_values__
                         -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                         -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                         -> (Tensor Build Data.Int.Int64, Tensor Build t,
                             Tensor Build Data.Int.Int64)
                         -- ^ (__output_indices__, __output_values__, __output_shape__)
                         --
                         -- * __output_indices__
                         --
                         -- * __output_values__
                         --
                         -- * __output_shape__
sparseReduceMaxSparse :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceMaxSparse = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceMaxSparse' OpParams
forall a. a -> a
id
sparseReduceMaxSparse' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                          OpParams ->
                          Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                          -> Tensor v'2 t -- ^ __input_values__
                          -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                          -> (Tensor Build Data.Int.Int64, Tensor Build t,
                              Tensor Build Data.Int.Int64)
                          -- ^ (__output_indices__, __output_values__, __output_shape__)
                          --
                          -- * __output_indices__
                          --
                          -- * __output_values__
                          --
                          -- * __output_shape__
sparseReduceMaxSparse' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceMaxSparse' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_values :: Tensor v'2 t
input_values input_shape :: Tensor v'3 Int64
input_shape
                       reduction_axes :: Tensor v'4 Int32
reduction_axes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_shape,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
reduction_axes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReduceMaxSparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_values" type_attr: "T" }
input_arg { name: "input_shape" type: DT_INT64 }
input_arg { name: "reduction_axes" type: DT_INT32 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseReduceSum :: forall v'1 v'2 v'3 v'4
                   t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Data.Word.Word32,
                                Data.Word.Word64, Data.Word.Word8, Double,
                                Float] t) =>
                   Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                   -> Tensor v'2 t -- ^ __input_values__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                   -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                   -> Tensor Build t -- ^ __output__
sparseReduceSum :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceSum = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceSum' OpParams
forall a. a -> a
id
sparseReduceSum' :: forall v'1 v'2 v'3 v'4
                    t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Data.Word.Word32,
                                 Data.Word.Word64, Data.Word.Word8, Double,
                                 Float] t) => OpParams ->
                    Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                    -> Tensor v'2 t -- ^ __input_values__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                    -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                    -> Tensor Build t -- ^ __output__
sparseReduceSum' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> Tensor Build t
sparseReduceSum' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_values :: Tensor v'2 t
input_values input_shape :: Tensor v'3 Int64
input_shape
                 reduction_axes :: Tensor v'4 Int32
reduction_axes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_shape,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
reduction_axes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReduceSum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_values" type_attr: "T" }
input_arg { name: "input_shape" type: DT_INT64 }
input_arg { name: "reduction_axes" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseReduceSumSparse :: forall v'1 v'2 v'3 v'4
                         t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) =>
                         Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                         -> Tensor v'2 t -- ^ __input_values__
                         -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                         -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                         -> (Tensor Build Data.Int.Int64, Tensor Build t,
                             Tensor Build Data.Int.Int64)
                         -- ^ (__output_indices__, __output_values__, __output_shape__)
                         --
                         -- * __output_indices__
                         --
                         -- * __output_values__
                         --
                         -- * __output_shape__
sparseReduceSumSparse :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceSumSparse = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceSumSparse' OpParams
forall a. a -> a
id
sparseReduceSumSparse' :: forall v'1 v'2 v'3 v'4
                          t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
                          Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                          -> Tensor v'2 t -- ^ __input_values__
                          -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __reduction_axes__
                          -> (Tensor Build Data.Int.Int64, Tensor Build t,
                              Tensor Build Data.Int.Int64)
                          -- ^ (__output_indices__, __output_values__, __output_shape__)
                          --
                          -- * __output_indices__
                          --
                          -- * __output_values__
                          --
                          -- * __output_shape__
sparseReduceSumSparse' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int32
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseReduceSumSparse' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_values :: Tensor v'2 t
input_values input_shape :: Tensor v'3 Int64
input_shape
                       reduction_axes :: Tensor v'4 Int32
reduction_axes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_shape,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
reduction_axes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReduceSumSparse"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_values" type_attr: "T" }
input_arg { name: "input_shape" type: DT_INT64 }
input_arg { name: "reduction_axes" type: DT_INT32 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseReorder :: forall v'1 v'2 v'3 t . (TensorType t) =>
                 Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                 -> Tensor v'2 t -- ^ __input_values__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                 -> (Tensor Build Data.Int.Int64, Tensor Build t)
                 -- ^ (__output_indices__, __output_values__)
                 --
                 -- * __output_indices__
                 --
                 -- * __output_values__
sparseReorder :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseReorder = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseReorder' OpParams
forall a. a -> a
id
sparseReorder' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                  Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                  -> Tensor v'2 t -- ^ __input_values__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __input_shape__
                  -> (Tensor Build Data.Int.Int64, Tensor Build t)
                  -- ^ (__output_indices__, __output_values__)
                  --
                  -- * __output_indices__
                  --
                  -- * __output_values__
sparseReorder' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseReorder' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_values :: Tensor v'2 t
input_values
               input_shape :: Tensor v'3 Int64
input_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build t))
-> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
input_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReorder"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_values" type_attr: "T" }
input_arg { name: "input_shape" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
sparseReshape :: Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __input_shape__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __new_shape__
                 -> (Tensor Build Data.Int.Int64, Tensor Build Data.Int.Int64)
                 -- ^ (__output_indices__, __output_shape__)
                 --
                 -- * __output_indices__
                 --
                 -- * __output_shape__
sparseReshape :: Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Int64)
sparseReshape = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *).
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Int64)
sparseReshape' OpParams
forall a. a -> a
id
sparseReshape' :: OpParams ->
                  Tensor v'1 Data.Int.Int64 -- ^ __input_indices__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __input_shape__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __new_shape__
                  -> (Tensor Build Data.Int.Int64, Tensor Build Data.Int.Int64)
                  -- ^ (__output_indices__, __output_shape__)
                  --
                  -- * __output_indices__
                  --
                  -- * __output_shape__
sparseReshape' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> (Tensor Build Int64, Tensor Build Int64)
sparseReshape' op'options :: OpParams
op'options input_indices :: Tensor v'1 Int64
input_indices input_shape :: Tensor v'2 Int64
input_shape
               new_shape :: Tensor v'3 Int64
new_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build Int64))
-> Build OpDef -> (Tensor Build Int64, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
input_indices,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
input_shape,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
new_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseReshape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_shape" type: DT_INT64 }
input_arg { name: "new_shape" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_shape" type: DT_INT64 }
-}
-- | 
sparseSegmentMean :: forall v'1 v'2 v'3 t tidx
                     tsegmentids . (OneOf '[Data.Word.Word16, Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tidx,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tsegmentids) =>
                     Tensor v'1 t -- ^ __data__
                     -> Tensor v'2 tidx -- ^ __indices__
                     -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                     -> Tensor Build t -- ^ __output__
sparseSegmentMean :: Tensor v'1 t
-> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentMean = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx
       tsegmentids.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentMean' OpParams
forall a. a -> a
id
sparseSegmentMean' :: forall v'1 v'2 v'3 t tidx
                      tsegmentids . (OneOf '[Data.Word.Word16, Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tidx,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tsegmentids) =>
                      OpParams ->
                      Tensor v'1 t -- ^ __data__
                      -> Tensor v'2 tidx -- ^ __indices__
                      -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                      -> Tensor Build t -- ^ __output__
sparseSegmentMean' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentMean' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentMean"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentMeanGrad :: forall v'1 v'2 v'3 v'4 t tidx
                         tsegmentids . (OneOf '[Double, Float] t,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tidx,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tsegmentids) =>
                         Tensor v'1 t -- ^ __grad__
                         -> Tensor v'2 tidx -- ^ __indices__
                         -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                         -> Tensor v'4 Data.Int.Int32 -- ^ __output_dim0__
                         -> Tensor Build t -- ^ __output__
sparseSegmentMeanGrad :: Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentMeanGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tidx tsegmentids.
(OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentMeanGrad' OpParams
forall a. a -> a
id
sparseSegmentMeanGrad' :: forall v'1 v'2 v'3 v'4 t tidx
                          tsegmentids . (OneOf '[Double, Float] t,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tidx,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tsegmentids) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __grad__
                          -> Tensor v'2 tidx -- ^ __indices__
                          -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __output_dim0__
                          -> Tensor Build t -- ^ __output__
sparseSegmentMeanGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentMeanGrad' op'options :: OpParams
op'options grad :: Tensor v'1 t
grad indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids
                       output_dim0 :: Tensor v'4 Int32
output_dim0 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
grad,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
output_dim0]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentMeanGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
input_arg { name: "output_dim0" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentMeanWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                    tsegmentids . (OneOf '[Data.Word.Word16,
                                                           Double, Float] t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tidx,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tnumsegments,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tsegmentids) =>
                                    Tensor v'1 t -- ^ __data__
                                    -> Tensor v'2 tidx -- ^ __indices__
                                    -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                    -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                    -> Tensor Build t -- ^ __output__
sparseSegmentMeanWithNumSegments :: Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentMeanWithNumSegments = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tidx tnumsegments tsegmentids.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tnumsegments,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentMeanWithNumSegments' OpParams
forall a. a -> a
id
sparseSegmentMeanWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                     tsegmentids . (OneOf '[Data.Word.Word16,
                                                            Double, Float] t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tidx,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tnumsegments,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tsegmentids) =>
                                     OpParams ->
                                     Tensor v'1 t -- ^ __data__
                                     -> Tensor v'2 tidx -- ^ __indices__
                                     -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                     -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                     -> Tensor Build t -- ^ __output__
sparseSegmentMeanWithNumSegments' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentMeanWithNumSegments' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids
                                  num_segments :: Tensor v'4 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids,
                                                             Tensor v'4 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentMeanWithNumSegments"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentSqrtN :: forall v'1 v'2 v'3 t tidx
                      tsegmentids . (OneOf '[Data.Word.Word16, Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tidx,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tsegmentids) =>
                      Tensor v'1 t -- ^ __data__
                      -> Tensor v'2 tidx -- ^ __indices__
                      -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                      -> Tensor Build t -- ^ __output__
sparseSegmentSqrtN :: Tensor v'1 t
-> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSqrtN = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx
       tsegmentids.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentSqrtN' OpParams
forall a. a -> a
id
sparseSegmentSqrtN' :: forall v'1 v'2 v'3 t tidx
                       tsegmentids . (OneOf '[Data.Word.Word16, Double,
                                              Float] t, OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tidx,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tsegmentids) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __data__
                       -> Tensor v'2 tidx -- ^ __indices__
                       -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                       -> Tensor Build t -- ^ __output__
sparseSegmentSqrtN' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentSqrtN' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentSqrtN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentSqrtNGrad :: forall v'1 v'2 v'3 v'4 t tidx
                          tsegmentids . (OneOf '[Double, Float] t,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tidx,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tsegmentids) =>
                          Tensor v'1 t -- ^ __grad__
                          -> Tensor v'2 tidx -- ^ __indices__
                          -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                          -> Tensor v'4 Data.Int.Int32 -- ^ __output_dim0__
                          -> Tensor Build t -- ^ __output__
sparseSegmentSqrtNGrad :: Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentSqrtNGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tidx tsegmentids.
(OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentSqrtNGrad' OpParams
forall a. a -> a
id
sparseSegmentSqrtNGrad' :: forall v'1 v'2 v'3 v'4 t tidx
                           tsegmentids . (OneOf '[Double, Float] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tidx,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tsegmentids) =>
                           OpParams ->
                           Tensor v'1 t -- ^ __grad__
                           -> Tensor v'2 tidx -- ^ __indices__
                           -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                           -> Tensor v'4 Data.Int.Int32 -- ^ __output_dim0__
                           -> Tensor Build t -- ^ __output__
sparseSegmentSqrtNGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 Int32
-> Tensor Build t
sparseSegmentSqrtNGrad' op'options :: OpParams
op'options grad :: Tensor v'1 t
grad indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids
                        output_dim0 :: Tensor v'4 Int32
output_dim0 | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
grad,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
output_dim0]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentSqrtNGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
input_arg { name: "output_dim0" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentSqrtNWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                     tsegmentids . (OneOf '[Data.Word.Word16,
                                                            Double, Float] t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tidx,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tnumsegments,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tsegmentids) =>
                                     Tensor v'1 t -- ^ __data__
                                     -> Tensor v'2 tidx -- ^ __indices__
                                     -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                     -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                     -> Tensor Build t -- ^ __output__
sparseSegmentSqrtNWithNumSegments :: Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSqrtNWithNumSegments = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tidx tnumsegments tsegmentids.
(OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx,
 OneOf '[Int32, Int64] tnumsegments,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSqrtNWithNumSegments' OpParams
forall a. a -> a
id
sparseSegmentSqrtNWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                      tsegmentids . (OneOf '[Data.Word.Word16,
                                                             Double, Float] t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tidx,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tnumsegments,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tsegmentids) =>
                                      OpParams ->
                                      Tensor v'1 t -- ^ __data__
                                      -> Tensor v'2 tidx -- ^ __indices__
                                      -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                      -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                      -> Tensor Build t -- ^ __output__
sparseSegmentSqrtNWithNumSegments' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSqrtNWithNumSegments' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids
                                   num_segments :: Tensor v'4 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids,
                                                             Tensor v'4 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentSqrtNWithNumSegments"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentSum :: forall v'1 v'2 v'3 t tidx
                    tsegmentids . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t,
                                   OneOf '[Data.Int.Int32, Data.Int.Int64] tidx,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tsegmentids) =>
                    Tensor v'1 t -- ^ __data__
                    -> Tensor v'2 tidx -- ^ __indices__
                    -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                    -> Tensor Build t -- ^ __output__
sparseSegmentSum :: Tensor v'1 t
-> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
sparseSegmentSum = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx
       tsegmentids.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentSum' OpParams
forall a. a -> a
id
sparseSegmentSum' :: forall v'1 v'2 v'3 t tidx
                     tsegmentids . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tidx,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tsegmentids) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __data__
                     -> Tensor v'2 tidx -- ^ __indices__
                     -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                     -> Tensor Build t -- ^ __output__
sparseSegmentSum' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
sparseSegmentSum' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentSum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSegmentSumWithNumSegments :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                   tsegmentids . (OneOf '[Data.Int.Int16,
                                                          Data.Int.Int32,
                                                          Data.Int.Int64,
                                                          Data.Int.Int8,
                                                          Data.Word.Word16,
                                                          Data.Word.Word32,
                                                          Data.Word.Word64,
                                                          Data.Word.Word8,
                                                          Double, Float] t,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tidx,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tnumsegments,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tsegmentids) =>
                                   Tensor v'1 t -- ^ __data__
                                   -> Tensor v'2 tidx -- ^ __indices__
                                   -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                   -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                   -> Tensor Build t -- ^ __output__
sparseSegmentSumWithNumSegments :: Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSumWithNumSegments = OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tidx tnumsegments tsegmentids.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tnumsegments,
 OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSumWithNumSegments' OpParams
forall a. a -> a
id
sparseSegmentSumWithNumSegments' :: forall v'1 v'2 v'3 v'4 t tidx tnumsegments
                                    tsegmentids . (OneOf '[Data.Int.Int16,
                                                           Data.Int.Int32,
                                                           Data.Int.Int64,
                                                           Data.Int.Int8,
                                                           Data.Word.Word16,
                                                           Data.Word.Word32,
                                                           Data.Word.Word64,
                                                           Data.Word.Word8,
                                                           Double, Float] t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tidx,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tnumsegments,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tsegmentids) =>
                                    OpParams ->
                                    Tensor v'1 t -- ^ __data__
                                    -> Tensor v'2 tidx -- ^ __indices__
                                    -> Tensor v'3 tsegmentids -- ^ __segment_ids__
                                    -> Tensor v'4 tnumsegments -- ^ __num_segments__
                                    -> Tensor Build t -- ^ __output__
sparseSegmentSumWithNumSegments' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor v'4 tnumsegments
-> Tensor Build t
sparseSegmentSumWithNumSegments' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' indices :: Tensor v'2 tidx
indices segment_ids :: Tensor v'3 tsegmentids
segment_ids
                                 num_segments :: Tensor v'4 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
indices,
                                                             Tensor v'3 tsegmentids -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tsegmentids
segment_ids,
                                                             Tensor v'4 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSegmentSumWithNumSegments"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsegmentids" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsegmentids -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsegmentids
forall a. HasCallStack => a
undefined :: tsegmentids)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "segment_ids" type_attr: "Tsegmentids" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tsegmentids"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSlice :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) =>
               Tensor v'1 Data.Int.Int64 -- ^ __indices__
               -> Tensor v'2 t -- ^ __values__
               -> Tensor v'3 Data.Int.Int64 -- ^ __shape__
               -> Tensor v'4 Data.Int.Int64 -- ^ __start__
               -> Tensor v'5 Data.Int.Int64 -- ^ __size__
               -> (Tensor Build Data.Int.Int64, Tensor Build t,
                   Tensor Build Data.Int.Int64)
               -- ^ (__output_indices__, __output_values__, __output_shape__)
               --
               -- * __output_indices__
               --
               -- * __output_values__
               --
               -- * __output_shape__
sparseSlice :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseSlice = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseSlice' OpParams
forall a. a -> a
id
sparseSlice' :: forall v'1 v'2 v'3 v'4 v'5 t . (TensorType t) => OpParams ->
                Tensor v'1 Data.Int.Int64 -- ^ __indices__
                -> Tensor v'2 t -- ^ __values__
                -> Tensor v'3 Data.Int.Int64 -- ^ __shape__
                -> Tensor v'4 Data.Int.Int64 -- ^ __start__
                -> Tensor v'5 Data.Int.Int64 -- ^ __size__
                -> (Tensor Build Data.Int.Int64, Tensor Build t,
                    Tensor Build Data.Int.Int64)
                -- ^ (__output_indices__, __output_values__, __output_shape__)
                --
                -- * __output_indices__
                --
                -- * __output_values__
                --
                -- * __output_shape__
sparseSlice' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseSlice' op'options :: OpParams
op'options indices :: Tensor v'1 Int64
indices values :: Tensor v'2 t
values shape :: Tensor v'3 Int64
shape start :: Tensor v'4 Int64
start size :: Tensor v'5 Int64
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
start,
                                                             Tensor v'5 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Int64
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSlice"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "shape" type: DT_INT64 }
input_arg { name: "start" type: DT_INT64 }
input_arg { name: "size" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
output_arg { name: "output_shape" type: DT_INT64 }
attr { name: "T" type: "type" }
-}
-- | 
sparseSliceGrad :: forall v'1 v'2 v'3 v'4
                   t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Data.Word.Word32,
                                Data.Word.Word64, Data.Word.Word8, Double,
                                Float] t) =>
                   Tensor v'1 t -- ^ __backprop_val_grad__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __input_indices__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __input_start__
                   -> Tensor v'4 Data.Int.Int64 -- ^ __output_indices__
                   -> Tensor Build t -- ^ __val_grad__
sparseSliceGrad :: Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
sparseSliceGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
sparseSliceGrad' OpParams
forall a. a -> a
id
sparseSliceGrad' :: forall v'1 v'2 v'3 v'4
                    t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Data.Word.Word32,
                                 Data.Word.Word64, Data.Word.Word8, Double,
                                 Float] t) => OpParams ->
                    Tensor v'1 t -- ^ __backprop_val_grad__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __input_indices__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __input_start__
                    -> Tensor v'4 Data.Int.Int64 -- ^ __output_indices__
                    -> Tensor Build t -- ^ __val_grad__
sparseSliceGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor Build t
sparseSliceGrad' op'options :: OpParams
op'options backprop_val_grad :: Tensor v'1 t
backprop_val_grad input_indices :: Tensor v'2 Int64
input_indices input_start :: Tensor v'3 Int64
input_start
                 output_indices :: Tensor v'4 Int64
output_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
backprop_val_grad,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
input_indices,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
input_start,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
output_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSliceGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "backprop_val_grad" type_attr: "T" }
input_arg { name: "input_indices" type: DT_INT64 }
input_arg { name: "input_start" type: DT_INT64 }
input_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "val_grad" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseSoftmax :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                 Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                 -> Tensor v'2 t -- ^ __sp_values__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                 -> Tensor Build t -- ^ __output__
sparseSoftmax :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build t
sparseSoftmax = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build t
sparseSoftmax' OpParams
forall a. a -> a
id
sparseSoftmax' :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                  OpParams ->
                  Tensor v'1 Data.Int.Int64 -- ^ __sp_indices__
                  -> Tensor v'2 t -- ^ __sp_values__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __sp_shape__
                  -> Tensor Build t -- ^ __output__
sparseSoftmax' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build t
sparseSoftmax' op'options :: OpParams
op'options sp_indices :: Tensor v'1 Int64
sp_indices sp_values :: Tensor v'2 t
sp_values sp_shape :: Tensor v'3 Int64
sp_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sp_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
sp_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
sp_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSoftmax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sp_indices" type: DT_INT64 }
input_arg { name: "sp_values" type_attr: "T" }
input_arg { name: "sp_shape" type: DT_INT64 }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
sparseSoftmaxCrossEntropyWithLogits :: forall v'1 v'2 t
                                       tlabels . (OneOf '[Data.Word.Word16,
                                                          Double, Float] t,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tlabels) =>
                                       Tensor v'1 t -- ^ __features__
                                       -> Tensor v'2 tlabels -- ^ __labels__
                                       -> (Tensor Build t, Tensor Build t)
                                       -- ^ (__loss__, __backprop__)
                                       --
                                       -- * __loss__
                                       --
                                       -- * __backprop__
sparseSoftmaxCrossEntropyWithLogits :: Tensor v'1 t
-> Tensor v'2 tlabels -> (Tensor Build t, Tensor Build t)
sparseSoftmaxCrossEntropyWithLogits = OpParams
-> Tensor v'1 t
-> Tensor v'2 tlabels
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t tlabels.
(OneOf '[Word16, Double, Float] t,
 OneOf '[Int32, Int64] tlabels) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tlabels
-> (Tensor Build t, Tensor Build t)
sparseSoftmaxCrossEntropyWithLogits' OpParams
forall a. a -> a
id
sparseSoftmaxCrossEntropyWithLogits' :: forall v'1 v'2 t
                                        tlabels . (OneOf '[Data.Word.Word16,
                                                           Double, Float] t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tlabels) =>
                                        OpParams ->
                                        Tensor v'1 t -- ^ __features__
                                        -> Tensor v'2 tlabels -- ^ __labels__
                                        -> (Tensor Build t, Tensor Build t)
                                        -- ^ (__loss__, __backprop__)
                                        --
                                        -- * __loss__
                                        --
                                        -- * __backprop__
sparseSoftmaxCrossEntropyWithLogits' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tlabels
-> (Tensor Build t, Tensor Build t)
sparseSoftmaxCrossEntropyWithLogits' op'options :: OpParams
op'options features :: Tensor v'1 t
features
                                     labels :: Tensor v'2 tlabels
labels | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
features,
                                                             Tensor v'2 tlabels -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tlabels
labels]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSoftmaxCrossEntropyWithLogits"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tlabels" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tlabels -> DataType
forall a. TensorType a => a -> DataType
tensorType (tlabels
forall a. HasCallStack => a
undefined :: tlabels)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "features" type_attr: "T" }
input_arg { name: "labels" type_attr: "Tlabels" }
output_arg { name: "loss" type_attr: "T" }
output_arg { name: "backprop" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "Tlabels"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseSparseMaximum :: forall v'1 v'2 v'3 v'4 v'5 v'6
                       t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
                       -> Tensor v'2 t -- ^ __a_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                       -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
                       -> Tensor v'5 t -- ^ __b_values__
                       -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
                       -> (Tensor Build Data.Int.Int64, Tensor Build t)
                       -- ^ (__output_indices__, __output_values__)
                       --
                       -- * __output_indices__
                       --
                       -- * __output_values__
sparseSparseMaximum :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMaximum = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMaximum' OpParams
forall a. a -> a
id
sparseSparseMaximum' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                        t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
                        -> Tensor v'2 t -- ^ __a_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                        -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
                        -> Tensor v'5 t -- ^ __b_values__
                        -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
                        -> (Tensor Build Data.Int.Int64, Tensor Build t)
                        -- ^ (__output_indices__, __output_values__)
                        --
                        -- * __output_indices__
                        --
                        -- * __output_values__
sparseSparseMaximum' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMaximum' op'options :: OpParams
op'options a_indices :: Tensor v'1 Int64
a_indices a_values :: Tensor v'2 t
a_values a_shape :: Tensor v'3 Int64
a_shape b_indices :: Tensor v'4 Int64
b_indices b_values :: Tensor v'5 t
b_values
                     b_shape :: Tensor v'6 Int64
b_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build t))
-> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
a_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
a_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
a_shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
b_indices,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
b_values,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
b_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSparseMaximum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a_indices" type: DT_INT64 }
input_arg { name: "a_values" type_attr: "T" }
input_arg { name: "a_shape" type: DT_INT64 }
input_arg { name: "b_indices" type: DT_INT64 }
input_arg { name: "b_values" type_attr: "T" }
input_arg { name: "b_shape" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseSparseMinimum :: forall v'1 v'2 v'3 v'4 v'5 v'6
                       t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word64, Data.Word.Word8, Double,
                                    Float] t) =>
                       Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
                       -> Tensor v'2 t -- ^ __a_values__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                       -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
                       -> Tensor v'5 t -- ^ __b_values__
                       -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
                       -> (Tensor Build Data.Int.Int64, Tensor Build t)
                       -- ^ (__output_indices__, __output_values__)
                       --
                       -- * __output_indices__
                       --
                       -- * __output_values__
sparseSparseMinimum :: Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMinimum = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMinimum' OpParams
forall a. a -> a
id
sparseSparseMinimum' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                        t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Int.Int16, Data.Int.Int32,
                                     Data.Int.Int64, Data.Int.Int8,
                                     Data.Word.Word16, Data.Word.Word32,
                                     Data.Word.Word64, Data.Word.Word8, Double,
                                     Float] t) => OpParams ->
                        Tensor v'1 Data.Int.Int64 -- ^ __a_indices__
                        -> Tensor v'2 t -- ^ __a_values__
                        -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                        -> Tensor v'4 Data.Int.Int64 -- ^ __b_indices__
                        -> Tensor v'5 t -- ^ __b_values__
                        -> Tensor v'6 Data.Int.Int64 -- ^ __b_shape__
                        -> (Tensor Build Data.Int.Int64, Tensor Build t)
                        -- ^ (__output_indices__, __output_values__)
                        --
                        -- * __output_indices__
                        --
                        -- * __output_values__
sparseSparseMinimum' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t)
sparseSparseMinimum' op'options :: OpParams
op'options a_indices :: Tensor v'1 Int64
a_indices a_values :: Tensor v'2 t
a_values a_shape :: Tensor v'3 Int64
a_shape b_indices :: Tensor v'4 Int64
b_indices b_values :: Tensor v'5 t
b_values
                     b_shape :: Tensor v'6 Int64
b_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Int64, Tensor Build t))
-> Build OpDef -> (Tensor Build Int64, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
a_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
a_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
a_shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
b_indices,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
b_values,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
b_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSparseMinimum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a_indices" type: DT_INT64 }
input_arg { name: "a_values" type_attr: "T" }
input_arg { name: "a_shape" type: DT_INT64 }
input_arg { name: "b_indices" type: DT_INT64 }
input_arg { name: "b_values" type_attr: "T" }
input_arg { name: "b_shape" type: DT_INT64 }
output_arg { name: "output_indices" type: DT_INT64 }
output_arg { name: "output_values" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
sparseSplit :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
               Data.Int.Int64 -- ^ __num_split__
               -> Tensor v'1 Data.Int.Int64 -- ^ __split_dim__
               -> Tensor v'2 Data.Int.Int64 -- ^ __indices__
               -> Tensor v'3 t -- ^ __values__
               -> Tensor v'4 Data.Int.Int64 -- ^ __shape__
               -> ([Tensor Build Data.Int.Int64], [Tensor Build t],
                   [Tensor Build Data.Int.Int64])
               -- ^ (__output_indices__, __output_values__, __output_shape__)
               --
               -- * __output_indices__
               --
               -- * __output_values__
               --
               -- * __output_shape__
sparseSplit :: Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
sparseSplit = OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
sparseSplit' OpParams
forall a. a -> a
id
sparseSplit' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                Data.Int.Int64 -- ^ __num_split__
                -> Tensor v'1 Data.Int.Int64 -- ^ __split_dim__
                -> Tensor v'2 Data.Int.Int64 -- ^ __indices__
                -> Tensor v'3 t -- ^ __values__
                -> Tensor v'4 Data.Int.Int64 -- ^ __shape__
                -> ([Tensor Build Data.Int.Int64], [Tensor Build t],
                    [Tensor Build Data.Int.Int64])
                -- ^ (__output_indices__, __output_values__, __output_shape__)
                --
                -- * __output_indices__
                --
                -- * __output_values__
                --
                -- * __output_shape__
sparseSplit' :: OpParams
-> Int64
-> Tensor v'1 Int64
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
sparseSplit' op'options :: OpParams
op'options num_split :: Int64
num_split split_dim :: Tensor v'1 Int64
split_dim indices :: Tensor v'2 Int64
indices values :: Tensor v'3 t
values
             shape :: Tensor v'4 Int64
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_split, Int64
num_split, Int64
num_split] (Build OpDef
 -> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64]))
-> Build OpDef
-> ([Tensor Build Int64], [Tensor Build t], [Tensor Build Int64])
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
split_dim,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
values,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseSplit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_split" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_split
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "split_dim" type: DT_INT64 }
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "shape" type: DT_INT64 }
output_arg {
  name: "output_indices" type: DT_INT64 number_attr: "num_split"
}
output_arg {
  name: "output_values" type_attr: "T" number_attr: "num_split"
}
output_arg {
  name: "output_shape" type: DT_INT64 number_attr: "num_split"
}
attr { name: "num_split" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | 
sparseTensorDenseAdd :: forall v'1 v'2 v'3 v'4 t
                        tindices . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word32,
                                            Data.Word.Word64, Data.Word.Word8,
                                            Double, Float] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tindices) =>
                        Tensor v'1 tindices -- ^ __a_indices__
                        -> Tensor v'2 t -- ^ __a_values__
                        -> Tensor v'3 tindices -- ^ __a_shape__
                        -> Tensor v'4 t -- ^ __b__
                        -> Tensor Build t -- ^ __output__
sparseTensorDenseAdd :: Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseAdd = OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tindices.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseAdd' OpParams
forall a. a -> a
id
sparseTensorDenseAdd' :: forall v'1 v'2 v'3 v'4 t
                         tindices . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tindices) =>
                         OpParams ->
                         Tensor v'1 tindices -- ^ __a_indices__
                         -> Tensor v'2 t -- ^ __a_values__
                         -> Tensor v'3 tindices -- ^ __a_shape__
                         -> Tensor v'4 t -- ^ __b__
                         -> Tensor Build t -- ^ __output__
sparseTensorDenseAdd' :: OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseAdd' op'options :: OpParams
op'options a_indices :: Tensor v'1 tindices
a_indices a_values :: Tensor v'2 t
a_values a_shape :: Tensor v'3 tindices
a_shape
                      b :: Tensor v'4 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tindices
a_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
a_values,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
a_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseTensorDenseAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a_indices" type_attr: "Tindices" }
input_arg { name: "a_values" type_attr: "T" }
input_arg { name: "a_shape" type_attr: "Tindices" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseTensorDenseMatMul :: forall v'1 v'2 v'3 v'4 t tindices . (TensorType t,
                                                                OneOf '[Data.Int.Int32,
                                                                        Data.Int.Int64] tindices) =>
                           Tensor v'1 tindices -- ^ __a_indices__
                           -> Tensor v'2 t -- ^ __a_values__
                           -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                           -> Tensor v'4 t -- ^ __b__
                           -> Tensor Build t -- ^ __product__
sparseTensorDenseMatMul :: Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseMatMul = OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tindices.
(TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseMatMul' OpParams
forall a. a -> a
id
sparseTensorDenseMatMul' :: forall v'1 v'2 v'3 v'4 t tindices . (TensorType t,
                                                                 OneOf '[Data.Int.Int32,
                                                                         Data.Int.Int64] tindices) =>
                            OpParams ->
                            Tensor v'1 tindices -- ^ __a_indices__
                            -> Tensor v'2 t -- ^ __a_values__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __a_shape__
                            -> Tensor v'4 t -- ^ __b__
                            -> Tensor Build t -- ^ __product__
sparseTensorDenseMatMul' :: OpParams
-> Tensor v'1 tindices
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 t
-> Tensor Build t
sparseTensorDenseMatMul' op'options :: OpParams
op'options a_indices :: Tensor v'1 tindices
a_indices a_values :: Tensor v'2 t
a_values a_shape :: Tensor v'3 Int64
a_shape
                         b :: Tensor v'4 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tindices
a_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
a_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
a_shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseTensorDenseMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a_indices" type_attr: "Tindices" }
input_arg { name: "a_values" type_attr: "T" }
input_arg { name: "a_shape" type: DT_INT64 }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "product" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "adjoint_a" type: "bool" default_value { b: false } }
attr { name: "adjoint_b" type: "bool" default_value { b: false } }
-}
-- | 
sparseTensorSliceDataset :: forall v'1 v'2 v'3 tvalues m' . (MonadBuild m',
                                                             TensorType tvalues) =>
                            Tensor v'1 Data.Int.Int64 -- ^ __indices__
                            -> Tensor v'2 tvalues -- ^ __values__
                            -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                            -> m' (Tensor Value Variant) -- ^ __handle__
sparseTensorSliceDataset :: Tensor v'1 Int64
-> Tensor v'2 tvalues
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
sparseTensorSliceDataset = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tvalues
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tvalues
       (m' :: * -> *).
(MonadBuild m', TensorType tvalues) =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tvalues
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
sparseTensorSliceDataset' OpParams
forall a. a -> a
id
sparseTensorSliceDataset' :: forall v'1 v'2 v'3 tvalues m' . (MonadBuild m',
                                                              TensorType tvalues) =>
                             OpParams ->
                             Tensor v'1 Data.Int.Int64 -- ^ __indices__
                             -> Tensor v'2 tvalues -- ^ __values__
                             -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                             -> m' (Tensor Value Variant) -- ^ __handle__
sparseTensorSliceDataset' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 tvalues
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
sparseTensorSliceDataset' op'options :: OpParams
op'options indices :: Tensor v'1 Int64
indices values :: Tensor v'2 tvalues
values
                          dense_shape :: Tensor v'3 Int64
dense_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 tvalues -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tvalues
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
dense_shape]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SparseTensorSliceDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tvalues" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tvalues -> DataType
forall a. TensorType a => a -> DataType
tensorType (tvalues
forall a. HasCallStack => a
undefined :: tvalues)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "Tvalues" }
input_arg { name: "dense_shape" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "Tvalues" type: "type" }
-}
-- | 
sparseTensorToCSRSparseMatrix :: forall v'1 v'2 v'3
                                 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Double, Float] t) =>
                                 Tensor v'1 Data.Int.Int64 -- ^ __indices__
                                 -> Tensor v'2 t -- ^ __values__
                                 -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                                 -> Tensor Build Variant -- ^ __sparse_matrix__
sparseTensorToCSRSparseMatrix :: Tensor v'1 Int64
-> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Variant
sparseTensorToCSRSparseMatrix = OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Variant
sparseTensorToCSRSparseMatrix' OpParams
forall a. a -> a
id
sparseTensorToCSRSparseMatrix' :: forall v'1 v'2 v'3
                                  t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Double, Float] t) => OpParams ->
                                  Tensor v'1 Data.Int.Int64 -- ^ __indices__
                                  -> Tensor v'2 t -- ^ __values__
                                  -> Tensor v'3 Data.Int.Int64 -- ^ __dense_shape__
                                  -> Tensor Build Variant -- ^ __sparse_matrix__
sparseTensorToCSRSparseMatrix' :: OpParams
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor Build Variant
sparseTensorToCSRSparseMatrix' op'options :: OpParams
op'options indices :: Tensor v'1 Int64
indices values :: Tensor v'2 t
values
                               dense_shape :: Tensor v'3 Int64
dense_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
dense_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseTensorToCSRSparseMatrix"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type: DT_INT64 }
input_arg { name: "values" type_attr: "T" }
input_arg { name: "dense_shape" type: DT_INT64 }
output_arg { name: "sparse_matrix" type: DT_VARIANT }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sparseToDense :: forall v'1 v'2 v'3 v'4 t tindices . (TensorType t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                 Tensor v'1 tindices -- ^ __sparse_indices__
                 -> Tensor v'2 tindices -- ^ __output_shape__
                 -> Tensor v'3 t -- ^ __sparse_values__
                 -> Tensor v'4 t -- ^ __default_value__
                 -> Tensor Build t -- ^ __dense__
sparseToDense :: Tensor v'1 tindices
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
sparseToDense = OpParams
-> Tensor v'1 tindices
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t tindices.
(TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 tindices
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
sparseToDense' OpParams
forall a. a -> a
id
sparseToDense' :: forall v'1 v'2 v'3 v'4 t tindices . (TensorType t,
                                                       OneOf '[Data.Int.Int32,
                                                               Data.Int.Int64] tindices) =>
                  OpParams ->
                  Tensor v'1 tindices -- ^ __sparse_indices__
                  -> Tensor v'2 tindices -- ^ __output_shape__
                  -> Tensor v'3 t -- ^ __sparse_values__
                  -> Tensor v'4 t -- ^ __default_value__
                  -> Tensor Build t -- ^ __dense__
sparseToDense' :: OpParams
-> Tensor v'1 tindices
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
sparseToDense' op'options :: OpParams
op'options sparse_indices :: Tensor v'1 tindices
sparse_indices output_shape :: Tensor v'2 tindices
output_shape sparse_values :: Tensor v'3 t
sparse_values
               default_value :: Tensor v'4 t
default_value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tindices
sparse_indices,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
output_shape,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
sparse_values,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
default_value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseToDense"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_indices" type_attr: "Tindices" }
input_arg { name: "output_shape" type_attr: "Tindices" }
input_arg { name: "sparse_values" type_attr: "T" }
input_arg { name: "default_value" type_attr: "T" }
output_arg { name: "dense" type_attr: "T" }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sparseToSparseSetOperation :: forall v'1 v'2 v'3 v'4 v'5 v'6
                              t . (OneOf '[Data.ByteString.ByteString,
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16,
                                           Data.Word.Word8] t) =>
                              ByteString -- ^ __set_operation__
                              -> Tensor v'1 Data.Int.Int64 -- ^ __set1_indices__
                              -> Tensor v'2 t -- ^ __set1_values__
                              -> Tensor v'3 Data.Int.Int64 -- ^ __set1_shape__
                              -> Tensor v'4 Data.Int.Int64 -- ^ __set2_indices__
                              -> Tensor v'5 t -- ^ __set2_values__
                              -> Tensor v'6 Data.Int.Int64 -- ^ __set2_shape__
                              -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                  Tensor Build Data.Int.Int64)
                              -- ^ (__result_indices__, __result_values__, __result_shape__)
                              --
                              -- * __result_indices__
                              --
                              -- * __result_values__
                              --
                              -- * __result_shape__
sparseToSparseSetOperation :: ByteString
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseToSparseSetOperation = OpParams
-> ByteString
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t.
OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t =>
OpParams
-> ByteString
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseToSparseSetOperation' OpParams
forall a. a -> a
id
sparseToSparseSetOperation' :: forall v'1 v'2 v'3 v'4 v'5 v'6
                               t . (OneOf '[Data.ByteString.ByteString,
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16,
                                            Data.Word.Word8] t) => OpParams ->
                               ByteString -- ^ __set_operation__
                               -> Tensor v'1 Data.Int.Int64 -- ^ __set1_indices__
                               -> Tensor v'2 t -- ^ __set1_values__
                               -> Tensor v'3 Data.Int.Int64 -- ^ __set1_shape__
                               -> Tensor v'4 Data.Int.Int64 -- ^ __set2_indices__
                               -> Tensor v'5 t -- ^ __set2_values__
                               -> Tensor v'6 Data.Int.Int64 -- ^ __set2_shape__
                               -> (Tensor Build Data.Int.Int64, Tensor Build t,
                                   Tensor Build Data.Int.Int64)
                               -- ^ (__result_indices__, __result_values__, __result_shape__)
                               --
                               -- * __result_indices__
                               --
                               -- * __result_values__
                               --
                               -- * __result_shape__
sparseToSparseSetOperation' :: OpParams
-> ByteString
-> Tensor v'1 Int64
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 t
-> Tensor v'6 Int64
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
sparseToSparseSetOperation' op'options :: OpParams
op'options set_operation :: ByteString
set_operation set1_indices :: Tensor v'1 Int64
set1_indices set1_values :: Tensor v'2 t
set1_values
                            set1_shape :: Tensor v'3 Int64
set1_shape set2_indices :: Tensor v'4 Int64
set2_indices set2_values :: Tensor v'5 t
set2_values
                            set2_shape :: Tensor v'6 Int64
set2_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
set1_indices,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
set1_values,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
set1_shape,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
set2_indices,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
set2_values,
                                                             Tensor v'6 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 Int64
set2_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SparseToSparseSetOperation"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "set_operation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
set_operation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "set1_indices" type: DT_INT64 }
input_arg { name: "set1_values" type_attr: "T" }
input_arg { name: "set1_shape" type: DT_INT64 }
input_arg { name: "set2_indices" type: DT_INT64 }
input_arg { name: "set2_values" type_attr: "T" }
input_arg { name: "set2_shape" type: DT_INT64 }
output_arg { name: "result_indices" type: DT_INT64 }
output_arg { name: "result_values" type_attr: "T" }
output_arg { name: "result_shape" type: DT_INT64 }
attr { name: "set_operation" type: "string" }
attr {
  name: "validate_indices" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_UINT8
      type: DT_UINT16
      type: DT_STRING
    }
  }
}
-}
-- | 
spence :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
spence :: Tensor v'1 t -> Tensor Build t
spence = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
spence' OpParams
forall a. a -> a
id
spence' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double, Float] t) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
spence' :: OpParams -> Tensor v'1 t -> Tensor Build t
spence' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Spence"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16 type: DT_HALF type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
-}
-- | 
split :: forall v'1 v'2 t . (TensorType t) => Data.Int.Int64 -- ^ __num_split__
         -> Tensor v'1 Data.Int.Int32 -- ^ __split_dim__
         -> Tensor v'2 t -- ^ __value__
         -> [Tensor Build t] -- ^ __output__
split :: Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
split = OpParams
-> Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
split' OpParams
forall a. a -> a
id
split' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
          Data.Int.Int64 -- ^ __num_split__
          -> Tensor v'1 Data.Int.Int32 -- ^ __split_dim__
          -> Tensor v'2 t -- ^ __value__
          -> [Tensor Build t] -- ^ __output__
split' :: OpParams
-> Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
split' op'options :: OpParams
op'options num_split :: Int64
num_split split_dim :: Tensor v'1 Int32
split_dim value :: Tensor v'2 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_split] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
split_dim,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Split"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_split" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_split
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "split_dim" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
output_arg {
  name: "output" type_attr: "T" number_attr: "num_split"
}
attr { name: "num_split" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | 
splitV :: forall v'1 v'2 v'3 t tlen . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tlen) =>
          Data.Int.Int64 -- ^ __num_split__
          -> Tensor v'1 t -- ^ __value__
          -> Tensor v'2 tlen -- ^ __size_splits__
          -> Tensor v'3 Data.Int.Int32 -- ^ __split_dim__
          -> [Tensor Build t] -- ^ __output__
splitV :: Int64
-> Tensor v'1 t
-> Tensor v'2 tlen
-> Tensor v'3 Int32
-> [Tensor Build t]
splitV = OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 tlen
-> Tensor v'3 Int32
-> [Tensor Build t]
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tlen.
(TensorType t, OneOf '[Int32, Int64] tlen) =>
OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 tlen
-> Tensor v'3 Int32
-> [Tensor Build t]
splitV' OpParams
forall a. a -> a
id
splitV' :: forall v'1 v'2 v'3 t tlen . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tlen) =>
           OpParams ->
           Data.Int.Int64 -- ^ __num_split__
           -> Tensor v'1 t -- ^ __value__
           -> Tensor v'2 tlen -- ^ __size_splits__
           -> Tensor v'3 Data.Int.Int32 -- ^ __split_dim__
           -> [Tensor Build t] -- ^ __output__
splitV' :: OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 tlen
-> Tensor v'3 Int32
-> [Tensor Build t]
splitV' op'options :: OpParams
op'options num_split :: Int64
num_split value :: Tensor v'1 t
value size_splits :: Tensor v'2 tlen
size_splits split_dim :: Tensor v'3 Int32
split_dim | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_split] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value,
                                                             Tensor v'2 tlen -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tlen
size_splits,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
split_dim]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SplitV"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tlen" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tlen -> DataType
forall a. TensorType a => a -> DataType
tensorType (tlen
forall a. HasCallStack => a
undefined :: tlen)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_split" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_split
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
input_arg { name: "size_splits" type_attr: "Tlen" }
input_arg { name: "split_dim" type: DT_INT32 }
output_arg {
  name: "output" type_attr: "T" number_attr: "num_split"
}
attr { name: "num_split" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
attr {
  name: "Tlen"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
sqlDataset :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
              [DataType] -- ^ __output_types__
              -> Tensor v'1 Data.ByteString.ByteString -- ^ __driver_name__
              -> Tensor v'2 Data.ByteString.ByteString -- ^ __data_source_name__
              -> Tensor v'3 Data.ByteString.ByteString -- ^ __query__
              -> m' (Tensor Value Variant) -- ^ __handle__
sqlDataset :: [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
sqlDataset = OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
sqlDataset' OpParams
forall a. a -> a
id
sqlDataset' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
               [DataType] -- ^ __output_types__
               -> Tensor v'1 Data.ByteString.ByteString -- ^ __driver_name__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __data_source_name__
               -> Tensor v'3 Data.ByteString.ByteString -- ^ __query__
               -> m' (Tensor Value Variant) -- ^ __handle__
sqlDataset' :: OpParams
-> [DataType]
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m' (Tensor Value Variant)
sqlDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types driver_name :: Tensor v'1 ByteString
driver_name data_source_name :: Tensor v'2 ByteString
data_source_name
            query :: Tensor v'3 ByteString
query | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
driver_name,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
data_source_name,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
query]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SqlDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "driver_name" type: DT_STRING }
input_arg { name: "data_source_name" type: DT_STRING }
input_arg { name: "query" type: DT_STRING }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
sqrt :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
sqrt :: Tensor v'1 t -> Tensor Build t
sqrt = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
sqrt' OpParams
forall a. a -> a
id
sqrt' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
sqrt' :: OpParams -> Tensor v'1 t -> Tensor Build t
sqrt' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sqrt"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
sqrtGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __y__
            -> Tensor v'2 t -- ^ __dy__
            -> Tensor Build t -- ^ __z__
sqrtGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sqrtGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sqrtGrad' OpParams
forall a. a -> a
id
sqrtGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __y__
             -> Tensor v'2 t -- ^ __dy__
             -> Tensor Build t -- ^ __z__
sqrtGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sqrtGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SqrtGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
square :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Double, Float] t) =>
          Tensor v'1 t -- ^ __x__
          -> Tensor Build t -- ^ __y__
square :: Tensor v'1 t -> Tensor Build t
square = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
square' OpParams
forall a. a -> a
id
square' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16, Double,
                                   Float] t) => OpParams ->
           Tensor v'1 t -- ^ __x__
           -> Tensor Build t -- ^ __y__
square' :: OpParams -> Tensor v'1 t -> Tensor Build t
square' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Square"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
squaredDifference :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Data.Int.Int32, Data.Int.Int64,
                                                 Data.Word.Word16, Double,
                                                 Float] t) =>
                     Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 t -- ^ __y__
                     -> Tensor Build t -- ^ __z__
squaredDifference :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
squaredDifference = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
squaredDifference' OpParams
forall a. a -> a
id
squaredDifference' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                  (Data.Complex.Complex Float),
                                                  Data.Int.Int32,
                                                  Data.Int.Int64,
                                                  Data.Word.Word16, Double,
                                                  Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __x__
                      -> Tensor v'2 t -- ^ __y__
                      -> Tensor Build t -- ^ __z__
squaredDifference' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
squaredDifference' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "SquaredDifference"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
squeeze :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
           -> Tensor Build t -- ^ __output__
squeeze :: Tensor v'1 t -> Tensor Build t
squeeze = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
squeeze' OpParams
forall a. a -> a
id
squeeze' :: forall v'1 t . (TensorType t) => OpParams ->
            Tensor v'1 t -- ^ __input__
            -> Tensor Build t -- ^ __output__
squeeze' :: OpParams -> Tensor v'1 t -> Tensor Build t
squeeze' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Squeeze"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "squeeze_dims"
  type: "list(int)"
  default_value { list { } }
  has_minimum: true
}
-}
-- | 
stack :: forall m' . (MonadBuild m') => DataType -- ^ __elem_type__
         -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
stack :: DataType -> m' (Tensor Ref ByteString)
stack = OpParams -> DataType -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> m' (Tensor Ref ByteString)
stack' OpParams
forall a. a -> a
id
stack' :: forall m' . (MonadBuild m') => OpParams ->
          DataType -- ^ __elem_type__
          -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
stack' :: OpParams -> DataType -> m' (Tensor Ref ByteString)
stack' op'options :: OpParams
op'options elem_type :: DataType
elem_type | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Stack"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "elem_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
elem_type
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr { name: "elem_type" type: "type" }
attr { name: "stack_name" type: "string" default_value { s: "" } }
-}
-- | 
stackClose :: forall m' . (MonadBuild m') =>
              Tensor Ref Data.ByteString.ByteString -- ^ __handle__
              -> m' (ControlNode)
stackClose :: Tensor Ref ByteString -> m' ControlNode
stackClose = OpParams -> Tensor Ref ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' ControlNode
stackClose' OpParams
forall a. a -> a
id
stackClose' :: forall m' . (MonadBuild m') => OpParams ->
               Tensor Ref Data.ByteString.ByteString -- ^ __handle__
               -> m' (ControlNode)
stackClose' :: OpParams -> Tensor Ref ByteString -> m' ControlNode
stackClose' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackClose"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
-}
-- | 
stackCloseV2 :: forall v'1 m' . (MonadBuild m') =>
                Tensor v'1 ResourceHandle -- ^ __handle__
                -> m' (ControlNode)
stackCloseV2 :: Tensor v'1 ResourceHandle -> m' ControlNode
stackCloseV2 = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
stackCloseV2' OpParams
forall a. a -> a
id
stackCloseV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                 Tensor v'1 ResourceHandle -- ^ __handle__
                 -> m' (ControlNode)
stackCloseV2' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
stackCloseV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackCloseV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
-}
-- | 
stackPop :: forall elem_type m' . (MonadBuild m', TensorType elem_type) =>
            Tensor Ref Data.ByteString.ByteString -- ^ __handle__
            -> m' (Tensor Value elem_type) -- ^ __elem__
stackPop :: Tensor Ref ByteString -> m' (Tensor Value elem_type)
stackPop = OpParams -> Tensor Ref ByteString -> m' (Tensor Value elem_type)
forall elem_type (m' :: * -> *).
(MonadBuild m', TensorType elem_type) =>
OpParams -> Tensor Ref ByteString -> m' (Tensor Value elem_type)
stackPop' OpParams
forall a. a -> a
id
stackPop' :: forall elem_type m' . (MonadBuild m', TensorType elem_type) =>
             OpParams ->
             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
             -> m' (Tensor Value elem_type) -- ^ __elem__
stackPop' :: OpParams -> Tensor Ref ByteString -> m' (Tensor Value elem_type)
stackPop' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value elem_type) -> m' (Tensor Value elem_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value elem_type) -> m' (Tensor Value elem_type))
-> Build (Tensor Value elem_type) -> m' (Tensor Value elem_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build (Tensor Value elem_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackPop"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "elem_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ elem_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (elem_type
forall a. HasCallStack => a
undefined :: elem_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
output_arg { name: "elem" type_attr: "elem_type" }
attr { name: "elem_type" type: "type" }
-}
-- | 
stackPopV2 :: forall v'1 elem_type m' . (MonadBuild m', TensorType elem_type) =>
              Tensor v'1 ResourceHandle -- ^ __handle__
              -> m' (Tensor Value elem_type) -- ^ __elem__
stackPopV2 :: Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
stackPopV2 = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
stackPopV2' OpParams
forall a. a -> a
id
stackPopV2' :: forall v'1 elem_type m' . (MonadBuild m',
                                          TensorType elem_type) => OpParams ->
               Tensor v'1 ResourceHandle -- ^ __handle__
               -> m' (Tensor Value elem_type) -- ^ __elem__
stackPopV2' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
stackPopV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value elem_type) -> m' (Tensor Value elem_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value elem_type) -> m' (Tensor Value elem_type))
-> Build (Tensor Value elem_type) -> m' (Tensor Value elem_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build (Tensor Value elem_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackPopV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "elem_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ elem_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (elem_type
forall a. HasCallStack => a
undefined :: elem_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "elem" type_attr: "elem_type" }
attr { name: "elem_type" type: "type" }
-}
-- | 
stackPush :: forall v'2 t m' . (MonadBuild m', TensorType t) =>
             Tensor Ref Data.ByteString.ByteString -- ^ __handle__
             -> Tensor v'2 t -- ^ __elem__
             -> m' (Tensor Value t) -- ^ __output__
stackPush :: Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
stackPush = OpParams
-> Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
forall (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
stackPush' OpParams
forall a. a -> a
id
stackPush' :: forall v'2 t m' . (MonadBuild m', TensorType t) => OpParams ->
              Tensor Ref Data.ByteString.ByteString -- ^ __handle__
              -> Tensor v'2 t -- ^ __elem__
              -> m' (Tensor Value t) -- ^ __output__
stackPush' :: OpParams
-> Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
stackPush' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle elem :: Tensor v'2 t
elem | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
elem]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackPush"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "elem" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "swap_memory" type: "bool" default_value { b: false }
}
-}
-- | 
stackPushV2 :: forall v'1 v'2 t m' . (MonadBuild m', TensorType t) =>
               Tensor v'1 ResourceHandle -- ^ __handle__
               -> Tensor v'2 t -- ^ __elem__
               -> m' (Tensor Value t) -- ^ __output__
stackPushV2 :: Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
stackPushV2 = OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
stackPushV2' OpParams
forall a. a -> a
id
stackPushV2' :: forall v'1 v'2 t m' . (MonadBuild m', TensorType t) =>
                OpParams ->
                Tensor v'1 ResourceHandle -- ^ __handle__
                -> Tensor v'2 t -- ^ __elem__
                -> m' (Tensor Value t) -- ^ __output__
stackPushV2' :: OpParams
-> Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
stackPushV2' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle elem :: Tensor v'2 t
elem | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
elem]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackPushV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "elem" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "swap_memory" type: "bool" default_value { b: false }
}
-}
-- | 
stackV2 :: forall v'1 m' . (MonadBuild m') => DataType -- ^ __elem_type__
           -> Tensor v'1 Data.Int.Int32 -- ^ __max_size__
           -> m' (Tensor Value ResourceHandle) -- ^ __handle__
stackV2 :: DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
stackV2 = OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
stackV2' OpParams
forall a. a -> a
id
stackV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
            DataType -- ^ __elem_type__
            -> Tensor v'1 Data.Int.Int32 -- ^ __max_size__
            -> m' (Tensor Value ResourceHandle) -- ^ __handle__
stackV2' :: OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
stackV2' op'options :: OpParams
op'options elem_type :: DataType
elem_type max_size :: Tensor v'1 Int32
max_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
max_size]
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StackV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "elem_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
elem_type
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "max_size" type: DT_INT32 }
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "elem_type" type: "type" }
attr { name: "stack_name" type: "string" default_value { s: "" } }
-}
-- | 
stage :: forall v'1 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
         TensorList (v'1) dtypes -- ^ __values__
         -> m' (ControlNode)
stage :: TensorList v'1 dtypes -> m' ControlNode
stage = OpParams -> TensorList v'1 dtypes -> m' ControlNode
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> TensorList v'1 dtypes -> m' ControlNode
stage' OpParams
forall a. a -> a
id
stage' :: forall v'1 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
          OpParams ->
          TensorList (v'1) dtypes -- ^ __values__
          -> m' (ControlNode)
stage' :: OpParams -> TensorList v'1 dtypes -> m' ControlNode
stage' op'options :: OpParams
op'options values :: TensorList v'1 dtypes
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 dtypes -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 dtypes
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Stage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
stageClear :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
              -> m' (ControlNode)
stageClear :: [DataType] -> m' ControlNode
stageClear = OpParams -> [DataType] -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' ControlNode
stageClear' OpParams
forall a. a -> a
id
stageClear' :: forall m' . (MonadBuild m') => OpParams ->
               [DataType] -- ^ __dtypes__
               -> m' (ControlNode)
stageClear' :: OpParams -> [DataType] -> m' ControlNode
stageClear' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StageClear"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
stagePeek :: forall v'1 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
             Tensor v'1 Data.Int.Int32 -- ^ __index__
             -> m' (TensorList (Value) dtypes) -- ^ __values__
stagePeek :: Tensor v'1 Int32 -> m' (TensorList Value dtypes)
stagePeek = OpParams -> Tensor v'1 Int32 -> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> Tensor v'1 Int32 -> m' (TensorList Value dtypes)
stagePeek' OpParams
forall a. a -> a
id
stagePeek' :: forall v'1 dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
              OpParams ->
              Tensor v'1 Data.Int.Int32 -- ^ __index__
              -> m' (TensorList (Value) dtypes) -- ^ __values__
stagePeek' :: OpParams -> Tensor v'1 Int32 -> m' (TensorList Value dtypes)
stagePeek' op'options :: OpParams
op'options index :: Tensor v'1 Int32
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
index]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StagePeek"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "index" type: DT_INT32 }
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
stageSize :: forall m' . (MonadBuild m') => [DataType] -- ^ __dtypes__
             -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
stageSize :: [DataType] -> m' (Tensor Value Int32)
stageSize = OpParams -> [DataType] -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> [DataType] -> m' (Tensor Value Int32)
stageSize' OpParams
forall a. a -> a
id
stageSize' :: forall m' . (MonadBuild m') => OpParams ->
              [DataType] -- ^ __dtypes__
              -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
stageSize' :: OpParams -> [DataType] -> m' (Tensor Value Int32)
stageSize' op'options :: OpParams
op'options dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StageSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "size" type: DT_INT32 }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr { name: "dtypes" type: "list(type)" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
statefulRandomBinomial :: forall v'1 v'2 v'3 v'4 v'5 s t dtype
                          m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] s,
                                OneOf '[Data.Int.Int32, Data.Int.Int64,
                                        Data.Word.Word16, Double, Float] t,
                                OneOf '[Data.Int.Int32, Data.Int.Int64,
                                        Data.Word.Word16, Double,
                                        Float] dtype) =>
                          Tensor v'1 ResourceHandle -- ^ __resource__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                          -> Tensor v'3 s -- ^ __shape__
                          -> Tensor v'4 t -- ^ __counts__
                          -> Tensor v'5 t -- ^ __probs__
                          -> m' (Tensor Value dtype) -- ^ __output__
statefulRandomBinomial :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 s
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Value dtype)
statefulRandomBinomial = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 s
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) s t dtype (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] s,
 OneOf '[Int32, Int64, Word16, Double, Float] t,
 OneOf '[Int32, Int64, Word16, Double, Float] dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 s
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Value dtype)
statefulRandomBinomial' OpParams
forall a. a -> a
id
statefulRandomBinomial' :: forall v'1 v'2 v'3 v'4 v'5 s t dtype
                           m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] s,
                                 OneOf '[Data.Int.Int32, Data.Int.Int64,
                                         Data.Word.Word16, Double, Float] t,
                                 OneOf '[Data.Int.Int32, Data.Int.Int64,
                                         Data.Word.Word16, Double,
                                         Float] dtype) => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __resource__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                           -> Tensor v'3 s -- ^ __shape__
                           -> Tensor v'4 t -- ^ __counts__
                           -> Tensor v'5 t -- ^ __probs__
                           -> m' (Tensor Value dtype) -- ^ __output__
statefulRandomBinomial' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 s
-> Tensor v'4 t
-> Tensor v'5 t
-> m' (Tensor Value dtype)
statefulRandomBinomial' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm shape :: Tensor v'3 s
shape counts :: Tensor v'4 t
counts
                        probs :: Tensor v'5 t
probs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 s
shape,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
counts,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
probs]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulRandomBinomial"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "counts" type_attr: "T" }
input_arg { name: "probs" type_attr: "T" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_DOUBLE }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
statefulStandardNormal :: forall v'1 v'2 dtype shape_dtype m' . (MonadBuild m',
                                                                 TensorType dtype,
                                                                 TensorType shape_dtype) =>
                          Tensor v'1 ResourceHandle -- ^ __resource__
                          -> Tensor v'2 shape_dtype -- ^ __shape__
                          -> m' (Tensor Value dtype) -- ^ __output__
statefulStandardNormal :: Tensor v'1 ResourceHandle
-> Tensor v'2 shape_dtype -> m' (Tensor Value dtype)
statefulStandardNormal = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 shape_dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype shape_dtype
       (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 shape_dtype
-> m' (Tensor Value dtype)
statefulStandardNormal' OpParams
forall a. a -> a
id
statefulStandardNormal' :: forall v'1 v'2 dtype shape_dtype m' . (MonadBuild m',
                                                                  TensorType dtype,
                                                                  TensorType shape_dtype) =>
                           OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __resource__
                           -> Tensor v'2 shape_dtype -- ^ __shape__
                           -> m' (Tensor Value dtype) -- ^ __output__
statefulStandardNormal' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 shape_dtype
-> m' (Tensor Value dtype)
statefulStandardNormal' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource shape :: Tensor v'2 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulStandardNormal"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_FLOAT }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statefulStandardNormalV2 :: forall v'1 v'2 v'3 dtype shape_dtype
                            m' . (MonadBuild m', TensorType dtype,
                                  TensorType shape_dtype) =>
                            Tensor v'1 ResourceHandle -- ^ __resource__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                            -> Tensor v'3 shape_dtype -- ^ __shape__
                            -> m' (Tensor Value dtype) -- ^ __output__
statefulStandardNormalV2 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulStandardNormalV2 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulStandardNormalV2' OpParams
forall a. a -> a
id
statefulStandardNormalV2' :: forall v'1 v'2 v'3 dtype shape_dtype
                             m' . (MonadBuild m', TensorType dtype,
                                   TensorType shape_dtype) => OpParams ->
                             Tensor v'1 ResourceHandle -- ^ __resource__
                             -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                             -> Tensor v'3 shape_dtype -- ^ __shape__
                             -> m' (Tensor Value dtype) -- ^ __output__
statefulStandardNormalV2' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulStandardNormalV2' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm
                          shape :: Tensor v'3 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulStandardNormalV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_FLOAT }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statefulTruncatedNormal :: forall v'1 v'2 v'3 dtype shape_dtype
                           m' . (MonadBuild m', TensorType dtype,
                                 TensorType shape_dtype) =>
                           Tensor v'1 ResourceHandle -- ^ __resource__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                           -> Tensor v'3 shape_dtype -- ^ __shape__
                           -> m' (Tensor Value dtype) -- ^ __output__
statefulTruncatedNormal :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulTruncatedNormal = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulTruncatedNormal' OpParams
forall a. a -> a
id
statefulTruncatedNormal' :: forall v'1 v'2 v'3 dtype shape_dtype
                            m' . (MonadBuild m', TensorType dtype,
                                  TensorType shape_dtype) => OpParams ->
                            Tensor v'1 ResourceHandle -- ^ __resource__
                            -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                            -> Tensor v'3 shape_dtype -- ^ __shape__
                            -> m' (Tensor Value dtype) -- ^ __output__
statefulTruncatedNormal' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulTruncatedNormal' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm
                         shape :: Tensor v'3 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulTruncatedNormal"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_FLOAT }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statefulUniform :: forall v'1 v'2 v'3 dtype shape_dtype m' . (MonadBuild m',
                                                              TensorType dtype,
                                                              TensorType shape_dtype) =>
                   Tensor v'1 ResourceHandle -- ^ __resource__
                   -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                   -> Tensor v'3 shape_dtype -- ^ __shape__
                   -> m' (Tensor Value dtype) -- ^ __output__
statefulUniform :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniform = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniform' OpParams
forall a. a -> a
id
statefulUniform' :: forall v'1 v'2 v'3 dtype shape_dtype m' . (MonadBuild m',
                                                               TensorType dtype,
                                                               TensorType shape_dtype) =>
                    OpParams ->
                    Tensor v'1 ResourceHandle -- ^ __resource__
                    -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                    -> Tensor v'3 shape_dtype -- ^ __shape__
                    -> m' (Tensor Value dtype) -- ^ __output__
statefulUniform' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniform' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm shape :: Tensor v'3 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulUniform"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_FLOAT }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statefulUniformFullInt :: forall v'1 v'2 v'3 dtype shape_dtype
                          m' . (MonadBuild m', TensorType dtype,
                                TensorType shape_dtype) =>
                          Tensor v'1 ResourceHandle -- ^ __resource__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                          -> Tensor v'3 shape_dtype -- ^ __shape__
                          -> m' (Tensor Value dtype) -- ^ __output__
statefulUniformFullInt :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniformFullInt = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniformFullInt' OpParams
forall a. a -> a
id
statefulUniformFullInt' :: forall v'1 v'2 v'3 dtype shape_dtype
                           m' . (MonadBuild m', TensorType dtype,
                                 TensorType shape_dtype) => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __resource__
                           -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                           -> Tensor v'3 shape_dtype -- ^ __shape__
                           -> m' (Tensor Value dtype) -- ^ __output__
statefulUniformFullInt' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> m' (Tensor Value dtype)
statefulUniformFullInt' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm shape :: Tensor v'3 shape_dtype
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_dtype
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulUniformFullInt"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "shape_dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_UINT64 }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statefulUniformInt :: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype
                      m' . (MonadBuild m', TensorType dtype,
                            TensorType shape_dtype) =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                      -> Tensor v'3 shape_dtype -- ^ __shape__
                      -> Tensor v'4 dtype -- ^ __minval__
                      -> Tensor v'5 dtype -- ^ __maxval__
                      -> m' (Tensor Value dtype) -- ^ __output__
statefulUniformInt :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
statefulUniformInt = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) dtype shape_dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype, TensorType shape_dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
statefulUniformInt' OpParams
forall a. a -> a
id
statefulUniformInt' :: forall v'1 v'2 v'3 v'4 v'5 dtype shape_dtype
                       m' . (MonadBuild m', TensorType dtype,
                             TensorType shape_dtype) => OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> Tensor v'2 Data.Int.Int64 -- ^ __algorithm__
                       -> Tensor v'3 shape_dtype -- ^ __shape__
                       -> Tensor v'4 dtype -- ^ __minval__
                       -> Tensor v'5 dtype -- ^ __maxval__
                       -> m' (Tensor Value dtype) -- ^ __output__
statefulUniformInt' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 shape_dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> m' (Tensor Value dtype)
statefulUniformInt' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource algorithm :: Tensor v'2 Int64
algorithm shape :: Tensor v'3 shape_dtype
shape minval :: Tensor v'4 dtype
minval
                    maxval :: Tensor v'5 dtype
maxval | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
algorithm,
                                                             Tensor v'3 shape_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_dtype
shape,
                                                             Tensor v'4 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 dtype
minval,
                                                             Tensor v'5 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 dtype
maxval]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatefulUniformInt"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_dtype
forall a. HasCallStack => a
undefined :: shape_dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
input_arg { name: "algorithm" type: DT_INT64 }
input_arg { name: "shape" type_attr: "shape_dtype" }
input_arg { name: "minval" type_attr: "dtype" }
input_arg { name: "maxval" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype" type: "type" default_value { type: DT_INT64 }
}
attr {
  name: "shape_dtype" type: "type" default_value { type: DT_INT64 }
}
-}
-- | 
statelessMultinomial :: forall v'1 v'2 v'3 t tseed
                        output_dtype . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tseed,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] output_dtype) =>
                        Tensor v'1 t -- ^ __logits__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __num_samples__
                        -> Tensor v'3 tseed -- ^ __seed__
                        -> Tensor Build output_dtype -- ^ __output__
statelessMultinomial :: Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 tseed
-> Tensor Build output_dtype
statelessMultinomial = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 tseed
-> Tensor Build output_dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tseed
       output_dtype.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tseed, OneOf '[Int32, Int64] output_dtype) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 tseed
-> Tensor Build output_dtype
statelessMultinomial' OpParams
forall a. a -> a
id
statelessMultinomial' :: forall v'1 v'2 v'3 t tseed
                         output_dtype . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] t,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tseed,
                                         OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] output_dtype) =>
                         OpParams ->
                         Tensor v'1 t -- ^ __logits__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __num_samples__
                         -> Tensor v'3 tseed -- ^ __seed__
                         -> Tensor Build output_dtype -- ^ __output__
statelessMultinomial' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> Tensor v'3 tseed
-> Tensor Build output_dtype
statelessMultinomial' op'options :: OpParams
op'options logits :: Tensor v'1 t
logits num_samples :: Tensor v'2 Int32
num_samples seed :: Tensor v'3 tseed
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build output_dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build output_dtype)
-> Build OpDef -> Tensor Build output_dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
logits,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_samples,
                                                             Tensor v'3 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tseed
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessMultinomial"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ output_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (output_dtype
forall a. HasCallStack => a
undefined :: output_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "logits" type_attr: "T" }
input_arg { name: "num_samples" type: DT_INT32 }
input_arg { name: "seed" type_attr: "Tseed" }
output_arg { name: "output" type_attr: "output_dtype" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "output_dtype"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessParameterizedTruncatedNormal :: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed
                                         dtype . (OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] s,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tseed,
                                                  OneOf '[Data.Word.Word16,
                                                          Double,
                                                          Float] dtype) =>
                                         Tensor v'1 s -- ^ __shape__
                                         -> Tensor v'2 tseed -- ^ __seed__
                                         -> Tensor v'3 dtype -- ^ __means__
                                         -> Tensor v'4 dtype -- ^ __stddevs__
                                         -> Tensor v'5 dtype -- ^ __minvals__
                                         -> Tensor v'6 dtype -- ^ __maxvals__
                                         -> Tensor Build dtype -- ^ __output__
statelessParameterizedTruncatedNormal :: Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> Tensor v'6 dtype
-> Tensor Build dtype
statelessParameterizedTruncatedNormal = OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> Tensor v'6 dtype
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) s tseed dtype.
(OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed,
 OneOf '[Word16, Double, Float] dtype) =>
OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> Tensor v'6 dtype
-> Tensor Build dtype
statelessParameterizedTruncatedNormal' OpParams
forall a. a -> a
id
statelessParameterizedTruncatedNormal' :: forall v'1 v'2 v'3 v'4 v'5 v'6 s tseed
                                          dtype . (OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] s,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tseed,
                                                   OneOf '[Data.Word.Word16,
                                                           Double,
                                                           Float] dtype) =>
                                          OpParams ->
                                          Tensor v'1 s -- ^ __shape__
                                          -> Tensor v'2 tseed -- ^ __seed__
                                          -> Tensor v'3 dtype -- ^ __means__
                                          -> Tensor v'4 dtype -- ^ __stddevs__
                                          -> Tensor v'5 dtype -- ^ __minvals__
                                          -> Tensor v'6 dtype -- ^ __maxvals__
                                          -> Tensor Build dtype -- ^ __output__
statelessParameterizedTruncatedNormal' :: OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor v'5 dtype
-> Tensor v'6 dtype
-> Tensor Build dtype
statelessParameterizedTruncatedNormal' op'options :: OpParams
op'options shape :: Tensor v'1 s
shape seed :: Tensor v'2 tseed
seed means :: Tensor v'3 dtype
means stddevs :: Tensor v'4 dtype
stddevs
                                       minvals :: Tensor v'5 dtype
minvals maxvals :: Tensor v'6 dtype
maxvals | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 s
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
means,
                                                             Tensor v'4 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 dtype
stddevs,
                                                             Tensor v'5 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 dtype
minvals,
                                                             Tensor v'6 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 dtype
maxvals]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessParameterizedTruncatedNormal"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "seed" type_attr: "Tseed" }
input_arg { name: "means" type_attr: "dtype" }
input_arg { name: "stddevs" type_attr: "dtype" }
input_arg { name: "minvals" type_attr: "dtype" }
input_arg { name: "maxvals" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
-}
-- | 
statelessRandomBinomial :: forall v'1 v'2 v'3 v'4 s tseed t
                           dtype . (OneOf '[Data.Int.Int32, Data.Int.Int64] s,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tseed,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double, Float] t,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] dtype) =>
                           Tensor v'1 s -- ^ __shape__
                           -> Tensor v'2 tseed -- ^ __seed__
                           -> Tensor v'3 t -- ^ __counts__
                           -> Tensor v'4 t -- ^ __probs__
                           -> Tensor Build dtype -- ^ __output__
statelessRandomBinomial :: Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build dtype
statelessRandomBinomial = OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) s tseed t dtype.
(OneOf '[Int32, Int64] s, OneOf '[Int32, Int64] tseed,
 OneOf '[Int32, Int64, Word16, Double, Float] t,
 OneOf '[Int32, Int64, Word16, Double, Float] dtype) =>
OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build dtype
statelessRandomBinomial' OpParams
forall a. a -> a
id
statelessRandomBinomial' :: forall v'1 v'2 v'3 v'4 s tseed t
                            dtype . (OneOf '[Data.Int.Int32, Data.Int.Int64] s,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tseed,
                                     OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Data.Word.Word16, Double, Float] t,
                                     OneOf '[Data.Int.Int32, Data.Int.Int64,
                                             Data.Word.Word16, Double,
                                             Float] dtype) => OpParams ->
                            Tensor v'1 s -- ^ __shape__
                            -> Tensor v'2 tseed -- ^ __seed__
                            -> Tensor v'3 t -- ^ __counts__
                            -> Tensor v'4 t -- ^ __probs__
                            -> Tensor Build dtype -- ^ __output__
statelessRandomBinomial' :: OpParams
-> Tensor v'1 s
-> Tensor v'2 tseed
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build dtype
statelessRandomBinomial' op'options :: OpParams
op'options shape :: Tensor v'1 s
shape seed :: Tensor v'2 tseed
seed counts :: Tensor v'3 t
counts probs :: Tensor v'4 t
probs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 s -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 s
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
counts,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
probs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomBinomial"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "S" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ s -> DataType
forall a. TensorType a => a -> DataType
tensorType (s
forall a. HasCallStack => a
undefined :: s)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "S" }
input_arg { name: "seed" type_attr: "Tseed" }
input_arg { name: "counts" type_attr: "T" }
input_arg { name: "probs" type_attr: "T" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "S"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_DOUBLE }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
-}
-- | 
statelessRandomGammaV2 :: forall v'1 v'2 v'3 dtype t
                          tseed . (OneOf '[Data.Word.Word16, Double,
                                           Float] dtype, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tseed) =>
                          Tensor v'1 t -- ^ __shape__
                          -> Tensor v'2 tseed -- ^ __seed__
                          -> Tensor v'3 dtype -- ^ __alpha__
                          -> Tensor Build dtype -- ^ __output__
statelessRandomGammaV2 :: Tensor v'1 t
-> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor Build dtype
statelessRandomGammaV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype t
       tseed.
(OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64] tseed) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor Build dtype
statelessRandomGammaV2' OpParams
forall a. a -> a
id
statelessRandomGammaV2' :: forall v'1 v'2 v'3 dtype t
                           tseed . (OneOf '[Data.Word.Word16, Double,
                                            Float] dtype,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tseed) =>
                           OpParams ->
                           Tensor v'1 t -- ^ __shape__
                           -> Tensor v'2 tseed -- ^ __seed__
                           -> Tensor v'3 dtype -- ^ __alpha__
                           -> Tensor Build dtype -- ^ __output__
statelessRandomGammaV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor Build dtype
statelessRandomGammaV2' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed alpha :: Tensor v'3 dtype
alpha | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
alpha]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomGammaV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
input_arg { name: "alpha" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list { type: DT_HALF type: DT_FLOAT type: DT_DOUBLE }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessRandomNormal :: forall v'1 v'2 dtype t
                         tseed . (OneOf '[Data.Word.Word16, Double,
                                          Float] dtype, OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] t,
                                  OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] tseed) =>
                         Tensor v'1 t -- ^ __shape__
                         -> Tensor v'2 tseed -- ^ __seed__
                         -> Tensor Build dtype -- ^ __output__
statelessRandomNormal :: Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomNormal = OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype t tseed.
(OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64] tseed) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomNormal' OpParams
forall a. a -> a
id
statelessRandomNormal' :: forall v'1 v'2 dtype t
                          tseed . (OneOf '[Data.Word.Word16, Double,
                                           Float] dtype, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tseed) => OpParams ->
                          Tensor v'1 t -- ^ __shape__
                          -> Tensor v'2 tseed -- ^ __seed__
                          -> Tensor Build dtype -- ^ __output__
statelessRandomNormal' :: OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomNormal' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomNormal"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessRandomPoisson :: forall v'1 v'2 v'3 rtype dtype t
                          tseed . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                           Data.Word.Word16, Double,
                                           Float] rtype, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Word.Word16,
                                                                 Double,
                                                                 Float] dtype,
                                   OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tseed) =>
                          Tensor v'1 t -- ^ __shape__
                          -> Tensor v'2 tseed -- ^ __seed__
                          -> Tensor v'3 rtype -- ^ __lam__
                          -> Tensor Build dtype -- ^ __output__
statelessRandomPoisson :: Tensor v'1 t
-> Tensor v'2 tseed -> Tensor v'3 rtype -> Tensor Build dtype
statelessRandomPoisson = OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 rtype
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) rtype dtype
       t tseed.
(OneOf '[Int32, Int64, Word16, Double, Float] rtype,
 OneOf '[Int32, Int64, Word16, Double, Float] dtype,
 OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 rtype
-> Tensor Build dtype
statelessRandomPoisson' OpParams
forall a. a -> a
id
statelessRandomPoisson' :: forall v'1 v'2 v'3 rtype dtype t
                           tseed . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] rtype,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] dtype,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tseed) =>
                           OpParams ->
                           Tensor v'1 t -- ^ __shape__
                           -> Tensor v'2 tseed -- ^ __seed__
                           -> Tensor v'3 rtype -- ^ __lam__
                           -> Tensor Build dtype -- ^ __output__
statelessRandomPoisson' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 rtype
-> Tensor Build dtype
statelessRandomPoisson' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed lam :: Tensor v'3 rtype
lam | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed,
                                                             Tensor v'3 rtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 rtype
lam]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomPoisson"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Rtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ rtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (rtype
forall a. HasCallStack => a
undefined :: rtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
input_arg { name: "lam" type_attr: "Rtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "Rtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessRandomUniform :: forall v'1 v'2 dtype t
                          tseed . (OneOf '[Data.Word.Word16, Double,
                                           Float] dtype, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tseed) =>
                          Tensor v'1 t -- ^ __shape__
                          -> Tensor v'2 tseed -- ^ __seed__
                          -> Tensor Build dtype -- ^ __output__
statelessRandomUniform :: Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniform = OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype t tseed.
(OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64] tseed) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniform' OpParams
forall a. a -> a
id
statelessRandomUniform' :: forall v'1 v'2 dtype t
                           tseed . (OneOf '[Data.Word.Word16, Double,
                                            Float] dtype,
                                    OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] tseed) =>
                           OpParams ->
                           Tensor v'1 t -- ^ __shape__
                           -> Tensor v'2 tseed -- ^ __seed__
                           -> Tensor Build dtype -- ^ __output__
statelessRandomUniform' :: OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniform' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomUniform"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessRandomUniformFullInt :: forall v'1 v'2 dtype t
                                 tseed . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64] dtype,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] t,
                                          OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64] tseed) =>
                                 Tensor v'1 t -- ^ __shape__
                                 -> Tensor v'2 tseed -- ^ __seed__
                                 -> Tensor Build dtype -- ^ __output__
statelessRandomUniformFullInt :: Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniformFullInt = OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype t tseed.
(OneOf '[Int32, Int64, Word32, Word64] dtype,
 OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64, Word32, Word64] tseed) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniformFullInt' OpParams
forall a. a -> a
id
statelessRandomUniformFullInt' :: forall v'1 v'2 dtype t
                                  tseed . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64] dtype,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] t,
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Word.Word32,
                                                   Data.Word.Word64] tseed) =>
                                  OpParams ->
                                  Tensor v'1 t -- ^ __shape__
                                  -> Tensor v'2 tseed -- ^ __seed__
                                  -> Tensor Build dtype -- ^ __output__
statelessRandomUniformFullInt' :: OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessRandomUniformFullInt' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomUniformFullInt"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_UINT64 }
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_UINT32 type: DT_UINT64
    }
  }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values {
    list {
      type: DT_INT32 type: DT_INT64 type: DT_UINT32 type: DT_UINT64
    }
  }
}
-}
-- | 
statelessRandomUniformInt :: forall v'1 v'2 v'3 v'4 dtype t
                             tseed . (OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] dtype,
                                      OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tseed) =>
                             Tensor v'1 t -- ^ __shape__
                             -> Tensor v'2 tseed -- ^ __seed__
                             -> Tensor v'3 dtype -- ^ __minval__
                             -> Tensor v'4 dtype -- ^ __maxval__
                             -> Tensor Build dtype -- ^ __output__
statelessRandomUniformInt :: Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor Build dtype
statelessRandomUniformInt = OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) dtype t tseed.
(OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64] tseed) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor Build dtype
statelessRandomUniformInt' OpParams
forall a. a -> a
id
statelessRandomUniformInt' :: forall v'1 v'2 v'3 v'4 dtype t
                              tseed . (OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] dtype,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] t,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tseed) =>
                              OpParams ->
                              Tensor v'1 t -- ^ __shape__
                              -> Tensor v'2 tseed -- ^ __seed__
                              -> Tensor v'3 dtype -- ^ __minval__
                              -> Tensor v'4 dtype -- ^ __maxval__
                              -> Tensor Build dtype -- ^ __output__
statelessRandomUniformInt' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tseed
-> Tensor v'3 dtype
-> Tensor v'4 dtype
-> Tensor Build dtype
statelessRandomUniformInt' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed minval :: Tensor v'3 dtype
minval
                           maxval :: Tensor v'4 dtype
maxval | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed,
                                                             Tensor v'3 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 dtype
minval,
                                                             Tensor v'4 dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 dtype
maxval]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessRandomUniformInt"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
input_arg { name: "minval" type_attr: "dtype" }
input_arg { name: "maxval" type_attr: "dtype" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
statelessTruncatedNormal :: forall v'1 v'2 dtype t
                            tseed . (OneOf '[Data.Word.Word16, Double,
                                             Float] dtype,
                                     OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] tseed) =>
                            Tensor v'1 t -- ^ __shape__
                            -> Tensor v'2 tseed -- ^ __seed__
                            -> Tensor Build dtype -- ^ __output__
statelessTruncatedNormal :: Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessTruncatedNormal = OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype t tseed.
(OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t,
 OneOf '[Int32, Int64] tseed) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessTruncatedNormal' OpParams
forall a. a -> a
id
statelessTruncatedNormal' :: forall v'1 v'2 dtype t
                             tseed . (OneOf '[Data.Word.Word16, Double,
                                              Float] dtype,
                                      OneOf '[Data.Int.Int32, Data.Int.Int64] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tseed) =>
                             OpParams ->
                             Tensor v'1 t -- ^ __shape__
                             -> Tensor v'2 tseed -- ^ __seed__
                             -> Tensor Build dtype -- ^ __output__
statelessTruncatedNormal' :: OpParams -> Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
statelessTruncatedNormal' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape seed :: Tensor v'2 tseed
seed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape,
                                                             Tensor v'2 tseed -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tseed
seed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StatelessTruncatedNormal"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tseed" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tseed -> DataType
forall a. TensorType a => a -> DataType
tensorType (tseed
forall a. HasCallStack => a
undefined :: tseed)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
input_arg { name: "seed" type_attr: "Tseed" }
output_arg { name: "output" type_attr: "dtype" }
attr {
  name: "dtype"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tseed"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
staticRegexFullMatch :: ByteString -- ^ __pattern__
                        -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                        -> Tensor Build Bool -- ^ __output__
staticRegexFullMatch :: ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
staticRegexFullMatch = OpParams
-> ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
forall (v'1 :: * -> *).
OpParams
-> ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
staticRegexFullMatch' OpParams
forall a. a -> a
id
staticRegexFullMatch' :: OpParams ->
                         ByteString -- ^ __pattern__
                         -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                         -> Tensor Build Bool -- ^ __output__
staticRegexFullMatch' :: OpParams
-> ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
staticRegexFullMatch' op'options :: OpParams
op'options pattern :: ByteString
pattern input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StaticRegexFullMatch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "pattern" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
pattern
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_BOOL }
attr { name: "pattern" type: "string" }
-}
-- | 
staticRegexReplace :: ByteString -- ^ __pattern__
                      -> ByteString -- ^ __rewrite__
                      -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                      -> Tensor Build Data.ByteString.ByteString -- ^ __output__
staticRegexReplace :: ByteString
-> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
staticRegexReplace = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
staticRegexReplace' OpParams
forall a. a -> a
id
staticRegexReplace' :: OpParams ->
                       ByteString -- ^ __pattern__
                       -> ByteString -- ^ __rewrite__
                       -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                       -> Tensor Build Data.ByteString.ByteString -- ^ __output__
staticRegexReplace' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
staticRegexReplace' op'options :: OpParams
op'options pattern :: ByteString
pattern rewrite :: ByteString
rewrite input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StaticRegexReplace"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "pattern" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
pattern
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "rewrite" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
rewrite
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr { name: "pattern" type: "string" }
attr { name: "rewrite" type: "string" }
attr {
  name: "replace_global" type: "bool" default_value { b: true }
}
-}
-- | 
statsAggregatorHandle :: forall m' . (MonadBuild m') =>
                         m' (Tensor Value ResourceHandle) -- ^ __handle__
statsAggregatorHandle :: m' (Tensor Value ResourceHandle)
statsAggregatorHandle = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorHandle' OpParams
forall a. a -> a
id
statsAggregatorHandle' :: forall m' . (MonadBuild m') => OpParams ->
                          m' (Tensor Value ResourceHandle) -- ^ __handle__
statsAggregatorHandle' :: OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorHandle' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatsAggregatorHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
statsAggregatorHandleV2 :: forall m' . (MonadBuild m') =>
                           m' (Tensor Value ResourceHandle) -- ^ __handle__
statsAggregatorHandleV2 :: m' (Tensor Value ResourceHandle)
statsAggregatorHandleV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorHandleV2' OpParams
forall a. a -> a
id
statsAggregatorHandleV2' :: forall m' . (MonadBuild m') => OpParams ->
                            m' (Tensor Value ResourceHandle) -- ^ __handle__
statsAggregatorHandleV2' :: OpParams -> m' (Tensor Value ResourceHandle)
statsAggregatorHandleV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatsAggregatorHandleV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
statsAggregatorSetSummaryWriter :: forall v'1 v'2 m' . (MonadBuild m') =>
                                   Tensor v'1 ResourceHandle -- ^ __stats_aggregator__
                                   -> Tensor v'2 ResourceHandle -- ^ __summary__
                                   -> m' (ControlNode)
statsAggregatorSetSummaryWriter :: Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle -> m' ControlNode
statsAggregatorSetSummaryWriter = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' ControlNode
statsAggregatorSetSummaryWriter' OpParams
forall a. a -> a
id
statsAggregatorSetSummaryWriter' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                    OpParams ->
                                    Tensor v'1 ResourceHandle -- ^ __stats_aggregator__
                                    -> Tensor v'2 ResourceHandle -- ^ __summary__
                                    -> m' (ControlNode)
statsAggregatorSetSummaryWriter' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ResourceHandle
-> m' ControlNode
statsAggregatorSetSummaryWriter' op'options :: OpParams
op'options stats_aggregator :: Tensor v'1 ResourceHandle
stats_aggregator
                                 summary :: Tensor v'2 ResourceHandle
summary | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
stats_aggregator,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
summary]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatsAggregatorSetSummaryWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "stats_aggregator" type: DT_RESOURCE }
input_arg { name: "summary" type: DT_RESOURCE }
-}
-- | 
statsAggregatorSummary :: forall v'1 m' . (MonadBuild m') =>
                          Tensor v'1 ResourceHandle -- ^ __iterator__
                          -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __summary__
statsAggregatorSummary :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
statsAggregatorSummary = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
statsAggregatorSummary' OpParams
forall a. a -> a
id
statsAggregatorSummary' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                           Tensor v'1 ResourceHandle -- ^ __iterator__
                           -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __summary__
statsAggregatorSummary' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
statsAggregatorSummary' op'options :: OpParams
op'options iterator :: Tensor v'1 ResourceHandle
iterator | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
iterator]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StatsAggregatorSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "iterator" type: DT_RESOURCE }
output_arg { name: "summary" type: DT_STRING }
-}
-- | 
stopGradient :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                -> Tensor Build t -- ^ __output__
stopGradient :: Tensor v'1 t -> Tensor Build t
stopGradient = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
stopGradient' OpParams
forall a. a -> a
id
stopGradient' :: forall v'1 t . (TensorType t) => OpParams ->
                 Tensor v'1 t -- ^ __input__
                 -> Tensor Build t -- ^ __output__
stopGradient' :: OpParams -> Tensor v'1 t -> Tensor Build t
stopGradient' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StopGradient"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
stridedSlice :: forall v'1 v'2 v'3 v'4 t index . (TensorType t,
                                                  OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] index) =>
                Tensor v'1 t -- ^ __input__
                -> Tensor v'2 index -- ^ __begin__
                -> Tensor v'3 index -- ^ __end__
                -> Tensor v'4 index -- ^ __strides__
                -> Tensor Build t -- ^ __output__
stridedSlice :: Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor Build t
stridedSlice = OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t index.
(TensorType t, OneOf '[Int32, Int64] index) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor Build t
stridedSlice' OpParams
forall a. a -> a
id
stridedSlice' :: forall v'1 v'2 v'3 v'4 t index . (TensorType t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] index) =>
                 OpParams ->
                 Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 index -- ^ __begin__
                 -> Tensor v'3 index -- ^ __end__
                 -> Tensor v'4 index -- ^ __strides__
                 -> Tensor Build t -- ^ __output__
stridedSlice' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor Build t
stridedSlice' op'options :: OpParams
op'options input :: Tensor v'1 t
input begin :: Tensor v'2 index
begin end :: Tensor v'3 index
end strides :: Tensor v'4 index
strides | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
end,
                                                             Tensor v'4 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 index
strides]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StridedSlice"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "end" type_attr: "Index" }
input_arg { name: "strides" type_attr: "Index" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "begin_mask" type: "int" default_value { i: 0 } }
attr { name: "end_mask" type: "int" default_value { i: 0 } }
attr { name: "ellipsis_mask" type: "int" default_value { i: 0 } }
attr { name: "new_axis_mask" type: "int" default_value { i: 0 } }
attr {
  name: "shrink_axis_mask" type: "int" default_value { i: 0 }
}
-}
-- | 
stridedSliceAssign :: forall v'2 v'3 v'4 v'5 t index m' . (MonadBuild m',
                                                           TensorType t,
                                                           OneOf '[Data.Int.Int32,
                                                                   Data.Int.Int64] index) =>
                      Tensor Ref t -- ^ __ref__
                      -> Tensor v'2 index -- ^ __begin__
                      -> Tensor v'3 index -- ^ __end__
                      -> Tensor v'4 index -- ^ __strides__
                      -> Tensor v'5 t -- ^ __value__
                      -> m' (Tensor Ref t) -- ^ __output_ref__
stridedSliceAssign :: Tensor Ref t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' (Tensor Ref t)
stridedSliceAssign = OpParams
-> Tensor Ref t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' (Tensor Ref t)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *)
       (v'5 :: * -> *) t index (m' :: * -> *).
(MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) =>
OpParams
-> Tensor Ref t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' (Tensor Ref t)
stridedSliceAssign' OpParams
forall a. a -> a
id
stridedSliceAssign' :: forall v'2 v'3 v'4 v'5 t index m' . (MonadBuild m',
                                                            TensorType t,
                                                            OneOf '[Data.Int.Int32,
                                                                    Data.Int.Int64] index) =>
                       OpParams ->
                       Tensor Ref t -- ^ __ref__
                       -> Tensor v'2 index -- ^ __begin__
                       -> Tensor v'3 index -- ^ __end__
                       -> Tensor v'4 index -- ^ __strides__
                       -> Tensor v'5 t -- ^ __value__
                       -> m' (Tensor Ref t) -- ^ __output_ref__
stridedSliceAssign' :: OpParams
-> Tensor Ref t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> m' (Tensor Ref t)
stridedSliceAssign' op'options :: OpParams
op'options ref :: Tensor Ref t
ref begin :: Tensor v'2 index
begin end :: Tensor v'3 index
end strides :: Tensor v'4 index
strides value :: Tensor v'5 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref t) -> m' (Tensor Ref t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref t) -> m' (Tensor Ref t))
-> Build (Tensor Ref t) -> m' (Tensor Ref t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref t
ref,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
end,
                                                             Tensor v'4 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 index
strides,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
value]
        [Int64] -> OpDef -> Build (Tensor Ref t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "StridedSliceAssign"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "ref" type_attr: "T" is_ref: true }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "end" type_attr: "Index" }
input_arg { name: "strides" type_attr: "Index" }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output_ref" type_attr: "T" is_ref: true }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "begin_mask" type: "int" default_value { i: 0 } }
attr { name: "end_mask" type: "int" default_value { i: 0 } }
attr { name: "ellipsis_mask" type: "int" default_value { i: 0 } }
attr { name: "new_axis_mask" type: "int" default_value { i: 0 } }
attr {
  name: "shrink_axis_mask" type: "int" default_value { i: 0 }
}
-}
-- | 
stridedSliceGrad :: forall v'1 v'2 v'3 v'4 v'5 t index . (TensorType t,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] index) =>
                    Tensor v'1 index -- ^ __shape__
                    -> Tensor v'2 index -- ^ __begin__
                    -> Tensor v'3 index -- ^ __end__
                    -> Tensor v'4 index -- ^ __strides__
                    -> Tensor v'5 t -- ^ __dy__
                    -> Tensor Build t -- ^ __output__
stridedSliceGrad :: Tensor v'1 index
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
stridedSliceGrad = OpParams
-> Tensor v'1 index
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t index.
(TensorType t, OneOf '[Int32, Int64] index) =>
OpParams
-> Tensor v'1 index
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
stridedSliceGrad' OpParams
forall a. a -> a
id
stridedSliceGrad' :: forall v'1 v'2 v'3 v'4 v'5 t index . (TensorType t,
                                                           OneOf '[Data.Int.Int32,
                                                                   Data.Int.Int64] index) =>
                     OpParams ->
                     Tensor v'1 index -- ^ __shape__
                     -> Tensor v'2 index -- ^ __begin__
                     -> Tensor v'3 index -- ^ __end__
                     -> Tensor v'4 index -- ^ __strides__
                     -> Tensor v'5 t -- ^ __dy__
                     -> Tensor Build t -- ^ __output__
stridedSliceGrad' :: OpParams
-> Tensor v'1 index
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
stridedSliceGrad' op'options :: OpParams
op'options shape :: Tensor v'1 index
shape begin :: Tensor v'2 index
begin end :: Tensor v'3 index
end strides :: Tensor v'4 index
strides dy :: Tensor v'5 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 index
shape,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
end,
                                                             Tensor v'4 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 index
strides,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StridedSliceGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "Index" }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "end" type_attr: "Index" }
input_arg { name: "strides" type_attr: "Index" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "begin_mask" type: "int" default_value { i: 0 } }
attr { name: "end_mask" type: "int" default_value { i: 0 } }
attr { name: "ellipsis_mask" type: "int" default_value { i: 0 } }
attr { name: "new_axis_mask" type: "int" default_value { i: 0 } }
attr {
  name: "shrink_axis_mask" type: "int" default_value { i: 0 }
}
-}
-- | 
stringFormat :: forall v'1 t . (TensorTypes t) =>
                TensorList (v'1) t -- ^ __inputs__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringFormat :: TensorList v'1 t -> Tensor Build ByteString
stringFormat = OpParams -> TensorList v'1 t -> Tensor Build ByteString
forall (v'1 :: * -> *) (t :: [*]).
TensorTypes t =>
OpParams -> TensorList v'1 t -> Tensor Build ByteString
stringFormat' OpParams
forall a. a -> a
id
stringFormat' :: forall v'1 t . (TensorTypes t) => OpParams ->
                 TensorList (v'1) t -- ^ __inputs__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringFormat' :: OpParams -> TensorList v'1 t -> Tensor Build ByteString
stringFormat' op'options :: OpParams
op'options inputs :: TensorList v'1 t
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 t
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringFormat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy t -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy t
forall k (t :: k). Proxy t
Proxy :: Proxy t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type_list_attr: "T" }
output_arg { name: "output" type: DT_STRING }
attr { name: "T" type: "list(type)" has_minimum: true }
attr { name: "template" type: "string" default_value { s: "%s" } }
attr {
  name: "placeholder" type: "string" default_value { s: "%s" }
}
attr { name: "summarize" type: "int" default_value { i: 3 } }
-}
-- | 
stringJoin :: [Tensor v'1 Data.ByteString.ByteString] -- ^ __inputs__
              -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringJoin :: [Tensor v'1 ByteString] -> Tensor Build ByteString
stringJoin = OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
stringJoin' OpParams
forall a. a -> a
id
stringJoin' :: OpParams ->
               [Tensor v'1 Data.ByteString.ByteString] -- ^ __inputs__
               -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringJoin' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
stringJoin' op'options :: OpParams
op'options
            inputs :: [Tensor v'1 ByteString]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ByteString]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 ByteString] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 ByteString]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringJoin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 ByteString] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ByteString]
inputs) :: Int64
{-
input_arg { name: "inputs" type: DT_STRING number_attr: "N" }
output_arg { name: "output" type: DT_STRING }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "separator" type: "string" default_value { s: "" } }
-}
-- | 
stringLength :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.Int.Int32 -- ^ __output__
stringLength :: Tensor v'1 ByteString -> Tensor Build Int32
stringLength = OpParams -> Tensor v'1 ByteString -> Tensor Build Int32
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build Int32
stringLength' OpParams
forall a. a -> a
id
stringLength' :: OpParams ->
                 Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> Tensor Build Data.Int.Int32 -- ^ __output__
stringLength' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int32
stringLength' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringLength"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_INT32 }
attr {
  name: "unit"
  type: "string"
  default_value { s: "BYTE" }
  allowed_values { list { s: "BYTE" s: "UTF8_CHAR" } }
}
-}
-- | 
stringLower :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
               -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringLower :: Tensor v'1 ByteString -> Tensor Build ByteString
stringLower = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringLower' OpParams
forall a. a -> a
id
stringLower' :: OpParams ->
                Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringLower' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringLower' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringLower"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr { name: "encoding" type: "string" default_value { s: "" } }
-}
-- | 
stringNGrams :: forall v'1 v'2 tsplits . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tsplits) =>
                ByteString -- ^ __left_pad__
                -> Data.Int.Int64 -- ^ __pad_width__
                -> Bool -- ^ __preserve_short_sequences__
                -> ByteString -- ^ __right_pad__
                -> ByteString -- ^ __separator__
                -> Tensor v'1 Data.ByteString.ByteString -- ^ __data__
                -> Tensor v'2 tsplits -- ^ __data_splits__
                -> (Tensor Build Data.ByteString.ByteString,
                    Tensor Build tsplits) -- ^ (__ngrams__, __ngrams_splits__)
                --
                -- * __ngrams__
                --
                -- * __ngrams_splits__
stringNGrams :: ByteString
-> Int64
-> Bool
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 tsplits
-> (Tensor Build ByteString, Tensor Build tsplits)
stringNGrams = OpParams
-> ByteString
-> Int64
-> Bool
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 tsplits
-> (Tensor Build ByteString, Tensor Build tsplits)
forall (v'1 :: * -> *) (v'2 :: * -> *) tsplits.
OneOf '[Int32, Int64] tsplits =>
OpParams
-> ByteString
-> Int64
-> Bool
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 tsplits
-> (Tensor Build ByteString, Tensor Build tsplits)
stringNGrams' OpParams
forall a. a -> a
id
stringNGrams' :: forall v'1 v'2 tsplits . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tsplits) =>
                 OpParams ->
                 ByteString -- ^ __left_pad__
                 -> Data.Int.Int64 -- ^ __pad_width__
                 -> Bool -- ^ __preserve_short_sequences__
                 -> ByteString -- ^ __right_pad__
                 -> ByteString -- ^ __separator__
                 -> Tensor v'1 Data.ByteString.ByteString -- ^ __data__
                 -> Tensor v'2 tsplits -- ^ __data_splits__
                 -> (Tensor Build Data.ByteString.ByteString,
                     Tensor Build tsplits) -- ^ (__ngrams__, __ngrams_splits__)
                 --
                 -- * __ngrams__
                 --
                 -- * __ngrams_splits__
stringNGrams' :: OpParams
-> ByteString
-> Int64
-> Bool
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 tsplits
-> (Tensor Build ByteString, Tensor Build tsplits)
stringNGrams' op'options :: OpParams
op'options left_pad :: ByteString
left_pad pad_width :: Int64
pad_width preserve_short_sequences :: Bool
preserve_short_sequences right_pad :: ByteString
right_pad
              separator :: ByteString
separator data' :: Tensor v'1 ByteString
data' data_splits :: Tensor v'2 tsplits
data_splits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build ByteString, Tensor Build tsplits)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build ByteString, Tensor Build tsplits))
-> Build OpDef -> (Tensor Build ByteString, Tensor Build tsplits)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
data',
                                                             Tensor v'2 tsplits -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tsplits
data_splits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringNGrams"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "left_pad" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
left_pad
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "pad_width" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
pad_width
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "preserve_short_sequences" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
preserve_short_sequences
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "right_pad" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
right_pad
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "separator" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
separator
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type: DT_STRING }
input_arg { name: "data_splits" type_attr: "Tsplits" }
output_arg { name: "ngrams" type: DT_STRING }
output_arg { name: "ngrams_splits" type_attr: "Tsplits" }
attr { name: "separator" type: "string" }
attr { name: "ngram_widths" type: "list(int)" has_minimum: true }
attr { name: "left_pad" type: "string" }
attr { name: "right_pad" type: "string" }
attr { name: "pad_width" type: "int" }
attr { name: "preserve_short_sequences" type: "bool" }
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
stringSplit :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
               -> Tensor v'2 Data.ByteString.ByteString -- ^ __delimiter__
               -> (Tensor Build Data.Int.Int64,
                   Tensor Build Data.ByteString.ByteString,
                   Tensor Build Data.Int.Int64)
               -- ^ (__indices__, __values__, __shape__)
               --
               -- * __indices__
               --
               -- * __values__
               --
               -- * __shape__
stringSplit :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplit = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplit' OpParams
forall a. a -> a
id
stringSplit' :: OpParams ->
                Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor v'2 Data.ByteString.ByteString -- ^ __delimiter__
                -> (Tensor Build Data.Int.Int64,
                    Tensor Build Data.ByteString.ByteString,
                    Tensor Build Data.Int.Int64)
                -- ^ (__indices__, __values__, __shape__)
                --
                -- * __indices__
                --
                -- * __values__
                --
                -- * __shape__
stringSplit' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplit' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input delimiter :: Tensor v'2 ByteString
delimiter | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build ByteString,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
delimiter]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringSplit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
input_arg { name: "delimiter" type: DT_STRING }
output_arg { name: "indices" type: DT_INT64 }
output_arg { name: "values" type: DT_STRING }
output_arg { name: "shape" type: DT_INT64 }
attr { name: "skip_empty" type: "bool" default_value { b: true } }
-}
-- | 
stringSplitV2 :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> Tensor v'2 Data.ByteString.ByteString -- ^ __sep__
                 -> (Tensor Build Data.Int.Int64,
                     Tensor Build Data.ByteString.ByteString,
                     Tensor Build Data.Int.Int64)
                 -- ^ (__indices__, __values__, __shape__)
                 --
                 -- * __indices__
                 --
                 -- * __values__
                 --
                 -- * __shape__
stringSplitV2 :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplitV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplitV2' OpParams
forall a. a -> a
id
stringSplitV2' :: OpParams ->
                  Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __sep__
                  -> (Tensor Build Data.Int.Int64,
                      Tensor Build Data.ByteString.ByteString,
                      Tensor Build Data.Int.Int64)
                  -- ^ (__indices__, __values__, __shape__)
                  --
                  -- * __indices__
                  --
                  -- * __values__
                  --
                  -- * __shape__
stringSplitV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
stringSplitV2' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input sep :: Tensor v'2 ByteString
sep | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build Int64, Tensor Build ByteString,
     Tensor Build Int64))
-> Build OpDef
-> (Tensor Build Int64, Tensor Build ByteString,
    Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
sep]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringSplitV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
input_arg { name: "sep" type: DT_STRING }
output_arg { name: "indices" type: DT_INT64 }
output_arg { name: "values" type: DT_STRING }
output_arg { name: "shape" type: DT_INT64 }
attr { name: "maxsplit" type: "int" default_value { i: -1 } }
-}
-- | 
stringStrip :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
               -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringStrip :: Tensor v'1 ByteString -> Tensor Build ByteString
stringStrip = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringStrip' OpParams
forall a. a -> a
id
stringStrip' :: OpParams ->
                Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringStrip' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringStrip' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringStrip"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
-}
-- | 
stringToHashBucket :: Data.Int.Int64 -- ^ __num_buckets__
                      -> Tensor v'1 Data.ByteString.ByteString -- ^ __string_tensor__
                      -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucket :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucket = OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucket' OpParams
forall a. a -> a
id
stringToHashBucket' :: OpParams ->
                       Data.Int.Int64 -- ^ __num_buckets__
                       -> Tensor v'1 Data.ByteString.ByteString -- ^ __string_tensor__
                       -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucket' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucket' op'options :: OpParams
op'options num_buckets :: Int64
num_buckets string_tensor :: Tensor v'1 ByteString
string_tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
string_tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringToHashBucket"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "string_tensor" type: DT_STRING }
output_arg { name: "output" type: DT_INT64 }
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
stringToHashBucketFast :: Data.Int.Int64 -- ^ __num_buckets__
                          -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                          -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucketFast :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketFast = OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketFast' OpParams
forall a. a -> a
id
stringToHashBucketFast' :: OpParams ->
                           Data.Int.Int64 -- ^ __num_buckets__
                           -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                           -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucketFast' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketFast' op'options :: OpParams
op'options num_buckets :: Int64
num_buckets input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringToHashBucketFast"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_INT64 }
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
-}
-- | 
stringToHashBucketStrong :: Data.Int.Int64 -- ^ __num_buckets__
                            -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                            -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucketStrong :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketStrong = OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
forall (v'1 :: * -> *).
OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketStrong' OpParams
forall a. a -> a
id
stringToHashBucketStrong' :: OpParams ->
                             Data.Int.Int64 -- ^ __num_buckets__
                             -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                             -> Tensor Build Data.Int.Int64 -- ^ __output__
stringToHashBucketStrong' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
stringToHashBucketStrong' op'options :: OpParams
op'options num_buckets :: Int64
num_buckets input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringToHashBucketStrong"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_buckets" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_buckets
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_INT64 }
attr {
  name: "num_buckets" type: "int" has_minimum: true minimum: 1
}
attr { name: "key" type: "list(int)" }
-}
-- | 
stringToNumber :: forall v'1 out_type . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                                 Double, Float] out_type) =>
                  Tensor v'1 Data.ByteString.ByteString -- ^ __string_tensor__
                  -> Tensor Build out_type -- ^ __output__
stringToNumber :: Tensor v'1 ByteString -> Tensor Build out_type
stringToNumber = OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
forall (v'1 :: * -> *) out_type.
OneOf '[Int32, Int64, Double, Float] out_type =>
OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
stringToNumber' OpParams
forall a. a -> a
id
stringToNumber' :: forall v'1 out_type . (OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64, Double,
                                                  Float] out_type) =>
                   OpParams ->
                   Tensor v'1 Data.ByteString.ByteString -- ^ __string_tensor__
                   -> Tensor Build out_type -- ^ __output__
stringToNumber' :: OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
stringToNumber' op'options :: OpParams
op'options string_tensor :: Tensor v'1 ByteString
string_tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
string_tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringToNumber"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "string_tensor" type: DT_STRING }
output_arg { name: "output" type_attr: "out_type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_FLOAT type: DT_DOUBLE type: DT_INT32 type: DT_INT64
    }
  }
}
-}
-- | 
stringUpper :: Tensor v'1 Data.ByteString.ByteString -- ^ __input__
               -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringUpper :: Tensor v'1 ByteString -> Tensor Build ByteString
stringUpper = OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringUpper' OpParams
forall a. a -> a
id
stringUpper' :: OpParams ->
                Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                -> Tensor Build Data.ByteString.ByteString -- ^ __output__
stringUpper' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
stringUpper' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "StringUpper"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr { name: "encoding" type: "string" default_value { s: "" } }
-}
-- | 
sub :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word32, Data.Word.Word8, Double,
                                   Float] t) => Tensor v'1 t -- ^ __x__
       -> Tensor v'2 t -- ^ __y__
       -> Tensor Build t -- ^ __z__
sub :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sub = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sub' OpParams
forall a. a -> a
id
sub' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                    (Data.Complex.Complex Float),
                                    Data.Int.Int16, Data.Int.Int32,
                                    Data.Int.Int64, Data.Int.Int8,
                                    Data.Word.Word16, Data.Word.Word32,
                                    Data.Word.Word8, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __y__
        -> Tensor Build t -- ^ __z__
sub' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
sub' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sub"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
      type: DT_UINT32
    }
  }
}
-}
-- | 
substr :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32, Data.Int.Int64] t) =>
          Tensor v'1 Data.ByteString.ByteString -- ^ __input__
          -> Tensor v'2 t -- ^ __pos__
          -> Tensor v'3 t -- ^ __len__
          -> Tensor Build Data.ByteString.ByteString -- ^ __output__
substr :: Tensor v'1 ByteString
-> Tensor v'2 t -> Tensor v'3 t -> Tensor Build ByteString
substr = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build ByteString
substr' OpParams
forall a. a -> a
id
substr' :: forall v'1 v'2 v'3 t . (OneOf '[Data.Int.Int32, Data.Int.Int64] t) =>
           OpParams ->
           Tensor v'1 Data.ByteString.ByteString -- ^ __input__
           -> Tensor v'2 t -- ^ __pos__
           -> Tensor v'3 t -- ^ __len__
           -> Tensor Build Data.ByteString.ByteString -- ^ __output__
substr' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor Build ByteString
substr' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input pos :: Tensor v'2 t
pos len :: Tensor v'3 t
len | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
pos,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
len]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Substr"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
input_arg { name: "pos" type_attr: "T" }
input_arg { name: "len" type_attr: "T" }
output_arg { name: "output" type: DT_STRING }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "unit"
  type: "string"
  default_value { s: "BYTE" }
  allowed_values { list { s: "BYTE" s: "UTF8_CHAR" } }
}
-}
-- | 
sum :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Int.Int16, Data.Int.Int32,
                                        Data.Int.Int64, Data.Int.Int8,
                                        Data.Word.Word16, Data.Word.Word32,
                                        Data.Word.Word64, Data.Word.Word8,
                                        Double, Float] t,
                                OneOf '[Data.Int.Int32, Data.Int.Int64] tidx) =>
       Tensor v'1 t -- ^ __input__
       -> Tensor v'2 tidx -- ^ __reduction_indices__
       -> Tensor Build t -- ^ __output__
sum :: Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
sum = OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
sum' OpParams
forall a. a -> a
id
sum' :: forall v'1 v'2 t tidx . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Int.Int16, Data.Int.Int32,
                                         Data.Int.Int64, Data.Int.Int8,
                                         Data.Word.Word16, Data.Word.Word32,
                                         Data.Word.Word64, Data.Word.Word8,
                                         Double, Float] t,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] tidx) => OpParams ->
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tidx -- ^ __reduction_indices__
        -> Tensor Build t -- ^ __output__
sum' :: OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
sum' op'options :: OpParams
op'options input :: Tensor v'1 t
input reduction_indices :: Tensor v'2 tidx
reduction_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
reduction_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Sum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "reduction_indices" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "T" }
attr { name: "keep_dims" type: "bool" default_value { b: false } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
summaryWriter :: forall m' . (MonadBuild m') =>
                 m' (Tensor Value ResourceHandle) -- ^ __writer__
summaryWriter :: m' (Tensor Value ResourceHandle)
summaryWriter = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
summaryWriter' OpParams
forall a. a -> a
id
summaryWriter' :: forall m' . (MonadBuild m') => OpParams ->
                  m' (Tensor Value ResourceHandle) -- ^ __writer__
summaryWriter' :: OpParams -> m' (Tensor Value ResourceHandle)
summaryWriter' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "SummaryWriter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "writer" type: DT_RESOURCE }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr { name: "container" type: "string" default_value { s: "" } }
-}
-- | 
svd :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Word.Word16,
                               Double, Float] t) => Tensor v'1 t -- ^ __input__
       -> (Tensor Build t, Tensor Build t, Tensor Build t)
       -- ^ (__s__, __u__, __v__)
       --
       -- * __s__
       --
       -- * __u__
       --
       -- * __v__
svd :: Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
svd = OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
svd' OpParams
forall a. a -> a
id
svd' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => OpParams ->
        Tensor v'1 t -- ^ __input__
        -> (Tensor Build t, Tensor Build t, Tensor Build t)
        -- ^ (__s__, __u__, __v__)
        --
        -- * __s__
        --
        -- * __u__
        --
        -- * __v__
svd' :: OpParams
-> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
svd' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Svd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "s" type_attr: "T" }
output_arg { name: "u" type_attr: "T" }
output_arg { name: "v" type_attr: "T" }
attr { name: "compute_uv" type: "bool" default_value { b: true } }
attr {
  name: "full_matrices" type: "bool" default_value { b: false }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_HALF
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
switch :: forall v'1 v'2 t . (TensorType t) => Tensor v'1 t -- ^ __data__
          -> Tensor v'2 Bool -- ^ __pred__
          -> (Tensor Build t, Tensor Build t)
          -- ^ (__output_false__, __output_true__)
          --
          -- * __output_false__
          --
          -- * __output_true__
switch :: Tensor v'1 t -> Tensor v'2 Bool -> (Tensor Build t, Tensor Build t)
switch = OpParams
-> Tensor v'1 t
-> Tensor v'2 Bool
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Bool
-> (Tensor Build t, Tensor Build t)
switch' OpParams
forall a. a -> a
id
switch' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
           Tensor v'1 t -- ^ __data__
           -> Tensor v'2 Bool -- ^ __pred__
           -> (Tensor Build t, Tensor Build t)
           -- ^ (__output_false__, __output_true__)
           --
           -- * __output_false__
           --
           -- * __output_true__
switch' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Bool
-> (Tensor Build t, Tensor Build t)
switch' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' pred :: Tensor v'2 Bool
pred | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Bool
pred]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Switch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "pred" type: DT_BOOL }
output_arg { name: "output_false" type_attr: "T" }
output_arg { name: "output_true" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
tFRecordDataset :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                   Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                   -> m' (Tensor Value Variant) -- ^ __handle__
tFRecordDataset :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
tFRecordDataset = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
tFRecordDataset' OpParams
forall a. a -> a
id
tFRecordDataset' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                    -> m' (Tensor Value Variant) -- ^ __handle__
tFRecordDataset' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
tFRecordDataset' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames compression_type :: Tensor v'2 ByteString
compression_type
                 buffer_size :: Tensor v'3 Int64
buffer_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
compression_type,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
buffer_size]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TFRecordDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
input_arg { name: "buffer_size" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
tFRecordReader :: forall m' . (MonadBuild m') =>
                  m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
tFRecordReader :: m' (Tensor Ref ByteString)
tFRecordReader = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
tFRecordReader' OpParams
forall a. a -> a
id
tFRecordReader' :: forall m' . (MonadBuild m') => OpParams ->
                   m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
tFRecordReader' :: OpParams -> m' (Tensor Ref ByteString)
tFRecordReader' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TFRecordReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "compression_type" type: "string" default_value { s: "" }
}
-}
-- | 
tFRecordReaderV2 :: forall m' . (MonadBuild m') =>
                    m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
tFRecordReaderV2 :: m' (Tensor Value ResourceHandle)
tFRecordReaderV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
tFRecordReaderV2' OpParams
forall a. a -> a
id
tFRecordReaderV2' :: forall m' . (MonadBuild m') => OpParams ->
                     m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
tFRecordReaderV2' :: OpParams -> m' (Tensor Value ResourceHandle)
tFRecordReaderV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TFRecordReaderV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr {
  name: "compression_type" type: "string" default_value { s: "" }
}
-}
-- | 
tPUCompilationResult :: Tensor Build Data.ByteString.ByteString -- ^ __output__
tPUCompilationResult :: Tensor Build ByteString
tPUCompilationResult = OpParams -> Tensor Build ByteString
tPUCompilationResult' OpParams
forall a. a -> a
id
tPUCompilationResult' :: OpParams ->
                         Tensor Build Data.ByteString.ByteString -- ^ __output__
tPUCompilationResult' :: OpParams -> Tensor Build ByteString
tPUCompilationResult' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TPUCompilationResult"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type: DT_STRING }
-}
-- | 
tPUEmbeddingActivations :: Data.Int.Int64 -- ^ __lookup_id__
                           -> Data.Int.Int64 -- ^ __table_id__
                           -> Tensor v'1 Float -- ^ __embedding_variable__
                           -> Tensor v'2 Float -- ^ __sliced_activations__
                           -> Tensor Build Float -- ^ __output__
tPUEmbeddingActivations :: Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor Build Float
tPUEmbeddingActivations = OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor Build Float
tPUEmbeddingActivations' OpParams
forall a. a -> a
id
tPUEmbeddingActivations' :: OpParams ->
                            Data.Int.Int64 -- ^ __lookup_id__
                            -> Data.Int.Int64 -- ^ __table_id__
                            -> Tensor v'1 Float -- ^ __embedding_variable__
                            -> Tensor v'2 Float -- ^ __sliced_activations__
                            -> Tensor Build Float -- ^ __output__
tPUEmbeddingActivations' :: OpParams
-> Int64
-> Int64
-> Tensor v'1 Float
-> Tensor v'2 Float
-> Tensor Build Float
tPUEmbeddingActivations' op'options :: OpParams
op'options lookup_id :: Int64
lookup_id table_id :: Int64
table_id embedding_variable :: Tensor v'1 Float
embedding_variable
                         sliced_activations :: Tensor v'2 Float
sliced_activations | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Float
embedding_variable,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
sliced_activations]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TPUEmbeddingActivations"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "lookup_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
lookup_id
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "table_id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
table_id
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "embedding_variable" type: DT_FLOAT }
input_arg { name: "sliced_activations" type: DT_FLOAT }
output_arg { name: "output" type: DT_FLOAT }
attr { name: "table_id" type: "int" has_minimum: true }
attr { name: "lookup_id" type: "int" has_minimum: true }
-}
-- | 
tPUOrdinalSelector :: forall m' . (MonadBuild m') =>
                      m' (Tensor Value Data.Int.Int32) -- ^ __device_ordinals__
tPUOrdinalSelector :: m' (Tensor Value Int32)
tPUOrdinalSelector = OpParams -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value Int32)
tPUOrdinalSelector' OpParams
forall a. a -> a
id
tPUOrdinalSelector' :: forall m' . (MonadBuild m') => OpParams ->
                       m' (Tensor Value Data.Int.Int32) -- ^ __device_ordinals__
tPUOrdinalSelector' :: OpParams -> m' (Tensor Value Int32)
tPUOrdinalSelector' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TPUOrdinalSelector"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "device_ordinals" type: DT_INT32 }
-}
-- | 
tPUReplicateMetadata :: forall m' . (MonadBuild m') =>
                        Data.Int.Int64 -- ^ __num_replicas__
                        -> m' (ControlNode)
tPUReplicateMetadata :: Int64 -> m' ControlNode
tPUReplicateMetadata = OpParams -> Int64 -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> m' ControlNode
tPUReplicateMetadata' OpParams
forall a. a -> a
id
tPUReplicateMetadata' :: forall m' . (MonadBuild m') => OpParams ->
                         Data.Int.Int64 -- ^ __num_replicas__
                         -> m' (ControlNode)
tPUReplicateMetadata' :: OpParams -> Int64 -> m' ControlNode
tPUReplicateMetadata' op'options :: OpParams
op'options num_replicas :: Int64
num_replicas | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TPUReplicateMetadata"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_replicas" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_replicas
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
attr { name: "num_replicas" type: "int" has_minimum: true }
attr {
  name: "num_cores_per_replica" type: "int" default_value { i: 1 }
}
attr { name: "topology" type: "string" default_value { s: "" } }
attr { name: "use_tpu" type: "bool" default_value { b: true } }
attr {
  name: "device_assignment"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "computation_shape"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "host_compute_core"
  type: "list(string)"
  default_value { list { } }
}
attr {
  name: "padding_map" type: "list(string)" default_value { list { } }
}
attr {
  name: "step_marker_location"
  type: "string"
  default_value { s: "STEP_MARK_AT_ENTRY" }
}
attr {
  name: "allow_soft_placement"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
tPUReplicatedInput :: forall v'1 t . (TensorType t) =>
                      [Tensor v'1 t] -- ^ __inputs__
                      -> Tensor Build t -- ^ __output__
tPUReplicatedInput :: [Tensor v'1 t] -> Tensor Build t
tPUReplicatedInput = OpParams -> [Tensor v'1 t] -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> [Tensor v'1 t] -> Tensor Build t
tPUReplicatedInput' OpParams
forall a. a -> a
id
tPUReplicatedInput' :: forall v'1 t . (TensorType t) => OpParams ->
                       [Tensor v'1 t] -- ^ __inputs__
                       -> Tensor Build t -- ^ __output__
tPUReplicatedInput' :: OpParams -> [Tensor v'1 t] -> Tensor Build t
tPUReplicatedInput' op'options :: OpParams
op'options
                    inputs :: [Tensor v'1 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
inputs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TPUReplicatedInput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
inputs) :: Int64
{-
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
attr {
  name: "is_mirrored_variable"
  type: "bool"
  default_value { b: false }
}
attr { name: "index" type: "int" default_value { i: -1 } }
attr { name: "is_packed" type: "bool" default_value { b: false } }
-}
-- | 
tPUReplicatedOutput :: forall v'1 t . (TensorType t) =>
                       Data.Int.Int64 -- ^ __num_replicas__
                       -> Tensor v'1 t -- ^ __input__
                       -> [Tensor Build t] -- ^ __outputs__
tPUReplicatedOutput :: Int64 -> Tensor v'1 t -> [Tensor Build t]
tPUReplicatedOutput = OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
tPUReplicatedOutput' OpParams
forall a. a -> a
id
tPUReplicatedOutput' :: forall v'1 t . (TensorType t) => OpParams ->
                        Data.Int.Int64 -- ^ __num_replicas__
                        -> Tensor v'1 t -- ^ __input__
                        -> [Tensor Build t] -- ^ __outputs__
tPUReplicatedOutput' :: OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
tPUReplicatedOutput' op'options :: OpParams
op'options num_replicas :: Int64
num_replicas input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_replicas] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TPUReplicatedOutput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_replicas" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_replicas
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg {
  name: "outputs" type_attr: "T" number_attr: "num_replicas"
}
attr {
  name: "num_replicas" type: "int" has_minimum: true minimum: 1
}
attr { name: "T" type: "type" }
-}
-- | 
takeDataset :: [DataType] -- ^ __output_types__
               -> Tensor v'1 Variant -- ^ __input_dataset__
               -> Tensor v'2 Data.Int.Int64 -- ^ __count__
               -> Tensor Build Variant -- ^ __handle__
takeDataset :: [DataType]
-> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor Build Variant
takeDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
takeDataset' OpParams
forall a. a -> a
id
takeDataset' :: OpParams ->
                [DataType] -- ^ __output_types__
                -> Tensor v'1 Variant -- ^ __input_dataset__
                -> Tensor v'2 Data.Int.Int64 -- ^ __count__
                -> Tensor Build Variant -- ^ __handle__
takeDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor Build Variant
takeDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset count :: Tensor v'2 Int64
count | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
count]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TakeDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "count" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
takeManySparseFromTensorsMap :: forall v'1 dtype m' . (MonadBuild m',
                                                       TensorType dtype) =>
                                Tensor v'1 Data.Int.Int64 -- ^ __sparse_handles__
                                -> m' ((Tensor Value Data.Int.Int64,
                                        Tensor Value dtype,
                                        Tensor Value Data.Int.Int64))
                                -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                                --
                                -- * __sparse_indices__
                                --
                                -- * __sparse_values__
                                --
                                -- * __sparse_shape__
takeManySparseFromTensorsMap :: Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
takeManySparseFromTensorsMap = OpParams
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall (v'1 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
takeManySparseFromTensorsMap' OpParams
forall a. a -> a
id
takeManySparseFromTensorsMap' :: forall v'1 dtype m' . (MonadBuild m',
                                                        TensorType dtype) =>
                                 OpParams ->
                                 Tensor v'1 Data.Int.Int64 -- ^ __sparse_handles__
                                 -> m' ((Tensor Value Data.Int.Int64,
                                         Tensor Value dtype,
                                         Tensor Value Data.Int.Int64))
                                 -- ^ (__sparse_indices__, __sparse_values__, __sparse_shape__)
                                 --
                                 -- * __sparse_indices__
                                 --
                                 -- * __sparse_values__
                                 --
                                 -- * __sparse_shape__
takeManySparseFromTensorsMap' :: OpParams
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
takeManySparseFromTensorsMap' op'options :: OpParams
op'options sparse_handles :: Tensor v'1 Int64
sparse_handles | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64))
-> Build
     (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
sparse_handles]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TakeManySparseFromTensorsMap"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sparse_handles" type: DT_INT64 }
output_arg { name: "sparse_indices" type: DT_INT64 }
output_arg { name: "sparse_values" type_attr: "dtype" }
output_arg { name: "sparse_shape" type: DT_INT64 }
attr { name: "dtype" type: "type" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
tan :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Double, Float] t) =>
       Tensor v'1 t -- ^ __x__
       -> Tensor Build t -- ^ __y__
tan :: Tensor v'1 t -> Tensor Build t
tan = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build t
tan' OpParams
forall a. a -> a
id
tan' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Double, Float] t) =>
        OpParams ->
        Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
tan' :: OpParams -> Tensor v'1 t -> Tensor Build t
tan' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Tan"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT8
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
tanh :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Word.Word16,
                                Double, Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor Build t -- ^ __y__
tanh :: Tensor v'1 t -> Tensor Build t
tanh = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
tanh' OpParams
forall a. a -> a
id
tanh' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Word.Word16,
                                 Double, Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor Build t -- ^ __y__
tanh' :: OpParams -> Tensor v'1 t -> Tensor Build t
tanh' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Tanh"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
tanhGrad :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            Tensor v'1 t -- ^ __y__
            -> Tensor v'2 t -- ^ __dy__
            -> Tensor Build t -- ^ __z__
tanhGrad :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tanhGrad = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tanhGrad' OpParams
forall a. a -> a
id
tanhGrad' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                         (Data.Complex.Complex Float),
                                         Data.Word.Word16, Double, Float] t) =>
             OpParams ->
             Tensor v'1 t -- ^ __y__
             -> Tensor v'2 t -- ^ __dy__
             -> Tensor Build t -- ^ __z__
tanhGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tanhGrad' op'options :: OpParams
op'options y :: Tensor v'1 t
y dy :: Tensor v'2 t
dy | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
y,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
dy]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TanhGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "y" type_attr: "T" }
input_arg { name: "dy" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
temporaryVariable :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                     Shape -- ^ __shape__
                     -> m' (Tensor Ref dtype) -- ^ __ref__
temporaryVariable :: Shape -> m' (Tensor Ref dtype)
temporaryVariable = OpParams -> Shape -> m' (Tensor Ref dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Ref dtype)
temporaryVariable' OpParams
forall a. a -> a
id
temporaryVariable' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                      OpParams ->
                      Shape -- ^ __shape__
                      -> m' (Tensor Ref dtype) -- ^ __ref__
temporaryVariable' :: OpParams -> Shape -> m' (Tensor Ref dtype)
temporaryVariable' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref dtype) -> m' (Tensor Ref dtype))
-> Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TemporaryVariable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "ref" type_attr: "dtype" is_ref: true }
attr { name: "shape" type: "shape" }
attr { name: "dtype" type: "type" }
attr { name: "var_name" type: "string" default_value { s: "" } }
-}
-- | 
tensorArray :: forall v'1 m' . (MonadBuild m') => DataType -- ^ __dtype__
               -> Tensor v'1 Data.Int.Int32 -- ^ __size__
               -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
tensorArray :: DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
tensorArray = OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
tensorArray' OpParams
forall a. a -> a
id
tensorArray' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                DataType -- ^ __dtype__
                -> Tensor v'1 Data.Int.Int32 -- ^ __size__
                -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __handle__
tensorArray' :: OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
tensorArray' op'options :: OpParams
op'options dtype :: DataType
dtype size :: Tensor v'1 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
size]
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArray"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "handle" type: DT_STRING is_ref: true }
attr { name: "dtype" type: "type" }
attr {
  name: "dynamic_size" type: "bool" default_value { b: false }
}
attr {
  name: "clear_after_read" type: "bool" default_value { b: true }
}
attr {
  name: "tensor_array_name" type: "string" default_value { s: "" }
}
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayClose :: forall m' . (MonadBuild m') =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> m' (ControlNode)
tensorArrayClose :: Tensor Ref ByteString -> m' ControlNode
tensorArrayClose = OpParams -> Tensor Ref ByteString -> m' ControlNode
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor Ref ByteString -> m' ControlNode
tensorArrayClose' OpParams
forall a. a -> a
id
tensorArrayClose' :: forall m' . (MonadBuild m') => OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> m' (ControlNode)
tensorArrayClose' :: OpParams -> Tensor Ref ByteString -> m' ControlNode
tensorArrayClose' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayClose"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
-}
-- | 
tensorArrayCloseV2 :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> m' (ControlNode)
tensorArrayCloseV2 :: Tensor v'1 ByteString -> m' ControlNode
tensorArrayCloseV2 = OpParams -> Tensor v'1 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' ControlNode
tensorArrayCloseV2' OpParams
forall a. a -> a
id
tensorArrayCloseV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> m' (ControlNode)
tensorArrayCloseV2' :: OpParams -> Tensor v'1 ByteString -> m' ControlNode
tensorArrayCloseV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayCloseV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
-}
-- | 
tensorArrayCloseV3 :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> m' (ControlNode)
tensorArrayCloseV3 :: Tensor v'1 ResourceHandle -> m' ControlNode
tensorArrayCloseV3 = OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
tensorArrayCloseV3' OpParams
forall a. a -> a
id
tensorArrayCloseV3' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> m' (ControlNode)
tensorArrayCloseV3' :: OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
tensorArrayCloseV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayCloseV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
-}
-- | 
tensorArrayConcat :: forall v'2 dtype m' . (MonadBuild m', TensorType dtype) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Float -- ^ __flow_in__
                     -> m' ((Tensor Value dtype, Tensor Value Data.Int.Int64))
                     -- ^ (__value__, __lengths__)
                     --
                     -- * __value__
                     --
                     -- * __lengths__
tensorArrayConcat :: Tensor Ref ByteString
-> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcat = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
forall (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcat' OpParams
forall a. a -> a
id
tensorArrayConcat' :: forall v'2 dtype m' . (MonadBuild m', TensorType dtype) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Float -- ^ __flow_in__
                      -> m' ((Tensor Value dtype, Tensor Value Data.Int.Int64))
                      -- ^ (__value__, __lengths__)
                      --
                      -- * __value__
                      --
                      -- * __lengths__
tensorArrayConcat' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcat' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value dtype, Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype, Tensor Value Int64)
 -> m' (Tensor Value dtype, Tensor Value Int64))
-> Build (Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value dtype, Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype, Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayConcat"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
output_arg { name: "lengths" type: DT_INT64 }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape_except0"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayConcatV2 :: forall v'1 v'2 dtype . (TensorType dtype) =>
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> Tensor v'2 Float -- ^ __flow_in__
                       -> (Tensor Build dtype, Tensor Build Data.Int.Int64)
                       -- ^ (__value__, __lengths__)
                       --
                       -- * __value__
                       --
                       -- * __lengths__
tensorArrayConcatV2 :: Tensor v'1 ByteString
-> Tensor v'2 Float -> (Tensor Build dtype, Tensor Build Int64)
tensorArrayConcatV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> (Tensor Build dtype, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype.
TensorType dtype =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> (Tensor Build dtype, Tensor Build Int64)
tensorArrayConcatV2' OpParams
forall a. a -> a
id
tensorArrayConcatV2' :: forall v'1 v'2 dtype . (TensorType dtype) => OpParams ->
                        Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                        -> Tensor v'2 Float -- ^ __flow_in__
                        -> (Tensor Build dtype, Tensor Build Data.Int.Int64)
                        -- ^ (__value__, __lengths__)
                        --
                        -- * __value__
                        --
                        -- * __lengths__
tensorArrayConcatV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> (Tensor Build dtype, Tensor Build Int64)
tensorArrayConcatV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build dtype, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build dtype, Tensor Build Int64))
-> Build OpDef -> (Tensor Build dtype, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArrayConcatV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
output_arg { name: "lengths" type: DT_INT64 }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape_except0"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayConcatV3 :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                  TensorType dtype) =>
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Float -- ^ __flow_in__
                       -> m' ((Tensor Value dtype, Tensor Value Data.Int.Int64))
                       -- ^ (__value__, __lengths__)
                       --
                       -- * __value__
                       --
                       -- * __lengths__
tensorArrayConcatV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcatV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcatV3' OpParams
forall a. a -> a
id
tensorArrayConcatV3' :: forall v'1 v'2 dtype m' . (MonadBuild m',
                                                   TensorType dtype) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __handle__
                        -> Tensor v'2 Float -- ^ __flow_in__
                        -> m' ((Tensor Value dtype,
                                Tensor Value Data.Int.Int64))
                        -- ^ (__value__, __lengths__)
                        --
                        -- * __value__
                        --
                        -- * __lengths__
tensorArrayConcatV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value dtype, Tensor Value Int64)
tensorArrayConcatV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value dtype, Tensor Value Int64)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype, Tensor Value Int64)
 -> m' (Tensor Value dtype, Tensor Value Int64))
-> Build (Tensor Value dtype, Tensor Value Int64)
-> m' (Tensor Value dtype, Tensor Value Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype, Tensor Value Int64)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayConcatV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
output_arg { name: "lengths" type: DT_INT64 }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape_except0"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayGather :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                                TensorType dtype) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                     -> Tensor v'3 Float -- ^ __flow_in__
                     -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayGather :: Tensor Ref ByteString
-> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
tensorArrayGather = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
forall (v'2 :: * -> *) (v'3 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayGather' OpParams
forall a. a -> a
id
tensorArrayGather' :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                                 TensorType dtype) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                      -> Tensor v'3 Float -- ^ __flow_in__
                      -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayGather' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayGather' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle indices :: Tensor v'2 Int32
indices flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGather"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayGatherV2 :: forall v'1 v'2 v'3 dtype . (TensorType dtype) =>
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                       -> Tensor v'3 Float -- ^ __flow_in__
                       -> Tensor Build dtype -- ^ __value__
tensorArrayGatherV2 :: Tensor v'1 ByteString
-> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
tensorArrayGatherV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype.
TensorType dtype =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
tensorArrayGatherV2' OpParams
forall a. a -> a
id
tensorArrayGatherV2' :: forall v'1 v'2 v'3 dtype . (TensorType dtype) =>
                        OpParams ->
                        Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                        -> Tensor v'3 Float -- ^ __flow_in__
                        -> Tensor Build dtype -- ^ __value__
tensorArrayGatherV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
tensorArrayGatherV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle indices :: Tensor v'2 Int32
indices flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArrayGatherV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayGatherV3 :: forall v'1 v'2 v'3 dtype m' . (MonadBuild m',
                                                      TensorType dtype) =>
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                       -> Tensor v'3 Float -- ^ __flow_in__
                       -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayGatherV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
tensorArrayGatherV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayGatherV3' OpParams
forall a. a -> a
id
tensorArrayGatherV3' :: forall v'1 v'2 v'3 dtype m' . (MonadBuild m',
                                                       TensorType dtype) =>
                        OpParams ->
                        Tensor v'1 ResourceHandle -- ^ __handle__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                        -> Tensor v'3 Float -- ^ __flow_in__
                        -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayGatherV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayGatherV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle indices :: Tensor v'2 Int32
indices flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGatherV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayGrad :: forall v'1 v'2 m' . (MonadBuild m') =>
                   ByteString -- ^ __source__
                   -> Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                   -> Tensor v'2 Float -- ^ __flow_in__
                   -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __grad_handle__
tensorArrayGrad :: ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Ref ByteString)
tensorArrayGrad = OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Ref ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Ref ByteString)
tensorArrayGrad' OpParams
forall a. a -> a
id
tensorArrayGrad' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                    ByteString -- ^ __source__
                    -> Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Float -- ^ __flow_in__
                    -> m' (Tensor Ref Data.ByteString.ByteString) -- ^ __grad_handle__
tensorArrayGrad' :: OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Ref ByteString)
tensorArrayGrad' op'options :: OpParams
op'options source :: ByteString
source handle :: Tensor v'1 ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGrad"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "source" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
source
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "grad_handle" type: DT_STRING is_ref: true }
attr { name: "source" type: "string" }
-}
-- | 
tensorArrayGradV2 :: forall v'1 v'2 m' . (MonadBuild m') =>
                     ByteString -- ^ __source__
                     -> Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Float -- ^ __flow_in__
                     -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __grad_handle__
tensorArrayGradV2 :: ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Value ByteString)
tensorArrayGradV2 = OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Value ByteString)
tensorArrayGradV2' OpParams
forall a. a -> a
id
tensorArrayGradV2' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                      ByteString -- ^ __source__
                      -> Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Float -- ^ __flow_in__
                      -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __grad_handle__
tensorArrayGradV2' :: OpParams
-> ByteString
-> Tensor v'1 ByteString
-> Tensor v'2 Float
-> m' (Tensor Value ByteString)
tensorArrayGradV2' op'options :: OpParams
op'options source :: ByteString
source handle :: Tensor v'1 ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGradV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "source" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
source
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "grad_handle" type: DT_STRING }
attr { name: "source" type: "string" }
-}
-- | 
tensorArrayGradV3 :: forall v'1 v'2 m' . (MonadBuild m') =>
                     ByteString -- ^ __source__
                     -> Tensor v'1 ResourceHandle -- ^ __handle__
                     -> Tensor v'2 Float -- ^ __flow_in__
                     -> m' ((Tensor Value ResourceHandle, Tensor Value Float))
                     -- ^ (__grad_handle__, __flow_out__)
                     --
                     -- * __grad_handle__
                     --
                     -- * __flow_out__
tensorArrayGradV3 :: ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradV3 = OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradV3' OpParams
forall a. a -> a
id
tensorArrayGradV3' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                      ByteString -- ^ __source__
                      -> Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Float -- ^ __flow_in__
                      -> m' ((Tensor Value ResourceHandle, Tensor Value Float))
                      -- ^ (__grad_handle__, __flow_out__)
                      --
                      -- * __grad_handle__
                      --
                      -- * __flow_out__
tensorArrayGradV3' :: OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradV3' op'options :: OpParams
op'options source :: ByteString
source handle :: Tensor v'1 ResourceHandle
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Float)
 -> m' (Tensor Value ResourceHandle, Tensor Value Float))
-> Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64]
-> OpDef -> Build (Tensor Value ResourceHandle, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGradV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "source" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
source
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "grad_handle" type: DT_RESOURCE }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "source" type: "string" }
-}
-- | 
tensorArrayGradWithShape :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                            ByteString -- ^ __source__
                            -> Tensor v'1 ResourceHandle -- ^ __handle__
                            -> Tensor v'2 Float -- ^ __flow_in__
                            -> Tensor v'3 Data.Int.Int32 -- ^ __shape_to_prepend__
                            -> m' ((Tensor Value ResourceHandle,
                                    Tensor Value Float))
                            -- ^ (__grad_handle__, __flow_out__)
                            --
                            -- * __grad_handle__
                            --
                            -- * __flow_out__
tensorArrayGradWithShape :: ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradWithShape = OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradWithShape' OpParams
forall a. a -> a
id
tensorArrayGradWithShape' :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                             OpParams ->
                             ByteString -- ^ __source__
                             -> Tensor v'1 ResourceHandle -- ^ __handle__
                             -> Tensor v'2 Float -- ^ __flow_in__
                             -> Tensor v'3 Data.Int.Int32 -- ^ __shape_to_prepend__
                             -> m' ((Tensor Value ResourceHandle,
                                     Tensor Value Float))
                             -- ^ (__grad_handle__, __flow_out__)
                             --
                             -- * __grad_handle__
                             --
                             -- * __flow_out__
tensorArrayGradWithShape' :: OpParams
-> ByteString
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> Tensor v'3 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayGradWithShape' op'options :: OpParams
op'options source :: ByteString
source handle :: Tensor v'1 ResourceHandle
handle flow_in :: Tensor v'2 Float
flow_in
                          shape_to_prepend :: Tensor v'3 Int32
shape_to_prepend | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Float)
 -> m' (Tensor Value ResourceHandle, Tensor Value Float))
-> Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
shape_to_prepend]
        [Int64]
-> OpDef -> Build (Tensor Value ResourceHandle, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayGradWithShape"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "source" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
source
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "flow_in" type: DT_FLOAT }
input_arg { name: "shape_to_prepend" type: DT_INT32 }
output_arg { name: "grad_handle" type: DT_RESOURCE }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "source" type: "string" }
-}
-- | 
tensorArrayPack :: forall v'2 dtype m' . (MonadBuild m', TensorType dtype) =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                   -> Tensor v'2 Float -- ^ __flow_in__
                   -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayPack :: Tensor Ref ByteString
-> Tensor v'2 Float -> m' (Tensor Value dtype)
tensorArrayPack = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype)
forall (v'2 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype)
tensorArrayPack' OpParams
forall a. a -> a
id
tensorArrayPack' :: forall v'2 dtype m' . (MonadBuild m', TensorType dtype) =>
                    OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Float -- ^ __flow_in__
                    -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayPack' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value dtype)
tensorArrayPack' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayPack"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorArrayRead :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                              TensorType dtype) =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                   -> Tensor v'3 Float -- ^ __flow_in__
                   -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayRead :: Tensor Ref ByteString
-> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
tensorArrayRead = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
forall (v'2 :: * -> *) (v'3 :: * -> *) dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayRead' OpParams
forall a. a -> a
id
tensorArrayRead' :: forall v'2 v'3 dtype m' . (MonadBuild m',
                                               TensorType dtype) => OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                    -> Tensor v'3 Float -- ^ __flow_in__
                    -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayRead' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayRead' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle index :: Tensor v'2 Int32
index flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayRead"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
tensorArrayReadV2 :: forall v'1 v'2 v'3 dtype . (TensorType dtype) =>
                     Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                     -> Tensor v'3 Float -- ^ __flow_in__
                     -> Tensor Build dtype -- ^ __value__
tensorArrayReadV2 :: Tensor v'1 ByteString
-> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
tensorArrayReadV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype.
TensorType dtype =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
tensorArrayReadV2' OpParams
forall a. a -> a
id
tensorArrayReadV2' :: forall v'1 v'2 v'3 dtype . (TensorType dtype) =>
                      OpParams ->
                      Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 Float -- ^ __flow_in__
                      -> Tensor Build dtype -- ^ __value__
tensorArrayReadV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> Tensor Build dtype
tensorArrayReadV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle index :: Tensor v'2 Int32
index flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dtype)
-> Build OpDef -> Tensor Build dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArrayReadV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
tensorArrayReadV3 :: forall v'1 v'2 v'3 dtype m' . (MonadBuild m',
                                                    TensorType dtype) =>
                     Tensor v'1 ResourceHandle -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                     -> Tensor v'3 Float -- ^ __flow_in__
                     -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayReadV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
tensorArrayReadV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) dtype
       (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayReadV3' OpParams
forall a. a -> a
id
tensorArrayReadV3' :: forall v'1 v'2 v'3 dtype m' . (MonadBuild m',
                                                     TensorType dtype) =>
                      OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 Float -- ^ __flow_in__
                      -> m' (Tensor Value dtype) -- ^ __value__
tensorArrayReadV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 Float
-> m' (Tensor Value dtype)
tensorArrayReadV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle index :: Tensor v'2 Int32
index flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayReadV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "value" type_attr: "dtype" }
attr { name: "dtype" type: "type" }
-}
-- | 
tensorArrayScatter :: forall v'2 v'3 v'4 t m' . (MonadBuild m', TensorType t) =>
                      Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                      -> Tensor v'3 t -- ^ __value__
                      -> Tensor v'4 Float -- ^ __flow_in__
                      -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayScatter :: Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatter = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatter' OpParams
forall a. a -> a
id
tensorArrayScatter' :: forall v'2 v'3 v'4 t m' . (MonadBuild m',
                                                  TensorType t) => OpParams ->
                       Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                       -> Tensor v'3 t -- ^ __value__
                       -> Tensor v'4 Float -- ^ __flow_in__
                       -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayScatter' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatter' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle indices :: Tensor v'2 Int32
indices value :: Tensor v'3 t
value flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayScatter"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayScatterV2 :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                        Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                        -> Tensor v'3 t -- ^ __value__
                        -> Tensor v'4 Float -- ^ __flow_in__
                        -> Tensor Build Float -- ^ __flow_out__
tensorArrayScatterV2 :: Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayScatterV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayScatterV2' OpParams
forall a. a -> a
id
tensorArrayScatterV2' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                         OpParams ->
                         Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                         -> Tensor v'3 t -- ^ __value__
                         -> Tensor v'4 Float -- ^ __flow_in__
                         -> Tensor Build Float -- ^ __flow_out__
tensorArrayScatterV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayScatterV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle indices :: Tensor v'2 Int32
indices value :: Tensor v'3 t
value
                      flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArrayScatterV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayScatterV3 :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                       TensorType t) =>
                        Tensor v'1 ResourceHandle -- ^ __handle__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                        -> Tensor v'3 t -- ^ __value__
                        -> Tensor v'4 Float -- ^ __flow_in__
                        -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayScatterV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatterV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatterV3' OpParams
forall a. a -> a
id
tensorArrayScatterV3' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                        TensorType t) =>
                         OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __handle__
                         -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                         -> Tensor v'3 t -- ^ __value__
                         -> Tensor v'4 Float -- ^ __flow_in__
                         -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayScatterV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayScatterV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle indices :: Tensor v'2 Int32
indices value :: Tensor v'3 t
value
                      flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayScatterV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArraySize :: forall v'2 m' . (MonadBuild m') =>
                   Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                   -> Tensor v'2 Float -- ^ __flow_in__
                   -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
tensorArraySize :: Tensor Ref ByteString
-> Tensor v'2 Float -> m' (Tensor Value Int32)
tensorArraySize = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
forall (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
tensorArraySize' OpParams
forall a. a -> a
id
tensorArraySize' :: forall v'2 m' . (MonadBuild m') => OpParams ->
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Float -- ^ __flow_in__
                    -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
tensorArraySize' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
tensorArraySize' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArraySize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
tensorArraySizeV2 :: Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Float -- ^ __flow_in__
                     -> Tensor Build Data.Int.Int32 -- ^ __size__
tensorArraySizeV2 :: Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
tensorArraySizeV2 = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
tensorArraySizeV2' OpParams
forall a. a -> a
id
tensorArraySizeV2' :: OpParams ->
                      Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Float -- ^ __flow_in__
                      -> Tensor Build Data.Int.Int32 -- ^ __size__
tensorArraySizeV2' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 Float -> Tensor Build Int32
tensorArraySizeV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArraySizeV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
tensorArraySizeV3 :: forall v'1 v'2 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __handle__
                     -> Tensor v'2 Float -- ^ __flow_in__
                     -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
tensorArraySizeV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Float -> m' (Tensor Value Int32)
tensorArraySizeV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
tensorArraySizeV3' OpParams
forall a. a -> a
id
tensorArraySizeV3' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Float -- ^ __flow_in__
                      -> m' (Tensor Value Data.Int.Int32) -- ^ __size__
tensorArraySizeV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Int32)
tensorArraySizeV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle flow_in :: Tensor v'2 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArraySizeV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "size" type: DT_INT32 }
-}
-- | 
tensorArraySplit :: forall v'2 v'3 v'4 t m' . (MonadBuild m', TensorType t) =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 t -- ^ __value__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                    -> Tensor v'4 Float -- ^ __flow_in__
                    -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArraySplit :: Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplit = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplit' OpParams
forall a. a -> a
id
tensorArraySplit' :: forall v'2 v'3 v'4 t m' . (MonadBuild m', TensorType t) =>
                     OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 t -- ^ __value__
                     -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                     -> Tensor v'4 Float -- ^ __flow_in__
                     -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArraySplit' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplit' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle value :: Tensor v'2 t
value lengths :: Tensor v'3 Int64
lengths flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
lengths,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArraySplit"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "lengths" type: DT_INT64 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArraySplitV2 :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                      Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 t -- ^ __value__
                      -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                      -> Tensor v'4 Float -- ^ __flow_in__
                      -> Tensor Build Float -- ^ __flow_out__
tensorArraySplitV2 :: Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> Tensor Build Float
tensorArraySplitV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> Tensor Build Float
tensorArraySplitV2' OpParams
forall a. a -> a
id
tensorArraySplitV2' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> Tensor v'2 t -- ^ __value__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                       -> Tensor v'4 Float -- ^ __flow_in__
                       -> Tensor Build Float -- ^ __flow_out__
tensorArraySplitV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> Tensor Build Float
tensorArraySplitV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle value :: Tensor v'2 t
value lengths :: Tensor v'3 Int64
lengths flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
lengths,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArraySplitV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "lengths" type: DT_INT64 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArraySplitV3 :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                     TensorType t) =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 t -- ^ __value__
                      -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                      -> Tensor v'4 Float -- ^ __flow_in__
                      -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArraySplitV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplitV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplitV3' OpParams
forall a. a -> a
id
tensorArraySplitV3' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                      TensorType t) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 t -- ^ __value__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                       -> Tensor v'4 Float -- ^ __flow_in__
                       -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArraySplitV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 t
-> Tensor v'3 Int64
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArraySplitV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle value :: Tensor v'2 t
value lengths :: Tensor v'3 Int64
lengths flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
lengths,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArraySplitV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "lengths" type: DT_INT64 }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayUnpack :: forall v'2 v'3 t m' . (MonadBuild m', TensorType t) =>
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 t -- ^ __value__
                     -> Tensor v'3 Float -- ^ __flow_in__
                     -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayUnpack :: Tensor Ref ByteString
-> Tensor v'2 t -> Tensor v'3 Float -> m' (Tensor Value Float)
tensorArrayUnpack = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Float
-> m' (Tensor Value Float)
forall (v'2 :: * -> *) (v'3 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Float
-> m' (Tensor Value Float)
tensorArrayUnpack' OpParams
forall a. a -> a
id
tensorArrayUnpack' :: forall v'2 v'3 t m' . (MonadBuild m', TensorType t) =>
                      OpParams ->
                      Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 t -- ^ __value__
                      -> Tensor v'3 Float -- ^ __flow_in__
                      -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayUnpack' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 t
-> Tensor v'3 Float
-> m' (Tensor Value Float)
tensorArrayUnpack' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle value :: Tensor v'2 t
value flow_in :: Tensor v'3 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
value,
                                                             Tensor v'3 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayUnpack"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayV2 :: forall v'1 m' . (MonadBuild m') => DataType -- ^ __dtype__
                 -> Tensor v'1 Data.Int.Int32 -- ^ __size__
                 -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __handle__
tensorArrayV2 :: DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
tensorArrayV2 = OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
tensorArrayV2' OpParams
forall a. a -> a
id
tensorArrayV2' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                  DataType -- ^ __dtype__
                  -> Tensor v'1 Data.Int.Int32 -- ^ __size__
                  -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __handle__
tensorArrayV2' :: OpParams
-> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
tensorArrayV2' op'options :: OpParams
op'options dtype :: DataType
dtype size :: Tensor v'1 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
size]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "handle" type: DT_STRING }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
attr {
  name: "dynamic_size" type: "bool" default_value { b: false }
}
attr {
  name: "clear_after_read" type: "bool" default_value { b: true }
}
attr {
  name: "tensor_array_name" type: "string" default_value { s: "" }
}
-}
-- | 
tensorArrayV3 :: forall v'1 m' . (MonadBuild m') => DataType -- ^ __dtype__
                 -> Tensor v'1 Data.Int.Int32 -- ^ __size__
                 -> m' ((Tensor Value ResourceHandle, Tensor Value Float))
                 -- ^ (__handle__, __flow__)
                 --
                 -- * __handle__
                 --
                 -- * __flow__
tensorArrayV3 :: DataType
-> Tensor v'1 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayV3 = OpParams
-> DataType
-> Tensor v'1 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> DataType
-> Tensor v'1 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayV3' OpParams
forall a. a -> a
id
tensorArrayV3' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                  DataType -- ^ __dtype__
                  -> Tensor v'1 Data.Int.Int32 -- ^ __size__
                  -> m' ((Tensor Value ResourceHandle, Tensor Value Float))
                  -- ^ (__handle__, __flow__)
                  --
                  -- * __handle__
                  --
                  -- * __flow__
tensorArrayV3' :: OpParams
-> DataType
-> Tensor v'1 Int32
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
tensorArrayV3' op'options :: OpParams
op'options dtype :: DataType
dtype size :: Tensor v'1 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle, Tensor Value Float)
 -> m' (Tensor Value ResourceHandle, Tensor Value Float))
-> Build (Tensor Value ResourceHandle, Tensor Value Float)
-> m' (Tensor Value ResourceHandle, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
size]
        [Int64]
-> OpDef -> Build (Tensor Value ResourceHandle, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "handle" type: DT_RESOURCE }
output_arg { name: "flow" type: DT_FLOAT }
attr { name: "dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
attr {
  name: "dynamic_size" type: "bool" default_value { b: false }
}
attr {
  name: "clear_after_read" type: "bool" default_value { b: true }
}
attr {
  name: "identical_element_shapes"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "tensor_array_name" type: "string" default_value { s: "" }
}
-}
-- | 
tensorArrayWrite :: forall v'2 v'3 v'4 t m' . (MonadBuild m', TensorType t) =>
                    Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                    -> Tensor v'3 t -- ^ __value__
                    -> Tensor v'4 Float -- ^ __flow_in__
                    -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayWrite :: Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWrite = OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'2 :: * -> *) (v'3 :: * -> *) (v'4 :: * -> *) t
       (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWrite' OpParams
forall a. a -> a
id
tensorArrayWrite' :: forall v'2 v'3 v'4 t m' . (MonadBuild m', TensorType t) =>
                     OpParams ->
                     Tensor Ref Data.ByteString.ByteString -- ^ __handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                     -> Tensor v'3 t -- ^ __value__
                     -> Tensor v'4 Float -- ^ __flow_in__
                     -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayWrite' :: OpParams
-> Tensor Ref ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWrite' op'options :: OpParams
op'options handle :: Tensor Ref ByteString
handle index :: Tensor v'2 Int32
index value :: Tensor v'3 t
value flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor Ref ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor Ref ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayWrite"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING is_ref: true }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayWriteV2 :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
                      Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 t -- ^ __value__
                      -> Tensor v'4 Float -- ^ __flow_in__
                      -> Tensor Build Float -- ^ __flow_out__
tensorArrayWriteV2 :: Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayWriteV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayWriteV2' OpParams
forall a. a -> a
id
tensorArrayWriteV2' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                       Tensor v'1 Data.ByteString.ByteString -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                       -> Tensor v'3 t -- ^ __value__
                       -> Tensor v'4 Float -- ^ __flow_in__
                       -> Tensor Build Float -- ^ __flow_out__
tensorArrayWriteV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> Tensor Build Float
tensorArrayWriteV2' op'options :: OpParams
op'options handle :: Tensor v'1 ByteString
handle index :: Tensor v'2 Int32
index value :: Tensor v'3 t
value flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Float
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Float)
-> Build OpDef -> Tensor Build Float
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorArrayWriteV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_STRING }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorArrayWriteV3 :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                     TensorType t) =>
                      Tensor v'1 ResourceHandle -- ^ __handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 t -- ^ __value__
                      -> Tensor v'4 Float -- ^ __flow_in__
                      -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayWriteV3 :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWriteV3 = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWriteV3' OpParams
forall a. a -> a
id
tensorArrayWriteV3' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                      TensorType t) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __handle__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                       -> Tensor v'3 t -- ^ __value__
                       -> Tensor v'4 Float -- ^ __flow_in__
                       -> m' (Tensor Value Float) -- ^ __flow_out__
tensorArrayWriteV3' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int32
-> Tensor v'3 t
-> Tensor v'4 Float
-> m' (Tensor Value Float)
tensorArrayWriteV3' op'options :: OpParams
op'options handle :: Tensor v'1 ResourceHandle
handle index :: Tensor v'2 Int32
index value :: Tensor v'3 t
value flow_in :: Tensor v'4 Float
flow_in | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
value,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
flow_in]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorArrayWriteV3"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "handle" type: DT_RESOURCE }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "value" type_attr: "T" }
input_arg { name: "flow_in" type: DT_FLOAT }
output_arg { name: "flow_out" type: DT_FLOAT }
attr { name: "T" type: "type" }
-}
-- | 
tensorDataset :: forall v'1 toutput_types m' . (MonadBuild m',
                                                TensorTypes toutput_types) =>
                 TensorList (v'1) toutput_types -- ^ __components__
                 -> m' (Tensor Value Variant) -- ^ __handle__
tensorDataset :: TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorDataset = OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (toutput_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes toutput_types) =>
OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorDataset' OpParams
forall a. a -> a
id
tensorDataset' :: forall v'1 toutput_types m' . (MonadBuild m',
                                                 TensorTypes toutput_types) =>
                  OpParams ->
                  TensorList (v'1) toutput_types -- ^ __components__
                  -> m' (Tensor Value Variant) -- ^ __handle__
tensorDataset' :: OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorDataset' op'options :: OpParams
op'options components :: TensorList v'1 toutput_types
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 toutput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 toutput_types
components]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutput_types
forall k (t :: k). Proxy t
Proxy :: Proxy toutput_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "components" type_list_attr: "Toutput_types" }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "Toutput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
tensorForestCreateTreeVariable :: forall v'1 v'2 m' . (MonadBuild m') =>
                                  Tensor v'1 ResourceHandle -- ^ __tree_handle__
                                  -> Tensor v'2 Data.ByteString.ByteString -- ^ __tree_config__
                                  -> m' (ControlNode)
tensorForestCreateTreeVariable :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString -> m' ControlNode
tensorForestCreateTreeVariable = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
tensorForestCreateTreeVariable' OpParams
forall a. a -> a
id
tensorForestCreateTreeVariable' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                   OpParams ->
                                   Tensor v'1 ResourceHandle -- ^ __tree_handle__
                                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __tree_config__
                                   -> m' (ControlNode)
tensorForestCreateTreeVariable' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
tensorForestCreateTreeVariable' op'options :: OpParams
op'options tree_handle :: Tensor v'1 ResourceHandle
tree_handle
                                tree_config :: Tensor v'2 ByteString
tree_config | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tree_config]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestCreateTreeVariable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
input_arg { name: "tree_config" type: DT_STRING }
-}
-- | 
tensorForestTreeDeserialize :: forall v'1 v'2 m' . (MonadBuild m') =>
                               Tensor v'1 ResourceHandle -- ^ __tree_handle__
                               -> Tensor v'2 Data.ByteString.ByteString -- ^ __tree_config__
                               -> m' (ControlNode)
tensorForestTreeDeserialize :: Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString -> m' ControlNode
tensorForestTreeDeserialize = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
tensorForestTreeDeserialize' OpParams
forall a. a -> a
id
tensorForestTreeDeserialize' :: forall v'1 v'2 m' . (MonadBuild m') =>
                                OpParams ->
                                Tensor v'1 ResourceHandle -- ^ __tree_handle__
                                -> Tensor v'2 Data.ByteString.ByteString -- ^ __tree_config__
                                -> m' (ControlNode)
tensorForestTreeDeserialize' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 ByteString
-> m' ControlNode
tensorForestTreeDeserialize' op'options :: OpParams
op'options tree_handle :: Tensor v'1 ResourceHandle
tree_handle
                             tree_config :: Tensor v'2 ByteString
tree_config | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
tree_config]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreeDeserialize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
input_arg { name: "tree_config" type: DT_STRING }
-}
-- | 
tensorForestTreeIsInitializedOp :: forall v'1 m' . (MonadBuild m') =>
                                   Tensor v'1 ResourceHandle -- ^ __tree_handle__
                                   -> m' (Tensor Value Bool) -- ^ __is_initialized__
tensorForestTreeIsInitializedOp :: Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
tensorForestTreeIsInitializedOp = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
tensorForestTreeIsInitializedOp' OpParams
forall a. a -> a
id
tensorForestTreeIsInitializedOp' :: forall v'1 m' . (MonadBuild m') =>
                                    OpParams ->
                                    Tensor v'1 ResourceHandle -- ^ __tree_handle__
                                    -> m' (Tensor Value Bool) -- ^ __is_initialized__
tensorForestTreeIsInitializedOp' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
tensorForestTreeIsInitializedOp' op'options :: OpParams
op'options tree_handle :: Tensor v'1 ResourceHandle
tree_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreeIsInitializedOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
output_arg { name: "is_initialized" type: DT_BOOL }
-}
-- | 
tensorForestTreePredict :: forall v'1 v'2 m' . (MonadBuild m') =>
                           Data.Int.Int64 -- ^ __logits_dimension__
                           -> Tensor v'1 ResourceHandle -- ^ __tree_handle__
                           -> Tensor v'2 Float -- ^ __dense_features__
                           -> m' (Tensor Value Float) -- ^ __logits__
tensorForestTreePredict :: Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Float)
tensorForestTreePredict = OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Float)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Float)
tensorForestTreePredict' OpParams
forall a. a -> a
id
tensorForestTreePredict' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                            Data.Int.Int64 -- ^ __logits_dimension__
                            -> Tensor v'1 ResourceHandle -- ^ __tree_handle__
                            -> Tensor v'2 Float -- ^ __dense_features__
                            -> m' (Tensor Value Float) -- ^ __logits__
tensorForestTreePredict' :: OpParams
-> Int64
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Float
-> m' (Tensor Value Float)
tensorForestTreePredict' op'options :: OpParams
op'options logits_dimension :: Int64
logits_dimension tree_handle :: Tensor v'1 ResourceHandle
tree_handle
                         dense_features :: Tensor v'2 Float
dense_features | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Float) -> m' (Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Float) -> m' (Tensor Value Float))
-> Build (Tensor Value Float) -> m' (Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle,
                                                             Tensor v'2 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Float
dense_features]
        [Int64] -> OpDef -> Build (Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreePredict"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "logits_dimension" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
logits_dimension
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
input_arg { name: "dense_features" type: DT_FLOAT }
output_arg { name: "logits" type: DT_FLOAT }
attr { name: "logits_dimension" type: "int" }
-}
-- | 
tensorForestTreeResourceHandleOp :: forall m' . (MonadBuild m') =>
                                    m' (Tensor Value ResourceHandle) -- ^ __resource__
tensorForestTreeResourceHandleOp :: m' (Tensor Value ResourceHandle)
tensorForestTreeResourceHandleOp = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
tensorForestTreeResourceHandleOp' OpParams
forall a. a -> a
id
tensorForestTreeResourceHandleOp' :: forall m' . (MonadBuild m') => OpParams ->
                                     m' (Tensor Value ResourceHandle) -- ^ __resource__
tensorForestTreeResourceHandleOp' :: OpParams -> m' (Tensor Value ResourceHandle)
tensorForestTreeResourceHandleOp' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreeResourceHandleOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resource" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
tensorForestTreeSerialize :: forall v'1 m' . (MonadBuild m') =>
                             Tensor v'1 ResourceHandle -- ^ __tree_handle__
                             -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __tree_config__
tensorForestTreeSerialize :: Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
tensorForestTreeSerialize = OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
tensorForestTreeSerialize' OpParams
forall a. a -> a
id
tensorForestTreeSerialize' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                              Tensor v'1 ResourceHandle -- ^ __tree_handle__
                              -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __tree_config__
tensorForestTreeSerialize' :: OpParams
-> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
tensorForestTreeSerialize' op'options :: OpParams
op'options tree_handle :: Tensor v'1 ResourceHandle
tree_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreeSerialize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
output_arg { name: "tree_config" type: DT_STRING }
-}
-- | 
tensorForestTreeSize :: forall v'1 m' . (MonadBuild m') =>
                        Tensor v'1 ResourceHandle -- ^ __tree_handle__
                        -> m' (Tensor Value Data.Int.Int32) -- ^ __tree_size__
tensorForestTreeSize :: Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
tensorForestTreeSize = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
tensorForestTreeSize' OpParams
forall a. a -> a
id
tensorForestTreeSize' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __tree_handle__
                         -> m' (Tensor Value Data.Int.Int32) -- ^ __tree_size__
tensorForestTreeSize' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
tensorForestTreeSize' op'options :: OpParams
op'options tree_handle :: Tensor v'1 ResourceHandle
tree_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
tree_handle]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorForestTreeSize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tree_handle" type: DT_RESOURCE }
output_arg { name: "tree_size" type: DT_INT32 }
-}
-- | 
tensorListConcat :: forall v'1 element_dtype . (TensorType element_dtype) =>
                    Tensor v'1 Variant -- ^ __input_handle__
                    -> (Tensor Build element_dtype, Tensor Build Data.Int.Int64)
                    -- ^ (__tensor__, __lengths__)
                    --
                    -- * __tensor__
                    --
                    -- * __lengths__
tensorListConcat :: Tensor v'1 Variant
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcat = OpParams
-> Tensor v'1 Variant
-> (Tensor Build element_dtype, Tensor Build Int64)
forall (v'1 :: * -> *) element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcat' OpParams
forall a. a -> a
id
tensorListConcat' :: forall v'1 element_dtype . (TensorType element_dtype) =>
                     OpParams ->
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> (Tensor Build element_dtype,
                         Tensor Build Data.Int.Int64)
                     -- ^ (__tensor__, __lengths__)
                     --
                     -- * __tensor__
                     --
                     -- * __lengths__
tensorListConcat' :: OpParams
-> Tensor v'1 Variant
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcat' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64))
-> Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListConcat"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
output_arg { name: "tensor" type_attr: "element_dtype" }
output_arg { name: "lengths" type: DT_INT64 }
attr { name: "element_dtype" type: "type" }
attr {
  name: "element_shape"
  type: "shape"
  default_value { shape { unknown_rank: true } }
}
-}
-- | 
tensorListConcatLists :: DataType -- ^ __element_dtype__
                         -> Tensor v'1 Variant -- ^ __input_a__
                         -> Tensor v'2 Variant -- ^ __input_b__
                         -> Tensor Build Variant -- ^ __output__
tensorListConcatLists :: DataType
-> Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor Build Variant
tensorListConcatLists = OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
tensorListConcatLists' OpParams
forall a. a -> a
id
tensorListConcatLists' :: OpParams ->
                          DataType -- ^ __element_dtype__
                          -> Tensor v'1 Variant -- ^ __input_a__
                          -> Tensor v'2 Variant -- ^ __input_b__
                          -> Tensor Build Variant -- ^ __output__
tensorListConcatLists' :: OpParams
-> DataType
-> Tensor v'1 Variant
-> Tensor v'2 Variant
-> Tensor Build Variant
tensorListConcatLists' op'options :: OpParams
op'options element_dtype :: DataType
element_dtype input_a :: Tensor v'1 Variant
input_a
                       input_b :: Tensor v'2 Variant
input_b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_a,
                                                             Tensor v'2 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Variant
input_b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListConcatLists"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
element_dtype
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_a" type: DT_VARIANT }
input_arg { name: "input_b" type: DT_VARIANT }
output_arg { name: "output" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListConcatV2 :: forall v'1 v'2 v'3 element_dtype
                      shape_type . (TensorType element_dtype,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] shape_type) =>
                      Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor v'2 shape_type -- ^ __element_shape__
                      -> Tensor v'3 Data.Int.Int64 -- ^ __leading_dims__
                      -> (Tensor Build element_dtype,
                          Tensor Build Data.Int.Int64)
                      -- ^ (__tensor__, __lengths__)
                      --
                      -- * __tensor__
                      --
                      -- * __lengths__
tensorListConcatV2 :: Tensor v'1 Variant
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcatV2 = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> (Tensor Build element_dtype, Tensor Build Int64)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype shape_type.
(TensorType element_dtype, OneOf '[Int32, Int64] shape_type) =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcatV2' OpParams
forall a. a -> a
id
tensorListConcatV2' :: forall v'1 v'2 v'3 element_dtype
                       shape_type . (TensorType element_dtype,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] shape_type) =>
                       OpParams ->
                       Tensor v'1 Variant -- ^ __input_handle__
                       -> Tensor v'2 shape_type -- ^ __element_shape__
                       -> Tensor v'3 Data.Int.Int64 -- ^ __leading_dims__
                       -> (Tensor Build element_dtype,
                           Tensor Build Data.Int.Int64)
                       -- ^ (__tensor__, __lengths__)
                       --
                       -- * __tensor__
                       --
                       -- * __lengths__
tensorListConcatV2' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> (Tensor Build element_dtype, Tensor Build Int64)
tensorListConcatV2' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle element_shape :: Tensor v'2 shape_type
element_shape
                    leading_dims :: Tensor v'3 Int64
leading_dims | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64))
-> Build OpDef -> (Tensor Build element_dtype, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 shape_type
element_shape,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
leading_dims]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListConcatV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "element_shape" type_attr: "shape_type" }
input_arg { name: "leading_dims" type: DT_INT64 }
output_arg { name: "tensor" type_attr: "element_dtype" }
output_arg { name: "lengths" type: DT_INT64 }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListElementShape :: forall v'1 shape_type . (OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] shape_type) =>
                          Tensor v'1 Variant -- ^ __input_handle__
                          -> Tensor Build shape_type -- ^ __element_shape__
tensorListElementShape :: Tensor v'1 Variant -> Tensor Build shape_type
tensorListElementShape = OpParams -> Tensor v'1 Variant -> Tensor Build shape_type
forall (v'1 :: * -> *) shape_type.
OneOf '[Int32, Int64] shape_type =>
OpParams -> Tensor v'1 Variant -> Tensor Build shape_type
tensorListElementShape' OpParams
forall a. a -> a
id
tensorListElementShape' :: forall v'1 shape_type . (OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] shape_type) =>
                           OpParams ->
                           Tensor v'1 Variant -- ^ __input_handle__
                           -> Tensor Build shape_type -- ^ __element_shape__
tensorListElementShape' :: OpParams -> Tensor v'1 Variant -> Tensor Build shape_type
tensorListElementShape' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build shape_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build shape_type)
-> Build OpDef -> Tensor Build shape_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListElementShape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
output_arg { name: "element_shape" type_attr: "shape_type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListFromTensor :: forall v'1 v'2 element_dtype
                        shape_type . (TensorType element_dtype,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] shape_type) =>
                        Tensor v'1 element_dtype -- ^ __tensor__
                        -> Tensor v'2 shape_type -- ^ __element_shape__
                        -> Tensor Build Variant -- ^ __output_handle__
tensorListFromTensor :: Tensor v'1 element_dtype
-> Tensor v'2 shape_type -> Tensor Build Variant
tensorListFromTensor = OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) element_dtype shape_type.
(TensorType element_dtype, OneOf '[Int32, Int64] shape_type) =>
OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor Build Variant
tensorListFromTensor' OpParams
forall a. a -> a
id
tensorListFromTensor' :: forall v'1 v'2 element_dtype
                         shape_type . (TensorType element_dtype,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] shape_type) =>
                         OpParams ->
                         Tensor v'1 element_dtype -- ^ __tensor__
                         -> Tensor v'2 shape_type -- ^ __element_shape__
                         -> Tensor Build Variant -- ^ __output_handle__
tensorListFromTensor' :: OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor Build Variant
tensorListFromTensor' op'options :: OpParams
op'options tensor :: Tensor v'1 element_dtype
tensor element_shape :: Tensor v'2 shape_type
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 element_dtype
tensor,
                                                             Tensor v'2 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 shape_type
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListFromTensor"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "element_dtype" }
input_arg { name: "element_shape" type_attr: "shape_type" }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListGather :: forall v'1 v'2 v'3
                    element_dtype . (TensorType element_dtype) =>
                    Tensor v'1 Variant -- ^ __input_handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                    -> Tensor v'3 Data.Int.Int32 -- ^ __element_shape__
                    -> Tensor Build element_dtype -- ^ __values__
tensorListGather :: Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGather = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGather' OpParams
forall a. a -> a
id
tensorListGather' :: forall v'1 v'2 v'3
                     element_dtype . (TensorType element_dtype) => OpParams ->
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                     -> Tensor v'3 Data.Int.Int32 -- ^ __element_shape__
                     -> Tensor Build element_dtype -- ^ __values__
tensorListGather' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGather' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle indices :: Tensor v'2 Int32
indices
                  element_shape :: Tensor v'3 Int32
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build element_dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build element_dtype)
-> Build OpDef -> Tensor Build element_dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListGather"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "element_shape" type: DT_INT32 }
output_arg { name: "values" type_attr: "element_dtype" }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListGetItem :: forall v'1 v'2 v'3
                     element_dtype . (TensorType element_dtype) =>
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                     -> Tensor v'3 Data.Int.Int32 -- ^ __element_shape__
                     -> Tensor Build element_dtype -- ^ __item__
tensorListGetItem :: Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGetItem = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGetItem' OpParams
forall a. a -> a
id
tensorListGetItem' :: forall v'1 v'2 v'3
                      element_dtype . (TensorType element_dtype) => OpParams ->
                      Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 Data.Int.Int32 -- ^ __element_shape__
                      -> Tensor Build element_dtype -- ^ __item__
tensorListGetItem' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 Int32
-> Tensor Build element_dtype
tensorListGetItem' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle index :: Tensor v'2 Int32
index
                   element_shape :: Tensor v'3 Int32
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build element_dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build element_dtype)
-> Build OpDef -> Tensor Build element_dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListGetItem"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "element_shape" type: DT_INT32 }
output_arg { name: "item" type_attr: "element_dtype" }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListLength :: Tensor v'1 Variant -- ^ __input_handle__
                    -> Tensor Build Data.Int.Int32 -- ^ __length__
tensorListLength :: Tensor v'1 Variant -> Tensor Build Int32
tensorListLength = OpParams -> Tensor v'1 Variant -> Tensor Build Int32
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Int32
tensorListLength' OpParams
forall a. a -> a
id
tensorListLength' :: OpParams ->
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor Build Data.Int.Int32 -- ^ __length__
tensorListLength' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
tensorListLength' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListLength"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
output_arg { name: "length" type: DT_INT32 }
-}
-- | 
tensorListPopBack :: forall v'1 v'2
                     element_dtype . (TensorType element_dtype) =>
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __element_shape__
                     -> (Tensor Build Variant, Tensor Build element_dtype)
                     -- ^ (__output_handle__, __tensor__)
                     --
                     -- * __output_handle__
                     --
                     -- * __tensor__
tensorListPopBack :: Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Variant, Tensor Build element_dtype)
tensorListPopBack = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Variant, Tensor Build element_dtype)
forall (v'1 :: * -> *) (v'2 :: * -> *) element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Variant, Tensor Build element_dtype)
tensorListPopBack' OpParams
forall a. a -> a
id
tensorListPopBack' :: forall v'1 v'2
                      element_dtype . (TensorType element_dtype) => OpParams ->
                      Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __element_shape__
                      -> (Tensor Build Variant, Tensor Build element_dtype)
                      -- ^ (__output_handle__, __tensor__)
                      --
                      -- * __output_handle__
                      --
                      -- * __tensor__
tensorListPopBack' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> (Tensor Build Variant, Tensor Build element_dtype)
tensorListPopBack' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle element_shape :: Tensor v'2 Int32
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build Variant, Tensor Build element_dtype)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build Variant, Tensor Build element_dtype))
-> Build OpDef
-> (Tensor Build Variant, Tensor Build element_dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListPopBack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "element_shape" type: DT_INT32 }
output_arg { name: "output_handle" type: DT_VARIANT }
output_arg { name: "tensor" type_attr: "element_dtype" }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListPushBack :: forall v'1 v'2
                      element_dtype . (TensorType element_dtype) =>
                      Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor v'2 element_dtype -- ^ __tensor__
                      -> Tensor Build Variant -- ^ __output_handle__
tensorListPushBack :: Tensor v'1 Variant
-> Tensor v'2 element_dtype -> Tensor Build Variant
tensorListPushBack = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
tensorListPushBack' OpParams
forall a. a -> a
id
tensorListPushBack' :: forall v'1 v'2
                       element_dtype . (TensorType element_dtype) => OpParams ->
                       Tensor v'1 Variant -- ^ __input_handle__
                       -> Tensor v'2 element_dtype -- ^ __tensor__
                       -> Tensor Build Variant -- ^ __output_handle__
tensorListPushBack' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor Build Variant
tensorListPushBack' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle tensor :: Tensor v'2 element_dtype
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 element_dtype
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListPushBack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "tensor" type_attr: "element_dtype" }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListPushBackBatch :: forall v'1 v'2
                           element_dtype . (TensorType element_dtype) =>
                           Tensor v'1 Variant -- ^ __input_handles__
                           -> Tensor v'2 element_dtype -- ^ __tensor__
                           -> Tensor Build Variant -- ^ __output_handles__
tensorListPushBackBatch :: Tensor v'1 Variant
-> Tensor v'2 element_dtype -> Tensor Build Variant
tensorListPushBackBatch = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
tensorListPushBackBatch' OpParams
forall a. a -> a
id
tensorListPushBackBatch' :: forall v'1 v'2
                            element_dtype . (TensorType element_dtype) =>
                            OpParams ->
                            Tensor v'1 Variant -- ^ __input_handles__
                            -> Tensor v'2 element_dtype -- ^ __tensor__
                            -> Tensor Build Variant -- ^ __output_handles__
tensorListPushBackBatch' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor Build Variant
tensorListPushBackBatch' op'options :: OpParams
op'options input_handles :: Tensor v'1 Variant
input_handles tensor :: Tensor v'2 element_dtype
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handles,
                                                             Tensor v'2 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 element_dtype
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListPushBackBatch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handles" type: DT_VARIANT }
input_arg { name: "tensor" type_attr: "element_dtype" }
output_arg { name: "output_handles" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListReserve :: forall v'1 v'2 shape_type . (OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] shape_type) =>
                     DataType -- ^ __element_dtype__
                     -> Tensor v'1 shape_type -- ^ __element_shape__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __num_elements__
                     -> Tensor Build Variant -- ^ __handle__
tensorListReserve :: DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
tensorListReserve = OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) shape_type.
OneOf '[Int32, Int64] shape_type =>
OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
tensorListReserve' OpParams
forall a. a -> a
id
tensorListReserve' :: forall v'1 v'2 shape_type . (OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] shape_type) =>
                      OpParams ->
                      DataType -- ^ __element_dtype__
                      -> Tensor v'1 shape_type -- ^ __element_shape__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __num_elements__
                      -> Tensor Build Variant -- ^ __handle__
tensorListReserve' :: OpParams
-> DataType
-> Tensor v'1 shape_type
-> Tensor v'2 Int32
-> Tensor Build Variant
tensorListReserve' op'options :: OpParams
op'options element_dtype :: DataType
element_dtype element_shape :: Tensor v'1 shape_type
element_shape
                   num_elements :: Tensor v'2 Int32
num_elements | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 shape_type
element_shape,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
num_elements]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListReserve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
element_dtype
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "element_shape" type_attr: "shape_type" }
input_arg { name: "num_elements" type: DT_INT32 }
output_arg { name: "handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListResize :: Tensor v'1 Variant -- ^ __input_handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                    -> Tensor Build Variant -- ^ __output_handle__
tensorListResize :: Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
tensorListResize = OpParams
-> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *).
OpParams
-> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
tensorListResize' OpParams
forall a. a -> a
id
tensorListResize' :: OpParams ->
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __size__
                     -> Tensor Build Variant -- ^ __output_handle__
tensorListResize' :: OpParams
-> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build Variant
tensorListResize' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle size :: Tensor v'2 Int32
size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
size]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListResize"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "size" type: DT_INT32 }
output_arg { name: "output_handle" type: DT_VARIANT }
-}
-- | 
tensorListScatter :: forall v'1 v'2 v'3 element_dtype
                     shape_type . (TensorType element_dtype,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] shape_type) =>
                     Tensor v'1 element_dtype -- ^ __tensor__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                     -> Tensor v'3 shape_type -- ^ __element_shape__
                     -> Tensor Build Variant -- ^ __output_handle__
tensorListScatter :: Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor Build Variant
tensorListScatter = OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype shape_type.
(TensorType element_dtype, OneOf '[Int32, Int64] shape_type) =>
OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor Build Variant
tensorListScatter' OpParams
forall a. a -> a
id
tensorListScatter' :: forall v'1 v'2 v'3 element_dtype
                      shape_type . (TensorType element_dtype,
                                    OneOf '[Data.Int.Int32,
                                            Data.Int.Int64] shape_type) =>
                      OpParams ->
                      Tensor v'1 element_dtype -- ^ __tensor__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                      -> Tensor v'3 shape_type -- ^ __element_shape__
                      -> Tensor Build Variant -- ^ __output_handle__
tensorListScatter' :: OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor Build Variant
tensorListScatter' op'options :: OpParams
op'options tensor :: Tensor v'1 element_dtype
tensor indices :: Tensor v'2 Int32
indices element_shape :: Tensor v'3 shape_type
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 element_dtype
tensor,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_type
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListScatter"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "element_dtype" }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "element_shape" type_attr: "shape_type" }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListScatterIntoExistingList :: forall v'1 v'2 v'3
                                     element_dtype . (TensorType element_dtype) =>
                                     Tensor v'1 Variant -- ^ __input_handle__
                                     -> Tensor v'2 element_dtype -- ^ __tensor__
                                     -> Tensor v'3 Data.Int.Int32 -- ^ __indices__
                                     -> Tensor Build Variant -- ^ __output_handle__
tensorListScatterIntoExistingList :: Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor v'3 Int32
-> Tensor Build Variant
tensorListScatterIntoExistingList = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor v'3 Int32
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor v'3 Int32
-> Tensor Build Variant
tensorListScatterIntoExistingList' OpParams
forall a. a -> a
id
tensorListScatterIntoExistingList' :: forall v'1 v'2 v'3
                                      element_dtype . (TensorType element_dtype) =>
                                      OpParams ->
                                      Tensor v'1 Variant -- ^ __input_handle__
                                      -> Tensor v'2 element_dtype -- ^ __tensor__
                                      -> Tensor v'3 Data.Int.Int32 -- ^ __indices__
                                      -> Tensor Build Variant -- ^ __output_handle__
tensorListScatterIntoExistingList' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 element_dtype
-> Tensor v'3 Int32
-> Tensor Build Variant
tensorListScatterIntoExistingList' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle tensor :: Tensor v'2 element_dtype
tensor
                                   indices :: Tensor v'3 Int32
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 element_dtype
tensor,
                                                             Tensor v'3 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int32
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListScatterIntoExistingList"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "tensor" type_attr: "element_dtype" }
input_arg { name: "indices" type: DT_INT32 }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListScatterV2 :: forall v'1 v'2 v'3 v'4 element_dtype
                       shape_type . (TensorType element_dtype,
                                     OneOf '[Data.Int.Int32,
                                             Data.Int.Int64] shape_type) =>
                       Tensor v'1 element_dtype -- ^ __tensor__
                       -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                       -> Tensor v'3 shape_type -- ^ __element_shape__
                       -> Tensor v'4 Data.Int.Int32 -- ^ __num_elements__
                       -> Tensor Build Variant -- ^ __output_handle__
tensorListScatterV2 :: Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor v'4 Int32
-> Tensor Build Variant
tensorListScatterV2 = OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor v'4 Int32
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) element_dtype shape_type.
(TensorType element_dtype, OneOf '[Int32, Int64] shape_type) =>
OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor v'4 Int32
-> Tensor Build Variant
tensorListScatterV2' OpParams
forall a. a -> a
id
tensorListScatterV2' :: forall v'1 v'2 v'3 v'4 element_dtype
                        shape_type . (TensorType element_dtype,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] shape_type) =>
                        OpParams ->
                        Tensor v'1 element_dtype -- ^ __tensor__
                        -> Tensor v'2 Data.Int.Int32 -- ^ __indices__
                        -> Tensor v'3 shape_type -- ^ __element_shape__
                        -> Tensor v'4 Data.Int.Int32 -- ^ __num_elements__
                        -> Tensor Build Variant -- ^ __output_handle__
tensorListScatterV2' :: OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 Int32
-> Tensor v'3 shape_type
-> Tensor v'4 Int32
-> Tensor Build Variant
tensorListScatterV2' op'options :: OpParams
op'options tensor :: Tensor v'1 element_dtype
tensor indices :: Tensor v'2 Int32
indices element_shape :: Tensor v'3 shape_type
element_shape
                     num_elements :: Tensor v'4 Int32
num_elements | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 element_dtype
tensor,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
indices,
                                                             Tensor v'3 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 shape_type
element_shape,
                                                             Tensor v'4 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int32
num_elements]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListScatterV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "element_dtype" }
input_arg { name: "indices" type: DT_INT32 }
input_arg { name: "element_shape" type_attr: "shape_type" }
input_arg { name: "num_elements" type: DT_INT32 }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListSetItem :: forall v'1 v'2 v'3
                     element_dtype . (TensorType element_dtype) =>
                     Tensor v'1 Variant -- ^ __input_handle__
                     -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                     -> Tensor v'3 element_dtype -- ^ __item__
                     -> Tensor Build Variant -- ^ __output_handle__
tensorListSetItem :: Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 element_dtype
-> Tensor Build Variant
tensorListSetItem = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 element_dtype
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 element_dtype
-> Tensor Build Variant
tensorListSetItem' OpParams
forall a. a -> a
id
tensorListSetItem' :: forall v'1 v'2 v'3
                      element_dtype . (TensorType element_dtype) => OpParams ->
                      Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor v'2 Data.Int.Int32 -- ^ __index__
                      -> Tensor v'3 element_dtype -- ^ __item__
                      -> Tensor Build Variant -- ^ __output_handle__
tensorListSetItem' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor v'3 element_dtype
-> Tensor Build Variant
tensorListSetItem' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle index :: Tensor v'2 Int32
index item :: Tensor v'3 element_dtype
item | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
index,
                                                             Tensor v'3 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 element_dtype
item]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListSetItem"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "index" type: DT_INT32 }
input_arg { name: "item" type_attr: "element_dtype" }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
-}
-- | 
tensorListSplit :: forall v'1 v'2 v'3 element_dtype
                   shape_type . (TensorType element_dtype,
                                 OneOf '[Data.Int.Int32,
                                         Data.Int.Int64] shape_type) =>
                   Tensor v'1 element_dtype -- ^ __tensor__
                   -> Tensor v'2 shape_type -- ^ __element_shape__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                   -> Tensor Build Variant -- ^ __output_handle__
tensorListSplit :: Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> Tensor Build Variant
tensorListSplit = OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       element_dtype shape_type.
(TensorType element_dtype, OneOf '[Int32, Int64] shape_type) =>
OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> Tensor Build Variant
tensorListSplit' OpParams
forall a. a -> a
id
tensorListSplit' :: forall v'1 v'2 v'3 element_dtype
                    shape_type . (TensorType element_dtype,
                                  OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] shape_type) =>
                    OpParams ->
                    Tensor v'1 element_dtype -- ^ __tensor__
                    -> Tensor v'2 shape_type -- ^ __element_shape__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __lengths__
                    -> Tensor Build Variant -- ^ __output_handle__
tensorListSplit' :: OpParams
-> Tensor v'1 element_dtype
-> Tensor v'2 shape_type
-> Tensor v'3 Int64
-> Tensor Build Variant
tensorListSplit' op'options :: OpParams
op'options tensor :: Tensor v'1 element_dtype
tensor element_shape :: Tensor v'2 shape_type
element_shape lengths :: Tensor v'3 Int64
lengths | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 element_dtype -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 element_dtype
tensor,
                                                             Tensor v'2 shape_type -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 shape_type
element_shape,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
lengths]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListSplit"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ shape_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (shape_type
forall a. HasCallStack => a
undefined :: shape_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "element_dtype" }
input_arg { name: "element_shape" type_attr: "shape_type" }
input_arg { name: "lengths" type: DT_INT64 }
output_arg { name: "output_handle" type: DT_VARIANT }
attr { name: "element_dtype" type: "type" }
attr {
  name: "shape_type"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorListStack :: forall v'1 v'2 element_dtype . (TensorType element_dtype) =>
                   Tensor v'1 Variant -- ^ __input_handle__
                   -> Tensor v'2 Data.Int.Int32 -- ^ __element_shape__
                   -> Tensor Build element_dtype -- ^ __tensor__
tensorListStack :: Tensor v'1 Variant
-> Tensor v'2 Int32 -> Tensor Build element_dtype
tensorListStack = OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build element_dtype
forall (v'1 :: * -> *) (v'2 :: * -> *) element_dtype.
TensorType element_dtype =>
OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build element_dtype
tensorListStack' OpParams
forall a. a -> a
id
tensorListStack' :: forall v'1 v'2 element_dtype . (TensorType element_dtype) =>
                    OpParams ->
                    Tensor v'1 Variant -- ^ __input_handle__
                    -> Tensor v'2 Data.Int.Int32 -- ^ __element_shape__
                    -> Tensor Build element_dtype -- ^ __tensor__
tensorListStack' :: OpParams
-> Tensor v'1 Variant
-> Tensor v'2 Int32
-> Tensor Build element_dtype
tensorListStack' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle element_shape :: Tensor v'2 Int32
element_shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build element_dtype
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build element_dtype)
-> Build OpDef -> Tensor Build element_dtype
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
element_shape]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorListStack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "element_dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ element_dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (element_dtype
forall a. HasCallStack => a
undefined :: element_dtype)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
input_arg { name: "element_shape" type: DT_INT32 }
output_arg { name: "tensor" type_attr: "element_dtype" }
attr { name: "element_dtype" type: "type" }
attr { name: "num_elements" type: "int" default_value { i: -1 } }
-}
-- | 
tensorScatterAdd :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> Tensor v'3 t -- ^ __updates__
                    -> Tensor Build t -- ^ __output__
tensorScatterAdd :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterAdd = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
tensorScatterAdd' OpParams
forall a. a -> a
id
tensorScatterAdd' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __tensor__
                     -> Tensor v'2 tindices -- ^ __indices__
                     -> Tensor v'3 t -- ^ __updates__
                     -> Tensor Build t -- ^ __output__
tensorScatterAdd' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
tensorScatterAdd' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorScatterAdd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorScatterMax :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> Tensor v'3 t -- ^ __updates__
                    -> Tensor Build t -- ^ __output__
tensorScatterMax :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMax = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
tensorScatterMax' OpParams
forall a. a -> a
id
tensorScatterMax' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __tensor__
                     -> Tensor v'2 tindices -- ^ __indices__
                     -> Tensor v'3 t -- ^ __updates__
                     -> Tensor Build t -- ^ __output__
tensorScatterMax' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
tensorScatterMax' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorScatterMax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorScatterMin :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> Tensor v'3 t -- ^ __updates__
                    -> Tensor Build t -- ^ __output__
tensorScatterMin :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterMin = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
tensorScatterMin' OpParams
forall a. a -> a
id
tensorScatterMin' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __tensor__
                     -> Tensor v'2 tindices -- ^ __indices__
                     -> Tensor v'3 t -- ^ __updates__
                     -> Tensor Build t -- ^ __output__
tensorScatterMin' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
tensorScatterMin' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorScatterMin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorScatterSub :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                    Tensor v'1 t -- ^ __tensor__
                    -> Tensor v'2 tindices -- ^ __indices__
                    -> Tensor v'3 t -- ^ __updates__
                    -> Tensor Build t -- ^ __output__
tensorScatterSub :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterSub = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
tensorScatterSub' OpParams
forall a. a -> a
id
tensorScatterSub' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                      OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tindices) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __tensor__
                     -> Tensor v'2 tindices -- ^ __indices__
                     -> Tensor v'3 t -- ^ __updates__
                     -> Tensor Build t -- ^ __output__
tensorScatterSub' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
tensorScatterSub' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorScatterSub"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorScatterUpdate :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] tindices) =>
                       Tensor v'1 t -- ^ __tensor__
                       -> Tensor v'2 tindices -- ^ __indices__
                       -> Tensor v'3 t -- ^ __updates__
                       -> Tensor Build t -- ^ __output__
tensorScatterUpdate :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
tensorScatterUpdate = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tpaddings.
(TensorType t, OneOf '[Int32, Int64] tpaddings) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tpaddings
-> Tensor v'3 t
-> Tensor Build t
tensorScatterUpdate' OpParams
forall a. a -> a
id
tensorScatterUpdate' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __tensor__
                        -> Tensor v'2 tindices -- ^ __indices__
                        -> Tensor v'3 t -- ^ __updates__
                        -> Tensor Build t -- ^ __output__
tensorScatterUpdate' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 t
-> Tensor Build t
tensorScatterUpdate' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor indices :: Tensor v'2 tindices
indices updates :: Tensor v'3 t
updates | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
indices,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
updates]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorScatterUpdate"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "indices" type_attr: "Tindices" }
input_arg { name: "updates" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tensorSliceDataset :: forall v'1 toutput_types m' . (MonadBuild m',
                                                     TensorTypes toutput_types) =>
                      TensorList (v'1) toutput_types -- ^ __components__
                      -> m' (Tensor Value Variant) -- ^ __handle__
tensorSliceDataset :: TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorSliceDataset = OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (toutput_types :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes toutput_types) =>
OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorSliceDataset' OpParams
forall a. a -> a
id
tensorSliceDataset' :: forall v'1 toutput_types m' . (MonadBuild m',
                                                      TensorTypes toutput_types) =>
                       OpParams ->
                       TensorList (v'1) toutput_types -- ^ __components__
                       -> m' (Tensor Value Variant) -- ^ __handle__
tensorSliceDataset' :: OpParams
-> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
tensorSliceDataset' op'options :: OpParams
op'options components :: TensorList v'1 toutput_types
components | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 toutput_types -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 toutput_types
components]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TensorSliceDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutput_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutput_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutput_types
forall k (t :: k). Proxy t
Proxy :: Proxy toutput_types)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "components" type_list_attr: "Toutput_types" }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "Toutput_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
tensorStridedSliceUpdate :: forall v'1 v'2 v'3 v'4 v'5 t index . (TensorType t,
                                                                  OneOf '[Data.Int.Int32,
                                                                          Data.Int.Int64] index) =>
                            Tensor v'1 t -- ^ __input__
                            -> Tensor v'2 index -- ^ __begin__
                            -> Tensor v'3 index -- ^ __end__
                            -> Tensor v'4 index -- ^ __strides__
                            -> Tensor v'5 t -- ^ __value__
                            -> Tensor Build t -- ^ __output__
tensorStridedSliceUpdate :: Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
tensorStridedSliceUpdate = OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t index.
(TensorType t, OneOf '[Int32, Int64] index) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
tensorStridedSliceUpdate' OpParams
forall a. a -> a
id
tensorStridedSliceUpdate' :: forall v'1 v'2 v'3 v'4 v'5 t index . (TensorType t,
                                                                   OneOf '[Data.Int.Int32,
                                                                           Data.Int.Int64] index) =>
                             OpParams ->
                             Tensor v'1 t -- ^ __input__
                             -> Tensor v'2 index -- ^ __begin__
                             -> Tensor v'3 index -- ^ __end__
                             -> Tensor v'4 index -- ^ __strides__
                             -> Tensor v'5 t -- ^ __value__
                             -> Tensor Build t -- ^ __output__
tensorStridedSliceUpdate' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 index
-> Tensor v'3 index
-> Tensor v'4 index
-> Tensor v'5 t
-> Tensor Build t
tensorStridedSliceUpdate' op'options :: OpParams
op'options input :: Tensor v'1 t
input begin :: Tensor v'2 index
begin end :: Tensor v'3 index
end strides :: Tensor v'4 index
strides
                          value :: Tensor v'5 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 index
begin,
                                                             Tensor v'3 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 index
end,
                                                             Tensor v'4 index -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 index
strides,
                                                             Tensor v'5 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorStridedSliceUpdate"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Index" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ index -> DataType
forall a. TensorType a => a -> DataType
tensorType (index
forall a. HasCallStack => a
undefined :: index)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "begin" type_attr: "Index" }
input_arg { name: "end" type_attr: "Index" }
input_arg { name: "strides" type_attr: "Index" }
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Index"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr { name: "begin_mask" type: "int" default_value { i: 0 } }
attr { name: "end_mask" type: "int" default_value { i: 0 } }
attr { name: "ellipsis_mask" type: "int" default_value { i: 0 } }
attr { name: "new_axis_mask" type: "int" default_value { i: 0 } }
attr {
  name: "shrink_axis_mask" type: "int" default_value { i: 0 }
}
-}
-- | 
tensorSummary :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __tensor__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
tensorSummary :: Tensor v'1 t -> Tensor Build ByteString
tensorSummary = OpParams -> Tensor v'1 t -> Tensor Build ByteString
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build ByteString
tensorSummary' OpParams
forall a. a -> a
id
tensorSummary' :: forall v'1 t . (TensorType t) => OpParams ->
                  Tensor v'1 t -- ^ __tensor__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
tensorSummary' :: OpParams -> Tensor v'1 t -> Tensor Build ByteString
tensorSummary' op'options :: OpParams
op'options tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorSummary"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tensor" type_attr: "T" }
output_arg { name: "summary" type: DT_STRING }
attr { name: "T" type: "type" }
attr { name: "description" type: "string" default_value { s: "" } }
attr {
  name: "labels" type: "list(string)" default_value { list { } }
}
attr {
  name: "display_name" type: "string" default_value { s: "" }
}
-}
-- | 
tensorSummaryV2 :: forall v'1 v'2 v'3 t . (TensorType t) =>
                   Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                   -> Tensor v'2 t -- ^ __tensor__
                   -> Tensor v'3 Data.ByteString.ByteString -- ^ __serialized_summary_metadata__
                   -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
tensorSummaryV2 :: Tensor v'1 ByteString
-> Tensor v'2 t -> Tensor v'3 ByteString -> Tensor Build ByteString
tensorSummaryV2 = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 ByteString
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 ByteString
-> Tensor Build ByteString
tensorSummaryV2' OpParams
forall a. a -> a
id
tensorSummaryV2' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __tag__
                    -> Tensor v'2 t -- ^ __tensor__
                    -> Tensor v'3 Data.ByteString.ByteString -- ^ __serialized_summary_metadata__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __summary__
tensorSummaryV2' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 t
-> Tensor v'3 ByteString
-> Tensor Build ByteString
tensorSummaryV2' op'options :: OpParams
op'options tag :: Tensor v'1 ByteString
tag tensor :: Tensor v'2 t
tensor
                 serialized_summary_metadata :: Tensor v'3 ByteString
serialized_summary_metadata | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
tag,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
tensor,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
serialized_summary_metadata]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TensorSummaryV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "serialized_summary_metadata" type: DT_STRING }
output_arg { name: "summary" type: DT_STRING }
attr { name: "T" type: "type" }
-}
-- | 
textLineDataset :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                   Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                   -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                   -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                   -> m' (Tensor Value Variant) -- ^ __handle__
textLineDataset :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
textLineDataset = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
textLineDataset' OpParams
forall a. a -> a
id
textLineDataset' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __filenames__
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __compression_type__
                    -> Tensor v'3 Data.Int.Int64 -- ^ __buffer_size__
                    -> m' (Tensor Value Variant) -- ^ __handle__
textLineDataset' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 Int64
-> m' (Tensor Value Variant)
textLineDataset' op'options :: OpParams
op'options filenames :: Tensor v'1 ByteString
filenames compression_type :: Tensor v'2 ByteString
compression_type
                 buffer_size :: Tensor v'3 Int64
buffer_size | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filenames,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
compression_type,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
buffer_size]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TextLineDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filenames" type: DT_STRING }
input_arg { name: "compression_type" type: DT_STRING }
input_arg { name: "buffer_size" type: DT_INT64 }
output_arg { name: "handle" type: DT_VARIANT }
-}
-- | 
textLineReader :: forall m' . (MonadBuild m') =>
                  m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
textLineReader :: m' (Tensor Ref ByteString)
textLineReader = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
textLineReader' OpParams
forall a. a -> a
id
textLineReader' :: forall m' . (MonadBuild m') => OpParams ->
                   m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
textLineReader' :: OpParams -> m' (Tensor Ref ByteString)
textLineReader' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TextLineReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr {
  name: "skip_header_lines" type: "int" default_value { i: 0 }
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
textLineReaderV2 :: forall m' . (MonadBuild m') =>
                    m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
textLineReaderV2 :: m' (Tensor Value ResourceHandle)
textLineReaderV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
textLineReaderV2' OpParams
forall a. a -> a
id
textLineReaderV2' :: forall m' . (MonadBuild m') => OpParams ->
                     m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
textLineReaderV2' :: OpParams -> m' (Tensor Value ResourceHandle)
textLineReaderV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TextLineReaderV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_RESOURCE }
attr {
  name: "skip_header_lines" type: "int" default_value { i: 0 }
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
threadPoolDataset :: forall v'1 v'2 m' . (MonadBuild m') =>
                     [DataType] -- ^ __output_types__
                     -> Tensor v'1 Variant -- ^ __input_dataset__
                     -> Tensor v'2 ResourceHandle -- ^ __thread_pool__
                     -> m' (Tensor Value Variant) -- ^ __handle__
threadPoolDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
threadPoolDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
threadPoolDataset' OpParams
forall a. a -> a
id
threadPoolDataset' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
                      [DataType] -- ^ __output_types__
                      -> Tensor v'1 Variant -- ^ __input_dataset__
                      -> Tensor v'2 ResourceHandle -- ^ __thread_pool__
                      -> m' (Tensor Value Variant) -- ^ __handle__
threadPoolDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 ResourceHandle
-> m' (Tensor Value Variant)
threadPoolDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset
                   thread_pool :: Tensor v'2 ResourceHandle
thread_pool | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Variant) -> m' (Tensor Value Variant))
-> Build (Tensor Value Variant) -> m' (Tensor Value Variant)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ResourceHandle
thread_pool]
        [Int64] -> OpDef -> Build (Tensor Value Variant)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ThreadPoolDataset"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "thread_pool" type: DT_RESOURCE }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
threadPoolHandle :: forall m' . (MonadBuild m') =>
                    ByteString -- ^ __display_name__
                    -> Data.Int.Int64 -- ^ __num_threads__
                    -> m' (Tensor Value ResourceHandle) -- ^ __handle__
threadPoolHandle :: ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
threadPoolHandle = OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
threadPoolHandle' OpParams
forall a. a -> a
id
threadPoolHandle' :: forall m' . (MonadBuild m') => OpParams ->
                     ByteString -- ^ __display_name__
                     -> Data.Int.Int64 -- ^ __num_threads__
                     -> m' (Tensor Value ResourceHandle) -- ^ __handle__
threadPoolHandle' :: OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
threadPoolHandle' op'options :: OpParams
op'options display_name :: ByteString
display_name num_threads :: Int64
num_threads | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ThreadPoolHandle"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "display_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
display_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_threads" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_threads
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "handle" type: DT_RESOURCE }
attr { name: "num_threads" type: "int" }
attr {
  name: "max_intra_op_parallelism" type: "int" default_value { i: 1 }
}
attr { name: "display_name" type: "string" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
threadUnsafeUnigramCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                                       Data.Int.Int64 -- ^ __num_sampled__
                                       -> Data.Int.Int64 -- ^ __num_true__
                                       -> Data.Int.Int64 -- ^ __range_max__
                                       -> Bool -- ^ __unique__
                                       -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                       -> m' ((Tensor Value Data.Int.Int64,
                                               Tensor Value Float,
                                               Tensor Value Float))
                                       -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                       --
                                       -- * __sampled_candidates__
                                       --
                                       -- * __true_expected_count__
                                       --
                                       -- * __sampled_expected_count__
threadUnsafeUnigramCandidateSampler :: Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
threadUnsafeUnigramCandidateSampler = OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
threadUnsafeUnigramCandidateSampler' OpParams
forall a. a -> a
id
threadUnsafeUnigramCandidateSampler' :: forall v'1 m' . (MonadBuild m') =>
                                        OpParams ->
                                        Data.Int.Int64 -- ^ __num_sampled__
                                        -> Data.Int.Int64 -- ^ __num_true__
                                        -> Data.Int.Int64 -- ^ __range_max__
                                        -> Bool -- ^ __unique__
                                        -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                                        -> m' ((Tensor Value Data.Int.Int64,
                                                Tensor Value Float,
                                                Tensor Value Float))
                                        -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                                        --
                                        -- * __sampled_candidates__
                                        --
                                        -- * __true_expected_count__
                                        --
                                        -- * __sampled_expected_count__
threadUnsafeUnigramCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
threadUnsafeUnigramCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true range_max :: Int64
range_max
                                     unique :: Bool
unique true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "ThreadUnsafeUnigramCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "range_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
range_max
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "range_max" type: "int" has_minimum: true minimum: 1 }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
tile :: forall v'1 v'2 t tmultiples . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tmultiples) =>
        Tensor v'1 t -- ^ __input__
        -> Tensor v'2 tmultiples -- ^ __multiples__
        -> Tensor Build t -- ^ __output__
tile :: Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
tile = OpParams -> Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
tile' OpParams
forall a. a -> a
id
tile' :: forall v'1 v'2 t tmultiples . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tmultiples) =>
         OpParams ->
         Tensor v'1 t -- ^ __input__
         -> Tensor v'2 tmultiples -- ^ __multiples__
         -> Tensor Build t -- ^ __output__
tile' :: OpParams -> Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
tile' op'options :: OpParams
op'options input :: Tensor v'1 t
input multiples :: Tensor v'2 tmultiples
multiples | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tmultiples -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tmultiples
multiples]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Tile"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tmultiples" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tmultiples -> DataType
forall a. TensorType a => a -> DataType
tensorType (tmultiples
forall a. HasCallStack => a
undefined :: tmultiples)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "multiples" type_attr: "Tmultiples" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tmultiples"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tileGrad :: forall v'1 v'2 t . (TensorType t) => Tensor v'1 t -- ^ __input__
            -> Tensor v'2 Data.Int.Int32 -- ^ __multiples__
            -> Tensor Build t -- ^ __output__
tileGrad :: Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
tileGrad = OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
tileGrad' OpParams
forall a. a -> a
id
tileGrad' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
             Tensor v'1 t -- ^ __input__
             -> Tensor v'2 Data.Int.Int32 -- ^ __multiples__
             -> Tensor Build t -- ^ __output__
tileGrad' :: OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
tileGrad' op'options :: OpParams
op'options input :: Tensor v'1 t
input multiples :: Tensor v'2 Int32
multiples | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
multiples]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TileGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "multiples" type: DT_INT32 }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
timestamp :: forall m' . (MonadBuild m') => m' (Tensor Value Double) -- ^ __ts__
timestamp :: m' (Tensor Value Double)
timestamp = OpParams -> m' (Tensor Value Double)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value Double)
timestamp' OpParams
forall a. a -> a
id
timestamp' :: forall m' . (MonadBuild m') => OpParams ->
              m' (Tensor Value Double) -- ^ __ts__
timestamp' :: OpParams -> m' (Tensor Value Double)
timestamp' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Double) -> m' (Tensor Value Double)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Double) -> m' (Tensor Value Double))
-> Build (Tensor Value Double) -> m' (Tensor Value Double)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Double)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Timestamp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "ts" type: DT_DOUBLE }
-}
-- | 
toBool :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
          -> Tensor Build Bool -- ^ __output__
toBool :: Tensor v'1 t -> Tensor Build Bool
toBool = OpParams -> Tensor v'1 t -> Tensor Build Bool
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build Bool
toBool' OpParams
forall a. a -> a
id
toBool' :: forall v'1 t . (TensorType t) => OpParams ->
           Tensor v'1 t -- ^ __input__
           -> Tensor Build Bool -- ^ __output__
toBool' :: OpParams -> Tensor v'1 t -> Tensor Build Bool
toBool' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Bool
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Bool)
-> Build OpDef -> Tensor Build Bool
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ToBool"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type: DT_BOOL }
attr { name: "T" type: "type" }
-}
-- | 
topK :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                Data.Int.Int8, Data.Word.Word16,
                                Data.Word.Word32, Data.Word.Word64,
                                Data.Word.Word8, Double, Float] t) =>
        Data.Int.Int64 -- ^ __k__
        -> Tensor v'1 t -- ^ __input__
        -> (Tensor Build t, Tensor Build Data.Int.Int32)
        -- ^ (__values__, __indices__)
        --
        -- * __values__
        --
        -- * __indices__
topK :: Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
topK = OpParams
-> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
forall (v'1 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
topK' OpParams
forall a. a -> a
id
topK' :: forall v'1 t . (OneOf '[Data.Int.Int16, Data.Int.Int32, Data.Int.Int64,
                                 Data.Int.Int8, Data.Word.Word16,
                                 Data.Word.Word32, Data.Word.Word64,
                                 Data.Word.Word8, Double, Float] t) =>
         OpParams ->
         Data.Int.Int64 -- ^ __k__
         -> Tensor v'1 t -- ^ __input__
         -> (Tensor Build t, Tensor Build Data.Int.Int32)
         -- ^ (__values__, __indices__)
         --
         -- * __values__
         --
         -- * __indices__
topK' :: OpParams
-> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
topK' op'options :: OpParams
op'options k :: Int64
k input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Int32))
-> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TopK"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "k" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
k
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "values" type_attr: "T" }
output_arg { name: "indices" type: DT_INT32 }
attr { name: "k" type: "int" has_minimum: true }
attr { name: "sorted" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
topKV2 :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) => Tensor v'1 t -- ^ __input__
          -> Tensor v'2 Data.Int.Int32 -- ^ __k__
          -> (Tensor Build t, Tensor Build Data.Int.Int32)
          -- ^ (__values__, __indices__)
          --
          -- * __values__
          --
          -- * __indices__
topKV2 :: Tensor v'1 t
-> Tensor v'2 Int32 -> (Tensor Build t, Tensor Build Int32)
topKV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build t, Tensor Build Int32)
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build t, Tensor Build Int32)
topKV2' OpParams
forall a. a -> a
id
topKV2' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
           Tensor v'1 t -- ^ __input__
           -> Tensor v'2 Data.Int.Int32 -- ^ __k__
           -> (Tensor Build t, Tensor Build Data.Int.Int32)
           -- ^ (__values__, __indices__)
           --
           -- * __values__
           --
           -- * __indices__
topKV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int32
-> (Tensor Build t, Tensor Build Int32)
topKV2' op'options :: OpParams
op'options input :: Tensor v'1 t
input k :: Tensor v'2 Int32
k | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Int32))
-> Build OpDef -> (Tensor Build t, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
k]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TopKV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "k" type: DT_INT32 }
output_arg { name: "values" type_attr: "T" }
output_arg { name: "indices" type: DT_INT32 }
attr { name: "sorted" type: "bool" default_value { b: true } }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
transpose :: forall v'1 v'2 t tperm . (TensorType t, OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tperm) =>
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 tperm -- ^ __perm__
             -> Tensor Build t -- ^ __y__
transpose :: Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
transpose = OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t tidx.
(TensorType t, OneOf '[Int32, Int64] tidx) =>
OpParams -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
transpose' OpParams
forall a. a -> a
id
transpose' :: forall v'1 v'2 t tperm . (TensorType t, OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64] tperm) =>
              OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor v'2 tperm -- ^ __perm__
              -> Tensor Build t -- ^ __y__
transpose' :: OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
transpose' op'options :: OpParams
op'options x :: Tensor v'1 t
x perm :: Tensor v'2 tperm
perm | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 tperm -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tperm
perm]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Transpose"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tperm" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tperm -> DataType
forall a. TensorType a => a -> DataType
tensorType (tperm
forall a. HasCallStack => a
undefined :: tperm)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "perm" type_attr: "Tperm" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tperm"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tridiagonalMatMul :: forall v'1 v'2 v'3 v'4
                     t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Double,
                                  Float] t) => Tensor v'1 t -- ^ __superdiag__
                     -> Tensor v'2 t -- ^ __maindiag__
                     -> Tensor v'3 t -- ^ __subdiag__
                     -> Tensor v'4 t -- ^ __rhs__
                     -> Tensor Build t -- ^ __output__
tridiagonalMatMul :: Tensor v'1 t
-> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
tridiagonalMatMul = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
tridiagonalMatMul' OpParams
forall a. a -> a
id
tridiagonalMatMul' :: forall v'1 v'2 v'3 v'4
                      t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Double,
                                   Float] t) => OpParams ->
                      Tensor v'1 t -- ^ __superdiag__
                      -> Tensor v'2 t -- ^ __maindiag__
                      -> Tensor v'3 t -- ^ __subdiag__
                      -> Tensor v'4 t -- ^ __rhs__
                      -> Tensor Build t -- ^ __output__
tridiagonalMatMul' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 t
-> Tensor v'4 t
-> Tensor Build t
tridiagonalMatMul' op'options :: OpParams
op'options superdiag :: Tensor v'1 t
superdiag maindiag :: Tensor v'2 t
maindiag subdiag :: Tensor v'3 t
subdiag
                   rhs :: Tensor v'4 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
superdiag,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
maindiag,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
subdiag,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TridiagonalMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "superdiag" type_attr: "T" }
input_arg { name: "maindiag" type_attr: "T" }
input_arg { name: "subdiag" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
tridiagonalSolve :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Double, Float] t) =>
                    Tensor v'1 t -- ^ __diagonals__
                    -> Tensor v'2 t -- ^ __rhs__
                    -> Tensor Build t -- ^ __output__
tridiagonalSolve :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tridiagonalSolve = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tridiagonalSolve' OpParams
forall a. a -> a
id
tridiagonalSolve' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                                 (Data.Complex.Complex Float),
                                                 Double, Float] t) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __diagonals__
                     -> Tensor v'2 t -- ^ __rhs__
                     -> Tensor Build t -- ^ __output__
tridiagonalSolve' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
tridiagonalSolve' op'options :: OpParams
op'options diagonals :: Tensor v'1 t
diagonals rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
diagonals,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TridiagonalSolve"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "diagonals" type_attr: "T" }
input_arg { name: "rhs" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "partial_pivoting" type: "bool" default_value { b: true }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_DOUBLE
      type: DT_FLOAT
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
truncateDiv :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word8,
                                           Double, Float] t) =>
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __z__
truncateDiv :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateDiv = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateDiv' OpParams
forall a. a -> a
id
truncateDiv' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                            (Data.Complex.Complex Float),
                                            Data.Int.Int16, Data.Int.Int32,
                                            Data.Int.Int64, Data.Int.Int8,
                                            Data.Word.Word16, Data.Word.Word8,
                                            Double, Float] t) => OpParams ->
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __y__
                -> Tensor Build t -- ^ __z__
truncateDiv' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateDiv' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TruncateDiv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
truncateMod :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                           Data.Word.Word16, Double,
                                           Float] t) => Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor Build t -- ^ __z__
truncateMod :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateMod = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateMod' OpParams
forall a. a -> a
id
truncateMod' :: forall v'1 v'2 t . (OneOf '[Data.Int.Int32, Data.Int.Int64,
                                            Data.Word.Word16, Double,
                                            Float] t) => OpParams ->
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __y__
                -> Tensor Build t -- ^ __z__
truncateMod' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
truncateMod' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "TruncateMod"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_INT32
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
    }
  }
}
-}
-- | 
truncatedNormal :: forall v'1 dtype t m' . (MonadBuild m',
                                            OneOf '[Data.Word.Word16, Double,
                                                    Float] dtype,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] t) =>
                   Tensor v'1 t -- ^ __shape__
                   -> m' (Tensor Value dtype) -- ^ __output__
truncatedNormal :: Tensor v'1 t -> m' (Tensor Value dtype)
truncatedNormal = OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
forall (v'1 :: * -> *) dtype t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Double, Float] dtype,
 OneOf '[Int32, Int64] t) =>
OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
truncatedNormal' OpParams
forall a. a -> a
id
truncatedNormal' :: forall v'1 dtype t m' . (MonadBuild m',
                                             OneOf '[Data.Word.Word16, Double,
                                                     Float] dtype,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] t) =>
                    OpParams ->
                    Tensor v'1 t -- ^ __shape__
                    -> m' (Tensor Value dtype) -- ^ __output__
truncatedNormal' :: OpParams -> Tensor v'1 t -> m' (Tensor Value dtype)
truncatedNormal' op'options :: OpParams
op'options shape :: Tensor v'1 t
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
shape]
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TruncatedNormal"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "shape" type_attr: "T" }
output_arg { name: "output" type_attr: "dtype" }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
attr {
  name: "dtype"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
tryRpc :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
          Tensor v'1 Data.ByteString.ByteString -- ^ __address__
          -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
          -> Tensor v'3 Data.ByteString.ByteString -- ^ __request__
          -> m' ((Tensor Value Data.ByteString.ByteString,
                  Tensor Value Data.Int.Int32,
                  Tensor Value Data.ByteString.ByteString))
          -- ^ (__response__, __status_code__, __status_message__)
          --
          -- * __response__
          --
          -- * __status_code__
          --
          -- * __status_message__
tryRpc :: Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
tryRpc = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
tryRpc' OpParams
forall a. a -> a
id
tryRpc' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
           Tensor v'1 Data.ByteString.ByteString -- ^ __address__
           -> Tensor v'2 Data.ByteString.ByteString -- ^ __method__
           -> Tensor v'3 Data.ByteString.ByteString -- ^ __request__
           -> m' ((Tensor Value Data.ByteString.ByteString,
                   Tensor Value Data.Int.Int32,
                   Tensor Value Data.ByteString.ByteString))
           -- ^ (__response__, __status_code__, __status_message__)
           --
           -- * __response__
           --
           -- * __status_code__
           --
           -- * __status_message__
tryRpc' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 ByteString
-> Tensor v'3 ByteString
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
tryRpc' op'options :: OpParams
op'options address :: Tensor v'1 ByteString
address method :: Tensor v'2 ByteString
method request :: Tensor v'3 ByteString
request | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build
  (Tensor Value ByteString, Tensor Value Int32,
   Tensor Value ByteString)
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build
   (Tensor Value ByteString, Tensor Value Int32,
    Tensor Value ByteString)
 -> m'
      (Tensor Value ByteString, Tensor Value Int32,
       Tensor Value ByteString))
-> Build
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
-> m'
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
address,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
method,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
request]
        [Int64]
-> OpDef
-> Build
     (Tensor Value ByteString, Tensor Value Int32,
      Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "TryRpc"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "address" type: DT_STRING }
input_arg { name: "method" type: DT_STRING }
input_arg { name: "request" type: DT_STRING }
output_arg { name: "response" type: DT_STRING }
output_arg { name: "status_code" type: DT_INT32 }
output_arg { name: "status_message" type: DT_STRING }
attr { name: "protocol" type: "string" default_value { s: "" } }
attr { name: "fail_fast" type: "bool" default_value { b: true } }
attr { name: "timeout_in_ms" type: "int" default_value { i: 0 } }
-}
-- | 
unbatch :: forall v'1 v'2 v'3 t . (TensorType t) =>
           Data.Int.Int64 -- ^ __timeout_micros__
           -> Tensor v'1 t -- ^ __batched_tensor__
           -> Tensor v'2 Data.Int.Int64 -- ^ __batch_index__
           -> Tensor v'3 Data.Int.Int64 -- ^ __id__
           -> Tensor Build t -- ^ __unbatched_tensor__
unbatch :: Int64
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
unbatch = OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
TensorType t =>
OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
unbatch' OpParams
forall a. a -> a
id
unbatch' :: forall v'1 v'2 v'3 t . (TensorType t) => OpParams ->
            Data.Int.Int64 -- ^ __timeout_micros__
            -> Tensor v'1 t -- ^ __batched_tensor__
            -> Tensor v'2 Data.Int.Int64 -- ^ __batch_index__
            -> Tensor v'3 Data.Int.Int64 -- ^ __id__
            -> Tensor Build t -- ^ __unbatched_tensor__
unbatch' :: OpParams
-> Int64
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor Build t
unbatch' op'options :: OpParams
op'options timeout_micros :: Int64
timeout_micros batched_tensor :: Tensor v'1 t
batched_tensor batch_index :: Tensor v'2 Int64
batch_index
         id :: Tensor v'3 Int64
id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
batched_tensor,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_index,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
id]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Unbatch"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "timeout_micros" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
timeout_micros
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "batched_tensor" type_attr: "T" }
input_arg { name: "batch_index" type: DT_INT64 }
input_arg { name: "id" type: DT_INT64 }
output_arg { name: "unbatched_tensor" type_attr: "T" }
attr { name: "timeout_micros" type: "int" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr { name: "T" type: "type" }
-}
-- | 
unbatchDataset :: [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor Build Variant -- ^ __handle__
unbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
unbatchDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
unbatchDataset' OpParams
forall a. a -> a
id
unbatchDataset' :: OpParams ->
                   [DataType] -- ^ __output_types__
                   -> Tensor v'1 Variant -- ^ __input_dataset__
                   -> Tensor Build Variant -- ^ __handle__
unbatchDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
unbatchDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnbatchDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
unbatchGrad :: forall v'1 v'2 v'3 v'4 t . (TensorType t) =>
               Tensor v'1 t -- ^ __original_input__
               -> Tensor v'2 Data.Int.Int64 -- ^ __batch_index__
               -> Tensor v'3 t -- ^ __grad__
               -> Tensor v'4 Data.Int.Int64 -- ^ __id__
               -> Tensor Build t -- ^ __batched_grad__
unbatchGrad :: Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor Build t
unbatchGrad = OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
TensorType t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor Build t
unbatchGrad' OpParams
forall a. a -> a
id
unbatchGrad' :: forall v'1 v'2 v'3 v'4 t . (TensorType t) => OpParams ->
                Tensor v'1 t -- ^ __original_input__
                -> Tensor v'2 Data.Int.Int64 -- ^ __batch_index__
                -> Tensor v'3 t -- ^ __grad__
                -> Tensor v'4 Data.Int.Int64 -- ^ __id__
                -> Tensor Build t -- ^ __batched_grad__
unbatchGrad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 Int64
-> Tensor Build t
unbatchGrad' op'options :: OpParams
op'options original_input :: Tensor v'1 t
original_input batch_index :: Tensor v'2 Int64
batch_index grad :: Tensor v'3 t
grad id :: Tensor v'4 Int64
id | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
original_input,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
batch_index,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
grad,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
id]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnbatchGrad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "original_input" type_attr: "T" }
input_arg { name: "batch_index" type: DT_INT64 }
input_arg { name: "grad" type_attr: "T" }
input_arg { name: "id" type: DT_INT64 }
output_arg { name: "batched_grad" type_attr: "T" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr { name: "T" type: "type" }
-}
-- | 
uncompressElement :: forall v'1 output_types . (TensorTypes output_types) =>
                     Tensor v'1 Variant -- ^ __compressed__
                     -> TensorList (Build) output_types -- ^ __components__
uncompressElement :: Tensor v'1 Variant -> TensorList Build output_types
uncompressElement = OpParams -> Tensor v'1 Variant -> TensorList Build output_types
forall (v'1 :: * -> *) (output_types :: [*]).
TensorTypes output_types =>
OpParams -> Tensor v'1 Variant -> TensorList Build output_types
uncompressElement' OpParams
forall a. a -> a
id
uncompressElement' :: forall v'1 output_types . (TensorTypes output_types) =>
                      OpParams ->
                      Tensor v'1 Variant -- ^ __compressed__
                      -> TensorList (Build) output_types -- ^ __components__
uncompressElement' :: OpParams -> Tensor v'1 Variant -> TensorList Build output_types
uncompressElement' op'options :: OpParams
op'options compressed :: Tensor v'1 Variant
compressed | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> TensorList Build output_types
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build output_types)
-> Build OpDef -> TensorList Build output_types
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
compressed]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UncompressElement"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy output_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy output_types
forall k (t :: k). Proxy t
Proxy :: Proxy output_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "compressed" type: DT_VARIANT }
output_arg { name: "components" type_list_attr: "output_types" }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
unicodeDecode :: forall v'1 tsplits . (OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tsplits) =>
                 ByteString -- ^ __input_encoding__
                 -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                 -> (Tensor Build tsplits, Tensor Build Data.Int.Int32)
                 -- ^ (__row_splits__, __char_values__)
                 --
                 -- * __row_splits__
                 --
                 -- * __char_values__
unicodeDecode :: ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32)
unicodeDecode = OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32)
forall (v'1 :: * -> *) tsplits.
OneOf '[Int32, Int64] tsplits =>
OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32)
unicodeDecode' OpParams
forall a. a -> a
id
unicodeDecode' :: forall v'1 tsplits . (OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tsplits) =>
                  OpParams ->
                  ByteString -- ^ __input_encoding__
                  -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                  -> (Tensor Build tsplits, Tensor Build Data.Int.Int32)
                  -- ^ (__row_splits__, __char_values__)
                  --
                  -- * __row_splits__
                  --
                  -- * __char_values__
unicodeDecode' :: OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32)
unicodeDecode' op'options :: OpParams
op'options input_encoding :: ByteString
input_encoding input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build tsplits, Tensor Build Int32)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build tsplits, Tensor Build Int32))
-> Build OpDef -> (Tensor Build tsplits, Tensor Build Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnicodeDecode"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_encoding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
input_encoding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "row_splits" type_attr: "Tsplits" }
output_arg { name: "char_values" type: DT_INT32 }
attr { name: "input_encoding" type: "string" }
attr {
  name: "errors"
  type: "string"
  default_value { s: "replace" }
  allowed_values { list { s: "strict" s: "replace" s: "ignore" } }
}
attr {
  name: "replacement_char" type: "int" default_value { i: 65533 }
}
attr {
  name: "replace_control_characters"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unicodeDecodeWithOffsets :: forall v'1 tsplits . (OneOf '[Data.Int.Int32,
                                                          Data.Int.Int64] tsplits) =>
                            ByteString -- ^ __input_encoding__
                            -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                            -> (Tensor Build tsplits,
                                Tensor Build Data.Int.Int32,
                                Tensor Build Data.Int.Int64)
                            -- ^ (__row_splits__, __char_values__, __char_to_byte_starts__)
                            --
                            -- * __row_splits__
                            --
                            -- * __char_values__
                            --
                            -- * __char_to_byte_starts__
unicodeDecodeWithOffsets :: ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
unicodeDecodeWithOffsets = OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
forall (v'1 :: * -> *) tsplits.
OneOf '[Int32, Int64] tsplits =>
OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
unicodeDecodeWithOffsets' OpParams
forall a. a -> a
id
unicodeDecodeWithOffsets' :: forall v'1 tsplits . (OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tsplits) =>
                             OpParams ->
                             ByteString -- ^ __input_encoding__
                             -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                             -> (Tensor Build tsplits,
                                 Tensor Build Data.Int.Int32,
                                 Tensor Build Data.Int.Int64)
                             -- ^ (__row_splits__, __char_values__, __char_to_byte_starts__)
                             --
                             -- * __row_splits__
                             --
                             -- * __char_values__
                             --
                             -- * __char_to_byte_starts__
unicodeDecodeWithOffsets' :: OpParams
-> ByteString
-> Tensor v'1 ByteString
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
unicodeDecodeWithOffsets' op'options :: OpParams
op'options input_encoding :: ByteString
input_encoding input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64))
-> Build OpDef
-> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnicodeDecodeWithOffsets"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_encoding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
input_encoding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "row_splits" type_attr: "Tsplits" }
output_arg { name: "char_values" type: DT_INT32 }
output_arg { name: "char_to_byte_starts" type: DT_INT64 }
attr { name: "input_encoding" type: "string" }
attr {
  name: "errors"
  type: "string"
  default_value { s: "replace" }
  allowed_values { list { s: "strict" s: "replace" s: "ignore" } }
}
attr {
  name: "replacement_char" type: "int" default_value { i: 65533 }
}
attr {
  name: "replace_control_characters"
  type: "bool"
  default_value { b: false }
}
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unicodeEncode :: forall v'1 v'2 tsplits . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tsplits) =>
                 ByteString -- ^ __output_encoding__
                 -> Tensor v'1 Data.Int.Int32 -- ^ __input_values__
                 -> Tensor v'2 tsplits -- ^ __input_splits__
                 -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unicodeEncode :: ByteString
-> Tensor v'1 Int32
-> Tensor v'2 tsplits
-> Tensor Build ByteString
unicodeEncode = OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 tsplits
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) tsplits.
OneOf '[Int32, Int64] tsplits =>
OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 tsplits
-> Tensor Build ByteString
unicodeEncode' OpParams
forall a. a -> a
id
unicodeEncode' :: forall v'1 v'2 tsplits . (OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tsplits) =>
                  OpParams ->
                  ByteString -- ^ __output_encoding__
                  -> Tensor v'1 Data.Int.Int32 -- ^ __input_values__
                  -> Tensor v'2 tsplits -- ^ __input_splits__
                  -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unicodeEncode' :: OpParams
-> ByteString
-> Tensor v'1 Int32
-> Tensor v'2 tsplits
-> Tensor Build ByteString
unicodeEncode' op'options :: OpParams
op'options output_encoding :: ByteString
output_encoding input_values :: Tensor v'1 Int32
input_values
               input_splits :: Tensor v'2 tsplits
input_splits | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
input_values,
                                                             Tensor v'2 tsplits -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tsplits
input_splits]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnicodeEncode"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tsplits" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tsplits -> DataType
forall a. TensorType a => a -> DataType
tensorType (tsplits
forall a. HasCallStack => a
undefined :: tsplits)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_encoding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
output_encoding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_values" type: DT_INT32 }
input_arg { name: "input_splits" type_attr: "Tsplits" }
output_arg { name: "output" type: DT_STRING }
attr {
  name: "errors"
  type: "string"
  default_value { s: "replace" }
  allowed_values { list { s: "ignore" s: "replace" s: "strict" } }
}
attr {
  name: "output_encoding"
  type: "string"
  allowed_values {
    list { s: "UTF-8" s: "UTF-16-BE" s: "UTF-32-BE" }
  }
}
attr {
  name: "replacement_char" type: "int" default_value { i: 65533 }
}
attr {
  name: "Tsplits"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unicodeScript :: Tensor v'1 Data.Int.Int32 -- ^ __input__
                 -> Tensor Build Data.Int.Int32 -- ^ __output__
unicodeScript :: Tensor v'1 Int32 -> Tensor Build Int32
unicodeScript = OpParams -> Tensor v'1 Int32 -> Tensor Build Int32
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Int32 -> Tensor Build Int32
unicodeScript' OpParams
forall a. a -> a
id
unicodeScript' :: OpParams ->
                  Tensor v'1 Data.Int.Int32 -- ^ __input__
                  -> Tensor Build Data.Int.Int32 -- ^ __output__
unicodeScript' :: OpParams -> Tensor v'1 Int32 -> Tensor Build Int32
unicodeScript' op'options :: OpParams
op'options input :: Tensor v'1 Int32
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnicodeScript"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_INT32 }
output_arg { name: "output" type: DT_INT32 }
-}
-- | 
unicodeTranscode :: ByteString -- ^ __input_encoding__
                    -> ByteString -- ^ __output_encoding__
                    -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                    -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unicodeTranscode :: ByteString
-> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
unicodeTranscode = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
forall (v'1 :: * -> *).
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
unicodeTranscode' OpParams
forall a. a -> a
id
unicodeTranscode' :: OpParams ->
                     ByteString -- ^ __input_encoding__
                     -> ByteString -- ^ __output_encoding__
                     -> Tensor v'1 Data.ByteString.ByteString -- ^ __input__
                     -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unicodeTranscode' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 ByteString
-> Tensor Build ByteString
unicodeTranscode' op'options :: OpParams
op'options input_encoding :: ByteString
input_encoding output_encoding :: ByteString
output_encoding
                  input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnicodeTranscode"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "input_encoding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
input_encoding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_encoding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
output_encoding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_STRING }
output_arg { name: "output" type: DT_STRING }
attr { name: "input_encoding" type: "string" }
attr {
  name: "output_encoding"
  type: "string"
  allowed_values {
    list { s: "UTF-8" s: "UTF-16-BE" s: "UTF-32-BE" }
  }
}
attr {
  name: "errors"
  type: "string"
  default_value { s: "replace" }
  allowed_values { list { s: "strict" s: "replace" s: "ignore" } }
}
attr {
  name: "replacement_char" type: "int" default_value { i: 65533 }
}
attr {
  name: "replace_control_characters"
  type: "bool"
  default_value { b: false }
}
-}
-- | 
uniformCandidateSampler :: forall v'1 m' . (MonadBuild m') =>
                           Data.Int.Int64 -- ^ __num_sampled__
                           -> Data.Int.Int64 -- ^ __num_true__
                           -> Data.Int.Int64 -- ^ __range_max__
                           -> Bool -- ^ __unique__
                           -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                           -> m' ((Tensor Value Data.Int.Int64,
                                   Tensor Value Float, Tensor Value Float))
                           -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                           --
                           -- * __sampled_candidates__
                           --
                           -- * __true_expected_count__
                           --
                           -- * __sampled_expected_count__
uniformCandidateSampler :: Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
uniformCandidateSampler = OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
uniformCandidateSampler' OpParams
forall a. a -> a
id
uniformCandidateSampler' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                            Data.Int.Int64 -- ^ __num_sampled__
                            -> Data.Int.Int64 -- ^ __num_true__
                            -> Data.Int.Int64 -- ^ __range_max__
                            -> Bool -- ^ __unique__
                            -> Tensor v'1 Data.Int.Int64 -- ^ __true_classes__
                            -> m' ((Tensor Value Data.Int.Int64,
                                    Tensor Value Float, Tensor Value Float))
                            -- ^ (__sampled_candidates__, __true_expected_count__, __sampled_expected_count__)
                            --
                            -- * __sampled_candidates__
                            --
                            -- * __true_expected_count__
                            --
                            -- * __sampled_expected_count__
uniformCandidateSampler' :: OpParams
-> Int64
-> Int64
-> Int64
-> Bool
-> Tensor v'1 Int64
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
uniformCandidateSampler' op'options :: OpParams
op'options num_sampled :: Int64
num_sampled num_true :: Int64
num_true range_max :: Int64
range_max unique :: Bool
unique
                         true_classes :: Tensor v'1 Int64
true_classes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float))
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
-> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int64
true_classes]
        [Int64]
-> OpDef
-> Build
     (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "UniformCandidateSampler"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_sampled" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_sampled
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_true" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_true
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "range_max" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
range_max
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "unique" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
unique
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "true_classes" type: DT_INT64 }
output_arg { name: "sampled_candidates" type: DT_INT64 }
output_arg { name: "true_expected_count" type: DT_FLOAT }
output_arg { name: "sampled_expected_count" type: DT_FLOAT }
attr { name: "num_true" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "num_sampled" type: "int" has_minimum: true minimum: 1
}
attr { name: "unique" type: "bool" }
attr { name: "range_max" type: "int" has_minimum: true minimum: 1 }
attr { name: "seed" type: "int" default_value { i: 0 } }
attr { name: "seed2" type: "int" default_value { i: 0 } }
-}
-- | 
unique :: forall v'1 t out_idx . (TensorType t, OneOf '[Data.Int.Int32,
                                                        Data.Int.Int64] out_idx) =>
          Tensor v'1 t -- ^ __x__
          -> (Tensor Build t, Tensor Build out_idx) -- ^ (__y__, __idx__)
          --
          -- * __y__
          --
          -- * __idx__
unique :: Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
unique = OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
forall (v'1 :: * -> *) t out_idx.
(TensorType t, OneOf '[Int32, Int64] out_idx) =>
OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
unique' OpParams
forall a. a -> a
id
unique' :: forall v'1 t out_idx . (TensorType t, OneOf '[Data.Int.Int32,
                                                         Data.Int.Int64] out_idx) =>
           OpParams ->
           Tensor v'1 t -- ^ __x__
           -> (Tensor Build t, Tensor Build out_idx) -- ^ (__y__, __idx__)
           --
           -- * __y__
           --
           -- * __idx__
unique' :: OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
unique' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build out_idx))
-> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Unique"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_idx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_idx -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_idx
forall a. HasCallStack => a
undefined :: out_idx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "idx" type_attr: "out_idx" }
attr { name: "T" type: "type" }
attr {
  name: "out_idx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
uniqueDataset :: [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor Build Variant -- ^ __handle__
uniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
uniqueDataset = OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
uniqueDataset' OpParams
forall a. a -> a
id
uniqueDataset' :: OpParams ->
                  [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor Build Variant -- ^ __handle__
uniqueDataset' :: OpParams
-> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
uniqueDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UniqueDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
uniqueV2 :: forall v'1 v'2 t taxis out_idx . (TensorType t,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] taxis,
                                              OneOf '[Data.Int.Int32,
                                                      Data.Int.Int64] out_idx) =>
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 taxis -- ^ __axis__
            -> (Tensor Build t, Tensor Build out_idx) -- ^ (__y__, __idx__)
            --
            -- * __y__
            --
            -- * __idx__
uniqueV2 :: Tensor v'1 t
-> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx)
uniqueV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx)
forall (v'1 :: * -> *) (v'2 :: * -> *) t taxis out_idx.
(TensorType t, OneOf '[Int32, Int64] taxis,
 OneOf '[Int32, Int64] out_idx) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx)
uniqueV2' OpParams
forall a. a -> a
id
uniqueV2' :: forall v'1 v'2 t taxis out_idx . (TensorType t,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] taxis,
                                               OneOf '[Data.Int.Int32,
                                                       Data.Int.Int64] out_idx) =>
             OpParams ->
             Tensor v'1 t -- ^ __x__
             -> Tensor v'2 taxis -- ^ __axis__
             -> (Tensor Build t, Tensor Build out_idx) -- ^ (__y__, __idx__)
             --
             -- * __y__
             --
             -- * __idx__
uniqueV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx)
uniqueV2' op'options :: OpParams
op'options x :: Tensor v'1 t
x axis :: Tensor v'2 taxis
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build out_idx))
-> Build OpDef -> (Tensor Build t, Tensor Build out_idx)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 taxis -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 taxis
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UniqueV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Taxis" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ taxis -> DataType
forall a. TensorType a => a -> DataType
tensorType (taxis
forall a. HasCallStack => a
undefined :: taxis)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_idx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_idx -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_idx
forall a. HasCallStack => a
undefined :: out_idx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "axis" type_attr: "Taxis" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "idx" type_attr: "out_idx" }
attr { name: "T" type: "type" }
attr {
  name: "Taxis"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "out_idx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
uniqueWithCounts :: forall v'1 t out_idx . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] out_idx) =>
                    Tensor v'1 t -- ^ __x__
                    -> (Tensor Build t, Tensor Build out_idx,
                        Tensor Build out_idx) -- ^ (__y__, __idx__, __count__)
                    --
                    -- * __y__
                    --
                    -- * __idx__
                    --
                    -- * __count__
uniqueWithCounts :: Tensor v'1 t
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCounts = OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall (v'1 :: * -> *) t out_idx.
(TensorType t, OneOf '[Int32, Int64] out_idx) =>
OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCounts' OpParams
forall a. a -> a
id
uniqueWithCounts' :: forall v'1 t out_idx . (TensorType t,
                                             OneOf '[Data.Int.Int32,
                                                     Data.Int.Int64] out_idx) =>
                     OpParams ->
                     Tensor v'1 t -- ^ __x__
                     -> (Tensor Build t, Tensor Build out_idx,
                         Tensor Build out_idx) -- ^ (__y__, __idx__, __count__)
                     --
                     -- * __y__
                     --
                     -- * __idx__
                     --
                     -- * __count__
uniqueWithCounts' :: OpParams
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCounts' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx))
-> Build OpDef
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UniqueWithCounts"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_idx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_idx -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_idx
forall a. HasCallStack => a
undefined :: out_idx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "idx" type_attr: "out_idx" }
output_arg { name: "count" type_attr: "out_idx" }
attr { name: "T" type: "type" }
attr {
  name: "out_idx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
uniqueWithCountsV2 :: forall v'1 v'2 t taxis out_idx . (TensorType t,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] taxis,
                                                        OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64] out_idx) =>
                      Tensor v'1 t -- ^ __x__
                      -> Tensor v'2 taxis -- ^ __axis__
                      -> (Tensor Build t, Tensor Build out_idx,
                          Tensor Build out_idx) -- ^ (__y__, __idx__, __count__)
                      --
                      -- * __y__
                      --
                      -- * __idx__
                      --
                      -- * __count__
uniqueWithCountsV2 :: Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCountsV2 = OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall (v'1 :: * -> *) (v'2 :: * -> *) t taxis out_idx.
(TensorType t, OneOf '[Int32, Int64] taxis,
 OneOf '[Int32, Int64] out_idx) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCountsV2' OpParams
forall a. a -> a
id
uniqueWithCountsV2' :: forall v'1 v'2 t taxis out_idx . (TensorType t,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] taxis,
                                                         OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] out_idx) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __x__
                       -> Tensor v'2 taxis -- ^ __axis__
                       -> (Tensor Build t, Tensor Build out_idx,
                           Tensor Build out_idx)
                       -- ^ (__y__, __idx__, __count__)
                       --
                       -- * __y__
                       --
                       -- * __idx__
                       --
                       -- * __count__
uniqueWithCountsV2' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 taxis
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
uniqueWithCountsV2' op'options :: OpParams
op'options x :: Tensor v'1 t
x axis :: Tensor v'2 taxis
axis | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx))
-> Build OpDef
-> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 taxis -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 taxis
axis]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UniqueWithCountsV2"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Taxis" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ taxis -> DataType
forall a. TensorType a => a -> DataType
tensorType (taxis
forall a. HasCallStack => a
undefined :: taxis)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_idx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_idx -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_idx
forall a. HasCallStack => a
undefined :: out_idx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "axis" type_attr: "Taxis" }
output_arg { name: "y" type_attr: "T" }
output_arg { name: "idx" type_attr: "out_idx" }
output_arg { name: "count" type_attr: "out_idx" }
attr { name: "T" type: "type" }
attr {
  name: "Taxis"
  type: "type"
  default_value { type: DT_INT64 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "out_idx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unpack :: forall v'1 t . (TensorType t) => Data.Int.Int64 -- ^ __num__
          -> Tensor v'1 t -- ^ __value__
          -> [Tensor Build t] -- ^ __output__
unpack :: Int64 -> Tensor v'1 t -> [Tensor Build t]
unpack = OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
unpack' OpParams
forall a. a -> a
id
unpack' :: forall v'1 t . (TensorType t) => OpParams ->
           Data.Int.Int64 -- ^ __num__
           -> Tensor v'1 t -- ^ __value__
           -> [Tensor Build t] -- ^ __output__
unpack' :: OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
unpack' op'options :: OpParams
op'options num :: Int64
num value :: Tensor v'1 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Unpack"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "value" type_attr: "T" }
output_arg { name: "output" type_attr: "T" number_attr: "num" }
attr { name: "num" type: "int" has_minimum: true }
attr { name: "T" type: "type" }
attr { name: "axis" type: "int" default_value { i: 0 } }
-}
-- | 
unravelIndex :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tidx) =>
                Tensor v'1 tidx -- ^ __indices__
                -> Tensor v'2 tidx -- ^ __dims__
                -> Tensor Build tidx -- ^ __output__
unravelIndex :: Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
unravelIndex = OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Int32, Int64] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
unravelIndex' OpParams
forall a. a -> a
id
unravelIndex' :: forall v'1 v'2 tidx . (OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tidx) =>
                 OpParams ->
                 Tensor v'1 tidx -- ^ __indices__
                 -> Tensor v'2 tidx -- ^ __dims__
                 -> Tensor Build tidx -- ^ __output__
unravelIndex' :: OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
unravelIndex' op'options :: OpParams
op'options indices :: Tensor v'1 tidx
indices dims :: Tensor v'2 tidx
dims | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build tidx
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build tidx)
-> Build OpDef -> Tensor Build tidx
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 tidx
indices,
                                                             Tensor v'2 tidx -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tidx
dims]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnravelIndex"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tidx" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tidx -> DataType
forall a. TensorType a => a -> DataType
tensorType (tidx
forall a. HasCallStack => a
undefined :: tidx)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "indices" type_attr: "Tidx" }
input_arg { name: "dims" type_attr: "Tidx" }
output_arg { name: "output" type_attr: "Tidx" }
attr {
  name: "Tidx"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unsortedSegmentJoin :: forall v'1 v'2 v'3 tindices
                       tnumsegments . (OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tindices,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tnumsegments) =>
                       Tensor v'1 Data.ByteString.ByteString -- ^ __inputs__
                       -> Tensor v'2 tindices -- ^ __segment_ids__
                       -> Tensor v'3 tnumsegments -- ^ __num_segments__
                       -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unsortedSegmentJoin :: Tensor v'1 ByteString
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build ByteString
unsortedSegmentJoin = OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build ByteString
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) tindices
       tnumsegments.
(OneOf '[Int32, Int64] tindices,
 OneOf '[Int32, Int64] tnumsegments) =>
OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build ByteString
unsortedSegmentJoin' OpParams
forall a. a -> a
id
unsortedSegmentJoin' :: forall v'1 v'2 v'3 tindices
                        tnumsegments . (OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tindices,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tnumsegments) =>
                        OpParams ->
                        Tensor v'1 Data.ByteString.ByteString -- ^ __inputs__
                        -> Tensor v'2 tindices -- ^ __segment_ids__
                        -> Tensor v'3 tnumsegments -- ^ __num_segments__
                        -> Tensor Build Data.ByteString.ByteString -- ^ __output__
unsortedSegmentJoin' :: OpParams
-> Tensor v'1 ByteString
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build ByteString
unsortedSegmentJoin' op'options :: OpParams
op'options inputs :: Tensor v'1 ByteString
inputs segment_ids :: Tensor v'2 tindices
segment_ids
                     num_segments :: Tensor v'3 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build ByteString
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build ByteString)
-> Build OpDef -> Tensor Build ByteString
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
inputs,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids,
                                                             Tensor v'3 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnsortedSegmentJoin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "inputs" type: DT_STRING }
input_arg { name: "segment_ids" type_attr: "Tindices" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type: DT_STRING }
attr { name: "separator" type: "string" default_value { s: "" } }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unsortedSegmentMax :: forall v'1 v'2 v'3 t tindices
                      tnumsegments . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tindices,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tnumsegments) =>
                      Tensor v'1 t -- ^ __data__
                      -> Tensor v'2 tindices -- ^ __segment_ids__
                      -> Tensor v'3 tnumsegments -- ^ __num_segments__
                      -> Tensor Build t -- ^ __output__
unsortedSegmentMax :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
unsortedSegmentMax = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx
       tsegmentids.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
unsortedSegmentMax' OpParams
forall a. a -> a
id
unsortedSegmentMax' :: forall v'1 v'2 v'3 t tindices
                       tnumsegments . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tnumsegments) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __data__
                       -> Tensor v'2 tindices -- ^ __segment_ids__
                       -> Tensor v'3 tnumsegments -- ^ __num_segments__
                       -> Tensor Build t -- ^ __output__
unsortedSegmentMax' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentMax' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids
                    num_segments :: Tensor v'3 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids,
                                                             Tensor v'3 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnsortedSegmentMax"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unsortedSegmentMin :: forall v'1 v'2 v'3 t tindices
                      tnumsegments . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tindices,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tnumsegments) =>
                      Tensor v'1 t -- ^ __data__
                      -> Tensor v'2 tindices -- ^ __segment_ids__
                      -> Tensor v'3 tnumsegments -- ^ __num_segments__
                      -> Tensor Build t -- ^ __output__
unsortedSegmentMin :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
unsortedSegmentMin = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tidx
       tsegmentids.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t,
 OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tidx
-> Tensor v'3 tsegmentids
-> Tensor Build t
unsortedSegmentMin' OpParams
forall a. a -> a
id
unsortedSegmentMin' :: forall v'1 v'2 v'3 t tindices
                       tnumsegments . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tnumsegments) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __data__
                       -> Tensor v'2 tindices -- ^ __segment_ids__
                       -> Tensor v'3 tnumsegments -- ^ __num_segments__
                       -> Tensor Build t -- ^ __output__
unsortedSegmentMin' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentMin' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids
                    num_segments :: Tensor v'3 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids,
                                                             Tensor v'3 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnsortedSegmentMin"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unsortedSegmentProd :: forall v'1 v'2 v'3 t tindices
                       tnumsegments . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tnumsegments) =>
                       Tensor v'1 t -- ^ __data__
                       -> Tensor v'2 tindices -- ^ __segment_ids__
                       -> Tensor v'3 tnumsegments -- ^ __num_segments__
                       -> Tensor Build t -- ^ __output__
unsortedSegmentProd :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
unsortedSegmentProd = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       tnumsegments.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices,
 OneOf '[Int32, Int64] tnumsegments) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentProd' OpParams
forall a. a -> a
id
unsortedSegmentProd' :: forall v'1 v'2 v'3 t tindices
                        tnumsegments . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word32,
                                                Data.Word.Word64,
                                                Data.Word.Word8, Double,
                                                Float] t,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tindices,
                                        OneOf '[Data.Int.Int32,
                                                Data.Int.Int64] tnumsegments) =>
                        OpParams ->
                        Tensor v'1 t -- ^ __data__
                        -> Tensor v'2 tindices -- ^ __segment_ids__
                        -> Tensor v'3 tnumsegments -- ^ __num_segments__
                        -> Tensor Build t -- ^ __output__
unsortedSegmentProd' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentProd' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids
                     num_segments :: Tensor v'3 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids,
                                                             Tensor v'3 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnsortedSegmentProd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unsortedSegmentSum :: forall v'1 v'2 v'3 t tindices
                      tnumsegments . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tindices,
                                      OneOf '[Data.Int.Int32,
                                              Data.Int.Int64] tnumsegments) =>
                      Tensor v'1 t -- ^ __data__
                      -> Tensor v'2 tindices -- ^ __segment_ids__
                      -> Tensor v'3 tnumsegments -- ^ __num_segments__
                      -> Tensor Build t -- ^ __output__
unsortedSegmentSum :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
unsortedSegmentSum = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices
       tnumsegments.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices,
 OneOf '[Int32, Int64] tnumsegments) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentSum' OpParams
forall a. a -> a
id
unsortedSegmentSum' :: forall v'1 v'2 v'3 t tindices
                       tnumsegments . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word32,
                                               Data.Word.Word64,
                                               Data.Word.Word8, Double,
                                               Float] t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] tindices,
                                       OneOf '[Data.Int.Int32,
                                               Data.Int.Int64] tnumsegments) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __data__
                       -> Tensor v'2 tindices -- ^ __segment_ids__
                       -> Tensor v'3 tnumsegments -- ^ __num_segments__
                       -> Tensor Build t -- ^ __output__
unsortedSegmentSum' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tnumsegments
-> Tensor Build t
unsortedSegmentSum' op'options :: OpParams
op'options data' :: Tensor v'1 t
data' segment_ids :: Tensor v'2 tindices
segment_ids
                    num_segments :: Tensor v'3 tnumsegments
num_segments | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
segment_ids,
                                                             Tensor v'3 tnumsegments -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tnumsegments
num_segments]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnsortedSegmentSum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tnumsegments" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tnumsegments -> DataType
forall a. TensorType a => a -> DataType
tensorType (tnumsegments
forall a. HasCallStack => a
undefined :: tnumsegments)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "segment_ids" type_attr: "Tindices" }
input_arg { name: "num_segments" type_attr: "Tnumsegments" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "Tnumsegments"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
unstage :: forall dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
           m' (TensorList (Value) dtypes) -- ^ __values__
unstage :: m' (TensorList Value dtypes)
unstage = OpParams -> m' (TensorList Value dtypes)
forall (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams -> m' (TensorList Value dtypes)
unstage' OpParams
forall a. a -> a
id
unstage' :: forall dtypes m' . (MonadBuild m', TensorTypes dtypes) =>
            OpParams ->
            m' (TensorList (Value) dtypes) -- ^ __values__
unstage' :: OpParams -> m' (TensorList Value dtypes)
unstage' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Unstage"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "values" type_list_attr: "dtypes" }
attr {
  name: "capacity"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "memory_limit"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
unwrapDatasetVariant :: Tensor v'1 Variant -- ^ __input_handle__
                        -> Tensor Build Variant -- ^ __output_handle__
unwrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
unwrapDatasetVariant = OpParams -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Variant
unwrapDatasetVariant' OpParams
forall a. a -> a
id
unwrapDatasetVariant' :: OpParams ->
                         Tensor v'1 Variant -- ^ __input_handle__
                         -> Tensor Build Variant -- ^ __output_handle__
unwrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
unwrapDatasetVariant' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UnwrapDatasetVariant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
output_arg { name: "output_handle" type: DT_VARIANT }
-}
-- | 
upperBound :: forall v'1 v'2 t out_type . (TensorType t, OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64] out_type) =>
              Tensor v'1 t -- ^ __sorted_inputs__
              -> Tensor v'2 t -- ^ __values__
              -> Tensor Build out_type -- ^ __output__
upperBound :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
upperBound = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
forall (v'1 :: * -> *) (v'2 :: * -> *) t out_type.
(TensorType t, OneOf '[Int32, Int64] out_type) =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
upperBound' OpParams
forall a. a -> a
id
upperBound' :: forall v'1 v'2 t out_type . (TensorType t,
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] out_type) =>
               OpParams ->
               Tensor v'1 t -- ^ __sorted_inputs__
               -> Tensor v'2 t -- ^ __values__
               -> Tensor Build out_type -- ^ __output__
upperBound' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
upperBound' op'options :: OpParams
op'options sorted_inputs :: Tensor v'1 t
sorted_inputs values :: Tensor v'2 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build out_type
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build out_type)
-> Build OpDef -> Tensor Build out_type
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
sorted_inputs,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "UpperBound"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "sorted_inputs" type_attr: "T" }
input_arg { name: "values" type_attr: "T" }
output_arg { name: "output" type_attr: "out_type" }
attr { name: "T" type: "type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
varHandleOp :: forall m' . (MonadBuild m') => DataType -- ^ __dtype__
               -> Shape -- ^ __shape__
               -> m' (Tensor Value ResourceHandle) -- ^ __resource__
varHandleOp :: DataType -> Shape -> m' (Tensor Value ResourceHandle)
varHandleOp = OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
varHandleOp' OpParams
forall a. a -> a
id
varHandleOp' :: forall m' . (MonadBuild m') => OpParams ->
                DataType -- ^ __dtype__
                -> Shape -- ^ __shape__
                -> m' (Tensor Value ResourceHandle) -- ^ __resource__
varHandleOp' :: OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
varHandleOp' op'options :: OpParams
op'options dtype :: DataType
dtype shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "VarHandleOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ DataType
dtype
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resource" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
attr { name: "dtype" type: "type" }
attr { name: "shape" type: "shape" }
attr {
  name: "allowed_devices"
  type: "list(string)"
  default_value { list { } }
}
-}
-- | 
varIsInitializedOp :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 ResourceHandle -- ^ __resource__
                      -> m' (Tensor Value Bool) -- ^ __is_initialized__
varIsInitializedOp :: Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
varIsInitializedOp = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
varIsInitializedOp' OpParams
forall a. a -> a
id
varIsInitializedOp' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __resource__
                       -> m' (Tensor Value Bool) -- ^ __is_initialized__
varIsInitializedOp' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
varIsInitializedOp' op'options :: OpParams
op'options resource :: Tensor v'1 ResourceHandle
resource | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Bool) -> m' (Tensor Value Bool))
-> Build (Tensor Value Bool) -> m' (Tensor Value Bool)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
resource]
        [Int64] -> OpDef -> Build (Tensor Value Bool)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "VarIsInitializedOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "resource" type: DT_RESOURCE }
output_arg { name: "is_initialized" type: DT_BOOL }
-}
-- | 
variable :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
            Shape -- ^ __shape__
            -> m' (Tensor Ref dtype) -- ^ __ref__
variable :: Shape -> m' (Tensor Ref dtype)
variable = OpParams -> Shape -> m' (Tensor Ref dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Ref dtype)
variable' OpParams
forall a. a -> a
id
variable' :: forall dtype m' . (MonadBuild m', TensorType dtype) => OpParams ->
             Shape -- ^ __shape__
             -> m' (Tensor Ref dtype) -- ^ __ref__
variable' :: OpParams -> Shape -> m' (Tensor Ref dtype)
variable' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref dtype) -> m' (Tensor Ref dtype))
-> Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "Variable"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "ref" type_attr: "dtype" is_ref: true }
attr { name: "shape" type: "shape" }
attr { name: "dtype" type: "type" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
variableShape :: forall v'1 out_type m' . (MonadBuild m',
                                           OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] out_type) =>
                 Tensor v'1 ResourceHandle -- ^ __input__
                 -> m' (Tensor Value out_type) -- ^ __output__
variableShape :: Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
variableShape = OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
forall (v'1 :: * -> *) out_type (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64] out_type) =>
OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
variableShape' OpParams
forall a. a -> a
id
variableShape' :: forall v'1 out_type m' . (MonadBuild m',
                                            OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] out_type) =>
                  OpParams ->
                  Tensor v'1 ResourceHandle -- ^ __input__
                  -> m' (Tensor Value out_type) -- ^ __output__
variableShape' :: OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
variableShape' op'options :: OpParams
op'options input :: Tensor v'1 ResourceHandle
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value out_type) -> m' (Tensor Value out_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value out_type) -> m' (Tensor Value out_type))
-> Build (Tensor Value out_type) -> m' (Tensor Value out_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
input]
        [Int64] -> OpDef -> Build (Tensor Value out_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "VariableShape"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ out_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (out_type
forall a. HasCallStack => a
undefined :: out_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type: DT_RESOURCE }
output_arg { name: "output" type_attr: "out_type" }
attr {
  name: "out_type"
  type: "type"
  default_value { type: DT_INT32 }
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | 
variableV2 :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
              Shape -- ^ __shape__
              -> m' (Tensor Ref dtype) -- ^ __ref__
variableV2 :: Shape -> m' (Tensor Ref dtype)
variableV2 = OpParams -> Shape -> m' (Tensor Ref dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Ref dtype)
variableV2' OpParams
forall a. a -> a
id
variableV2' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
               OpParams ->
               Shape -- ^ __shape__
               -> m' (Tensor Ref dtype) -- ^ __ref__
variableV2' :: OpParams -> Shape -> m' (Tensor Ref dtype)
variableV2' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref dtype) -> m' (Tensor Ref dtype))
-> Build (Tensor Ref dtype) -> m' (Tensor Ref dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "VariableV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "ref" type_attr: "dtype" is_ref: true }
attr { name: "shape" type: "shape" }
attr { name: "dtype" type: "type" }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
where' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Bool,
                                  Data.Int.Int16, Data.Int.Int32,
                                  Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t) => Tensor v'1 t -- ^ __input__
          -> Tensor Build Data.Int.Int64 -- ^ __index__
where' :: Tensor v'1 t -> Tensor Build Int64
where' = OpParams -> Tensor v'1 t -> Tensor Build Int64
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8,
    Word16, Word32, Word64, Word8, Double, Float]
  t =>
OpParams -> Tensor v'1 t -> Tensor Build Int64
where'' OpParams
forall a. a -> a
id
where'' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Bool,
                                   Data.Int.Int16, Data.Int.Int32,
                                   Data.Int.Int64, Data.Int.Int8,
                                   Data.Word.Word16, Data.Word.Word32,
                                   Data.Word.Word64, Data.Word.Word8, Double,
                                   Float] t) => OpParams ->
           Tensor v'1 t -- ^ __input__
           -> Tensor Build Data.Int.Int64 -- ^ __index__
where'' :: OpParams -> Tensor v'1 t -> Tensor Build Int64
where'' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int64
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int64)
-> Build OpDef -> Tensor Build Int64
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Where"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "index" type: DT_INT64 }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_BOOL }
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
      type: DT_BOOL
    }
  }
}
-}
-- | 
wholeFileReader :: forall m' . (MonadBuild m') =>
                   m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
wholeFileReader :: m' (Tensor Ref ByteString)
wholeFileReader = OpParams -> m' (Tensor Ref ByteString)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Ref ByteString)
wholeFileReader' OpParams
forall a. a -> a
id
wholeFileReader' :: forall m' . (MonadBuild m') => OpParams ->
                    m' (Tensor Ref Data.ByteString.ByteString) -- ^ __reader_handle__
wholeFileReader' :: OpParams -> m' (Tensor Ref ByteString)
wholeFileReader' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString))
-> Build (Tensor Ref ByteString) -> m' (Tensor Ref ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Ref ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WholeFileReader"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_STRING is_ref: true }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
wholeFileReaderV2 :: forall m' . (MonadBuild m') =>
                     m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
wholeFileReaderV2 :: m' (Tensor Value ResourceHandle)
wholeFileReaderV2 = OpParams -> m' (Tensor Value ResourceHandle)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value ResourceHandle)
wholeFileReaderV2' OpParams
forall a. a -> a
id
wholeFileReaderV2' :: forall m' . (MonadBuild m') => OpParams ->
                      m' (Tensor Value ResourceHandle) -- ^ __reader_handle__
wholeFileReaderV2' :: OpParams -> m' (Tensor Value ResourceHandle)
wholeFileReaderV2' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ResourceHandle)
 -> m' (Tensor Value ResourceHandle))
-> Build (Tensor Value ResourceHandle)
-> m' (Tensor Value ResourceHandle)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value ResourceHandle)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WholeFileReaderV2"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "reader_handle" type: DT_RESOURCE }
attr { name: "container" type: "string" default_value { s: "" } }
attr { name: "shared_name" type: "string" default_value { s: "" } }
-}
-- | 
windowDataset :: [DataType] -- ^ __output_types__
                 -> Tensor v'1 Variant -- ^ __input_dataset__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __size__
                 -> Tensor v'3 Data.Int.Int64 -- ^ __shift__
                 -> Tensor v'4 Data.Int.Int64 -- ^ __stride__
                 -> Tensor v'5 Bool -- ^ __drop_remainder__
                 -> Tensor Build Variant -- ^ __handle__
windowDataset :: [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Bool
-> Tensor Build Variant
windowDataset = OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Bool
-> Tensor Build Variant
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *).
OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Bool
-> Tensor Build Variant
windowDataset' OpParams
forall a. a -> a
id
windowDataset' :: OpParams ->
                  [DataType] -- ^ __output_types__
                  -> Tensor v'1 Variant -- ^ __input_dataset__
                  -> Tensor v'2 Data.Int.Int64 -- ^ __size__
                  -> Tensor v'3 Data.Int.Int64 -- ^ __shift__
                  -> Tensor v'4 Data.Int.Int64 -- ^ __stride__
                  -> Tensor v'5 Bool -- ^ __drop_remainder__
                  -> Tensor Build Variant -- ^ __handle__
windowDataset' :: OpParams
-> [DataType]
-> Tensor v'1 Variant
-> Tensor v'2 Int64
-> Tensor v'3 Int64
-> Tensor v'4 Int64
-> Tensor v'5 Bool
-> Tensor Build Variant
windowDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types input_dataset :: Tensor v'1 Variant
input_dataset size :: Tensor v'2 Int64
size shift :: Tensor v'3 Int64
shift stride :: Tensor v'4 Int64
stride
               drop_remainder :: Tensor v'5 Bool
drop_remainder | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_dataset,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
size,
                                                             Tensor v'3 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Int64
shift,
                                                             Tensor v'4 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Int64
stride,
                                                             Tensor v'5 Bool -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Bool
drop_remainder]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "WindowDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_dataset" type: DT_VARIANT }
input_arg { name: "size" type: DT_INT64 }
input_arg { name: "shift" type: DT_INT64 }
input_arg { name: "stride" type: DT_INT64 }
input_arg { name: "drop_remainder" type: DT_BOOL }
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
-}
-- | 
workerHeartbeat :: forall v'1 m' . (MonadBuild m') =>
                   Tensor v'1 Data.ByteString.ByteString -- ^ __request__
                   -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __response__
workerHeartbeat :: Tensor v'1 ByteString -> m' (Tensor Value ByteString)
workerHeartbeat = OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ByteString)
workerHeartbeat' OpParams
forall a. a -> a
id
workerHeartbeat' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                    Tensor v'1 Data.ByteString.ByteString -- ^ __request__
                    -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __response__
workerHeartbeat' :: OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ByteString)
workerHeartbeat' op'options :: OpParams
op'options request :: Tensor v'1 ByteString
request | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
request]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WorkerHeartbeat"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "request" type: DT_STRING }
output_arg { name: "response" type: DT_STRING }
-}
-- | 
wrapDatasetVariant :: Tensor v'1 Variant -- ^ __input_handle__
                      -> Tensor Build Variant -- ^ __output_handle__
wrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
wrapDatasetVariant = OpParams -> Tensor v'1 Variant -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams -> Tensor v'1 Variant -> Tensor Build Variant
wrapDatasetVariant' OpParams
forall a. a -> a
id
wrapDatasetVariant' :: OpParams ->
                       Tensor v'1 Variant -- ^ __input_handle__
                       -> Tensor Build Variant -- ^ __output_handle__
wrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
wrapDatasetVariant' op'options :: OpParams
op'options input_handle :: Tensor v'1 Variant
input_handle | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Variant -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Variant
input_handle]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "WrapDatasetVariant"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input_handle" type: DT_VARIANT }
output_arg { name: "output_handle" type: DT_VARIANT }
-}
-- | 
writeAudioSummary :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __writer__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                     -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                     -> Tensor v'4 Float -- ^ __tensor__
                     -> Tensor v'5 Float -- ^ __sample_rate__
                     -> m' (ControlNode)
writeAudioSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' ControlNode
writeAudioSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' ControlNode
writeAudioSummary' OpParams
forall a. a -> a
id
writeAudioSummary' :: forall v'1 v'2 v'3 v'4 v'5 m' . (MonadBuild m') =>
                      OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                      -> Tensor v'4 Float -- ^ __tensor__
                      -> Tensor v'5 Float -- ^ __sample_rate__
                      -> m' (ControlNode)
writeAudioSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 Float
-> Tensor v'5 Float
-> m' ControlNode
writeAudioSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tag :: Tensor v'3 ByteString
tag tensor :: Tensor v'4 Float
tensor
                   sample_rate :: Tensor v'5 Float
sample_rate | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Float
tensor,
                                                             Tensor v'5 Float -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Float
sample_rate]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteAudioSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type: DT_FLOAT }
input_arg { name: "sample_rate" type: DT_FLOAT }
attr {
  name: "max_outputs"
  type: "int"
  default_value { i: 3 }
  has_minimum: true
  minimum: 1
}
-}
-- | 
writeFile :: forall v'1 v'2 m' . (MonadBuild m') =>
             Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
             -> Tensor v'2 Data.ByteString.ByteString -- ^ __contents__
             -> m' (ControlNode)
writeFile :: Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
writeFile = OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
writeFile' OpParams
forall a. a -> a
id
writeFile' :: forall v'1 v'2 m' . (MonadBuild m') => OpParams ->
              Tensor v'1 Data.ByteString.ByteString -- ^ __filename__
              -> Tensor v'2 Data.ByteString.ByteString -- ^ __contents__
              -> m' (ControlNode)
writeFile' :: OpParams
-> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
writeFile' op'options :: OpParams
op'options filename :: Tensor v'1 ByteString
filename contents :: Tensor v'2 ByteString
contents | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
filename,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
contents]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteFile"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "filename" type: DT_STRING }
input_arg { name: "contents" type: DT_STRING }
-}
-- | 
writeGraphSummary :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                     Tensor v'1 ResourceHandle -- ^ __writer__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                     -> Tensor v'3 Data.ByteString.ByteString -- ^ __tensor__
                     -> m' (ControlNode)
writeGraphSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
writeGraphSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
writeGraphSummary' OpParams
forall a. a -> a
id
writeGraphSummary' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __tensor__
                      -> m' (ControlNode)
writeGraphSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
writeGraphSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tensor :: Tensor v'3 ByteString
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteGraphSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tensor" type: DT_STRING }
-}
-- | 
writeHistogramSummary :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                        OneOf '[Data.Int.Int16,
                                                                Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Int.Int8,
                                                                Data.Word.Word16,
                                                                Data.Word.Word32,
                                                                Data.Word.Word64,
                                                                Data.Word.Word8,
                                                                Double,
                                                                Float] t) =>
                         Tensor v'1 ResourceHandle -- ^ __writer__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                         -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                         -> Tensor v'4 t -- ^ __values__
                         -> m' (ControlNode)
writeHistogramSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeHistogramSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeHistogramSummary' OpParams
forall a. a -> a
id
writeHistogramSummary' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                         OneOf '[Data.Int.Int16,
                                                                 Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Int.Int8,
                                                                 Data.Word.Word16,
                                                                 Data.Word.Word32,
                                                                 Data.Word.Word64,
                                                                 Data.Word.Word8,
                                                                 Double,
                                                                 Float] t) =>
                          OpParams ->
                          Tensor v'1 ResourceHandle -- ^ __writer__
                          -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                          -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                          -> Tensor v'4 t -- ^ __values__
                          -> m' (ControlNode)
writeHistogramSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeHistogramSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tag :: Tensor v'3 ByteString
tag values :: Tensor v'4 t
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
values]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteHistogramSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "values" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
writeImageSummary :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                        OneOf '[Data.Word.Word16,
                                                                Data.Word.Word8,
                                                                Float] t) =>
                     Tensor v'1 ResourceHandle -- ^ __writer__
                     -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                     -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                     -> Tensor v'4 t -- ^ __tensor__
                     -> Tensor v'5 Data.Word.Word8 -- ^ __bad_color__
                     -> m' (ControlNode)
writeImageSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> Tensor v'5 Word8
-> m' ControlNode
writeImageSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> Tensor v'5 Word8
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Word16, Word8, Float] t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> Tensor v'5 Word8
-> m' ControlNode
writeImageSummary' OpParams
forall a. a -> a
id
writeImageSummary' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                         OneOf '[Data.Word.Word16,
                                                                 Data.Word.Word8,
                                                                 Float] t) =>
                      OpParams ->
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                      -> Tensor v'4 t -- ^ __tensor__
                      -> Tensor v'5 Data.Word.Word8 -- ^ __bad_color__
                      -> m' (ControlNode)
writeImageSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> Tensor v'5 Word8
-> m' ControlNode
writeImageSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tag :: Tensor v'3 ByteString
tag tensor :: Tensor v'4 t
tensor
                   bad_color :: Tensor v'5 Word8
bad_color | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
tensor,
                                                             Tensor v'5 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 Word8
bad_color]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteImageSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "bad_color" type: DT_UINT8 }
attr {
  name: "max_images"
  type: "int"
  default_value { i: 3 }
  has_minimum: true
  minimum: 1
}
attr {
  name: "T"
  type: "type"
  default_value { type: DT_FLOAT }
  allowed_values {
    list { type: DT_UINT8 type: DT_FLOAT type: DT_HALF }
  }
}
-}
-- | 
writeRawProtoSummary :: forall v'1 v'2 v'3 m' . (MonadBuild m') =>
                        Tensor v'1 ResourceHandle -- ^ __writer__
                        -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                        -> Tensor v'3 Data.ByteString.ByteString -- ^ __tensor__
                        -> m' (ControlNode)
writeRawProtoSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
writeRawProtoSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
writeRawProtoSummary' OpParams
forall a. a -> a
id
writeRawProtoSummary' :: forall v'1 v'2 v'3 m' . (MonadBuild m') => OpParams ->
                         Tensor v'1 ResourceHandle -- ^ __writer__
                         -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                         -> Tensor v'3 Data.ByteString.ByteString -- ^ __tensor__
                         -> m' (ControlNode)
writeRawProtoSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> m' ControlNode
writeRawProtoSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tensor :: Tensor v'3 ByteString
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteRawProtoSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tensor" type: DT_STRING }
-}
-- | 
writeScalarSummary :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                     OneOf '[Data.Int.Int16,
                                                             Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Int.Int8,
                                                             Data.Word.Word16,
                                                             Data.Word.Word32,
                                                             Data.Word.Word64,
                                                             Data.Word.Word8,
                                                             Double,
                                                             Float] t) =>
                      Tensor v'1 ResourceHandle -- ^ __writer__
                      -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                      -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                      -> Tensor v'4 t -- ^ __value__
                      -> m' (ControlNode)
writeScalarSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeScalarSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t (m' :: * -> *).
(MonadBuild m',
 OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeScalarSummary' OpParams
forall a. a -> a
id
writeScalarSummary' :: forall v'1 v'2 v'3 v'4 t m' . (MonadBuild m',
                                                      OneOf '[Data.Int.Int16,
                                                              Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Int.Int8,
                                                              Data.Word.Word16,
                                                              Data.Word.Word32,
                                                              Data.Word.Word64,
                                                              Data.Word.Word8,
                                                              Double,
                                                              Float] t) =>
                       OpParams ->
                       Tensor v'1 ResourceHandle -- ^ __writer__
                       -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                       -> Tensor v'3 Data.ByteString.ByteString -- ^ __tag__
                       -> Tensor v'4 t -- ^ __value__
                       -> m' (ControlNode)
writeScalarSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 ByteString
-> Tensor v'4 t
-> m' ControlNode
writeScalarSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tag :: Tensor v'3 ByteString
tag value :: Tensor v'4 t
value | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 ByteString
tag,
                                                             Tensor v'4 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 t
value]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteScalarSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "value" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
writeSummary :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                   TensorType t) =>
                Tensor v'1 ResourceHandle -- ^ __writer__
                -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                -> Tensor v'3 t -- ^ __tensor__
                -> Tensor v'4 Data.ByteString.ByteString -- ^ __tag__
                -> Tensor v'5 Data.ByteString.ByteString -- ^ __summary_metadata__
                -> m' (ControlNode)
writeSummary :: Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
writeSummary = OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
writeSummary' OpParams
forall a. a -> a
id
writeSummary' :: forall v'1 v'2 v'3 v'4 v'5 t m' . (MonadBuild m',
                                                    TensorType t) => OpParams ->
                 Tensor v'1 ResourceHandle -- ^ __writer__
                 -> Tensor v'2 Data.Int.Int64 -- ^ __step__
                 -> Tensor v'3 t -- ^ __tensor__
                 -> Tensor v'4 Data.ByteString.ByteString -- ^ __tag__
                 -> Tensor v'5 Data.ByteString.ByteString -- ^ __summary_metadata__
                 -> m' (ControlNode)
writeSummary' :: OpParams
-> Tensor v'1 ResourceHandle
-> Tensor v'2 Int64
-> Tensor v'3 t
-> Tensor v'4 ByteString
-> Tensor v'5 ByteString
-> m' ControlNode
writeSummary' op'options :: OpParams
op'options writer :: Tensor v'1 ResourceHandle
writer step :: Tensor v'2 Int64
step tensor :: Tensor v'3 t
tensor tag :: Tensor v'4 ByteString
tag
              summary_metadata :: Tensor v'5 ByteString
summary_metadata | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ResourceHandle -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ResourceHandle
writer,
                                                             Tensor v'2 Int64 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int64
step,
                                                             Tensor v'3 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 t
tensor,
                                                             Tensor v'4 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 ByteString
tag,
                                                             Tensor v'5 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 ByteString
summary_metadata]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "WriteSummary"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "writer" type: DT_RESOURCE }
input_arg { name: "step" type: DT_INT64 }
input_arg { name: "tensor" type_attr: "T" }
input_arg { name: "tag" type: DT_STRING }
input_arg { name: "summary_metadata" type: DT_STRING }
attr { name: "T" type: "type" }
-}
-- | 
xdivy :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __y__
         -> Tensor Build t -- ^ __z__
xdivy :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xdivy = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xdivy' OpParams
forall a. a -> a
id
xdivy' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor v'2 t -- ^ __y__
          -> Tensor Build t -- ^ __z__
xdivy' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xdivy' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Xdivy"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | Helper operator for performing XLA-style broadcasts
--
-- Broadcasts `lhs` and `rhs` to the same rank, by adding size 1 dimensions to
-- whichever of `lhs` and `rhs` has the lower rank, using XLA's broadcasting rules
-- for binary operators.
xlaBroadcastHelper :: forall v'1 v'2 v'3 t
                      tindices . (OneOf '[(Data.Complex.Complex Double),
                                          (Data.Complex.Complex Float),
                                          Data.Int.Int16, Data.Int.Int32,
                                          Data.Int.Int64, Data.Int.Int8,
                                          Data.Word.Word16, Data.Word.Word32,
                                          Data.Word.Word64, Data.Word.Word8,
                                          Double, Float] t,
                                  OneOf '[Data.Int.Int32,
                                          Data.Int.Int64] tindices) =>
                      Tensor v'1 t -- ^ __lhs__: the LHS input tensor
                      -> Tensor v'2 t -- ^ __rhs__: the RHS input tensor
                      -> Tensor v'3 tindices -- ^ __broadcast_dims__: an XLA-style broadcast dimension specification
                      -> (Tensor Build t, Tensor Build t)
                      -- ^ (__lhs_output__, __rhs_output__)
                      --
                      -- * __lhs_output__: the broadcasted LHS tensor
                      --
                      -- * __rhs_output__: the broadcasted RHS tensor
xlaBroadcastHelper :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> (Tensor Build t, Tensor Build t)
xlaBroadcastHelper = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> (Tensor Build t, Tensor Build t)
xlaBroadcastHelper' OpParams
forall a. a -> a
id
xlaBroadcastHelper' :: forall v'1 v'2 v'3 t
                       tindices . (OneOf '[(Data.Complex.Complex Double),
                                           (Data.Complex.Complex Float),
                                           Data.Int.Int16, Data.Int.Int32,
                                           Data.Int.Int64, Data.Int.Int8,
                                           Data.Word.Word16, Data.Word.Word32,
                                           Data.Word.Word64, Data.Word.Word8,
                                           Double, Float] t,
                                   OneOf '[Data.Int.Int32,
                                           Data.Int.Int64] tindices) =>
                       OpParams ->
                       Tensor v'1 t -- ^ __lhs__: the LHS input tensor
                       -> Tensor v'2 t -- ^ __rhs__: the RHS input tensor
                       -> Tensor v'3 tindices -- ^ __broadcast_dims__: an XLA-style broadcast dimension specification
                       -> (Tensor Build t, Tensor Build t)
                       -- ^ (__lhs_output__, __rhs_output__)
                       --
                       -- * __lhs_output__: the broadcasted LHS tensor
                       --
                       -- * __rhs_output__: the broadcasted RHS tensor
xlaBroadcastHelper' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> (Tensor Build t, Tensor Build t)
xlaBroadcastHelper' op'options :: OpParams
op'options lhs :: Tensor v'1 t
lhs rhs :: Tensor v'2 t
rhs broadcast_dims :: Tensor v'3 tindices
broadcast_dims | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
lhs,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
broadcast_dims]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaBroadcastHelper"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "lhs" description: "the LHS input tensor" type_attr: "T"
}
input_arg {
  name: "rhs" description: "the RHS input tensor" type_attr: "T"
}
input_arg {
  name: "broadcast_dims"
  description: "an XLA-style broadcast dimension specification"
  type_attr: "Tindices"
}
output_arg {
  name: "lhs_output"
  description: "the broadcasted LHS tensor"
  type_attr: "T"
}
output_arg {
  name: "rhs_output"
  description: "the broadcasted RHS tensor"
  type_attr: "T"
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | Operator that connects the output of an XLA computation to other consumer graph nodes.
xlaClusterOutput :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
                    -> Tensor Build t -- ^ __outputs__
xlaClusterOutput :: Tensor v'1 t -> Tensor Build t
xlaClusterOutput = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
xlaClusterOutput' OpParams
forall a. a -> a
id
xlaClusterOutput' :: forall v'1 t . (TensorType t) => OpParams ->
                     Tensor v'1 t -- ^ __input__
                     -> Tensor Build t -- ^ __outputs__
xlaClusterOutput' :: OpParams -> Tensor v'1 t -> Tensor Build t
xlaClusterOutput' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaClusterOutput"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "outputs" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | Wraps the XLA ConvGeneralDilated operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution
-- .
xlaConv :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
           tindices . (OneOf '[(Data.Complex.Complex Double),
                               (Data.Complex.Complex Float), Data.Int.Int16,
                               Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                               Data.Word.Word16, Data.Word.Word32,
                               Data.Word.Word64, Data.Word.Word8, Double,
                               Float] t, OneOf '[Data.Int.Int32,
                                                 Data.Int.Int64] tindices) =>
           ByteString -- ^ __dimension_numbers__: a serialized xla::ConvolutionDimensionNumbers proto.
           -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
           -> Tensor v'1 t -- ^ __lhs__: the input tensor
           -> Tensor v'2 t -- ^ __rhs__: the kernel tensor
           -> Tensor v'3 tindices -- ^ __window_strides__: the inter-window strides
           -> Tensor v'4 tindices -- ^ __padding__: the padding to apply at the start and end of each input dimensions
           -> Tensor v'5 tindices -- ^ __lhs_dilation__: dilation to apply between input elements
           -> Tensor v'6 tindices -- ^ __rhs_dilation__: dilation to apply between kernel elements
           -> Tensor v'7 tindices -- ^ __feature_group_count__: number of feature groups for grouped convolution.
           -> Tensor Build t -- ^ __output__
xlaConv :: ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor v'6 tindices
-> Tensor v'7 tindices
-> Tensor Build t
xlaConv = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor v'6 tindices
-> Tensor v'7 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) (v'7 :: * -> *) t
       tindices.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor v'6 tindices
-> Tensor v'7 tindices
-> Tensor Build t
xlaConv' OpParams
forall a. a -> a
id
xlaConv' :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 t
            tindices . (OneOf '[(Data.Complex.Complex Double),
                                (Data.Complex.Complex Float), Data.Int.Int16,
                                Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                Data.Word.Word16, Data.Word.Word32,
                                Data.Word.Word64, Data.Word.Word8, Double,
                                Float] t, OneOf '[Data.Int.Int32,
                                                  Data.Int.Int64] tindices) =>
            OpParams ->
            ByteString -- ^ __dimension_numbers__: a serialized xla::ConvolutionDimensionNumbers proto.
            -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
            -> Tensor v'1 t -- ^ __lhs__: the input tensor
            -> Tensor v'2 t -- ^ __rhs__: the kernel tensor
            -> Tensor v'3 tindices -- ^ __window_strides__: the inter-window strides
            -> Tensor v'4 tindices -- ^ __padding__: the padding to apply at the start and end of each input dimensions
            -> Tensor v'5 tindices -- ^ __lhs_dilation__: dilation to apply between input elements
            -> Tensor v'6 tindices -- ^ __rhs_dilation__: dilation to apply between kernel elements
            -> Tensor v'7 tindices -- ^ __feature_group_count__: number of feature groups for grouped convolution.
            -> Tensor Build t -- ^ __output__
xlaConv' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor v'6 tindices
-> Tensor v'7 tindices
-> Tensor Build t
xlaConv' op'options :: OpParams
op'options dimension_numbers :: ByteString
dimension_numbers precision_config :: ByteString
precision_config lhs :: Tensor v'1 t
lhs rhs :: Tensor v'2 t
rhs window_strides :: Tensor v'3 tindices
window_strides
         padding :: Tensor v'4 tindices
padding lhs_dilation :: Tensor v'5 tindices
lhs_dilation rhs_dilation :: Tensor v'6 tindices
rhs_dilation
         feature_group_count :: Tensor v'7 tindices
feature_group_count | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
lhs,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
window_strides,
                                                             Tensor v'4 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tindices
padding,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
lhs_dilation,
                                                             Tensor v'6 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'6 tindices
rhs_dilation,
                                                             Tensor v'7 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'7 tindices
feature_group_count]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaConv"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dimension_numbers" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
dimension_numbers
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "precision_config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
precision_config
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "lhs" description: "the input tensor" type_attr: "T"
}
input_arg {
  name: "rhs" description: "the kernel tensor" type_attr: "T"
}
input_arg {
  name: "window_strides"
  description: "the inter-window strides"
  type_attr: "Tindices"
}
input_arg {
  name: "padding"
  description: "the padding to apply at the start and end of each input dimensions"
  type_attr: "Tindices"
}
input_arg {
  name: "lhs_dilation"
  description: "dilation to apply between input elements"
  type_attr: "Tindices"
}
input_arg {
  name: "rhs_dilation"
  description: "dilation to apply between kernel elements"
  type_attr: "Tindices"
}
input_arg {
  name: "feature_group_count"
  description: "number of feature groups for grouped convolution."
  type_attr: "Tindices"
}
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
attr {
  name: "dimension_numbers"
  type: "string"
  description: "a serialized xla::ConvolutionDimensionNumbers proto."
}
attr {
  name: "precision_config"
  type: "string"
  description: "a serialized xla::PrecisionConfig proto."
}
-}
-- | Takes the packed uint32 input and unpacks the input to uint8 to do
--
-- Dequantization on device.
xlaDequantize :: forall v'1 m' . (MonadBuild m') =>
                 Float -- ^ __max_range__: The maximum scalar value possibly produced for the input.
                 -> Float -- ^ __min_range__: The minimum scalar value possibly produced for the input.
                 -> ByteString -- ^ __mode__: String to determine the dequantize mode in {"MIN_COMBINED", "MIN_FIRST", "SCALED"}.
                 -> Bool -- ^ __transpose_output__: Boolean to determine if output is transposed. transpose_output
                         -- is faster when input is large and rank of input is higher than 1.
                 -> Tensor v'1 Data.Word.Word32 -- ^ __input__: Input tensors whose types is uint32, shape is [d0, ..., dn].
                 -> m' (Tensor Value Data.Word.Word16) -- ^ __output__: Output tensors whose types is bloat16. If transpose_output is true,
                 -- output shape is [dn * 4, dn-1, ..., d1, d0]. If transpose_output
                 -- is false, output shape is [d0,..., dn * 4].
xlaDequantize :: Float
-> Float
-> ByteString
-> Bool
-> Tensor v'1 Word32
-> m' (Tensor Value Word16)
xlaDequantize = OpParams
-> Float
-> Float
-> ByteString
-> Bool
-> Tensor v'1 Word32
-> m' (Tensor Value Word16)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams
-> Float
-> Float
-> ByteString
-> Bool
-> Tensor v'1 Word32
-> m' (Tensor Value Word16)
xlaDequantize' OpParams
forall a. a -> a
id
xlaDequantize' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                  Float -- ^ __max_range__: The maximum scalar value possibly produced for the input.
                  -> Float -- ^ __min_range__: The minimum scalar value possibly produced for the input.
                  -> ByteString -- ^ __mode__: String to determine the dequantize mode in {"MIN_COMBINED", "MIN_FIRST", "SCALED"}.
                  -> Bool -- ^ __transpose_output__: Boolean to determine if output is transposed. transpose_output
                          -- is faster when input is large and rank of input is higher than 1.
                  -> Tensor v'1 Data.Word.Word32 -- ^ __input__: Input tensors whose types is uint32, shape is [d0, ..., dn].
                  -> m' (Tensor Value Data.Word.Word16) -- ^ __output__: Output tensors whose types is bloat16. If transpose_output is true,
                  -- output shape is [dn * 4, dn-1, ..., d1, d0]. If transpose_output
                  -- is false, output shape is [d0,..., dn * 4].
xlaDequantize' :: OpParams
-> Float
-> Float
-> ByteString
-> Bool
-> Tensor v'1 Word32
-> m' (Tensor Value Word16)
xlaDequantize' op'options :: OpParams
op'options max_range :: Float
max_range min_range :: Float
min_range mode :: ByteString
mode transpose_output :: Bool
transpose_output
               input :: Tensor v'1 Word32
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Word16) -> m' (Tensor Value Word16)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Word16) -> m' (Tensor Value Word16))
-> Build (Tensor Value Word16) -> m' (Tensor Value Word16)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Word32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Word32
input]
        [Int64] -> OpDef -> Build (Tensor Value Word16)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "XlaDequantize"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_range" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
max_range
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "min_range" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
min_range
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "mode" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
mode
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "transpose_output" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
transpose_output
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input"
  description: "Input tensors whose types is uint32, shape is [d0, ..., dn]."
  type: DT_UINT32
}
output_arg {
  name: "output"
  description: "Output tensors whose types is bloat16. If transpose_output is true,\noutput shape is [dn * 4, dn-1, ..., d1, d0]. If transpose_output\nis false, output shape is [d0,..., dn * 4]."
  type: DT_BFLOAT16
}
attr {
  name: "min_range"
  type: "float"
  description: "The minimum scalar value possibly produced for the input."
}
attr {
  name: "max_range"
  type: "float"
  description: "The maximum scalar value possibly produced for the input."
}
attr {
  name: "mode"
  type: "string"
  description: "String to determine the dequantize mode in {\"MIN_COMBINED\", \"MIN_FIRST\", \"SCALED\"}."
}
attr {
  name: "transpose_output"
  type: "bool"
  description: "Boolean to determine if output is transposed. transpose_output\nis faster when input is large and rank of input is higher than 1."
}
-}
-- | Wraps the XLA DotGeneral operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral
-- .
xlaDot :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int16, Data.Int.Int32,
                                      Data.Int.Int64, Data.Int.Int8,
                                      Data.Word.Word16, Data.Word.Word32,
                                      Data.Word.Word64, Data.Word.Word8, Double,
                                      Float] t) =>
          ByteString -- ^ __dimension_numbers__: a serialized xla::DotDimensionNumbers proto.
          -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
          -> Tensor v'1 t -- ^ __lhs__: the LHS tensor
          -> Tensor v'2 t -- ^ __rhs__: the RHS tensor
          -> Tensor Build t -- ^ __output__
xlaDot :: ByteString
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaDot = OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor Build t
xlaDot' OpParams
forall a. a -> a
id
xlaDot' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int16, Data.Int.Int32,
                                       Data.Int.Int64, Data.Int.Int8,
                                       Data.Word.Word16, Data.Word.Word32,
                                       Data.Word.Word64, Data.Word.Word8,
                                       Double, Float] t) => OpParams ->
           ByteString -- ^ __dimension_numbers__: a serialized xla::DotDimensionNumbers proto.
           -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
           -> Tensor v'1 t -- ^ __lhs__: the LHS tensor
           -> Tensor v'2 t -- ^ __rhs__: the RHS tensor
           -> Tensor Build t -- ^ __output__
xlaDot' :: OpParams
-> ByteString
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor Build t
xlaDot' op'options :: OpParams
op'options dimension_numbers :: ByteString
dimension_numbers precision_config :: ByteString
precision_config lhs :: Tensor v'1 t
lhs
        rhs :: Tensor v'2 t
rhs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
lhs,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
rhs]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaDot"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dimension_numbers" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
dimension_numbers
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "precision_config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
precision_config
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "lhs" description: "the LHS tensor" type_attr: "T"
}
input_arg {
  name: "rhs" description: "the RHS tensor" type_attr: "T"
}
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "dimension_numbers"
  type: "string"
  description: "a serialized xla::DotDimensionNumbers proto."
}
attr {
  name: "precision_config"
  type: "string"
  description: "a serialized xla::PrecisionConfig proto."
}
-}
-- | Wraps the XLA DynamicSlice operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice
-- .
-- 
-- DynamicSlice extracts a sub-array from the input array at dynamic
-- start_indices. The size of the slice in each dimension is passed in
-- size_indices, which specify the end point of exclusive slice intervals in each
-- dimension -- [start, start + size). The shape of start_indices must have rank 1,
-- with dimension size equal to the rank of operand.
xlaDynamicSlice :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tindices) =>
                   Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
                   -> Tensor v'2 tindices -- ^ __start_indices__: List of N integers containing the slice size for each
                                          -- dimension. Each value must be strictly greater than zero, and start + size
                                          -- must be less than or equal to the size of the dimension to avoid
                                          -- implementation defined behavior.
                   -> Tensor v'3 tindices -- ^ __size_indices__
                   -> Tensor Build t -- ^ __output__
xlaDynamicSlice :: Tensor v'1 t
-> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
xlaDynamicSlice = OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindex.
(TensorType t, OneOf '[Int32, Int64] tindex) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 tindex
-> Tensor v'3 tindex
-> Tensor Build t
xlaDynamicSlice' OpParams
forall a. a -> a
id
xlaDynamicSlice' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                     OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64] tindices) =>
                    OpParams ->
                    Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
                    -> Tensor v'2 tindices -- ^ __start_indices__: List of N integers containing the slice size for each
                                           -- dimension. Each value must be strictly greater than zero, and start + size
                                           -- must be less than or equal to the size of the dimension to avoid
                                           -- implementation defined behavior.
                    -> Tensor v'3 tindices -- ^ __size_indices__
                    -> Tensor Build t -- ^ __output__
xlaDynamicSlice' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
xlaDynamicSlice' op'options :: OpParams
op'options input :: Tensor v'1 t
input start_indices :: Tensor v'2 tindices
start_indices
                 size_indices :: Tensor v'3 tindices
size_indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
start_indices,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
size_indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaDynamicSlice"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "A `Tensor` of type T." type_attr: "T"
}
input_arg {
  name: "start_indices"
  description: "List of N integers containing the slice size for each\ndimension. Each value must be strictly greater than zero, and start + size\nmust be less than or equal to the size of the dimension to avoid\nimplementation defined behavior."
  type_attr: "Tindices"
}
input_arg { name: "size_indices" type_attr: "Tindices" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | Wraps the XLA DynamicUpdateSlice operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice
-- .
-- 
-- XlaDynamicUpdateSlice generates a result which is the value of the `input`
-- operand, with a slice update overwritten at `indices`. The shape of `update`
-- determines the shape of the sub-array of the result which is updated. The shape
-- of indices must be rank == 1, with dimension size equal to the rank of `input`.
-- 
-- Handling of out-of-bounds slice indices is implementation-defined.
xlaDynamicUpdateSlice :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                          OneOf '[Data.Int.Int32,
                                                                  Data.Int.Int64] tindices) =>
                         Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
                         -> Tensor v'2 t -- ^ __update__: A `Tensor` of type T. Same rank as `input`.
                         -> Tensor v'3 tindices -- ^ __indices__: A vector of indices into `input`. Must have length equal to the rank of
                                                -- `input`.
                         -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaDynamicUpdateSlice :: Tensor v'1 t
-> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
xlaDynamicUpdateSlice = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices.
(TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
xlaDynamicUpdateSlice' OpParams
forall a. a -> a
id
xlaDynamicUpdateSlice' :: forall v'1 v'2 v'3 t tindices . (TensorType t,
                                                           OneOf '[Data.Int.Int32,
                                                                   Data.Int.Int64] tindices) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
                          -> Tensor v'2 t -- ^ __update__: A `Tensor` of type T. Same rank as `input`.
                          -> Tensor v'3 tindices -- ^ __indices__: A vector of indices into `input`. Must have length equal to the rank of
                                                 -- `input`.
                          -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaDynamicUpdateSlice' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor Build t
xlaDynamicUpdateSlice' op'options :: OpParams
op'options input :: Tensor v'1 t
input update :: Tensor v'2 t
update indices :: Tensor v'3 tindices
indices | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
update,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
indices]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaDynamicUpdateSlice"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "A `Tensor` of type T." type_attr: "T"
}
input_arg {
  name: "update"
  description: "A `Tensor` of type T. Same rank as `input`."
  type_attr: "T"
}
input_arg {
  name: "indices"
  description: "A vector of indices into `input`. Must have length equal to the rank of\n`input`."
  type_attr: "Tindices"
}
output_arg {
  name: "output" description: "A `Tensor` of type T." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | An op which supports basic einsum op with 2 inputs and 1 output.
--
-- This op has better TPU performance since it doesn't have explicitly reshape and
-- transpose operations as tf.einsum does.
xlaEinsum :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Float),
                                         Data.Word.Word16, Float] t) =>
             ByteString -- ^ __equation__
             -> Tensor v'1 t -- ^ __a__
             -> Tensor v'2 t -- ^ __b__
             -> Tensor Build t -- ^ __product__
xlaEinsum :: ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaEinsum = OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Float, Word16, Float] t =>
OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaEinsum' OpParams
forall a. a -> a
id
xlaEinsum' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Float),
                                          Data.Word.Word16, Float] t) =>
              OpParams ->
              ByteString -- ^ __equation__
              -> Tensor v'1 t -- ^ __a__
              -> Tensor v'2 t -- ^ __b__
              -> Tensor Build t -- ^ __product__
xlaEinsum' :: OpParams
-> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlaEinsum' op'options :: OpParams
op'options equation :: ByteString
equation a :: Tensor v'1 t
a b :: Tensor v'2 t
b | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaEinsum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "equation" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
equation
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
output_arg { name: "product" type_attr: "T" }
attr { name: "equation" type: "string" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_COMPLEX64 type: DT_BFLOAT16 type: DT_FLOAT }
  }
}
-}
-- | Wraps the XLA Gather operator documented at
--
--   https://www.tensorflow.org/xla/operation_semantics#gather
xlaGather :: forall v'1 v'2 v'3 t
             tindices . (OneOf '[(Data.Complex.Complex Double),
                                 (Data.Complex.Complex Float), Data.Int.Int16,
                                 Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                 Data.Word.Word16, Data.Word.Word32,
                                 Data.Word.Word64, Data.Word.Word8, Double,
                                 Float] t, OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64] tindices) =>
             ByteString -- ^ __dimension_numbers__: A serialized xla::GatherDimensionNumbers proto.
             -> Bool -- ^ __indices_are_sorted__: Boolean indicating if the indices are sorted.
             -> Tensor v'1 t -- ^ __operand__: The array we're gathering from.
             -> Tensor v'2 tindices -- ^ __start_indices__: Array containing the starting indices of the slices we gather.
             -> Tensor v'3 tindices -- ^ __slice_sizes__: slice_sizes[i] is the bounds for the slice on dimension i.
             -> Tensor Build t -- ^ __output__
xlaGather :: ByteString
-> Bool
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
xlaGather = OpParams
-> ByteString
-> Bool
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t tindices.
(OneOf
   '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
     Word32, Word64, Word8, Double, Float]
   t,
 OneOf '[Int32, Int64] tindices) =>
OpParams
-> ByteString
-> Bool
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
xlaGather' OpParams
forall a. a -> a
id
xlaGather' :: forall v'1 v'2 v'3 t
              tindices . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t, OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64] tindices) =>
              OpParams ->
              ByteString -- ^ __dimension_numbers__: A serialized xla::GatherDimensionNumbers proto.
              -> Bool -- ^ __indices_are_sorted__: Boolean indicating if the indices are sorted.
              -> Tensor v'1 t -- ^ __operand__: The array we're gathering from.
              -> Tensor v'2 tindices -- ^ __start_indices__: Array containing the starting indices of the slices we gather.
              -> Tensor v'3 tindices -- ^ __slice_sizes__: slice_sizes[i] is the bounds for the slice on dimension i.
              -> Tensor Build t -- ^ __output__
xlaGather' :: OpParams
-> ByteString
-> Bool
-> Tensor v'1 t
-> Tensor v'2 tindices
-> Tensor v'3 tindices
-> Tensor Build t
xlaGather' op'options :: OpParams
op'options dimension_numbers :: ByteString
dimension_numbers indices_are_sorted :: Bool
indices_are_sorted operand :: Tensor v'1 t
operand start_indices :: Tensor v'2 tindices
start_indices
           slice_sizes :: Tensor v'3 tindices
slice_sizes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
operand,
                                                             Tensor v'2 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 tindices
start_indices,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
slice_sizes]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaGather"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dimension_numbers" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
dimension_numbers
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "indices_are_sorted" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
indices_are_sorted
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "operand"
  description: "The array we\'re gathering from."
  type_attr: "T"
}
input_arg {
  name: "start_indices"
  description: "Array containing the starting indices of the slices we gather."
  type_attr: "Tindices"
}
input_arg {
  name: "slice_sizes"
  description: "slice_sizes[i] is the bounds for the slice on dimension i."
  type_attr: "Tindices"
}
output_arg { name: "output" type_attr: "T" }
attr {
  name: "dimension_numbers"
  type: "string"
  description: "A serialized xla::GatherDimensionNumbers proto."
}
attr {
  name: "indices_are_sorted"
  type: "bool"
  description: "Boolean indicating if the indices are sorted."
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | Wraps the XLA Sort operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#sort
-- .
-- 
-- Sorts a tensor. Currently only sorts in ascending order are supported.
xlaKeyValueSort :: forall v'1 v'2 k v . (OneOf '[Data.Int.Int16, Data.Int.Int32,
                                                 Data.Int.Int64, Data.Int.Int8,
                                                 Data.Word.Word16,
                                                 Data.Word.Word32,
                                                 Data.Word.Word64,
                                                 Data.Word.Word8, Double,
                                                 Float] k, TensorType v) =>
                   Tensor v'1 k -- ^ __keys__: A `Tensor` of type K.
                   -> Tensor v'2 v -- ^ __values__: A `Tensor` of type V.
                   -> (Tensor Build k, Tensor Build v)
                   -- ^ (__sorted_keys__, __sorted_values__)
                   --
                   -- * __sorted_keys__: A `Tensor` of type K.
                   --
                   -- * __sorted_values__: A `Tensor` of type V.
xlaKeyValueSort :: Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
xlaKeyValueSort = OpParams
-> Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
forall (v'1 :: * -> *) (v'2 :: * -> *) k v.
(OneOf
   '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double,
     Float]
   k,
 TensorType v) =>
OpParams
-> Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
xlaKeyValueSort' OpParams
forall a. a -> a
id
xlaKeyValueSort' :: forall v'1 v'2 k v . (OneOf '[Data.Int.Int16,
                                                  Data.Int.Int32,
                                                  Data.Int.Int64, Data.Int.Int8,
                                                  Data.Word.Word16,
                                                  Data.Word.Word32,
                                                  Data.Word.Word64,
                                                  Data.Word.Word8, Double,
                                                  Float] k, TensorType v) =>
                    OpParams ->
                    Tensor v'1 k -- ^ __keys__: A `Tensor` of type K.
                    -> Tensor v'2 v -- ^ __values__: A `Tensor` of type V.
                    -> (Tensor Build k, Tensor Build v)
                    -- ^ (__sorted_keys__, __sorted_values__)
                    --
                    -- * __sorted_keys__: A `Tensor` of type K.
                    --
                    -- * __sorted_values__: A `Tensor` of type V.
xlaKeyValueSort' :: OpParams
-> Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
xlaKeyValueSort' op'options :: OpParams
op'options keys :: Tensor v'1 k
keys values :: Tensor v'2 v
values | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build k, Tensor Build v)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build k, Tensor Build v))
-> Build OpDef -> (Tensor Build k, Tensor Build v)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 k -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 k
keys,
                                                             Tensor v'2 v -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 v
values]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaKeyValueSort"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "K" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ k -> DataType
forall a. TensorType a => a -> DataType
tensorType (k
forall a. HasCallStack => a
undefined :: k)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "V" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ v -> DataType
forall a. TensorType a => a -> DataType
tensorType (v
forall a. HasCallStack => a
undefined :: v)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "keys" description: "A `Tensor` of type K." type_attr: "K"
}
input_arg {
  name: "values" description: "A `Tensor` of type V." type_attr: "V"
}
output_arg {
  name: "sorted_keys"
  description: "A `Tensor` of type K."
  type_attr: "K"
}
output_arg {
  name: "sorted_values"
  description: "A `Tensor` of type V."
  type_attr: "V"
}
attr {
  name: "K"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_INT64
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
attr { name: "V" type: "type" }
-}
-- | Wraps the XLA Pad operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#pad
-- .
xlaPad :: forall v'1 v'2 v'3 v'4 v'5 t tindices . (TensorType t,
                                                   OneOf '[Data.Int.Int32,
                                                           Data.Int.Int64] tindices) =>
          Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
          -> Tensor v'2 t -- ^ __padding_value__: A scalar `Tensor` of type T.
          -> Tensor v'3 tindices -- ^ __padding_low__: the padding to apply at the start of each input dimensions
          -> Tensor v'4 tindices -- ^ __padding_high__: the padding to apply at the end of each input dimension.
          -> Tensor v'5 tindices -- ^ __padding_interior__: the padding to apply between each input element.
          -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaPad :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor Build t
xlaPad = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) t tindices.
(TensorType t, OneOf '[Int32, Int64] tindices) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor Build t
xlaPad' OpParams
forall a. a -> a
id
xlaPad' :: forall v'1 v'2 v'3 v'4 v'5 t tindices . (TensorType t,
                                                    OneOf '[Data.Int.Int32,
                                                            Data.Int.Int64] tindices) =>
           OpParams ->
           Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
           -> Tensor v'2 t -- ^ __padding_value__: A scalar `Tensor` of type T.
           -> Tensor v'3 tindices -- ^ __padding_low__: the padding to apply at the start of each input dimensions
           -> Tensor v'4 tindices -- ^ __padding_high__: the padding to apply at the end of each input dimension.
           -> Tensor v'5 tindices -- ^ __padding_interior__: the padding to apply between each input element.
           -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaPad' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 tindices
-> Tensor v'4 tindices
-> Tensor v'5 tindices
-> Tensor Build t
xlaPad' op'options :: OpParams
op'options input :: Tensor v'1 t
input padding_value :: Tensor v'2 t
padding_value padding_low :: Tensor v'3 tindices
padding_low padding_high :: Tensor v'4 tindices
padding_high
        padding_interior :: Tensor v'5 tindices
padding_interior | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
padding_value,
                                                             Tensor v'3 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 tindices
padding_low,
                                                             Tensor v'4 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 tindices
padding_high,
                                                             Tensor v'5 tindices -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 tindices
padding_interior]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaPad"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tindices" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tindices -> DataType
forall a. TensorType a => a -> DataType
tensorType (tindices
forall a. HasCallStack => a
undefined :: tindices)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "A `Tensor` of type T." type_attr: "T"
}
input_arg {
  name: "padding_value"
  description: "A scalar `Tensor` of type T."
  type_attr: "T"
}
input_arg {
  name: "padding_low"
  description: "the padding to apply at the start of each input dimensions"
  type_attr: "Tindices"
}
input_arg {
  name: "padding_high"
  description: "the padding to apply at the end of each input dimension."
  type_attr: "Tindices"
}
input_arg {
  name: "padding_interior"
  description: "the padding to apply between each input element."
  type_attr: "Tindices"
}
output_arg {
  name: "output" description: "A `Tensor` of type T." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "Tindices"
  type: "type"
  allowed_values { list { type: DT_INT32 type: DT_INT64 } }
}
-}
-- | Receives the named tensor from another XLA computation. Wraps the XLA Recv
--
-- operator documented at
--  https://www.tensorflow.org/performance/xla/operation_semantics#recv .
xlaRecv :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
           Shape -- ^ __shape__: The shape of the tensor.
           -> ByteString -- ^ __tensor_name__: A string key that identifies the channel.
           -> m' (Tensor Value dtype) -- ^ __tensor__: The tensor to receive.
xlaRecv :: Shape -> ByteString -> m' (Tensor Value dtype)
xlaRecv = OpParams -> Shape -> ByteString -> m' (Tensor Value dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> ByteString -> m' (Tensor Value dtype)
xlaRecv' OpParams
forall a. a -> a
id
xlaRecv' :: forall dtype m' . (MonadBuild m', TensorType dtype) => OpParams ->
            Shape -- ^ __shape__: The shape of the tensor.
            -> ByteString -- ^ __tensor_name__: A string key that identifies the channel.
            -> m' (Tensor Value dtype) -- ^ __tensor__: The tensor to receive.
xlaRecv' :: OpParams -> Shape -> ByteString -> m' (Tensor Value dtype)
xlaRecv' op'options :: OpParams
op'options shape :: Shape
shape tensor_name :: ByteString
tensor_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "XlaRecv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "tensor"
  description: "The tensor to receive."
  type_attr: "dtype"
}
attr {
  name: "dtype" type: "type" description: "The type of the tensor."
}
attr {
  name: "tensor_name"
  type: "string"
  description: "A string key that identifies the channel."
}
attr {
  name: "shape" type: "shape" description: "The shape of the tensor."
}
-}
-- | Replica ID.
xlaReplicaId :: Tensor Build Data.Int.Int32 -- ^ __id__
xlaReplicaId :: Tensor Build Int32
xlaReplicaId = OpParams -> Tensor Build Int32
xlaReplicaId' OpParams
forall a. a -> a
id
xlaReplicaId' :: OpParams ->
                 Tensor Build Data.Int.Int32 -- ^ __id__
xlaReplicaId' :: OpParams -> Tensor Build Int32
xlaReplicaId' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build Int32
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Int32)
-> Build OpDef -> Tensor Build Int32
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaReplicaId"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "id" type: DT_INT32 }
-}
-- | Computes the eigen decomposition of a batch of self-adjoint matrices
--
-- (Note: Only real inputs are supported).
-- 
-- Computes the eigenvalues and eigenvectors of the innermost N-by-N matrices in
-- tensor such that tensor[...,:,:] * v[..., :,i] = e[..., i] * v[...,:,i], for
-- i=0...N-1.
xlaSelfAdjointEig :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                             (Data.Complex.Complex Float),
                                             Data.Int.Int16, Data.Int.Int32,
                                             Data.Int.Int64, Data.Int.Int8,
                                             Data.Word.Word16, Data.Word.Word32,
                                             Data.Word.Word64, Data.Word.Word8,
                                             Double, Float] t) =>
                     Float -- ^ __epsilon__: the tolerance ratio.
                     -> Bool -- ^ __lower__: a boolean specifies whether the calculation is done with the lower
                             -- triangular part or the upper triangular part.
                     -> Data.Int.Int64 -- ^ __max_iter__: maximum number of sweep update, i.e., the whole lower triangular
                                       -- part or upper triangular part based on parameter lower. Heuristically, it has
                                       -- been argued that approximately logN sweeps are needed in practice (Ref: Golub &
                                       -- van Loan "Matrix Computation").
                     -> Tensor v'1 t -- ^ __a__: the input tensor.
                     -> (Tensor Build t, Tensor Build t) -- ^ (__w__, __v__)
                     --
                     -- * __w__: The eigenvalues in ascending order, each repeated according to its
                     -- multiplicity.
                     --
                     -- * __v__: The column v[..., :, i] is the normalized eigenvector corresponding to the
                     -- eigenvalue w[..., i].
xlaSelfAdjointEig :: Float
-> Bool
-> Int64
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t)
xlaSelfAdjointEig = OpParams
-> Float
-> Bool
-> Int64
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Float
-> Bool
-> Int64
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t)
xlaSelfAdjointEig' OpParams
forall a. a -> a
id
xlaSelfAdjointEig' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                              (Data.Complex.Complex Float),
                                              Data.Int.Int16, Data.Int.Int32,
                                              Data.Int.Int64, Data.Int.Int8,
                                              Data.Word.Word16,
                                              Data.Word.Word32,
                                              Data.Word.Word64, Data.Word.Word8,
                                              Double, Float] t) => OpParams ->
                      Float -- ^ __epsilon__: the tolerance ratio.
                      -> Bool -- ^ __lower__: a boolean specifies whether the calculation is done with the lower
                              -- triangular part or the upper triangular part.
                      -> Data.Int.Int64 -- ^ __max_iter__: maximum number of sweep update, i.e., the whole lower triangular
                                        -- part or upper triangular part based on parameter lower. Heuristically, it has
                                        -- been argued that approximately logN sweeps are needed in practice (Ref: Golub &
                                        -- van Loan "Matrix Computation").
                      -> Tensor v'1 t -- ^ __a__: the input tensor.
                      -> (Tensor Build t, Tensor Build t) -- ^ (__w__, __v__)
                      --
                      -- * __w__: The eigenvalues in ascending order, each repeated according to its
                      -- multiplicity.
                      --
                      -- * __v__: The column v[..., :, i] is the normalized eigenvector corresponding to the
                      -- eigenvalue w[..., i].
xlaSelfAdjointEig' :: OpParams
-> Float
-> Bool
-> Int64
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t)
xlaSelfAdjointEig' op'options :: OpParams
op'options epsilon :: Float
epsilon lower :: Bool
lower max_iter :: Int64
max_iter a :: Tensor v'1 t
a | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSelfAdjointEig"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "epsilon" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
epsilon
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Bool
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "lower" (forall (f :: * -> *). Identical f => LensLike' f OpDef Bool)
-> Bool -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Bool
lower
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_iter" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_iter
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "a" description: "the input tensor." type_attr: "T"
}
output_arg {
  name: "w"
  description: "The eigenvalues in ascending order, each repeated according to its\nmultiplicity."
  type_attr: "T"
}
output_arg {
  name: "v"
  description: "The column v[..., :, i] is the normalized eigenvector corresponding to the\neigenvalue w[..., i]."
  type_attr: "T"
}
attr {
  name: "lower"
  type: "bool"
  description: "a boolean specifies whether the calculation is done with the lower\ntriangular part or the upper triangular part."
}
attr {
  name: "max_iter"
  type: "int"
  description: "maximum number of sweep update, i.e., the whole lower triangular\npart or upper triangular part based on parameter lower. Heuristically, it has\nbeen argued that approximately logN sweeps are needed in practice (Ref: Golub &\nvan Loan \"Matrix Computation\")."
}
attr {
  name: "epsilon" type: "float" description: "the tolerance ratio."
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | Sends the named tensor to another XLA computation. Wraps the XLA Send operator
--
-- documented at
--  https://www.tensorflow.org/performance/xla/operation_semantics#send .
xlaSend :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
           ByteString -- ^ __tensor_name__: A string key that identifies the channel.
           -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
           -> m' (ControlNode)
xlaSend :: ByteString -> Tensor v'1 t -> m' ControlNode
xlaSend = OpParams -> ByteString -> Tensor v'1 t -> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> ByteString -> Tensor v'1 t -> m' ControlNode
xlaSend' OpParams
forall a. a -> a
id
xlaSend' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
            ByteString -- ^ __tensor_name__: A string key that identifies the channel.
            -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
            -> m' (ControlNode)
xlaSend' :: OpParams -> ByteString -> Tensor v'1 t -> m' ControlNode
xlaSend' op'options :: OpParams
op'options tensor_name :: ByteString
tensor_name tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "XlaSend"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "tensor" description: "The tensor to send." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "tensor_name"
  type: "string"
  description: "A string key that identifies the channel."
}
-}
-- | An op which shards the input based on the given sharding attribute.
xlaSharding :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __input__
               -> Tensor Build t -- ^ __output__
xlaSharding :: Tensor v'1 t -> Tensor Build t
xlaSharding = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
xlaSharding' OpParams
forall a. a -> a
id
xlaSharding' :: forall v'1 t . (TensorType t) => OpParams ->
                Tensor v'1 t -- ^ __input__
                -> Tensor Build t -- ^ __output__
xlaSharding' :: OpParams -> Tensor v'1 t -> Tensor Build t
xlaSharding' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSharding"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | Wraps the XLA Sort operator, documented at
--
--  https://www.tensorflow.org/performance/xla/operation_semantics#sort
-- .
-- 
-- Sorts a tensor. Currently only sorts in ascending order are supported.
xlaSort :: forall v'1 t . (TensorType t) =>
           Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
           -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaSort :: Tensor v'1 t -> Tensor Build t
xlaSort = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
xlaSort' OpParams
forall a. a -> a
id
xlaSort' :: forall v'1 t . (TensorType t) => OpParams ->
            Tensor v'1 t -- ^ __input__: A `Tensor` of type T.
            -> Tensor Build t -- ^ __output__: A `Tensor` of type T.
xlaSort' :: OpParams -> Tensor v'1 t -> Tensor Build t
xlaSort' op'options :: OpParams
op'options input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSort"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "A `Tensor` of type T." type_attr: "T"
}
output_arg {
  name: "output" description: "A `Tensor` of type T." type_attr: "T"
}
attr { name: "T" type: "type" }
-}
-- | An op used by XLA SPMD partitioner to switch from automatic partitioning to
--
-- manual partitioning. It annotates the input (full-shape, to be automatically
-- partitioned) with the same sharding used by manual partitioning, and outputs a
-- shard-shaped tensor to be consumed by later manually-partitioned ops. If the
-- shape is not evenly partitionable, the padding region will be masked with 0s.
xlaSpmdFullToShardShape :: forall v'1 t . (TensorType t) =>
                           ByteString -- ^ __manual_sharding__
                           -> Tensor v'1 t -- ^ __input__
                           -> Tensor Build t -- ^ __output__
xlaSpmdFullToShardShape :: ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdFullToShardShape = OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdFullToShardShape' OpParams
forall a. a -> a
id
xlaSpmdFullToShardShape' :: forall v'1 t . (TensorType t) => OpParams ->
                            ByteString -- ^ __manual_sharding__
                            -> Tensor v'1 t -- ^ __input__
                            -> Tensor Build t -- ^ __output__
xlaSpmdFullToShardShape' :: OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdFullToShardShape' op'options :: OpParams
op'options manual_sharding :: ByteString
manual_sharding input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSpmdFullToShardShape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "manual_sharding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
manual_sharding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "manual_sharding" type: "string" }
-}
-- | An op used by XLA SPMD partitioner to switch from manual partitioning to
--
-- automatic partitioning. It converts the shard-shaped, manually partitioned input
-- into full-shaped tensor to be partitioned automatically with the same sharding
-- used by manual partitioning.
xlaSpmdShardToFullShape :: forall v'1 t . (TensorType t) =>
                           Shape -- ^ __full_shape__
                           -> ByteString -- ^ __manual_sharding__
                           -> Tensor v'1 t -- ^ __input__
                           -> Tensor Build t -- ^ __output__
xlaSpmdShardToFullShape :: Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdShardToFullShape = OpParams -> Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdShardToFullShape' OpParams
forall a. a -> a
id
xlaSpmdShardToFullShape' :: forall v'1 t . (TensorType t) => OpParams ->
                            Shape -- ^ __full_shape__
                            -> ByteString -- ^ __manual_sharding__
                            -> Tensor v'1 t -- ^ __input__
                            -> Tensor Build t -- ^ __output__
xlaSpmdShardToFullShape' :: OpParams -> Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
xlaSpmdShardToFullShape' op'options :: OpParams
op'options full_shape :: Shape
full_shape manual_sharding :: ByteString
manual_sharding
                         input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSpmdShardToFullShape"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "full_shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
full_shape
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "manual_sharding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
manual_sharding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
attr { name: "manual_sharding" type: "string" }
attr { name: "full_shape" type: "shape" }
-}
-- | Computes the eigen decomposition of a batch of self-adjoint matrices
--
-- (Note: Only real inputs are supported).
-- 
-- Computes the eigenvalues and eigenvectors of the innermost M-by-N matrices in
-- tensor such that tensor[...,:,:] = u[..., :, :] * Diag(s[..., :]) * Transpose(v[...,:,:]).
xlaSvd :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                  (Data.Complex.Complex Float), Data.Int.Int16,
                                  Data.Int.Int32, Data.Int.Int64, Data.Int.Int8,
                                  Data.Word.Word16, Data.Word.Word32,
                                  Data.Word.Word64, Data.Word.Word8, Double,
                                  Float] t) =>
          Float -- ^ __epsilon__: the tolerance ratio.
          -> Data.Int.Int64 -- ^ __max_iter__: maximum number of sweep update, i.e., the whole lower triangular
                            -- part or upper triangular part based on parameter lower. Heuristically, it has
                            -- been argued that approximately log(min (M, N)) sweeps are needed in practice
                            -- (Ref: Golub & van Loan "Matrix Computation").
          -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
          -> Tensor v'1 t -- ^ __a__: the input tensor.
          -> (Tensor Build t, Tensor Build t, Tensor Build t)
          -- ^ (__s__, __u__, __v__)
          --
          -- * __s__: Singular values. The values are sorted in reverse order of magnitude, so
          -- s[..., 0] is the largest value, s[..., 1] is the second largest, etc.
          --
          -- * __u__: Left singular vectors.
          --
          -- * __v__: Right singular vectors.
xlaSvd :: Float
-> Int64
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t, Tensor Build t)
xlaSvd = OpParams
-> Float
-> Int64
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t, Tensor Build t)
forall (v'1 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word32, Word64, Word8, Double, Float]
  t =>
OpParams
-> Float
-> Int64
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t, Tensor Build t)
xlaSvd' OpParams
forall a. a -> a
id
xlaSvd' :: forall v'1 t . (OneOf '[(Data.Complex.Complex Double),
                                   (Data.Complex.Complex Float), Data.Int.Int16,
                                   Data.Int.Int32, Data.Int.Int64,
                                   Data.Int.Int8, Data.Word.Word16,
                                   Data.Word.Word32, Data.Word.Word64,
                                   Data.Word.Word8, Double, Float] t) =>
           OpParams ->
           Float -- ^ __epsilon__: the tolerance ratio.
           -> Data.Int.Int64 -- ^ __max_iter__: maximum number of sweep update, i.e., the whole lower triangular
                             -- part or upper triangular part based on parameter lower. Heuristically, it has
                             -- been argued that approximately log(min (M, N)) sweeps are needed in practice
                             -- (Ref: Golub & van Loan "Matrix Computation").
           -> ByteString -- ^ __precision_config__: a serialized xla::PrecisionConfig proto.
           -> Tensor v'1 t -- ^ __a__: the input tensor.
           -> (Tensor Build t, Tensor Build t, Tensor Build t)
           -- ^ (__s__, __u__, __v__)
           --
           -- * __s__: Singular values. The values are sorted in reverse order of magnitude, so
           -- s[..., 0] is the largest value, s[..., 1] is the second largest, etc.
           --
           -- * __u__: Left singular vectors.
           --
           -- * __v__: Right singular vectors.
xlaSvd' :: OpParams
-> Float
-> Int64
-> ByteString
-> Tensor v'1 t
-> (Tensor Build t, Tensor Build t, Tensor Build t)
xlaSvd' op'options :: OpParams
op'options epsilon :: Float
epsilon max_iter :: Int64
max_iter precision_config :: ByteString
precision_config a :: Tensor v'1 t
a | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64]
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t))
-> Build OpDef -> (Tensor Build t, Tensor Build t, Tensor Build t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "XlaSvd"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Float
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "epsilon" (forall (f :: * -> *). Identical f => LensLike' f OpDef Float)
-> Float -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Float
epsilon
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "max_iter" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
max_iter
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "precision_config" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
precision_config
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "a" description: "the input tensor." type_attr: "T"
}
output_arg {
  name: "s"
  description: "Singular values. The values are sorted in reverse order of magnitude, so\ns[..., 0] is the largest value, s[..., 1] is the second largest, etc."
  type_attr: "T"
}
output_arg {
  name: "u" description: "Left singular vectors." type_attr: "T"
}
output_arg {
  name: "v" description: "Right singular vectors." type_attr: "T"
}
attr {
  name: "max_iter"
  type: "int"
  description: "maximum number of sweep update, i.e., the whole lower triangular\npart or upper triangular part based on parameter lower. Heuristically, it has\nbeen argued that approximately log(min (M, N)) sweeps are needed in practice\n(Ref: Golub & van Loan \"Matrix Computation\")."
}
attr {
  name: "epsilon" type: "float" description: "the tolerance ratio."
}
attr {
  name: "precision_config"
  type: "string"
  description: "a serialized xla::PrecisionConfig proto."
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_UINT8
      type: DT_INT16
      type: DT_INT8
      type: DT_COMPLEX64
      type: DT_INT64
      type: DT_QINT8
      type: DT_QUINT8
      type: DT_QINT32
      type: DT_BFLOAT16
      type: DT_UINT16
      type: DT_COMPLEX128
      type: DT_HALF
      type: DT_UINT32
      type: DT_UINT64
    }
  }
}
-}
-- | 
xlog1py :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Word.Word16, Double, Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor Build t -- ^ __z__
xlog1py :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlog1py = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlog1py' OpParams
forall a. a -> a
id
xlog1py' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                        (Data.Complex.Complex Float),
                                        Data.Word.Word16, Double, Float] t) =>
            OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor Build t -- ^ __z__
xlog1py' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlog1py' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Xlog1py"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
xlogy :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                     (Data.Complex.Complex Float),
                                     Data.Word.Word16, Double, Float] t) =>
         Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __y__
         -> Tensor Build t -- ^ __z__
xlogy :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlogy = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Complex Double, Complex Float, Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlogy' OpParams
forall a. a -> a
id
xlogy' :: forall v'1 v'2 t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Word.Word16, Double, Float] t) =>
          OpParams ->
          Tensor v'1 t -- ^ __x__
          -> Tensor v'2 t -- ^ __y__
          -> Tensor Build t -- ^ __z__
xlogy' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
xlogy' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Xlogy"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | 
zerosLike :: forall v'1 t . (TensorType t) => Tensor v'1 t -- ^ __x__
             -> Tensor Build t -- ^ __y__
zerosLike :: Tensor v'1 t -> Tensor Build t
zerosLike = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor Build t
zerosLike' OpParams
forall a. a -> a
id
zerosLike' :: forall v'1 t . (TensorType t) => OpParams ->
              Tensor v'1 t -- ^ __x__
              -> Tensor Build t -- ^ __y__
zerosLike' :: OpParams -> Tensor v'1 t -> Tensor Build t
zerosLike' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ZerosLike"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | 
zeta :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => Tensor v'1 t -- ^ __x__
        -> Tensor v'2 t -- ^ __q__
        -> Tensor Build t -- ^ __z__
zeta :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
zeta = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
zeta' OpParams
forall a. a -> a
id
zeta' :: forall v'1 v'2 t . (OneOf '[Double, Float] t) => OpParams ->
         Tensor v'1 t -- ^ __x__
         -> Tensor v'2 t -- ^ __q__
         -> Tensor Build t -- ^ __z__
zeta' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
zeta' op'options :: OpParams
op'options x :: Tensor v'1 t
x q :: Tensor v'2 t
q | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
q]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "Zeta"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "q" type_attr: "T" }
output_arg { name: "z" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
-}
-- | 
zipDataset :: [DataType] -- ^ __output_types__
              -> [Tensor v'1 Variant] -- ^ __input_datasets__
              -> Tensor Build Variant -- ^ __handle__
zipDataset :: [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
zipDataset = OpParams
-> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
forall (v'1 :: * -> *).
OpParams
-> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
zipDataset' OpParams
forall a. a -> a
id
zipDataset' :: OpParams ->
               [DataType] -- ^ __output_types__
               -> [Tensor v'1 Variant] -- ^ __input_datasets__
               -> Tensor Build Variant -- ^ __handle__
zipDataset' :: OpParams
-> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
zipDataset' op'options :: OpParams
op'options output_types :: [DataType]
output_types
            input_datasets :: [Tensor v'1 Variant]
input_datasets | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("input_datasets", [Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets)])] =
    [Int64] -> Build OpDef -> Tensor Build Variant
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build Variant)
-> Build OpDef -> Tensor Build Variant
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Variant] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Variant]
input_datasets]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "ZipDataset"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "output_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
output_types
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Variant] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Variant]
input_datasets) :: Int64
{-
input_arg {
  name: "input_datasets" type: DT_VARIANT number_attr: "N"
}
output_arg { name: "handle" type: DT_VARIANT }
attr {
  name: "output_types"
  type: "list(type)"
  has_minimum: true
  minimum: 1
}
attr {
  name: "output_shapes"
  type: "list(shape)"
  has_minimum: true
  minimum: 1
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | A graph node which represents an argument to a function.
_Arg :: forall t m' . (MonadBuild m', TensorType t) =>
        Data.Int.Int64 -- ^ __index__: This argument is the index-th argument of the function.
                       -- 
                       -- Attributes for shape inference:
                       -- 1. _output_shapes: this attribute should contain a list of TensorShapeProto
                       --    describing the shape(s) of the tensor(s) this _Arg node will produce. If set,
                       --    _Arg node's shape inference function will use it as the node's output shapes.
                       -- 2. _handle_dtypes and _handle_shapes: these attributes can be set on an _Arg
                       --    node producing resource output(s). If set, value of _handle_dtypes should
                       --    contain the dtype(s) of the resource(s) and value of _handle_shapes should
                       --    contain the shape(s) of the resource(s). If both attributes are set, _Arg
                       --    node's shape inference function will use their values as the node's output
                       --    handle's type(s) and shape(s).
        -> m' (Tensor Value t) -- ^ __output__: The argument.
_Arg :: Int64 -> m' (Tensor Value t)
_Arg = OpParams -> Int64 -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Int64 -> m' (Tensor Value t)
_Arg' OpParams
forall a. a -> a
id
_Arg' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
         Data.Int.Int64 -- ^ __index__: This argument is the index-th argument of the function.
                        -- 
                        -- Attributes for shape inference:
                        -- 1. _output_shapes: this attribute should contain a list of TensorShapeProto
                        --    describing the shape(s) of the tensor(s) this _Arg node will produce. If set,
                        --    _Arg node's shape inference function will use it as the node's output shapes.
                        -- 2. _handle_dtypes and _handle_shapes: these attributes can be set on an _Arg
                        --    node producing resource output(s). If set, value of _handle_dtypes should
                        --    contain the dtype(s) of the resource(s) and value of _handle_shapes should
                        --    contain the shape(s) of the resource(s). If both attributes are set, _Arg
                        --    node's shape inference function will use their values as the node's output
                        --    handle's type(s) and shape(s).
         -> m' (Tensor Value t) -- ^ __output__: The argument.
_Arg' :: OpParams -> Int64 -> m' (Tensor Value t)
_Arg' op'options :: OpParams
op'options index :: Int64
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_Arg"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "output" description: "The argument." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "index"
  type: "int"
  description: "This argument is the index-th argument of the function.\n\nAttributes for shape inference:\n1. _output_shapes: this attribute should contain a list of TensorShapeProto\n   describing the shape(s) of the tensor(s) this _Arg node will produce. If set,\n   _Arg node\'s shape inference function will use it as the node\'s output shapes.\n2. _handle_dtypes and _handle_shapes: these attributes can be set on an _Arg\n   node producing resource output(s). If set, value of _handle_dtypes should\n   contain the dtype(s) of the resource(s) and value of _handle_shapes should\n   contain the shape(s) of the resource(s). If both attributes are set, _Arg\n   node\'s shape inference function will use their values as the node\'s output\n   handle\'s type(s) and shape(s)."
  has_minimum: true
}
-}
-- | Converts an array of tensors to a list of tensors.
_ArrayToList :: forall v'1 t out_types . (TensorType t,
                                          TensorTypes out_types) =>
                [Tensor v'1 t] -- ^ __input__
                -> TensorList (Build) out_types -- ^ __output__
_ArrayToList :: [Tensor v'1 t] -> TensorList Build out_types
_ArrayToList = OpParams -> [Tensor v'1 t] -> TensorList Build out_types
forall (v'1 :: * -> *) t (out_types :: [*]).
(TensorType t, TensorTypes out_types) =>
OpParams -> [Tensor v'1 t] -> TensorList Build out_types
_ArrayToList' OpParams
forall a. a -> a
id
_ArrayToList' :: forall v'1 t out_types . (TensorType t,
                                           TensorTypes out_types) => OpParams ->
                 [Tensor v'1 t] -- ^ __input__
                 -> TensorList (Build) out_types -- ^ __output__
_ArrayToList' :: OpParams -> [Tensor v'1 t] -> TensorList Build out_types
_ArrayToList' op'options :: OpParams
op'options
              input :: [Tensor v'1 t]
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("input", [Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input)])] =
    [Int64] -> Build OpDef -> TensorList Build out_types
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> TensorList Build out_types)
-> Build OpDef -> TensorList Build out_types
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 t]
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_ArrayToList"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "out_types" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy out_types -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy out_types
forall k (t :: k). Proxy t
Proxy :: Proxy out_types)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 t]
input) :: Int64
{-
input_arg { name: "input" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_list_attr: "out_types" }
attr { name: "T" type: "type" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "out_types" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | An op that sets up the centralized structures for a distributed TPU
--
-- system.
_ConfigureDistributedTPU :: forall v'1 m' . (MonadBuild m') =>
                            [Tensor v'1 Data.Int.Int32] -- ^ __inputs__: A scalar tensor for each host indicating how many TPU chips
                                                        -- there are on the host.
                            -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __output__: A tensor containing a TPUHostConfiguration proto serialized to
                            -- a string, containing the information necessary to initialize the chips
                            -- in a host.
_ConfigureDistributedTPU :: [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_ConfigureDistributedTPU = OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_ConfigureDistributedTPU' OpParams
forall a. a -> a
id
_ConfigureDistributedTPU' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                             [Tensor v'1 Data.Int.Int32] -- ^ __inputs__: A scalar tensor for each host indicating how many TPU chips
                                                         -- there are on the host.
                             -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __output__: A tensor containing a TPUHostConfiguration proto serialized to
                             -- a string, containing the information necessary to initialize the chips
                             -- in a host.
_ConfigureDistributedTPU' :: OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_ConfigureDistributedTPU' op'options :: OpParams
op'options
                          inputs :: [Tensor v'1 Int32]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
inputs)])] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int32]
inputs]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ConfigureDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
inputs) :: Int64
{-
input_arg {
  name: "inputs"
  description: "A scalar tensor for each host indicating how many TPU chips\nthere are on the host."
  type: DT_INT32
  number_attr: "N"
}
output_arg {
  name: "output"
  description: "A tensor containing a TPUHostConfiguration proto serialized to\na string, containing the information necessary to initialize the chips\nin a host."
  type: DT_STRING
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
attr {
  name: "enable_whole_mesh_compilations"
  type: "bool"
  default_value { b: false }
  description: "Usually the master TPU worker is the only\nworker compile ops are sent, and the master worker is the only one which\ncan execute them. Other TPU clients distribute TPU compilation across all\nthe hosts of the mesh, and setting this flag to True enables such mesh\ninitialization mode."
}
-}
-- | A graph node which represents an argument to a function.
_DeviceArg :: forall t m' . (MonadBuild m', TensorType t) =>
              Data.Int.Int64 -- ^ __index__: This argument is the index-th argument of the function.
              -> m' (Tensor Value t) -- ^ __output__: The argument.
_DeviceArg :: Int64 -> m' (Tensor Value t)
_DeviceArg = OpParams -> Int64 -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Int64 -> m' (Tensor Value t)
_DeviceArg' OpParams
forall a. a -> a
id
_DeviceArg' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
               Data.Int.Int64 -- ^ __index__: This argument is the index-th argument of the function.
               -> m' (Tensor Value t) -- ^ __output__: The argument.
_DeviceArg' :: OpParams -> Int64 -> m' (Tensor Value t)
_DeviceArg' op'options :: OpParams
op'options index :: Int64
index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_DeviceArg"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "output" description: "The argument." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "index"
  type: "int"
  description: "This argument is the index-th argument of the function."
  has_minimum: true
}
-}
-- | A graph node which represents a return value of a function.
_DeviceRetval :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
                 Data.Int.Int64 -- ^ __index__: This return value is the index-th return value of the function.
                 -> Tensor v'1 t -- ^ __input__: The return value.
                 -> m' (ControlNode)
_DeviceRetval :: Int64 -> Tensor v'1 t -> m' ControlNode
_DeviceRetval = OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
_DeviceRetval' OpParams
forall a. a -> a
id
_DeviceRetval' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
                  Data.Int.Int64 -- ^ __index__: This return value is the index-th return value of the function.
                  -> Tensor v'1 t -- ^ __input__: The return value.
                  -> m' (ControlNode)
_DeviceRetval' :: OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
_DeviceRetval' op'options :: OpParams
op'options index :: Int64
index input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_DeviceRetval"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "The return value." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "index"
  type: "int"
  description: "This return value is the index-th return value of the function."
  has_minimum: true
}
-}
-- | An op that disconnects the TPUs on a host from a running distributed
--
-- TPU system.
_DisconnectHostFromDistributedTPUSystem :: forall m' . (MonadBuild m') =>
                                           m' (Tensor Value Data.Int.Int32) -- ^ __number_of_tpu_chips__: A scalar tensor containing the number of TPU
                                           -- chips on the host.
_DisconnectHostFromDistributedTPUSystem :: m' (Tensor Value Int32)
_DisconnectHostFromDistributedTPUSystem = OpParams -> m' (Tensor Value Int32)
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> m' (Tensor Value Int32)
_DisconnectHostFromDistributedTPUSystem' OpParams
forall a. a -> a
id
_DisconnectHostFromDistributedTPUSystem' :: forall m' . (MonadBuild m') =>
                                            OpParams ->
                                            m' (Tensor Value Data.Int.Int32) -- ^ __number_of_tpu_chips__: A scalar tensor containing the number of TPU
                                            -- chips on the host.
_DisconnectHostFromDistributedTPUSystem' :: OpParams -> m' (Tensor Value Int32)
_DisconnectHostFromDistributedTPUSystem' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_DisconnectHostFromDistributedTPUSystem"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "number_of_tpu_chips"
  description: "A scalar tensor containing the number of TPU\nchips on the host."
  type: DT_INT32
}
-}
-- | *NOTE*: Do not invoke this operator directly in Python. Grappler is
--
-- expected to create these operators.
_FusedBatchNormEx :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                     u . (OneOf '[Data.Word.Word16, Float] t,
                          OneOf '[Float] u) => Tensor v'1 t -- ^ __x__
                     -> Tensor v'2 u -- ^ __scale__
                     -> Tensor v'3 u -- ^ __offset__
                     -> Tensor v'4 u -- ^ __mean__
                     -> Tensor v'5 u -- ^ __variance__
                     -> [Tensor v'6 t] -- ^ __side_input__
                     -> (Tensor Build t, Tensor Build u, Tensor Build u,
                         Tensor Build u, Tensor Build u, Tensor Build u)
                     -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__, __reserve_space_3__)
                     --
                     -- * __y__
                     --
                     -- * __batch_mean__
                     --
                     -- * __batch_variance__
                     --
                     -- * __reserve_space_1__
                     --
                     -- * __reserve_space_2__
                     --
                     -- * __reserve_space_3__
_FusedBatchNormEx :: Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> [Tensor v'6 t]
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
_FusedBatchNormEx = OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> [Tensor v'6 t]
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) (v'5 :: * -> *) (v'6 :: * -> *) t u.
(OneOf '[Word16, Float] t, OneOf '[Float] u) =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> [Tensor v'6 t]
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
_FusedBatchNormEx' OpParams
forall a. a -> a
id
_FusedBatchNormEx' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t
                      u . (OneOf '[Data.Word.Word16, Float] t,
                           OneOf '[Float] u) => OpParams ->
                      Tensor v'1 t -- ^ __x__
                      -> Tensor v'2 u -- ^ __scale__
                      -> Tensor v'3 u -- ^ __offset__
                      -> Tensor v'4 u -- ^ __mean__
                      -> Tensor v'5 u -- ^ __variance__
                      -> [Tensor v'6 t] -- ^ __side_input__
                      -> (Tensor Build t, Tensor Build u, Tensor Build u,
                          Tensor Build u, Tensor Build u, Tensor Build u)
                      -- ^ (__y__, __batch_mean__, __batch_variance__, __reserve_space_1__, __reserve_space_2__, __reserve_space_3__)
                      --
                      -- * __y__
                      --
                      -- * __batch_mean__
                      --
                      -- * __batch_variance__
                      --
                      -- * __reserve_space_1__
                      --
                      -- * __reserve_space_2__
                      --
                      -- * __reserve_space_3__
_FusedBatchNormEx' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 u
-> Tensor v'3 u
-> Tensor v'4 u
-> Tensor v'5 u
-> [Tensor v'6 t]
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
_FusedBatchNormEx' op'options :: OpParams
op'options x :: Tensor v'1 t
x scale :: Tensor v'2 u
scale offset :: Tensor v'3 u
offset mean :: Tensor v'4 u
mean variance :: Tensor v'5 u
variance
                   side_input :: [Tensor v'6 t]
side_input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_side_inputs", [("side_input", [Tensor v'6 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'6 t]
side_input)])] =
    [Int64]
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef
 -> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
     Tensor Build u, Tensor Build u))
-> Build OpDef
-> (Tensor Build t, Tensor Build u, Tensor Build u, Tensor Build u,
    Tensor Build u, Tensor Build u)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 u
scale,
                                                             Tensor v'3 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 u
offset,
                                                             Tensor v'4 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 u
mean,
                                                             Tensor v'5 u -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'5 u
variance,
                                                             [Tensor v'6 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'6 t]
side_input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_FusedBatchNormEx"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "U" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ u -> DataType
forall a. TensorType a => a -> DataType
tensorType (u
forall a. HasCallStack => a
undefined :: u)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_side_inputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_side_inputs
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_side_inputs :: Int64
num_side_inputs = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'6 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'6 t]
side_input) :: Int64
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "scale" type_attr: "U" }
input_arg { name: "offset" type_attr: "U" }
input_arg { name: "mean" type_attr: "U" }
input_arg { name: "variance" type_attr: "U" }
input_arg {
  name: "side_input" type_attr: "T" number_attr: "num_side_inputs"
}
output_arg { name: "y" type_attr: "T" }
output_arg { name: "batch_mean" type_attr: "U" }
output_arg { name: "batch_variance" type_attr: "U" }
output_arg { name: "reserve_space_1" type_attr: "U" }
output_arg { name: "reserve_space_2" type_attr: "U" }
output_arg { name: "reserve_space_3" type_attr: "U" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_HALF type: DT_FLOAT } }
}
attr {
  name: "U" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
attr {
  name: "exponential_avg_factor"
  type: "float"
  default_value { f: 1.0 }
}
attr {
  name: "num_side_inputs"
  type: "int"
  default_value { i: 0 }
  has_minimum: true
}
attr {
  name: "activation_mode"
  type: "string"
  default_value { s: "Identity" }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr { name: "is_training" type: "bool" default_value { b: true } }
-}
-- | Performs a convolution followed by a specified series of operations.
--
-- The inputs to the convolution are `input` and `filter`. The series of operations
-- that follows is specified by the `fused_ops` attribute, which is a list of TF op
-- names specified as strings (e.g. "Relu"). They are performed in order, where the
-- (first) input to each op is the output of the preceding op. The first input and
-- the output of each fused_op must be of type T.
-- 
-- Currently supported fused_op combinations are: [X] and [X,A], where X is one of
-- {"BiasAdd","FusedBatchNorm"} and A is one of {"Elu","Relu","Relu6"}.
-- 
-- * The first input to op X is the Conv2D result, and the additional input(s) to X
-- are specified by `args`.
-- * If there is an op A specified, the output of op X is the input to op A, and op
-- A produces the _FusedConv2D output. Otherwise, op X produces the _FusedConv2D
-- output.
-- 
-- *NOTE*: Do not invoke this operator directly in Python. Grappler is expected to
-- create these operators.
_FusedConv2D :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                ByteString -- ^ __padding__
                -> Tensor v'1 t -- ^ __input__
                -> Tensor v'2 t -- ^ __filter__
                -> [Tensor v'3 t] -- ^ __args__
                -> Tensor Build t -- ^ __output__
_FusedConv2D :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedConv2D = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
_FusedConv2D' OpParams
forall a. a -> a
id
_FusedConv2D' :: forall v'1 v'2 v'3 t . (OneOf '[Double, Float] t) =>
                 OpParams ->
                 ByteString -- ^ __padding__
                 -> Tensor v'1 t -- ^ __input__
                 -> Tensor v'2 t -- ^ __filter__
                 -> [Tensor v'3 t] -- ^ __args__
                 -> Tensor Build t -- ^ __output__
_FusedConv2D' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
_FusedConv2D' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
              args :: [Tensor v'3 t]
args | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_args", [("args", [Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             [Tensor v'3 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t]
args]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_FusedConv2D"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_args" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_args
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_args :: Int64
num_args = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args) :: Int64
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "args" type_attr: "T" number_attr: "num_args" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values { list { type: DT_FLOAT type: DT_DOUBLE } }
}
attr { name: "num_args" type: "int" has_minimum: true }
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" s: "EXPLICIT" } }
}
attr {
  name: "explicit_paddings"
  type: "list(int)"
  default_value { list { } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "use_cudnn_on_gpu" type: "bool" default_value { b: true }
}
attr {
  name: "fused_ops" type: "list(string)" default_value { list { } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
-}
-- | 
_FusedDepthwiseConv2dNative :: forall v'1 v'2 v'3 t . (OneOf '[Data.Word.Word16,
                                                               Double,
                                                               Float] t) =>
                               ByteString -- ^ __padding__
                               -> Tensor v'1 t -- ^ __input__
                               -> Tensor v'2 t -- ^ __filter__
                               -> [Tensor v'3 t] -- ^ __args__
                               -> Tensor Build t -- ^ __output__
_FusedDepthwiseConv2dNative :: ByteString
-> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedDepthwiseConv2dNative = OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
_FusedDepthwiseConv2dNative' OpParams
forall a. a -> a
id
_FusedDepthwiseConv2dNative' :: forall v'1 v'2 v'3
                                t . (OneOf '[Data.Word.Word16, Double,
                                             Float] t) => OpParams ->
                                ByteString -- ^ __padding__
                                -> Tensor v'1 t -- ^ __input__
                                -> Tensor v'2 t -- ^ __filter__
                                -> [Tensor v'3 t] -- ^ __args__
                                -> Tensor Build t -- ^ __output__
_FusedDepthwiseConv2dNative' :: OpParams
-> ByteString
-> Tensor v'1 t
-> Tensor v'2 t
-> [Tensor v'3 t]
-> Tensor Build t
_FusedDepthwiseConv2dNative' op'options :: OpParams
op'options padding :: ByteString
padding input :: Tensor v'1 t
input filter :: Tensor v'2 t
filter
                             args :: [Tensor v'3 t]
args | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_args", [("args", [Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
filter,
                                                             [Tensor v'3 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t]
args]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_FusedDepthwiseConv2dNative"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "padding" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
padding
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_args" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_args
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_args :: Int64
num_args = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args) :: Int64
{-
input_arg { name: "input" type_attr: "T" }
input_arg { name: "filter" type_attr: "T" }
input_arg { name: "args" type_attr: "T" number_attr: "num_args" }
output_arg { name: "output" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF type: DT_BFLOAT16 type: DT_FLOAT type: DT_DOUBLE
    }
  }
}
attr { name: "num_args" type: "int" has_minimum: true }
attr { name: "strides" type: "list(int)" }
attr {
  name: "padding"
  type: "string"
  allowed_values { list { s: "SAME" s: "VALID" } }
}
attr {
  name: "data_format"
  type: "string"
  default_value { s: "NHWC" }
  allowed_values { list { s: "NHWC" s: "NCHW" } }
}
attr {
  name: "dilations"
  type: "list(int)"
  default_value { list { i: 1 i: 1 i: 1 i: 1 } }
}
attr {
  name: "fused_ops" type: "list(string)" default_value { list { } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
-}
-- | Performs a MatMul followed by a specified series of operations.
--
-- The inputs to the MatMul are specified by `a` and `b`. The series of operations
-- that follows is specified by the `fused_ops` attribute, which is a list of TF op
-- names specified as strings (e.g. "Relu"). They are performed in order, where the
-- (first) input to each op is the output of the preceding op. The first input and
-- the output of each fused_op must be of type T.
-- 
-- Currently supported fused_op combinations are: ["BiasAdd"] and ["BiasAdd",A],
-- where A is one of {"Elu","Relu","Relu6"}.
-- 
-- * The first input to BiasAdd is the Conv2D result, and the additional BiasAdd
-- input is specified by `args`.
-- * If there is an op A specified, the output of the BiasAdd is the input to op A,
-- and op A produces the _FusedConv2D output. Otherwise, the BiasAdd produces the
-- _FusedConv2D output.
-- 
-- *NOTE*: Do not invoke this operator directly in Python. Grappler is
-- expected to create these operators.
_FusedMatMul :: forall v'1 v'2 v'3 t . (OneOf '[Float] t) =>
                Tensor v'1 t -- ^ __a__
                -> Tensor v'2 t -- ^ __b__
                -> [Tensor v'3 t] -- ^ __args__
                -> Tensor Build t -- ^ __product__
_FusedMatMul :: Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedMatMul = OpParams
-> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *) t.
OneOf '[Float] t =>
OpParams
-> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedMatMul' OpParams
forall a. a -> a
id
_FusedMatMul' :: forall v'1 v'2 v'3 t . (OneOf '[Float] t) => OpParams ->
                 Tensor v'1 t -- ^ __a__
                 -> Tensor v'2 t -- ^ __b__
                 -> [Tensor v'3 t] -- ^ __args__
                 -> Tensor Build t -- ^ __product__
_FusedMatMul' :: OpParams
-> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
_FusedMatMul' op'options :: OpParams
op'options a :: Tensor v'1 t
a b :: Tensor v'2 t
b
              args :: [Tensor v'3 t]
args | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("num_args", [("args", [Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args)])] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
a,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
b,
                                                             [Tensor v'3 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'3 t]
args]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_FusedMatMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_args" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_args
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    num_args :: Int64
num_args = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'3 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'3 t]
args) :: Int64
{-
input_arg { name: "a" type_attr: "T" }
input_arg { name: "b" type_attr: "T" }
input_arg { name: "args" type_attr: "T" number_attr: "num_args" }
output_arg { name: "product" type_attr: "T" }
attr {
  name: "transpose_a" type: "bool" default_value { b: false }
}
attr {
  name: "transpose_b" type: "bool" default_value { b: false }
}
attr {
  name: "T" type: "type" allowed_values { list { type: DT_FLOAT } }
}
attr { name: "num_args" type: "int" has_minimum: true }
attr {
  name: "fused_ops" type: "list(string)" default_value { list { } }
}
attr { name: "epsilon" type: "float" default_value { f: 1.0e-4 } }
-}
-- | Cast x of type SrcT to y of DstT.
--
-- _HostCast requires its input and produces its output in host memory.
_HostCast :: forall v'1 srcT dstT . (TensorType srcT, TensorType dstT) =>
             Tensor v'1 srcT -- ^ __x__
             -> Tensor Build dstT -- ^ __y__
_HostCast :: Tensor v'1 srcT -> Tensor Build dstT
_HostCast = OpParams -> Tensor v'1 srcT -> Tensor Build dstT
forall (v'1 :: * -> *) srcT dstT.
(TensorType srcT, TensorType dstT) =>
OpParams -> Tensor v'1 srcT -> Tensor Build dstT
_HostCast' OpParams
forall a. a -> a
id
_HostCast' :: forall v'1 srcT dstT . (TensorType srcT, TensorType dstT) =>
              OpParams ->
              Tensor v'1 srcT -- ^ __x__
              -> Tensor Build dstT -- ^ __y__
_HostCast' :: OpParams -> Tensor v'1 srcT -> Tensor Build dstT
_HostCast' op'options :: OpParams
op'options x :: Tensor v'1 srcT
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build dstT
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build dstT)
-> Build OpDef -> Tensor Build dstT
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 srcT -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 srcT
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_HostCast"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "SrcT" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ srcT -> DataType
forall a. TensorType a => a -> DataType
tensorType (srcT
forall a. HasCallStack => a
undefined :: srcT)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "DstT" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dstT -> DataType
forall a. TensorType a => a -> DataType
tensorType (dstT
forall a. HasCallStack => a
undefined :: dstT)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "SrcT" }
output_arg { name: "y" type_attr: "DstT" }
attr { name: "SrcT" type: "type" }
attr { name: "DstT" type: "type" }
attr { name: "Truncate" type: "bool" default_value { b: false } }
-}
-- | Receives the named tensor from send_device on recv_device.
--
-- _HostRecv produces its output on host memory whereas _Recv produces its
-- output on device memory.
_HostRecv :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
             ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
             -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
             -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
             -> ByteString -- ^ __tensor_name__: The name of the tensor to receive.
             -> m' (Tensor Value tensor_type) -- ^ __tensor__: The tensor to receive.
_HostRecv :: ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_HostRecv = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
forall tensor_type (m' :: * -> *).
(MonadBuild m', TensorType tensor_type) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_HostRecv' OpParams
forall a. a -> a
id
_HostRecv' :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
              OpParams ->
              ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
              -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
              -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
              -> ByteString -- ^ __tensor_name__: The name of the tensor to receive.
              -> m' (Tensor Value tensor_type) -- ^ __tensor__: The tensor to receive.
_HostRecv' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_HostRecv' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation
           tensor_name :: ByteString
tensor_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type))
-> Build (Tensor Value tensor_type)
-> m' (Tensor Value tensor_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value tensor_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_HostRecv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tensor_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (tensor_type
forall a. HasCallStack => a
undefined :: tensor_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "tensor"
  description: "The tensor to receive."
  type_attr: "tensor_type"
}
attr { name: "tensor_type" type: "type" }
attr {
  name: "tensor_name"
  type: "string"
  description: "The name of the tensor to receive."
}
attr {
  name: "send_device"
  type: "string"
  description: "The name of the device sending the tensor."
}
attr {
  name: "send_device_incarnation"
  type: "int"
  description: "The current incarnation of send_device."
}
attr {
  name: "recv_device"
  type: "string"
  description: "The name of the device receiving the tensor."
}
attr {
  name: "client_terminated"
  type: "bool"
  default_value { b: false }
  description: "If set to true, this indicates that the node was added\nto the graph as a result of a client-side feed or fetch of Tensor data,\nin which case the corresponding send or recv is expected to be managed\nlocally by the caller."
}
-}
-- | Sends the named tensor from send_device to recv_device.
--
-- _HostSend requires its input on host memory whereas _Send requires its
-- input on device memory.
_HostSend :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
             ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
             -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
             -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
             -> ByteString -- ^ __tensor_name__: The name of the tensor to send.
             -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
             -> m' (ControlNode)
_HostSend :: ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_HostSend = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_HostSend' OpParams
forall a. a -> a
id
_HostSend' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
              ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
              -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
              -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
              -> ByteString -- ^ __tensor_name__: The name of the tensor to send.
              -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
              -> m' (ControlNode)
_HostSend' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_HostSend' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation
           tensor_name :: ByteString
tensor_name tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_HostSend"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "tensor" description: "The tensor to send." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "tensor_name"
  type: "string"
  description: "The name of the tensor to send."
}
attr {
  name: "send_device"
  type: "string"
  description: "The name of the device sending the tensor."
}
attr {
  name: "send_device_incarnation"
  type: "int"
  description: "The current incarnation of send_device."
}
attr {
  name: "recv_device"
  type: "string"
  description: "The name of the device receiving the tensor."
}
attr {
  name: "client_terminated"
  type: "bool"
  default_value { b: false }
  description: "If set to true, this indicates that the node was added\nto the graph as a result of a client-side feed or fetch of Tensor data,\nin which case the corresponding send or recv is expected to be managed\nlocally by the caller."
}
-}
-- | An op that connects each chip on the host to a centralized UberDriver to allow
--
-- them to operate as a distributed system with chips in other hosts.
_InitializeHostForDistributedTPU :: forall v'1 m' . (MonadBuild m') =>
                                    Tensor v'1 Data.ByteString.ByteString -- ^ __input__: A string containing the address of the UberDriver to connect to.
                                    -> m' (Tensor Value Data.Int.Int32) -- ^ __tpu_ids__: A vector containing the global TPU id of each TPU on the host.
_InitializeHostForDistributedTPU :: Tensor v'1 ByteString -> m' (Tensor Value Int32)
_InitializeHostForDistributedTPU = OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Int32)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Int32)
_InitializeHostForDistributedTPU' OpParams
forall a. a -> a
id
_InitializeHostForDistributedTPU' :: forall v'1 m' . (MonadBuild m') =>
                                     OpParams ->
                                     Tensor v'1 Data.ByteString.ByteString -- ^ __input__: A string containing the address of the UberDriver to connect to.
                                     -> m' (Tensor Value Data.Int.Int32) -- ^ __tpu_ids__: A vector containing the global TPU id of each TPU on the host.
_InitializeHostForDistributedTPU' :: OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Int32)
_InitializeHostForDistributedTPU' op'options :: OpParams
op'options input :: Tensor v'1 ByteString
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value Int32) -> m' (Tensor Value Int32))
-> Build (Tensor Value Int32) -> m' (Tensor Value Int32)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
input]
        [Int64] -> OpDef -> Build (Tensor Value Int32)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_InitializeHostForDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input"
  description: "A string containing the address of the UberDriver to connect to."
  type: DT_STRING
}
output_arg {
  name: "tpu_ids"
  description: "A vector containing the global TPU id of each TPU on the host."
  type: DT_INT32
}
attr {
  name: "enable_whole_mesh_compilations"
  type: "bool"
  default_value { b: false }
  description: "Usually the master TPU worker is the only\nworker compile ops are sent, and the master worker is the only one which\ncan execute them. Other TPU clients distribute TPU compilation across all\nthe hosts of the mesh, and setting this flag to True enables such mesh\ninitialization mode."
}
-}
-- | Converts a list of tensors to an array of tensors.
_ListToArray :: forall v'1 tin t . (TensorTypes tin, TensorType t) =>
                Data.Int.Int64 -- ^ __N__
                -> TensorList (v'1) tin -- ^ __input__
                -> [Tensor Build t] -- ^ __output__
_ListToArray :: Int64 -> TensorList v'1 tin -> [Tensor Build t]
_ListToArray = OpParams -> Int64 -> TensorList v'1 tin -> [Tensor Build t]
forall (v'1 :: * -> *) (tin :: [*]) t.
(TensorTypes tin, TensorType t) =>
OpParams -> Int64 -> TensorList v'1 tin -> [Tensor Build t]
_ListToArray' OpParams
forall a. a -> a
id
_ListToArray' :: forall v'1 tin t . (TensorTypes tin, TensorType t) =>
                 OpParams ->
                 Data.Int.Int64 -- ^ __N__
                 -> TensorList (v'1) tin -- ^ __input__
                 -> [Tensor Build t] -- ^ __output__
_ListToArray' :: OpParams -> Int64 -> TensorList v'1 tin -> [Tensor Build t]
_ListToArray' op'options :: OpParams
op'options n :: Int64
n input :: TensorList v'1 tin
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
n] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tin -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tin
input]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_ListToArray"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tin" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tin -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tin
forall k (t :: k). Proxy t
Proxy :: Proxy tin)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "input" type_list_attr: "Tin" }
output_arg { name: "output" type_attr: "T" number_attr: "N" }
attr {
  name: "Tin" type: "list(type)" has_minimum: true minimum: 1
}
attr { name: "T" type: "type" }
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | Returns the max of x and y (i.e. x > y ? x : y) element-wise.
--
-- *NOTE*: `Maximum` supports broadcasting. More about broadcasting
-- [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
_MklMaximum :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                   Data.Int.Int64,
                                                   Data.Word.Word16, Double,
                                                   Float] t) =>
               Tensor v'1 t -- ^ __x__
               -> Tensor v'2 t -- ^ __y__
               -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
               -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
               -> (Tensor Build t, Tensor Build Data.Word.Word8)
               -- ^ (__z__, __mkl_z__)
               --
               -- * __z__
               --
               -- * __mkl_z__
_MklMaximum :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMaximum = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf '[Int32, Int64, Word16, Double, Float] t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMaximum' OpParams
forall a. a -> a
id
_MklMaximum' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[Data.Int.Int32,
                                                    Data.Int.Int64,
                                                    Data.Word.Word16, Double,
                                                    Float] t) => OpParams ->
                Tensor v'1 t -- ^ __x__
                -> Tensor v'2 t -- ^ __y__
                -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
                -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
                -> (Tensor Build t, Tensor Build Data.Word.Word8)
                -- ^ (__z__, __mkl_z__)
                --
                -- * __z__
                --
                -- * __mkl_z__
_MklMaximum' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMaximum' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y mkl_x :: Tensor v'3 Word8
mkl_x mkl_y :: Tensor v'4 Word8
mkl_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Word8))
-> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y,
                                                             Tensor v'3 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Word8
mkl_x,
                                                             Tensor v'4 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Word8
mkl_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_MklMaximum"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
input_arg { name: "mkl_x" type: DT_UINT8 }
input_arg { name: "mkl_y" type: DT_UINT8 }
output_arg { name: "z" type_attr: "T" }
output_arg { name: "mkl_z" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_BFLOAT16
    }
  }
}
-}
-- | Returns x * y element-wise.
--
-- *NOTE*: `Mul` supports broadcasting. More about broadcasting
-- [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
_MklMul :: forall v'1 v'2 v'3 v'4 t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int16, Data.Int.Int32,
                                               Data.Int.Int64, Data.Int.Int8,
                                               Data.Word.Word16,
                                               Data.Word.Word8, Double,
                                               Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
           -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
           -> (Tensor Build t, Tensor Build Data.Word.Word8)
           -- ^ (__z__, __mkl_z__)
           --
           -- * __z__
           --
           -- * __mkl_z__
_MklMul :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMul = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16,
    Word8, Double, Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMul' OpParams
forall a. a -> a
id
_MklMul' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int16, Data.Int.Int32,
                                                Data.Int.Int64, Data.Int.Int8,
                                                Data.Word.Word16,
                                                Data.Word.Word8, Double,
                                                Float] t) => OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
            -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
            -> (Tensor Build t, Tensor Build Data.Word.Word8)
            -- ^ (__z__, __mkl_z__)
            --
            -- * __z__
            --
            -- * __mkl_z__
_MklMul' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklMul' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y mkl_x :: Tensor v'3 Word8
mkl_x mkl_y :: Tensor v'4 Word8
mkl_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Word8))
-> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y,
                                                             Tensor v'3 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Word8
mkl_x,
                                                             Tensor v'4 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Word8
mkl_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_MklMul"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
input_arg { name: "mkl_x" type: DT_UINT8 }
input_arg { name: "mkl_y" type: DT_UINT8 }
output_arg { name: "z" type_attr: "T" }
output_arg { name: "mkl_z" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_UINT8
      type: DT_INT8
      type: DT_UINT16
      type: DT_INT16
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | Returns (x - y)(x - y) element-wise.
--
-- *NOTE*: `SquaredDifference` supports broadcasting. More about broadcasting
-- [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
_MklSquaredDifference :: forall v'1 v'2 v'3 v'4
                         t . (OneOf '[(Data.Complex.Complex Double),
                                      (Data.Complex.Complex Float),
                                      Data.Int.Int32, Data.Int.Int64,
                                      Data.Word.Word16, Double, Float] t) =>
                         Tensor v'1 t -- ^ __x__
                         -> Tensor v'2 t -- ^ __y__
                         -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
                         -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
                         -> (Tensor Build t, Tensor Build Data.Word.Word8)
                         -- ^ (__z__, __mkl_z__)
                         --
                         -- * __z__
                         --
                         -- * __mkl_z__
_MklSquaredDifference :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSquaredDifference = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSquaredDifference' OpParams
forall a. a -> a
id
_MklSquaredDifference' :: forall v'1 v'2 v'3 v'4
                          t . (OneOf '[(Data.Complex.Complex Double),
                                       (Data.Complex.Complex Float),
                                       Data.Int.Int32, Data.Int.Int64,
                                       Data.Word.Word16, Double, Float] t) =>
                          OpParams ->
                          Tensor v'1 t -- ^ __x__
                          -> Tensor v'2 t -- ^ __y__
                          -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
                          -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
                          -> (Tensor Build t, Tensor Build Data.Word.Word8)
                          -- ^ (__z__, __mkl_z__)
                          --
                          -- * __z__
                          --
                          -- * __mkl_z__
_MklSquaredDifference' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSquaredDifference' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y mkl_x :: Tensor v'3 Word8
mkl_x mkl_y :: Tensor v'4 Word8
mkl_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Word8))
-> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y,
                                                             Tensor v'3 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Word8
mkl_x,
                                                             Tensor v'4 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Word8
mkl_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_MklSquaredDifference"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
input_arg { name: "mkl_x" type: DT_UINT8 }
input_arg { name: "mkl_y" type: DT_UINT8 }
output_arg { name: "z" type_attr: "T" }
output_arg { name: "mkl_z" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | Returns x - y element-wise.
--
-- *NOTE*: `Sub` supports broadcasting. More about broadcasting
-- [here](http://docs.scipy.org/doc/numpy/user/basics.broadcasting.html)
_MklSub :: forall v'1 v'2 v'3 v'4 t . (OneOf '[(Data.Complex.Complex Double),
                                               (Data.Complex.Complex Float),
                                               Data.Int.Int32, Data.Int.Int64,
                                               Data.Word.Word16, Double,
                                               Float] t) =>
           Tensor v'1 t -- ^ __x__
           -> Tensor v'2 t -- ^ __y__
           -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
           -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
           -> (Tensor Build t, Tensor Build Data.Word.Word8)
           -- ^ (__z__, __mkl_z__)
           --
           -- * __z__
           --
           -- * __mkl_z__
_MklSub :: Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSub = OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
forall (v'1 :: * -> *) (v'2 :: * -> *) (v'3 :: * -> *)
       (v'4 :: * -> *) t.
OneOf
  '[Complex Double, Complex Float, Int32, Int64, Word16, Double,
    Float]
  t =>
OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSub' OpParams
forall a. a -> a
id
_MklSub' :: forall v'1 v'2 v'3 v'4 t . (OneOf '[(Data.Complex.Complex Double),
                                                (Data.Complex.Complex Float),
                                                Data.Int.Int32, Data.Int.Int64,
                                                Data.Word.Word16, Double,
                                                Float] t) => OpParams ->
            Tensor v'1 t -- ^ __x__
            -> Tensor v'2 t -- ^ __y__
            -> Tensor v'3 Data.Word.Word8 -- ^ __mkl_x__
            -> Tensor v'4 Data.Word.Word8 -- ^ __mkl_y__
            -> (Tensor Build t, Tensor Build Data.Word.Word8)
            -- ^ (__z__, __mkl_z__)
            --
            -- * __z__
            --
            -- * __mkl_z__
_MklSub' :: OpParams
-> Tensor v'1 t
-> Tensor v'2 t
-> Tensor v'3 Word8
-> Tensor v'4 Word8
-> (Tensor Build t, Tensor Build Word8)
_MklSub' op'options :: OpParams
op'options x :: Tensor v'1 t
x y :: Tensor v'2 t
y mkl_x :: Tensor v'3 Word8
mkl_x mkl_y :: Tensor v'4 Word8
mkl_y | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> (Tensor Build t, Tensor Build Word8))
-> Build OpDef -> (Tensor Build t, Tensor Build Word8)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
y,
                                                             Tensor v'3 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'3 Word8
mkl_x,
                                                             Tensor v'4 Word8 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'4 Word8
mkl_y]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_MklSub"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
input_arg { name: "y" type_attr: "T" }
input_arg { name: "mkl_x" type: DT_UINT8 }
input_arg { name: "mkl_y" type: DT_UINT8 }
output_arg { name: "z" type_attr: "T" }
output_arg { name: "mkl_z" type: DT_UINT8 }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_BFLOAT16
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
      type: DT_COMPLEX64
      type: DT_COMPLEX128
    }
  }
}
-}
-- | Replacement node for NcclBroadcast.
--
-- Sends data of shape `shape` from the _NcclBroadcastSend op registered in the
-- same `shared_name`.
-- The graph should be constructed so that one device runs `_NcclBroadcastSend` and
-- `num_devices-1` devices run _NcclBroadcastRecv ops with shared_name value `c`.
-- Failure to do so will cause the graph execution to fail to complete.
_NcclBroadcastRecv :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Word.Word16,
                                                                Double,
                                                                Float] t) =>
                      Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                      -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same broadcast.
                      -> Tensor v'1 Data.Int.Int32 -- ^ __shape__: The shape of the output.
                      -> m' (Tensor Value t) -- ^ __output__: The broadcast data received from the NcclBroadcastSend op.
_NcclBroadcastRecv :: Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
_NcclBroadcastRecv = OpParams
-> Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
_NcclBroadcastRecv' OpParams
forall a. a -> a
id
_NcclBroadcastRecv' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Word.Word16,
                                                                 Double,
                                                                 Float] t) =>
                       OpParams ->
                       Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                       -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same broadcast.
                       -> Tensor v'1 Data.Int.Int32 -- ^ __shape__: The shape of the output.
                       -> m' (Tensor Value t) -- ^ __output__: The broadcast data received from the NcclBroadcastSend op.
_NcclBroadcastRecv' :: OpParams
-> Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
_NcclBroadcastRecv' op'options :: OpParams
op'options num_devices :: Int64
num_devices shared_name :: ByteString
shared_name
                    shape :: Tensor v'1 Int32
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 Int32
shape]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_NcclBroadcastRecv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "shape"
  description: "The shape of the output."
  type: DT_INT32
}
output_arg {
  name: "output"
  description: "The broadcast data received from the NcclBroadcastSend op."
  type_attr: "T"
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "num_devices"
  type: "int"
  description: "The number of devices participating in this reduction."
}
attr {
  name: "shared_name"
  type: "string"
  description: "Identifier that is shared between ops of the same broadcast."
}
-}
-- | Replacement node for NcclBroadcast.
--
-- Sends `input` to the _NcclBroadcastRecv ops registered in the same
-- `shared_name`.
-- The graph should be constructed so that one device runs `_NcclBroadcastSend` and
-- `num_devices-1` devices run _NcclBroadcastRecv ops with shared_name value `c`.
-- Failure to do so will cause the graph execution to fail to complete.
_NcclBroadcastSend :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                Data.Int.Int64,
                                                                Data.Word.Word16,
                                                                Double,
                                                                Float] t) =>
                      Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                      -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same broadcast.
                      -> Tensor v'1 t -- ^ __input__: The input to the broadcast.
                      -> m' (ControlNode)
_NcclBroadcastSend :: Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclBroadcastSend = OpParams -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclBroadcastSend' OpParams
forall a. a -> a
id
_NcclBroadcastSend' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                                 Data.Int.Int64,
                                                                 Data.Word.Word16,
                                                                 Double,
                                                                 Float] t) =>
                       OpParams ->
                       Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                       -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same broadcast.
                       -> Tensor v'1 t -- ^ __input__: The input to the broadcast.
                       -> m' (ControlNode)
_NcclBroadcastSend' :: OpParams -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclBroadcastSend' op'options :: OpParams
op'options num_devices :: Int64
num_devices shared_name :: ByteString
shared_name
                    input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_NcclBroadcastSend"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input"
  description: "The input to the broadcast."
  type_attr: "T"
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "num_devices"
  type: "int"
  description: "The number of devices participating in this reduction."
}
attr {
  name: "shared_name"
  type: "string"
  description: "Identifier that is shared between ops of the same broadcast."
}
-}
-- | Replacement node for NcclReduce.
--
-- Reduces 'input' from this op and the NcclReduceSend ops registered in the same
-- `shared_name`.
-- The graph should be constructed so that 'num_devices-1' devices run
-- `_NcclReduceSend` and one device runs _NcclReduceRecv op with shared_name value
-- `c`. Failure to do so will cause the graph execution to fail to complete.
_NcclReduceRecv :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Word.Word16,
                                                             Double,
                                                             Float] t) =>
                   Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                   -> ByteString -- ^ __reduction__: the reduction operation to perform.
                   -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same reduce.
                   -> Tensor v'1 t -- ^ __input__: The input to the reduction.
                   -> m' (Tensor Value t) -- ^ __data__: The reduced data received from this op and the NcclReduceSend op.
_NcclReduceRecv :: Int64
-> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
_NcclReduceRecv = OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
_NcclReduceRecv' OpParams
forall a. a -> a
id
_NcclReduceRecv' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                    OpParams ->
                    Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                    -> ByteString -- ^ __reduction__: the reduction operation to perform.
                    -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same reduce.
                    -> Tensor v'1 t -- ^ __input__: The input to the reduction.
                    -> m' (Tensor Value t) -- ^ __data__: The reduced data received from this op and the NcclReduceSend op.
_NcclReduceRecv' :: OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' (Tensor Value t)
_NcclReduceRecv' op'options :: OpParams
op'options num_devices :: Int64
num_devices reduction :: ByteString
reduction shared_name :: ByteString
shared_name
                 input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_NcclReduceRecv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "reduction" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
reduction
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input"
  description: "The input to the reduction."
  type_attr: "T"
}
output_arg {
  name: "data"
  description: "The reduced data received from this op and the NcclReduceSend op."
  type_attr: "T"
}
attr {
  name: "reduction"
  type: "string"
  description: "the reduction operation to perform."
  allowed_values { list { s: "min" s: "max" s: "prod" s: "sum" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "num_devices"
  type: "int"
  description: "The number of devices participating in this reduction."
}
attr {
  name: "shared_name"
  type: "string"
  description: "Identifier that is shared between ops of the same reduce."
}
-}
-- | Replacement node for NcclReduce.
--
-- Reduces `input` to the NcclReduceRecv op registered in the same `shared_name`.
-- The graph should be constructed so that 'num_devices-1' devices run
-- `_NcclReduceSend` and one device runs _NcclReduceRecv op with shared_name value
-- `c`. Failure to do so will cause the graph execution to fail to complete.
_NcclReduceSend :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                             Data.Int.Int64,
                                                             Data.Word.Word16,
                                                             Double,
                                                             Float] t) =>
                   Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                   -> ByteString -- ^ __reduction__: the reduction operation to perform.
                   -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same reduce.
                   -> Tensor v'1 t -- ^ __input__: The input to the reduction.
                   -> m' (ControlNode)
_NcclReduceSend :: Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' ControlNode
_NcclReduceSend = OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) =>
OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_NcclReduceSend' OpParams
forall a. a -> a
id
_NcclReduceSend' :: forall v'1 t m' . (MonadBuild m', OneOf '[Data.Int.Int32,
                                                              Data.Int.Int64,
                                                              Data.Word.Word16,
                                                              Double,
                                                              Float] t) =>
                    OpParams ->
                    Data.Int.Int64 -- ^ __num_devices__: The number of devices participating in this reduction.
                    -> ByteString -- ^ __reduction__: the reduction operation to perform.
                    -> ByteString -- ^ __shared_name__: Identifier that is shared between ops of the same reduce.
                    -> Tensor v'1 t -- ^ __input__: The input to the reduction.
                    -> m' (ControlNode)
_NcclReduceSend' :: OpParams
-> Int64
-> ByteString
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_NcclReduceSend' op'options :: OpParams
op'options num_devices :: Int64
num_devices reduction :: ByteString
reduction shared_name :: ByteString
shared_name
                 input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_NcclReduceSend"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_devices" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_devices
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "reduction" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
reduction
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shared_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
shared_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input"
  description: "The input to the reduction."
  type_attr: "T"
}
attr {
  name: "reduction"
  type: "string"
  description: "the reduction operation to perform."
  allowed_values { list { s: "min" s: "max" s: "prod" s: "sum" } }
}
attr {
  name: "T"
  type: "type"
  allowed_values {
    list {
      type: DT_HALF
      type: DT_FLOAT
      type: DT_DOUBLE
      type: DT_INT32
      type: DT_INT64
    }
  }
}
attr {
  name: "num_devices"
  type: "int"
  description: "The number of devices participating in this reduction."
}
attr {
  name: "shared_name"
  type: "string"
  description: "Identifier that is shared between ops of the same reduce."
}
-}
-- | Creates an empty Tensor with shape `shape` and type `dtype`.
--
-- The memory can optionally be initialized. This is usually useful in
-- conjunction with inplace operations.
_ParallelConcatStart :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                        Shape -- ^ __shape__: 1-D `Tensor` indicating the shape of the output.
                        -> m' (Tensor Value dtype) -- ^ __output__: An empty Tensor of the specified type.
_ParallelConcatStart :: Shape -> m' (Tensor Value dtype)
_ParallelConcatStart = OpParams -> Shape -> m' (Tensor Value dtype)
forall dtype (m' :: * -> *).
(MonadBuild m', TensorType dtype) =>
OpParams -> Shape -> m' (Tensor Value dtype)
_ParallelConcatStart' OpParams
forall a. a -> a
id
_ParallelConcatStart' :: forall dtype m' . (MonadBuild m', TensorType dtype) =>
                         OpParams ->
                         Shape -- ^ __shape__: 1-D `Tensor` indicating the shape of the output.
                         -> m' (Tensor Value dtype) -- ^ __output__: An empty Tensor of the specified type.
_ParallelConcatStart' :: OpParams -> Shape -> m' (Tensor Value dtype)
_ParallelConcatStart' op'options :: OpParams
op'options shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value dtype) -> m' (Tensor Value dtype))
-> Build (Tensor Value dtype) -> m' (Tensor Value dtype)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value dtype)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ParallelConcatStart"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtype" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ dtype -> DataType
forall a. TensorType a => a -> DataType
tensorType (dtype
forall a. HasCallStack => a
undefined :: dtype)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "output"
  description: "An empty Tensor of the specified type."
  type_attr: "dtype"
}
attr {
  name: "shape"
  type: "shape"
  description: "1-D `Tensor` indicating the shape of the output."
}
attr {
  name: "dtype"
  type: "type"
  description: "The element type of the returned tensor."
}
-}
-- | Updates input `value` at `loc` with `update`.
--
-- If you use this function you will almost certainly want to add
-- a control dependency as done in the implementation of parallel_stack to
-- avoid race conditions.
_ParallelConcatUpdate :: forall v'1 v'2 t . (TensorType t) =>
                         Data.Int.Int64 -- ^ __loc__: A scalar indicating the index of the first dimension such that
                                        -- value[loc, :] is updated.
                         -> Tensor v'1 t -- ^ __value__: A `Tensor` object that will be updated in-place.
                         -> Tensor v'2 t -- ^ __update__: A `Tensor` of rank one less than `value` if `loc` is a scalar,
                                         -- otherwise of rank equal to `value` that contains the new values
                                         -- for `value`.
                         -> Tensor Build t -- ^ __output__: `value` that has been updated accordingly.
_ParallelConcatUpdate :: Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_ParallelConcatUpdate = OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_ParallelConcatUpdate' OpParams
forall a. a -> a
id
_ParallelConcatUpdate' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
                          Data.Int.Int64 -- ^ __loc__: A scalar indicating the index of the first dimension such that
                                         -- value[loc, :] is updated.
                          -> Tensor v'1 t -- ^ __value__: A `Tensor` object that will be updated in-place.
                          -> Tensor v'2 t -- ^ __update__: A `Tensor` of rank one less than `value` if `loc` is a scalar,
                                          -- otherwise of rank equal to `value` that contains the new values
                                          -- for `value`.
                          -> Tensor Build t -- ^ __output__: `value` that has been updated accordingly.
_ParallelConcatUpdate' :: OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_ParallelConcatUpdate' op'options :: OpParams
op'options loc :: Int64
loc value :: Tensor v'1 t
value update :: Tensor v'2 t
update | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
value,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
update]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_ParallelConcatUpdate"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "loc" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
loc
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "value"
  description: "A `Tensor` object that will be updated in-place."
  type_attr: "T"
}
input_arg {
  name: "update"
  description: "A `Tensor` of rank one less than `value` if `loc` is a scalar,\notherwise of rank equal to `value` that contains the new values\nfor `value`."
  type_attr: "T"
}
output_arg {
  name: "output"
  description: "`value` that has been updated accordingly."
  type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "loc"
  type: "int"
  description: "A scalar indicating the index of the first dimension such that\nvalue[loc, :] is updated."
}
-}
-- | 
_ReadVariablesOp :: forall v'1 dtypes m' . (MonadBuild m',
                                            TensorTypes dtypes) =>
                    [Tensor v'1 ResourceHandle] -- ^ __resources__
                    -> m' (TensorList (Value) dtypes) -- ^ __values__
_ReadVariablesOp :: [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
_ReadVariablesOp = OpParams
-> [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
forall (v'1 :: * -> *) (dtypes :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes dtypes) =>
OpParams
-> [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
_ReadVariablesOp' OpParams
forall a. a -> a
id
_ReadVariablesOp' :: forall v'1 dtypes m' . (MonadBuild m',
                                             TensorTypes dtypes) => OpParams ->
                     [Tensor v'1 ResourceHandle] -- ^ __resources__
                     -> m' (TensorList (Value) dtypes) -- ^ __values__
_ReadVariablesOp' :: OpParams
-> [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
_ReadVariablesOp' op'options :: OpParams
op'options
                  resources :: [Tensor v'1 ResourceHandle]
resources | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("resources", [Tensor v'1 ResourceHandle] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ResourceHandle]
resources)])] =
    Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value dtypes) -> m' (TensorList Value dtypes))
-> Build (TensorList Value dtypes) -> m' (TensorList Value dtypes)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 ResourceHandle] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 ResourceHandle]
resources]
        [Int64] -> OpDef -> Build (TensorList Value dtypes)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ReadVariablesOp"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy dtypes -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy dtypes
forall k (t :: k). Proxy t
Proxy :: Proxy dtypes)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 ResourceHandle] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 ResourceHandle]
resources) :: Int64
{-
input_arg { name: "resources" type: DT_RESOURCE number_attr: "N" }
output_arg { name: "values" type_list_attr: "dtypes" }
attr { name: "N" type: "int" has_minimum: true }
attr {
  name: "dtypes" type: "list(type)" has_minimum: true minimum: 1
}
-}
-- | Receives the named tensor from send_device on recv_device.
_Recv :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
         ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
         -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
         -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
         -> ByteString -- ^ __tensor_name__: The name of the tensor to receive.
         -> m' (Tensor Value tensor_type) -- ^ __tensor__: The tensor to receive.
_Recv :: ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_Recv = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
forall tensor_type (m' :: * -> *).
(MonadBuild m', TensorType tensor_type) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_Recv' OpParams
forall a. a -> a
id
_Recv' :: forall tensor_type m' . (MonadBuild m', TensorType tensor_type) =>
          OpParams ->
          ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
          -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
          -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
          -> ByteString -- ^ __tensor_name__: The name of the tensor to receive.
          -> m' (Tensor Value tensor_type) -- ^ __tensor__: The tensor to receive.
_Recv' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> m' (Tensor Value tensor_type)
_Recv' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation
       tensor_name :: ByteString
tensor_name | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value tensor_type) -> m' (Tensor Value tensor_type))
-> Build (Tensor Value tensor_type)
-> m' (Tensor Value tensor_type)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value tensor_type)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_Recv"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_type" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ tensor_type -> DataType
forall a. TensorType a => a -> DataType
tensorType (tensor_type
forall a. HasCallStack => a
undefined :: tensor_type)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg {
  name: "tensor"
  description: "The tensor to receive."
  type_attr: "tensor_type"
}
attr { name: "tensor_type" type: "type" }
attr {
  name: "tensor_name"
  type: "string"
  description: "The name of the tensor to receive."
}
attr {
  name: "send_device"
  type: "string"
  description: "The name of the device sending the tensor."
}
attr {
  name: "send_device_incarnation"
  type: "int"
  description: "The current incarnation of send_device."
}
attr {
  name: "recv_device"
  type: "string"
  description: "The name of the device receiving the tensor."
}
attr {
  name: "client_terminated"
  type: "bool"
  default_value { b: false }
  description: "If set to true, this indicates that the node was added\nto the graph as a result of a client-side feed or fetch of Tensor data,\nin which case the corresponding send or recv is expected to be managed\nlocally by the caller."
}
-}
-- | A graph node which represents a return value of a function.
_Retval :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
           Data.Int.Int64 -- ^ __index__: This return value is the index-th return value of the function.
           -> Tensor v'1 t -- ^ __input__: The return value.
           -> m' (ControlNode)
_Retval :: Int64 -> Tensor v'1 t -> m' ControlNode
_Retval = OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
_Retval' OpParams
forall a. a -> a
id
_Retval' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
            Data.Int.Int64 -- ^ __index__: This return value is the index-th return value of the function.
            -> Tensor v'1 t -- ^ __input__: The return value.
            -> m' (ControlNode)
_Retval' :: OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
_Retval' op'options :: OpParams
op'options index :: Int64
index input :: Tensor v'1 t
input | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
input]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_Retval"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "index" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
index
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "input" description: "The return value." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "index"
  type: "int"
  description: "This return value is the index-th return value of the function."
  has_minimum: true
}
-}
-- | Allocates a mutable tensor that becomes available to appropriately annotated
--
-- downstream Ops as backing store for their output tensor allocations via the
-- ScopedAllocatorMgr.
-- Returns a reference to this value.
-- 
-- This is an experimental op for internal use only.  It is possible to use this
-- op in unsafe ways.
-- 
-- 'shapes' is a list of the shapes of the tensors that are to be allocated
-- by this ScopedAllocator.
-- 'shape' is the shape of the output of this Op, i.e. the 1D backing tensor
-- from which the individual allocated tensors are aliased.
-- 'sa_name' is the name assigned to the Node, for connectivity specification
-- and debugging.
-- 'id' is a non-negative integer 'scope_id' handled by the ScopedAllocatorMgr.
-- 'expected_call_count' is the number of individual tensors expected to
-- be allocated from the backing tensor.
_ScopedAllocator :: forall t m' . (MonadBuild m', TensorType t) =>
                    Data.Int.Int64 -- ^ __expected_call_count__
                    -> Data.Int.Int64 -- ^ __id__
                    -> ByteString -- ^ __sa_name__
                    -> Shape -- ^ __shape__
                    -> m' (Tensor Value t) -- ^ __output__
_ScopedAllocator :: Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
_ScopedAllocator = OpParams
-> Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
forall t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
_ScopedAllocator' OpParams
forall a. a -> a
id
_ScopedAllocator' :: forall t m' . (MonadBuild m', TensorType t) => OpParams ->
                     Data.Int.Int64 -- ^ __expected_call_count__
                     -> Data.Int.Int64 -- ^ __id__
                     -> ByteString -- ^ __sa_name__
                     -> Shape -- ^ __shape__
                     -> m' (Tensor Value t) -- ^ __output__
_ScopedAllocator' :: OpParams
-> Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
_ScopedAllocator' op'options :: OpParams
op'options expected_call_count :: Int64
expected_call_count id :: Int64
id sa_name :: ByteString
sa_name
                  shape :: Shape
shape | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ScopedAllocator"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "expected_call_count" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
expected_call_count
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sa_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
sa_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "output" type_attr: "T" }
attr { name: "shapes" type: "list(shape)" }
attr { name: "shape" type: "shape" }
attr { name: "T" type: "type" }
attr { name: "sa_name" type: "string" }
attr { name: "id" type: "int" }
attr { name: "expected_call_count" type: "int" }
-}
-- | Acts like a Concat Op that merges multiple tensors into one, however it must
--
-- only be used in conjunction with a ScopedAllocator which is backing the memory
-- of all of its input tensors so that actually it just outputs a read-only
-- reference to that ScopedAllocator's backing tensor.
-- 
-- This is an experimental op for internal use only.  It is possible to use this
-- op in unsafe ways.
-- 
-- 'backing' is the backing tensor, i.e. the output of an upstream ScopedAllocator.
-- 'inputs' is a list of nominal input tensors, all of which must be aliases
-- to regions of the backing tensor.  These will be outputs of upstream nodes
-- that allocate their outputs from the same ScopedAllocator.
-- 'shape' is the shape of the output, which will usually be the same shape as
-- the input backing tensor.
-- 'reshape' is true iff the output shape is to be different from that of
-- the input backing tensor.
-- 'sa_name' is the Node name of the upstream ScopedAllocator.
-- 'id' is the scope_id identifying the upstream ScopedAllocator.
-- 'N' is the number of nominal inputs to be concatenated.
_ScopedAllocatorConcat :: forall v'1 v'2 t m' . (MonadBuild m', TensorType t) =>
                          Data.Int.Int64 -- ^ __id__
                          -> ByteString -- ^ __sa_name__
                          -> Shape -- ^ __shape__
                          -> Tensor v'1 t -- ^ __backing__
                          -> [Tensor v'2 t] -- ^ __inputs__
                          -> m' (Tensor Value t) -- ^ __output__
_ScopedAllocatorConcat :: Int64
-> ByteString
-> Shape
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' (Tensor Value t)
_ScopedAllocatorConcat = OpParams
-> Int64
-> ByteString
-> Shape
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' (Tensor Value t)
forall (v'1 :: * -> *) (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Int64
-> ByteString
-> Shape
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' (Tensor Value t)
_ScopedAllocatorConcat' OpParams
forall a. a -> a
id
_ScopedAllocatorConcat' :: forall v'1 v'2 t m' . (MonadBuild m',
                                                  TensorType t) => OpParams ->
                           Data.Int.Int64 -- ^ __id__
                           -> ByteString -- ^ __sa_name__
                           -> Shape -- ^ __shape__
                           -> Tensor v'1 t -- ^ __backing__
                           -> [Tensor v'2 t] -- ^ __inputs__
                           -> m' (Tensor Value t) -- ^ __output__
_ScopedAllocatorConcat' :: OpParams
-> Int64
-> ByteString
-> Shape
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' (Tensor Value t)
_ScopedAllocatorConcat' op'options :: OpParams
op'options id :: Int64
id sa_name :: ByteString
sa_name shape :: Shape
shape backing :: Tensor v'1 t
backing
                        inputs :: [Tensor v'2 t]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
inputs)])] =
    Build (Tensor Value t) -> m' (Tensor Value t)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value t) -> m' (Tensor Value t))
-> Build (Tensor Value t) -> m' (Tensor Value t)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
backing,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
inputs]
        [Int64] -> OpDef -> Build (Tensor Value t)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ScopedAllocatorConcat"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
id
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sa_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
sa_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Shape
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "shape" (forall (f :: * -> *). Identical f => LensLike' f OpDef Shape)
-> Shape -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Shape
shape
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
inputs) :: Int64
{-
input_arg { name: "backing" type_attr: "T" }
input_arg { name: "inputs" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" }
attr { name: "shape" type: "shape" }
attr { name: "T" type: "type" }
attr { name: "reshape" type: "bool" default_value { b: false } }
attr { name: "sa_name" type: "string" }
attr { name: "id" type: "int" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
-}
-- | Acts roughly like a SplitV Op that splits one tensor into multiple tensors
--
-- but must only be used in conjunction with corresponding ScopedAllocator
-- and ScopedAllocatorConcat instances.  In practice it is provided as inputs
-- the backing tensor as first input, which contains the concatenated values,
-- and a list of alias tensors as its other input and it simply outputs that
-- second list.
-- 
-- This is an experimental op for internal use only.  It is possible to use this
-- op in unsafe ways.
-- 
-- 'concat' is the single output produced by an upstream ScopedAllocatorConcat
-- node.  This is actually the backing tensor from a ScopedAllocator node
-- upstream of the ScopedAllocatorConcat.
-- 'split' is a list of tensors aliased from the backing tensor.  It will
-- become the output of this ScopedAllocatorSplit node.
-- 'type' is the common DataType of all of the input and output tensors.
-- 'sa_name' is the Node name of the upstream ScopedAllocator.
-- 'id' is the scope_id identifying the upstream ScopedAllocator.
-- 'N' is the number of split tensors.
-- 'shapes' is a list of the split tensor shapes.
_ScopedAllocatorSplit :: forall v'1 v'2 t m' . (MonadBuild m', TensorType t) =>
                         Data.Int.Int64 -- ^ __id__
                         -> ByteString -- ^ __sa_name__
                         -> Tensor v'1 t -- ^ __concat__
                         -> [Tensor v'2 t] -- ^ __split__
                         -> m' ([Tensor Value t]) -- ^ __output__
_ScopedAllocatorSplit :: Int64
-> ByteString
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' [Tensor Value t]
_ScopedAllocatorSplit = OpParams
-> Int64
-> ByteString
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' [Tensor Value t]
forall (v'1 :: * -> *) (v'2 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> Int64
-> ByteString
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' [Tensor Value t]
_ScopedAllocatorSplit' OpParams
forall a. a -> a
id
_ScopedAllocatorSplit' :: forall v'1 v'2 t m' . (MonadBuild m', TensorType t) =>
                          OpParams ->
                          Data.Int.Int64 -- ^ __id__
                          -> ByteString -- ^ __sa_name__
                          -> Tensor v'1 t -- ^ __concat__
                          -> [Tensor v'2 t] -- ^ __split__
                          -> m' ([Tensor Value t]) -- ^ __output__
_ScopedAllocatorSplit' :: OpParams
-> Int64
-> ByteString
-> Tensor v'1 t
-> [Tensor v'2 t]
-> m' [Tensor Value t]
_ScopedAllocatorSplit' op'options :: OpParams
op'options id :: Int64
id sa_name :: ByteString
sa_name concat :: Tensor v'1 t
concat
                       split :: [Tensor v'2 t]
split | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("split", [Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
split)])] =
    Build [Tensor Value t] -> m' [Tensor Value t]
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build [Tensor Value t] -> m' [Tensor Value t])
-> Build [Tensor Value t] -> m' [Tensor Value t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
concat,
                                                             [Tensor v'2 t] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'2 t]
split]
        [Int64] -> OpDef -> Build [Tensor Value t]
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [Int64
n] (OpType -> OpDef
opDef "_ScopedAllocatorSplit"
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "id" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
id
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "sa_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
sa_name
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'2 t] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'2 t]
split) :: Int64
{-
input_arg { name: "concat" type_attr: "T" }
input_arg { name: "split" type_attr: "T" number_attr: "N" }
output_arg { name: "output" type_attr: "T" number_attr: "N" }
attr { name: "T" type: "type" }
attr { name: "sa_name" type: "string" }
attr { name: "id" type: "int" }
attr { name: "N" type: "int" has_minimum: true minimum: 2 }
attr { name: "shapes" type: "list(shape)" }
-}
-- | Sends the named tensor from send_device to recv_device.
_Send :: forall v'1 t m' . (MonadBuild m', TensorType t) =>
         ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
         -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
         -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
         -> ByteString -- ^ __tensor_name__: The name of the tensor to send.
         -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
         -> m' (ControlNode)
_Send :: ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_Send = OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
forall (v'1 :: * -> *) t (m' :: * -> *).
(MonadBuild m', TensorType t) =>
OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_Send' OpParams
forall a. a -> a
id
_Send' :: forall v'1 t m' . (MonadBuild m', TensorType t) => OpParams ->
          ByteString -- ^ __recv_device__: The name of the device receiving the tensor.
          -> ByteString -- ^ __send_device__: The name of the device sending the tensor.
          -> Data.Int.Int64 -- ^ __send_device_incarnation__: The current incarnation of send_device.
          -> ByteString -- ^ __tensor_name__: The name of the tensor to send.
          -> Tensor v'1 t -- ^ __tensor__: The tensor to send.
          -> m' (ControlNode)
_Send' :: OpParams
-> ByteString
-> ByteString
-> Int64
-> ByteString
-> Tensor v'1 t
-> m' ControlNode
_Send' op'options :: OpParams
op'options recv_device :: ByteString
recv_device send_device :: ByteString
send_device send_device_incarnation :: Int64
send_device_incarnation tensor_name :: ByteString
tensor_name
       tensor :: Tensor v'1 t
tensor | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
tensor]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_Send"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "recv_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
recv_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
send_device
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "send_device_incarnation" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
send_device_incarnation
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "tensor_name" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
tensor_name
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "tensor" description: "The tensor to send." type_attr: "T"
}
attr { name: "T" type: "type" }
attr {
  name: "tensor_name"
  type: "string"
  description: "The name of the tensor to send."
}
attr {
  name: "send_device"
  type: "string"
  description: "The name of the device sending the tensor."
}
attr {
  name: "send_device_incarnation"
  type: "int"
  description: "The current incarnation of send_device."
}
attr {
  name: "recv_device"
  type: "string"
  description: "The name of the device receiving the tensor."
}
attr {
  name: "client_terminated"
  type: "bool"
  default_value { b: false }
  description: "If set to true, this indicates that the node was added\nto the graph as a result of a client-side feed or fetch of Tensor data,\nin which case the corresponding send or recv is expected to be managed\nlocally by the caller."
}
-}
-- | An op that informs a host of the global ids of all the of TPUs in the
--
-- system.
_SetGlobalTPUArray :: forall v'1 m' . (MonadBuild m') =>
                      Tensor v'1 Data.ByteString.ByteString -- ^ __topology__: A serialized tensorflow.tpu.TopologyProto that describes the TPU
                                                            -- topology.
                      -> m' (ControlNode)
_SetGlobalTPUArray :: Tensor v'1 ByteString -> m' ControlNode
_SetGlobalTPUArray = OpParams -> Tensor v'1 ByteString -> m' ControlNode
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> Tensor v'1 ByteString -> m' ControlNode
_SetGlobalTPUArray' OpParams
forall a. a -> a
id
_SetGlobalTPUArray' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                       Tensor v'1 Data.ByteString.ByteString -- ^ __topology__: A serialized tensorflow.tpu.TopologyProto that describes the TPU
                                                             -- topology.
                       -> m' (ControlNode)
_SetGlobalTPUArray' :: OpParams -> Tensor v'1 ByteString -> m' ControlNode
_SetGlobalTPUArray' op'options :: OpParams
op'options topology :: Tensor v'1 ByteString
topology | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
topology]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_SetGlobalTPUArray"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "topology"
  description: "A serialized tensorflow.tpu.TopologyProto that describes the TPU\ntopology."
  type: DT_STRING
}
-}
-- | An op that shuts down a running distributed TPU system. The Op returns
--
-- an error if no system is running. This Op must be run on the same
-- TPU_SYSTEM device as the corresponding _ConfigureDistributedTPU was run
-- to start the system, and must be run only after
-- _DisconnectHostFromDistributedTPUSystem has completed on every host in
-- the system.
_ShutdownDistributedTPU :: forall m' . (MonadBuild m') => m' (ControlNode)
_ShutdownDistributedTPU :: m' ControlNode
_ShutdownDistributedTPU = OpParams -> m' ControlNode
forall (m' :: * -> *). MonadBuild m' => OpParams -> m' ControlNode
_ShutdownDistributedTPU' OpParams
forall a. a -> a
id
_ShutdownDistributedTPU' :: forall m' . (MonadBuild m') => OpParams ->
                            m' (ControlNode)
_ShutdownDistributedTPU' :: OpParams -> m' ControlNode
_ShutdownDistributedTPU' op'options :: OpParams
op'options | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_ShutdownDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-

-}
-- | 
_SwitchN :: forall v'1 v'2 t . (TensorType t) =>
            Data.Int.Int64 -- ^ __num_outs__
            -> Tensor v'1 t -- ^ __data__
            -> Tensor v'2 Data.Int.Int32 -- ^ __output_index__
            -> [Tensor Build t] -- ^ __outputs__
_SwitchN :: Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
_SwitchN = OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
_SwitchN' OpParams
forall a. a -> a
id
_SwitchN' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
             Data.Int.Int64 -- ^ __num_outs__
             -> Tensor v'1 t -- ^ __data__
             -> Tensor v'2 Data.Int.Int32 -- ^ __output_index__
             -> [Tensor Build t] -- ^ __outputs__
_SwitchN' :: OpParams
-> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
_SwitchN' op'options :: OpParams
op'options num_outs :: Int64
num_outs data' :: Tensor v'1 t
data' output_index :: Tensor v'2 Int32
output_index | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> [Tensor Build t]
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [Int64
num_outs] (Build OpDef -> [Tensor Build t])
-> Build OpDef -> [Tensor Build t]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
data',
                                                             Tensor v'2 Int32 -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 Int32
output_index]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_SwitchN"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "num_outs" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
num_outs
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "data" type_attr: "T" }
input_arg { name: "output_index" type: DT_INT32 }
output_arg {
  name: "outputs" type_attr: "T" number_attr: "num_outs"
}
attr { name: "num_outs" type: "int" has_minimum: true minimum: 1 }
attr { name: "T" type: "type" }
-}
-- | *NOTE*: Do not invoke this operator directly in Python. Graph rewrite pass is
--
-- expected to create these operators.
_UnaryOpsComposition :: forall v'1 t . (OneOf '[Data.Word.Word16, Double,
                                                Float] t) =>
                        Tensor v'1 t -- ^ __x__
                        -> Tensor Build t -- ^ __y__
_UnaryOpsComposition :: Tensor v'1 t -> Tensor Build t
_UnaryOpsComposition = OpParams -> Tensor v'1 t -> Tensor Build t
forall (v'1 :: * -> *) t.
OneOf '[Word16, Double, Float] t =>
OpParams -> Tensor v'1 t -> Tensor Build t
_UnaryOpsComposition' OpParams
forall a. a -> a
id
_UnaryOpsComposition' :: forall v'1 t . (OneOf '[Data.Word.Word16, Double,
                                                 Float] t) => OpParams ->
                         Tensor v'1 t -- ^ __x__
                         -> Tensor Build t -- ^ __y__
_UnaryOpsComposition' :: OpParams -> Tensor v'1 t -> Tensor Build t
_UnaryOpsComposition' op'options :: OpParams
op'options x :: Tensor v'1 t
x | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
x]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_UnaryOpsComposition"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "x" type_attr: "T" }
output_arg { name: "y" type_attr: "T" }
attr {
  name: "T"
  type: "type"
  allowed_values {
    list { type: DT_FLOAT type: DT_HALF type: DT_DOUBLE }
  }
}
attr { name: "op_names" type: "list(string)" }
-}
-- | 
_VarHandlesOp :: forall m' . (MonadBuild m') => Data.Int.Int64 -- ^ __N__
                 -> [DataType] -- ^ __dtypes__
                 -> m' ([Tensor Value ResourceHandle]) -- ^ __resources__
_VarHandlesOp :: Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
_VarHandlesOp = OpParams -> Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
forall (m' :: * -> *).
MonadBuild m' =>
OpParams -> Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
_VarHandlesOp' OpParams
forall a. a -> a
id
_VarHandlesOp' :: forall m' . (MonadBuild m') => OpParams ->
                  Data.Int.Int64 -- ^ __N__
                  -> [DataType] -- ^ __dtypes__
                  -> m' ([Tensor Value ResourceHandle]) -- ^ __resources__
_VarHandlesOp' :: OpParams -> Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
_VarHandlesOp' op'options :: OpParams
op'options n :: Int64
n dtypes :: [DataType]
dtypes | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build [Tensor Value ResourceHandle]
-> m' [Tensor Value ResourceHandle]
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build [Tensor Value ResourceHandle]
 -> m' [Tensor Value ResourceHandle])
-> Build [Tensor Value ResourceHandle]
-> m' [Tensor Value ResourceHandle]
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence []
        [Int64] -> OpDef -> Build [Tensor Value ResourceHandle]
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [Int64
n] (OpType -> OpDef
opDef "_VarHandlesOp"
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "dtypes" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [DataType]
dtypes
                     OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
output_arg { name: "resources" type: DT_RESOURCE number_attr: "N" }
attr { name: "containers" type: "list(string)" }
attr { name: "shared_names" type: "list(string)" }
attr { name: "N" type: "int" has_minimum: true }
attr { name: "dtypes" type: "list(type)" }
attr { name: "shapes" type: "list(shape)" }
-}
-- | An op that blocks execution until a distributed TPU system has
--
-- started up. This Op must be run on the same TPU_SYSTEM device as
-- _ConfigureDistributedTPU, and takes an inputs the outputs from the
-- _InitializeHostForDistributedTPU Ops.
_WaitForDistributedTPU :: forall v'1 m' . (MonadBuild m') =>
                          [Tensor v'1 Data.Int.Int32] -- ^ __inputs__: For each initialized host, a vector giving the global TPU id
                                                      -- of each TPU on the host.
                          -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __topology__: A serialized tensorflow.tpu.TopologyProto that describes the TPU
                          -- topology.
_WaitForDistributedTPU :: [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_WaitForDistributedTPU = OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
forall (v'1 :: * -> *) (m' :: * -> *).
MonadBuild m' =>
OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_WaitForDistributedTPU' OpParams
forall a. a -> a
id
_WaitForDistributedTPU' :: forall v'1 m' . (MonadBuild m') => OpParams ->
                           [Tensor v'1 Data.Int.Int32] -- ^ __inputs__: For each initialized host, a vector giving the global TPU id
                                                       -- of each TPU on the host.
                           -> m' (Tensor Value Data.ByteString.ByteString) -- ^ __topology__: A serialized tensorflow.tpu.TopologyProto that describes the TPU
                           -- topology.
_WaitForDistributedTPU' :: OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
_WaitForDistributedTPU' op'options :: OpParams
op'options
                        inputs :: [Tensor v'1 Int32]
inputs | [(String, [(String, Int)])] -> Bool
eqLengthGuard [("N", [("inputs", [Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
inputs)])] =
    Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (Tensor Value ByteString) -> m' (Tensor Value ByteString))
-> Build (Tensor Value ByteString) -> m' (Tensor Value ByteString)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [[Tensor v'1 Int32] -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs [Tensor v'1 Int32]
inputs]
        [Int64] -> OpDef -> Build (Tensor Value ByteString)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_WaitForDistributedTPU"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "N" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
n
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
  where
    n :: Int64
n = Int -> Int64
forall a b. (Integral a, Num b) => a -> b
fromIntegral ([Tensor v'1 Int32] -> Int
forall (t :: * -> *) a. Foldable t => t a -> Int
length [Tensor v'1 Int32]
inputs) :: Int64
{-
input_arg {
  name: "inputs"
  description: "For each initialized host, a vector giving the global TPU id\nof each TPU on the host."
  type: DT_INT32
  number_attr: "N"
}
output_arg {
  name: "topology"
  description: "A serialized tensorflow.tpu.TopologyProto that describes the TPU\ntopology."
  type: DT_STRING
}
attr {
  name: "startup_timeout_sec"
  type: "int"
  default_value { i: 20 }
  description: "The number of seconds to wait for the TPU system\nto stabilize."
}
attr { name: "N" type: "int" has_minimum: true minimum: 1 }
-}
-- | XLA Merge Op. For use by the XLA JIT only.
--
-- Merges the outputs from the PartitionedCall node and the _XlaRun node.
-- Unlike the TensorFlow Merge op, which requires inputs of some types to be
-- placed on the host, the _XlaMerge op can merge inputs of all types when
-- placed on the device. This prevents the need for copy operations, in
-- particular when an XLA cluster has int32 outputs. The _XlaMerge up does not
-- have a value_index output that identifies the chosen input.
_XlaMerge :: forall v'1 v'2 t . (TensorType t) =>
             Tensor v'1 t -- ^ __partitioned_call__
             -> Tensor v'2 t -- ^ __xla_run__
             -> Tensor Build t -- ^ __output__
_XlaMerge :: Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_XlaMerge = OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
forall (v'1 :: * -> *) (v'2 :: * -> *) t.
TensorType t =>
OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_XlaMerge' OpParams
forall a. a -> a
id
_XlaMerge' :: forall v'1 v'2 t . (TensorType t) => OpParams ->
              Tensor v'1 t -- ^ __partitioned_call__
              -> Tensor v'2 t -- ^ __xla_run__
              -> Tensor Build t -- ^ __output__
_XlaMerge' :: OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
_XlaMerge' op'options :: OpParams
op'options partitioned_call :: Tensor v'1 t
partitioned_call xla_run :: Tensor v'2 t
xla_run | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    [Int64] -> Build OpDef -> Tensor Build t
forall a. PureResult a => [Int64] -> Build OpDef -> a
pureOp [] (Build OpDef -> Tensor Build t) -> Build OpDef -> Tensor Build t
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 t
partitioned_call,
                                                             Tensor v'2 t -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 t
xla_run]
        OpDef -> Build OpDef
forall (m :: * -> *) a. Monad m => a -> m a
return (OpType -> OpDef
opDef "_XlaMerge"
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef DataType
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "T" (forall (f :: * -> *). Identical f => LensLike' f OpDef DataType)
-> DataType -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ t -> DataType
forall a. TensorType a => a -> DataType
tensorType (t
forall a. HasCallStack => a
undefined :: t)
                OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "partitioned_call" type_attr: "T" }
input_arg { name: "xla_run" type_attr: "T" }
output_arg { name: "output" type_attr: "T" }
attr { name: "T" type: "type" }
-}
-- | A placeholder op to receive values from a running XLA computation.
_XlaRecvAtHost :: forall v'1 toutputs m' . (MonadBuild m',
                                            TensorTypes toutputs) =>
                  Data.Int.Int64 -- ^ __device_ordinal__: The device to use.
                  -> ByteString -- ^ __key__: A key that is unique in the computation and associates the send with the consumer in
                                -- the XLA computation.
                  -> Tensor v'1 Data.ByteString.ByteString -- ^ __dynamic_key__: The key sent at runtime by the compile node to identify which
                                                           -- execution the transfer corresponds to.
                  -> m' (TensorList (Value) toutputs) -- ^ __outputs__: A list of tensors that will be received from the XLA computation.
_XlaRecvAtHost :: Int64
-> ByteString
-> Tensor v'1 ByteString
-> m' (TensorList Value toutputs)
_XlaRecvAtHost = OpParams
-> Int64
-> ByteString
-> Tensor v'1 ByteString
-> m' (TensorList Value toutputs)
forall (v'1 :: * -> *) (toutputs :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes toutputs) =>
OpParams
-> Int64
-> ByteString
-> Tensor v'1 ByteString
-> m' (TensorList Value toutputs)
_XlaRecvAtHost' OpParams
forall a. a -> a
id
_XlaRecvAtHost' :: forall v'1 toutputs m' . (MonadBuild m',
                                             TensorTypes toutputs) =>
                   OpParams ->
                   Data.Int.Int64 -- ^ __device_ordinal__: The device to use.
                   -> ByteString -- ^ __key__: A key that is unique in the computation and associates the send with the consumer in
                                 -- the XLA computation.
                   -> Tensor v'1 Data.ByteString.ByteString -- ^ __dynamic_key__: The key sent at runtime by the compile node to identify which
                                                            -- execution the transfer corresponds to.
                   -> m' (TensorList (Value) toutputs) -- ^ __outputs__: A list of tensors that will be received from the XLA computation.
_XlaRecvAtHost' :: OpParams
-> Int64
-> ByteString
-> Tensor v'1 ByteString
-> m' (TensorList Value toutputs)
_XlaRecvAtHost' op'options :: OpParams
op'options device_ordinal :: Int64
device_ordinal key :: ByteString
key dynamic_key :: Tensor v'1 ByteString
dynamic_key | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value toutputs) -> m' (TensorList Value toutputs)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value toutputs)
 -> m' (TensorList Value toutputs))
-> Build (TensorList Value toutputs)
-> m' (TensorList Value toutputs)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [Tensor v'1 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'1 ByteString
dynamic_key]
        [Int64] -> OpDef -> Build (TensorList Value toutputs)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_XlaRecvAtHost"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Toutputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy toutputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy toutputs
forall k (t :: k). Proxy t
Proxy :: Proxy toutputs)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "device_ordinal" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
device_ordinal
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "dynamic_key"
  description: "The key sent at runtime by the compile node to identify which\nexecution the transfer corresponds to."
  type: DT_STRING
}
output_arg {
  name: "outputs"
  description: "A list of tensors that will be received from the XLA computation."
  type_list_attr: "Toutputs"
}
attr {
  name: "Toutputs"
  type: "list(type)"
  description: "The element types of each element in `outputs`."
  has_minimum: true
}
attr {
  name: "key"
  type: "string"
  description: "A key that is unique in the computation and associates the send with the consumer in\nthe XLA computation."
}
attr {
  name: "device_ordinal"
  type: "int"
  description: "The device to use."
}
-}
-- | XLA Run Op. For use by the XLA JIT only.
--
-- Executes a TensorFlow function previously compiled into a LocalExecutable by an
-- _XlaCompile op.
_XlaRun :: forall v'1 v'2 targs tresults m' . (MonadBuild m', TensorTypes targs,
                                               TensorTypes tresults) =>
           TensorList (v'1) targs -- ^ __args__
           -> Tensor v'2 Data.ByteString.ByteString -- ^ __key__
           -> m' (TensorList (Value) tresults) -- ^ __results__
_XlaRun :: TensorList v'1 targs
-> Tensor v'2 ByteString -> m' (TensorList Value tresults)
_XlaRun = OpParams
-> TensorList v'1 targs
-> Tensor v'2 ByteString
-> m' (TensorList Value tresults)
forall (v'1 :: * -> *) (v'2 :: * -> *) (targs :: [*])
       (tresults :: [*]) (m' :: * -> *).
(MonadBuild m', TensorTypes targs, TensorTypes tresults) =>
OpParams
-> TensorList v'1 targs
-> Tensor v'2 ByteString
-> m' (TensorList Value tresults)
_XlaRun' OpParams
forall a. a -> a
id
_XlaRun' :: forall v'1 v'2 targs tresults m' . (MonadBuild m',
                                                TensorTypes targs,
                                                TensorTypes tresults) =>
            OpParams ->
            TensorList (v'1) targs -- ^ __args__
            -> Tensor v'2 Data.ByteString.ByteString -- ^ __key__
            -> m' (TensorList (Value) tresults) -- ^ __results__
_XlaRun' :: OpParams
-> TensorList v'1 targs
-> Tensor v'2 ByteString
-> m' (TensorList Value tresults)
_XlaRun' op'options :: OpParams
op'options args :: TensorList v'1 targs
args key :: Tensor v'2 ByteString
key | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build (TensorList Value tresults) -> m' (TensorList Value tresults)
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build (TensorList Value tresults)
 -> m' (TensorList Value tresults))
-> Build (TensorList Value tresults)
-> m' (TensorList Value tresults)
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 targs -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 targs
args,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
key]
        [Int64] -> OpDef -> Build (TensorList Value tresults)
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_XlaRun"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Targs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy targs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy targs
forall k (t :: k). Proxy t
Proxy :: Proxy targs)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tresults" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tresults -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tresults
forall k (t :: k). Proxy t
Proxy :: Proxy tresults)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg { name: "args" type_list_attr: "Targs" }
input_arg { name: "key" type: DT_STRING }
output_arg { name: "results" type_list_attr: "Tresults" }
attr { name: "Targs" type: "list(type)" has_minimum: true }
attr { name: "Tresults" type: "list(type)" has_minimum: true }
-}
-- | A placeholder op to send values to a running XLA computation.
_XlaSendFromHost :: forall v'1 v'2 tinputs m' . (MonadBuild m',
                                                 TensorTypes tinputs) =>
                    Data.Int.Int64 -- ^ __device_ordinal__: The device to use.
                    -> ByteString -- ^ __key__: A key that is unique in the computation and associates the send with the consumer in
                                  -- the XLA computation.
                    -> TensorList (v'1) tinputs -- ^ __inputs__: A list of tensors that will be sent to the XLA computation.
                    -> Tensor v'2 Data.ByteString.ByteString -- ^ __dynamic_key__: The key sent at runtime by the compile node to identify which
                                                             -- execution the transfer corresponds to.
                    -> m' (ControlNode)
_XlaSendFromHost :: Int64
-> ByteString
-> TensorList v'1 tinputs
-> Tensor v'2 ByteString
-> m' ControlNode
_XlaSendFromHost = OpParams
-> Int64
-> ByteString
-> TensorList v'1 tinputs
-> Tensor v'2 ByteString
-> m' ControlNode
forall (v'1 :: * -> *) (v'2 :: * -> *) (tinputs :: [*])
       (m' :: * -> *).
(MonadBuild m', TensorTypes tinputs) =>
OpParams
-> Int64
-> ByteString
-> TensorList v'1 tinputs
-> Tensor v'2 ByteString
-> m' ControlNode
_XlaSendFromHost' OpParams
forall a. a -> a
id
_XlaSendFromHost' :: forall v'1 v'2 tinputs m' . (MonadBuild m',
                                                  TensorTypes tinputs) =>
                     OpParams ->
                     Data.Int.Int64 -- ^ __device_ordinal__: The device to use.
                     -> ByteString -- ^ __key__: A key that is unique in the computation and associates the send with the consumer in
                                   -- the XLA computation.
                     -> TensorList (v'1) tinputs -- ^ __inputs__: A list of tensors that will be sent to the XLA computation.
                     -> Tensor v'2 Data.ByteString.ByteString -- ^ __dynamic_key__: The key sent at runtime by the compile node to identify which
                                                              -- execution the transfer corresponds to.
                     -> m' (ControlNode)
_XlaSendFromHost' :: OpParams
-> Int64
-> ByteString
-> TensorList v'1 tinputs
-> Tensor v'2 ByteString
-> m' ControlNode
_XlaSendFromHost' op'options :: OpParams
op'options device_ordinal :: Int64
device_ordinal key :: ByteString
key inputs :: TensorList v'1 tinputs
inputs
                  dynamic_key :: Tensor v'2 ByteString
dynamic_key | [(String, [(String, Int)])] -> Bool
eqLengthGuard [] =
    Build ControlNode -> m' ControlNode
forall (m :: * -> *) a. MonadBuild m => Build a -> m a
build (Build ControlNode -> m' ControlNode)
-> Build ControlNode -> m' ControlNode
forall a b. (a -> b) -> a -> b
$ do
        [Output]
op'inputs <- ([[Output]] -> [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
fmap [[Output]] -> [Output]
forall (t :: * -> *) a. Foldable t => t [a] -> [a]
Prelude.concat (BuildT Identity [[Output]] -> BuildT Identity [Output])
-> BuildT Identity [[Output]] -> BuildT Identity [Output]
forall a b. (a -> b) -> a -> b
$ [BuildT Identity [Output]] -> BuildT Identity [[Output]]
forall (t :: * -> *) (m :: * -> *) a.
(Traversable t, Monad m) =>
t (m a) -> m (t a)
Prelude.sequence [TensorList v'1 tinputs -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs TensorList v'1 tinputs
inputs,
                                                             Tensor v'2 ByteString -> BuildT Identity [Output]
forall a. BuildInputs a => a -> BuildT Identity [Output]
buildInputs Tensor v'2 ByteString
dynamic_key]
        [Int64] -> OpDef -> Build ControlNode
forall a. BuildResult a => [Int64] -> OpDef -> Build a
buildOp [] (OpType -> OpDef
opDef "_XlaSendFromHost"
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef [DataType]
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "Tinputs" (forall (f :: * -> *). Identical f => LensLike' f OpDef [DataType])
-> [DataType] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Proxy tinputs -> [DataType]
forall (as :: [*]). TensorTypes as => Proxy as -> [DataType]
fromTensorTypes (Proxy tinputs
forall k (t :: k). Proxy t
Proxy :: Proxy tinputs)
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef Int64
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "device_ordinal" (forall (f :: * -> *). Identical f => LensLike' f OpDef Int64)
-> Int64 -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ Int64
device_ordinal
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Text -> Lens' OpDef ByteString
forall a. Attribute a => Text -> Lens' OpDef a
opAttr "key" (forall (f :: * -> *). Identical f => LensLike' f OpDef ByteString)
-> ByteString -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ ByteString
key
                    OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& OpParams
op'options OpDef -> OpParams -> OpDef
forall s t. s -> (s -> t) -> t
& Lens' OpDef [Output]
forall (f :: * -> *). Identical f => LensLike' f OpDef [Output]
opInputs (forall (f :: * -> *). Identical f => LensLike' f OpDef [Output])
-> [Output] -> OpParams
forall s t a b. Setter s t a b -> b -> s -> t
.~ [Output]
op'inputs)
{-
input_arg {
  name: "inputs"
  description: "A list of tensors that will be sent to the XLA computation."
  type_list_attr: "Tinputs"
}
input_arg {
  name: "dynamic_key"
  description: "The key sent at runtime by the compile node to identify which\nexecution the transfer corresponds to."
  type: DT_STRING
}
attr {
  name: "Tinputs"
  type: "list(type)"
  description: "The element types of each element in `inputs`."
  has_minimum: true
}
attr {
  name: "key"
  type: "string"
  description: "A key that is unique in the computation and associates the send with the consumer in\nthe XLA computation."
}
attr {
  name: "device_ordinal"
  type: "int"
  description: "The device to use."
}
-}