4396 lines
484 KiB
Plaintext
4396 lines
484 KiB
Plaintext
-- Hoogle documentation, generated by Haddock
|
|
-- See Hoogle, http://www.haskell.org/hoogle/
|
|
|
|
|
|
-- | Haskell wrappers for Core Tensorflow Ops.
|
|
--
|
|
-- Code generated signatures for the Ops in libtensorflow.
|
|
@package tensorflow-core-ops
|
|
@version 0.3.0.0
|
|
|
|
module TensorFlow.GenOps.Core
|
|
|
|
abort :: forall m'. MonadBuild m' => m' ControlNode
|
|
abort' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
|
|
|
|
abs :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
abs' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
accumulateNV2 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Shape -> [Tensor v'1 t] -> Tensor Build t
|
|
accumulateNV2' :: 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
|
|
|
|
accumulatorApplyGradient :: 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) => Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
|
|
accumulatorApplyGradient' :: 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
|
|
|
|
accumulatorNumAccumulated :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
accumulatorNumAccumulated' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
|
|
accumulatorSetGlobalStep :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
|
|
accumulatorSetGlobalStep' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int64 -> m' ControlNode
|
|
|
|
accumulatorTakeGradient :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
|
|
accumulatorTakeGradient' :: 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)
|
|
|
|
acos :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
acos' :: 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
|
|
|
|
acosh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
acosh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
add :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
add' :: 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
|
|
|
|
addManySparseToTensorsMap :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
|
|
addManySparseToTensorsMap' :: 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)
|
|
|
|
addN :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float, Variant] t => [Tensor v'1 t] -> Tensor Build t
|
|
addN' :: 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
|
|
|
|
addSparseToTensorsMap :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
|
|
addSparseToTensorsMap' :: 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)
|
|
|
|
addV2 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
addV2' :: 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
|
|
|
|
adjustContrast :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
|
|
adjustContrast' :: 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
|
|
|
|
adjustContrastv2 :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
adjustContrastv2' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
|
|
adjustHue :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
adjustHue' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
|
|
adjustSaturation :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
adjustSaturation' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
|
|
all :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
|
|
all' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
|
|
|
|
allCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
allCandidateSampler' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
|
|
allToAll :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Int64 -> Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
allToAll' :: 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
|
|
|
|
angle :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
|
|
angle' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
|
|
|
|
anonymousIterator :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
|
|
anonymousIterator' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
|
|
|
|
anonymousIteratorV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
anonymousIteratorV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
|
|
anonymousMemoryCache :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
anonymousMemoryCache' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
|
|
anonymousMultiDeviceIterator :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
anonymousMultiDeviceIterator' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
|
|
anonymousRandomSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
anonymousRandomSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
|
|
anonymousSeedGenerator :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Bool -> m' (Tensor Value ResourceHandle, Tensor Value Variant)
|
|
anonymousSeedGenerator' :: 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)
|
|
|
|
any :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
|
|
any' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 Bool -> Tensor v'2 tidx -> Tensor Build Bool
|
|
|
|
applyAdaMax :: 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) => 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' :: 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)
|
|
|
|
applyAdadelta :: 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) => 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' :: 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)
|
|
|
|
applyAdagrad :: 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) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> m' (Tensor Ref t)
|
|
applyAdagrad' :: 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)
|
|
|
|
applyAdagradDA :: 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) => 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' :: 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)
|
|
|
|
applyAdagradV2 :: 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) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
|
|
applyAdagradV2' :: 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)
|
|
|
|
applyAdam :: 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) => 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' :: 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)
|
|
|
|
applyAddSign :: 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) => 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' :: 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)
|
|
|
|
applyCenteredRMSProp :: 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) => 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' :: 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)
|
|
|
|
applyFtrl :: 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) => 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' :: 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)
|
|
|
|
applyFtrlV2 :: 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) => 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' :: 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)
|
|
|
|
applyGradientDescent :: 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) => Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
applyGradientDescent' :: 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)
|
|
|
|
applyMomentum :: 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) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
|
|
applyMomentum' :: 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)
|
|
|
|
applyPowerSign :: 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) => 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' :: 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)
|
|
|
|
applyProximalAdagrad :: 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) => 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' :: 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)
|
|
|
|
applyProximalGradientDescent :: 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) => Tensor Ref t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' (Tensor Ref t)
|
|
applyProximalGradientDescent' :: 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)
|
|
|
|
applyRMSProp :: 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) => 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' :: 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)
|
|
|
|
approximateEqual :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
approximateEqual' :: 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
|
|
|
|
argMax :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
|
|
argMax' :: 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 :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build output_type
|
|
argMin' :: 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
|
|
|
|
asString :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Double, Float] t => Tensor v'1 t -> Tensor Build ByteString
|
|
asString' :: 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
|
|
|
|
asin :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
asin' :: 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
|
|
|
|
asinh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
asinh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
assert :: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
|
|
assert' :: forall v'1 v'2 t m'. (MonadBuild m', TensorTypes t) => OpParams -> Tensor v'1 Bool -> TensorList v'2 t -> m' ControlNode
|
|
|
|
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
|
|
|
|
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
|
|
|
|
assign :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
|
|
assign' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
|
|
|
|
assignAdd :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
|
|
assignAdd' :: 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)
|
|
|
|
assignAddVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
assignAddVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
|
|
assignSub :: forall v'2 t m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => Tensor Ref t -> Tensor v'2 t -> m' (Tensor Ref t)
|
|
assignSub' :: 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)
|
|
|
|
assignSubVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
assignSubVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
|
|
assignVariableOp :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
assignVariableOp' :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 dtype -> m' ControlNode
|
|
|
|
atan :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
atan' :: 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
|
|
|
|
atan2 :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
atan2' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
atanh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
atanh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
audioSpectrogram :: Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
|
|
audioSpectrogram' :: OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor Build Float
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
avgPool :: forall v'1 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
avgPool' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
avgPool3D :: forall v'1 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
avgPool3D' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
avgPool3DGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
|
|
avgPool3DGrad' :: 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 :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
|
|
avgPoolGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor Build t
|
|
|
|
bandedTriangularSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
bandedTriangularSolve' :: 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
|
|
|
|
barrier :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
|
|
barrier' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
|
|
|
|
barrierClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
|
|
barrierClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
|
|
|
|
barrierIncompleteSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
barrierIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
|
|
barrierInsertMany :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> Tensor v'3 t -> m' ControlNode
|
|
barrierInsertMany' :: 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
|
|
|
|
barrierReadySize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
barrierReadySize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
|
|
barrierTakeMany :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value ByteString, TensorList Value component_types)
|
|
barrierTakeMany' :: 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)
|
|
|
|
batch :: forall v'1 t. TensorTypes t => Int64 -> Int64 -> Int64 -> Int64 -> TensorList v'1 t -> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
|
|
batch' :: forall v'1 t. TensorTypes t => OpParams -> Int64 -> Int64 -> Int64 -> Int64 -> TensorList v'1 t -> (TensorList Build t, Tensor Build Int64, Tensor Build Int64)
|
|
|
|
batchCholesky :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
batchCholesky' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchCholeskyGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchCholeskyGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
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
|
|
|
|
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
|
|
|
|
batchFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchFFT' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchFFT2D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchFFT3D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchIFFT :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchIFFT' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchIFFT2D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchIFFT2D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchIFFT3D :: Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
batchIFFT3D' :: OpParams -> Tensor v'1 (Complex Float) -> Tensor Build (Complex Float)
|
|
|
|
batchMatMul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchMatMul' :: 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 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchMatMulV2' :: 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
|
|
|
|
batchMatrixBandPart :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
|
|
batchMatrixBandPart' :: 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
|
|
|
|
batchMatrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
batchMatrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchMatrixDiag :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
batchMatrixDiag' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchMatrixDiagPart :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
batchMatrixDiagPart' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchMatrixInverse :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
batchMatrixInverse' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchMatrixSetDiag :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchMatrixSetDiag' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
batchMatrixSolve :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchMatrixSolve' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
batchMatrixSolveLs :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
|
|
batchMatrixSolveLs' :: 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
|
|
|
|
batchMatrixTriangularSolve :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
batchMatrixTriangularSolve' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
batchNormWithGlobalNormalization :: 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 => 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' :: 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
|
|
|
|
batchNormWithGlobalNormalizationGrad :: 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 => 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' :: 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)
|
|
|
|
batchSelfAdjointEig :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
batchSelfAdjointEig' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
batchSelfAdjointEigV2 :: forall v'1 t. OneOf '[Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
batchSelfAdjointEigV2' :: forall v'1 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
|
|
batchSvd :: forall v'1 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
|
|
batchSvd' :: 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)
|
|
|
|
batchToSpace :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => Int64 -> Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
batchToSpace' :: 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
|
|
|
|
batchToSpaceND :: forall v'1 v'2 v'3 t tblock_shape tcrops. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tcrops) => Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tcrops -> Tensor Build t
|
|
batchToSpaceND' :: 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
|
|
|
|
besselI0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselI0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselI0e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselI0e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselI1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselI1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselI1e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselI1e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselJ0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselJ0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselJ1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselJ1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselK0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselK0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselK0e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselK0e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselK1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselK1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselK1e :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselK1e' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselY0 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselY0' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
besselY1 :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
besselY1' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
betainc :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
betainc' :: 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
|
|
|
|
biasAdd :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
biasAdd' :: 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
|
|
|
|
biasAddGrad :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
biasAddGrad' :: 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
|
|
|
|
biasAddV1 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
biasAddV1' :: 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
|
|
|
|
bincount :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
bincount' :: 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
|
|
|
|
bitcast :: 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') => Tensor v'1 t -> Tensor Build type'
|
|
bitcast' :: 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'
|
|
|
|
bitwiseAnd :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
bitwiseAnd' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
bitwiseOr' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
bitwiseXor' :: 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
|
|
|
|
blockLSTM :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => 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' :: 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)
|
|
|
|
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 '[Word16, Float] t => 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' :: 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 :: 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 => 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' :: 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)
|
|
|
|
blockLSTMV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 t. OneOf '[Word16, Float] t => 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' :: 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)
|
|
|
|
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
|
|
|
|
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]
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
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])
|
|
|
|
boostedTreesCenterBias :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => 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' :: 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)
|
|
|
|
boostedTreesCreateEnsemble :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
|
|
boostedTreesCreateEnsemble' :: 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
|
|
|
|
boostedTreesCreateQuantileStreamResource :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int64 -> m' ControlNode
|
|
boostedTreesCreateQuantileStreamResource' :: 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
|
|
|
|
boostedTreesDeserializeEnsemble :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
|
|
boostedTreesDeserializeEnsemble' :: 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
|
|
|
|
boostedTreesEnsembleResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
boostedTreesEnsembleResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
boostedTreesExampleDebugOutputs :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value ByteString)
|
|
boostedTreesExampleDebugOutputs' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value ByteString)
|
|
|
|
boostedTreesFlushQuantileSummaries :: forall v'1 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
|
|
boostedTreesFlushQuantileSummaries' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
|
|
|
|
boostedTreesGetEnsembleStates :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32, Tensor Value Int32)
|
|
boostedTreesGetEnsembleStates' :: 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)
|
|
|
|
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]
|
|
|
|
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
|
|
|
|
boostedTreesPredict :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value Float)
|
|
boostedTreesPredict' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> [Tensor v'2 Int32] -> m' (Tensor Value Float)
|
|
|
|
boostedTreesQuantileStreamResourceAddSummaries :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
|
|
boostedTreesQuantileStreamResourceAddSummaries' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
|
|
|
|
boostedTreesQuantileStreamResourceDeserialize :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
|
|
boostedTreesQuantileStreamResourceDeserialize' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> [Tensor v'2 Float] -> m' ControlNode
|
|
|
|
boostedTreesQuantileStreamResourceFlush :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
|
|
boostedTreesQuantileStreamResourceFlush' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
|
|
|
|
boostedTreesQuantileStreamResourceGetBucketBoundaries :: forall v'1 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
|
|
boostedTreesQuantileStreamResourceGetBucketBoundaries' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' [Tensor Value Float]
|
|
|
|
boostedTreesQuantileStreamResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
boostedTreesQuantileStreamResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
boostedTreesSerializeEnsemble :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value ByteString)
|
|
boostedTreesSerializeEnsemble' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64, Tensor Value ByteString)
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
boostedTreesTrainingPredict :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => 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' :: 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)
|
|
|
|
boostedTreesUpdateEnsemble :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 v'9 m'. MonadBuild m' => 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' :: 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
|
|
|
|
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 -> [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' :: 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
|
|
|
|
broadcastArgs :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
broadcastArgs' :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
broadcastGradientArgs :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
|
|
broadcastGradientArgs' :: 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)
|
|
|
|
broadcastTo :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
broadcastTo' :: 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
|
|
|
|
bucketize :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> Tensor Build Int32
|
|
bucketize' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Int32
|
|
|
|
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
|
|
|
|
cSRSparseMatrixComponents :: forall v'1 v'2 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> Tensor v'2 Int32 -> (Tensor Build Int32, Tensor Build Int32, Tensor Build type')
|
|
cSRSparseMatrixComponents' :: 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')
|
|
|
|
cSRSparseMatrixToDense :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> Tensor Build type'
|
|
cSRSparseMatrixToDense' :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => OpParams -> Tensor v'1 Variant -> Tensor Build type'
|
|
|
|
cSRSparseMatrixToSparseTensor :: forall v'1 type'. OneOf '[Complex Double, Complex Float, Double, Float] type' => Tensor v'1 Variant -> (Tensor Build Int64, Tensor Build type', Tensor Build Int64)
|
|
cSRSparseMatrixToSparseTensor' :: 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)
|
|
|
|
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 '[ByteString, Int32, Int64, Double, Float] output_types) => 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' :: 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)
|
|
|
|
cTCBeamSearchDecoder :: forall v'1 v'2 t. OneOf '[Double, Float] t => Int64 -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> ([Tensor Build Int64], [Tensor Build Int64], [Tensor Build Int64], Tensor Build t)
|
|
cTCBeamSearchDecoder' :: 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)
|
|
|
|
cTCGreedyDecoder :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build Int64, Tensor Build Int64, Tensor Build Int64, Tensor Build t)
|
|
cTCGreedyDecoder' :: 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)
|
|
|
|
cTCLoss :: forall v'1 v'2 v'3 v'4 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> (Tensor Build t, Tensor Build t)
|
|
cTCLoss' :: 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)
|
|
|
|
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)
|
|
|
|
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
|
|
|
|
cacheDatasetV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ResourceHandle -> m' (Tensor Value Variant)
|
|
cacheDatasetV2' :: 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)
|
|
|
|
cast :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => Tensor v'1 srcT -> Tensor Build dstT
|
|
cast' :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => OpParams -> Tensor v'1 srcT -> Tensor Build dstT
|
|
|
|
ceil :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
ceil' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
checkNumerics :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
checkNumerics' :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
checkNumericsV2 :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
checkNumericsV2' :: forall v'1 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => OpParams -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
cholesky :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
cholesky' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
choleskyGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
choleskyGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
chooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
chooseFastestDataset' :: OpParams -> Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
|
|
clipByValue :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
clipByValue' :: 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
|
|
|
|
closeSummaryWriter :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
closeSummaryWriter' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
collectiveBcastRecv :: forall t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
|
|
collectiveBcastRecv' :: forall t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> Int64 -> Int64 -> Shape -> m' (Tensor Value t)
|
|
|
|
collectiveBcastSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Bool, Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
|
|
collectiveBcastSend' :: 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)
|
|
|
|
collectiveGather :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> Int64 -> Int64 -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
|
|
collectiveGather' :: 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)
|
|
|
|
collectivePermute :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
collectivePermute' :: 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
|
|
|
|
collectiveReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => ByteString -> Int64 -> Int64 -> Int64 -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
collectiveReduce' :: 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)
|
|
|
|
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)
|
|
|
|
compareAndBitpack :: forall v'1 v'2 t. OneOf '[Bool, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Word8
|
|
compareAndBitpack' :: 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
|
|
|
|
complex :: forall v'1 v'2 t tout. (OneOf '[Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build tout
|
|
complex' :: 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
|
|
|
|
complexAbs :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
|
|
complexAbs' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
|
|
|
|
compressElement :: forall v'1 input_types. TensorTypes input_types => TensorList v'1 input_types -> Tensor Build Variant
|
|
compressElement' :: forall v'1 input_types. TensorTypes input_types => OpParams -> TensorList v'1 input_types -> Tensor Build Variant
|
|
|
|
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)
|
|
|
|
concat :: forall v'1 v'2 t. TensorType t => Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
|
|
concat' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Int32 -> [Tensor v'2 t] -> Tensor Build t
|
|
|
|
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]
|
|
|
|
concatV2 :: forall v'1 v'2 t tidx. (TensorType t, OneOf '[Int32, Int64] tidx) => [Tensor v'1 t] -> Tensor v'2 tidx -> Tensor Build t
|
|
concatV2' :: 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
|
|
|
|
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
|
|
|
|
conditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Ref ByteString)
|
|
conditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
|
|
|
|
configureDistributedTPU :: forall m'. MonadBuild m' => m' (Tensor Value ByteString)
|
|
configureDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ByteString)
|
|
|
|
configureTPUEmbedding :: forall m'. MonadBuild m' => ByteString -> m' ControlNode
|
|
configureTPUEmbedding' :: forall m'. MonadBuild m' => OpParams -> ByteString -> m' ControlNode
|
|
|
|
conj :: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t => Tensor v'1 t -> Tensor Build t
|
|
conj' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Variant] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
conjugateTranspose :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
|
|
conjugateTranspose' :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
|
|
|
|
const :: forall dtype. TensorType dtype => Tensor Build dtype
|
|
const' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
|
|
|
|
consumeMutexLock :: forall v'1 m'. MonadBuild m' => Tensor v'1 Variant -> m' ControlNode
|
|
consumeMutexLock' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> m' ControlNode
|
|
|
|
controlTrigger :: forall m'. MonadBuild m' => m' ControlNode
|
|
controlTrigger' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
|
|
|
|
conv2D :: forall v'1 v'2 t. OneOf '[Int32, Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
conv2D' :: 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
|
|
|
|
conv2DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
conv2DBackpropFilter' :: 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
|
|
|
|
conv2DBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Int32, Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
conv2DBackpropInput' :: 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
|
|
|
|
conv3D :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
conv3D' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
conv3DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
conv3DBackpropFilter' :: 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
|
|
|
|
conv3DBackpropFilterV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
conv3DBackpropFilterV2' :: 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
|
|
|
|
conv3DBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
conv3DBackpropInput' :: 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
|
|
|
|
conv3DBackpropInputV2 :: forall v'1 v'2 v'3 t tshape. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tshape) => ByteString -> Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
conv3DBackpropInputV2' :: 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
|
|
|
|
copy :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
copy' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
copyHost :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
copyHost' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
cos :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
cos' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
cosh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
cosh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
countUpTo :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => Int64 -> Tensor Ref t -> m' (Tensor Value t)
|
|
countUpTo' :: forall t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => OpParams -> Int64 -> Tensor Ref t -> m' (Tensor Value t)
|
|
|
|
createSummaryDbWriter :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
|
|
createSummaryDbWriter' :: 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
|
|
|
|
createSummaryFileWriter :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 ByteString -> m' ControlNode
|
|
createSummaryFileWriter' :: 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
|
|
|
|
cropAndResize :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build Float
|
|
cropAndResize' :: 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
|
|
|
|
cropAndResizeGradBoxes :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Int32 -> Tensor Build Float
|
|
cropAndResizeGradBoxes' :: 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
|
|
|
|
cropAndResizeGradImage :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor Build t
|
|
cropAndResizeGradImage' :: 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
|
|
|
|
cross :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
cross' :: 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
|
|
|
|
crossReplicaSum :: forall v'1 v'2 t. OneOf '[Int32, Word16, Word32, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
crossReplicaSum' :: forall v'1 v'2 t. OneOf '[Int32, Word16, Word32, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
|
|
cudnnRNN :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
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 '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
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 '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
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 '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
cudnnRNNCanonicalToParams :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 t] -> [Tensor v'5 t] -> Tensor Build t
|
|
cudnnRNNCanonicalToParams' :: 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 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> [Tensor v'4 t] -> [Tensor v'5 t] -> Tensor Build t
|
|
cudnnRNNCanonicalToParamsV2' :: 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
|
|
|
|
cudnnRNNParamsSize :: forall v'1 v'2 v'3 s. OneOf '[Int32, Int64] s => DataType -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build s
|
|
cudnnRNNParamsSize' :: 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
|
|
|
|
cudnnRNNParamsToCanonical :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Int64 -> Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> ([Tensor Build t], [Tensor Build t])
|
|
cudnnRNNParamsToCanonical' :: 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])
|
|
|
|
cudnnRNNParamsToCanonicalV2 :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => 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' :: 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])
|
|
|
|
cudnnRNNV2 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
cudnnRNNV3 :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Word16, Double, Float] t) => 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' :: 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)
|
|
|
|
cumprod :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
cumprod' :: 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 :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
cumsum' :: 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
|
|
|
|
cumulativeLogsumexp :: forall v'1 v'2 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
cumulativeLogsumexp' :: 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
|
|
|
|
dataFormatDimMap :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
|
|
dataFormatDimMap' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
dataFormatVecPermute :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
|
|
dataFormatVecPermute' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
dataServiceDataset :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 m'. MonadBuild m' => [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' :: 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)
|
|
|
|
datasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
|
|
datasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
|
|
|
|
datasetFromGraph :: Tensor v'1 ByteString -> Tensor Build Variant
|
|
datasetFromGraph' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Variant
|
|
|
|
datasetToGraph :: Tensor v'1 Variant -> Tensor Build ByteString
|
|
datasetToGraph' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
|
|
|
|
datasetToGraphV2 :: Tensor v'1 Variant -> Tensor Build ByteString
|
|
datasetToGraphV2' :: OpParams -> Tensor v'1 Variant -> Tensor Build ByteString
|
|
|
|
datasetToSingleElement :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 Variant -> m' (TensorList Value output_types)
|
|
datasetToSingleElement' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 Variant -> m' (TensorList Value output_types)
|
|
|
|
datasetToTFRecord :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
|
|
datasetToTFRecord' :: 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
|
|
|
|
dawsn :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
dawsn' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
debugGradientIdentity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
debugGradientIdentity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
debugGradientRefIdentity :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
|
|
debugGradientRefIdentity' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
|
|
|
|
debugIdentity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
debugIdentity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
debugIdentityV2 :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
|
|
debugIdentityV2' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
debugNanCount :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Int64
|
|
debugNanCount' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Int64
|
|
|
|
debugNumericSummary :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Double
|
|
debugNumericSummary' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Double
|
|
|
|
debugNumericSummaryV2 :: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t) => Tensor v'1 t -> Tensor Build output_dtype
|
|
debugNumericSummaryV2' :: forall v'1 output_dtype t. (OneOf '[Double, Float] output_dtype, TensorType t) => OpParams -> Tensor v'1 t -> Tensor Build output_dtype
|
|
|
|
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
|
|
|
|
decodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
decodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
decodeBmp :: Tensor v'1 ByteString -> Tensor Build Word8
|
|
decodeBmp' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
|
|
|
|
decodeCSV :: forall v'1 v'2 oUT_TYPE. OneOfs '[ByteString, Int32, Int64, Double, Float] oUT_TYPE => Tensor v'1 ByteString -> TensorList v'2 oUT_TYPE -> TensorList Build oUT_TYPE
|
|
decodeCSV' :: 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
|
|
|
|
decodeCompressed :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
decodeCompressed' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
decodeGif :: Tensor v'1 ByteString -> Tensor Build Word8
|
|
decodeGif' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
|
|
|
|
decodeJSONExample :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
decodeJSONExample' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
decodeJpeg :: Tensor v'1 ByteString -> Tensor Build Word8
|
|
decodeJpeg' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Word8
|
|
|
|
decodePaddedRaw :: forall v'1 v'2 out_type. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor Build out_type
|
|
decodePaddedRaw' :: 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
|
|
|
|
decodePng :: forall v'1 dtype. OneOf '[Word16, Word8] dtype => Tensor v'1 ByteString -> Tensor Build dtype
|
|
decodePng' :: forall v'1 dtype. OneOf '[Word16, Word8] dtype => OpParams -> Tensor v'1 ByteString -> Tensor Build dtype
|
|
|
|
decodeProtoV2 :: forall v'1 output_types. TensorTypes output_types => ByteString -> Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
|
|
decodeProtoV2' :: forall v'1 output_types. TensorTypes output_types => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build Int32, TensorList Build output_types)
|
|
|
|
decodeRaw :: forall v'1 out_type. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] out_type => Tensor v'1 ByteString -> Tensor Build out_type
|
|
decodeRaw' :: 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
|
|
|
|
decodeWav :: Tensor v'1 ByteString -> (Tensor Build Float, Tensor Build Int32)
|
|
decodeWav' :: OpParams -> Tensor v'1 ByteString -> (Tensor Build Float, Tensor Build Int32)
|
|
|
|
deepCopy :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
|
|
deepCopy' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
deleteIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
deleteIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
deleteMemoryCache :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
deleteMemoryCache' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
deleteMultiDeviceIterator :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> [Tensor v'2 ResourceHandle] -> Tensor v'3 Variant -> m' ControlNode
|
|
deleteMultiDeviceIterator' :: 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
|
|
|
|
deleteRandomSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
deleteRandomSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
deleteSeedGenerator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
deleteSeedGenerator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
deleteSessionTensor :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
|
|
deleteSessionTensor' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
|
|
|
|
denseBincount :: forall v'1 v'2 v'3 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 t -> Tensor Build t
|
|
denseBincount' :: 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
|
|
|
|
denseCountSparseOutput :: forall v'1 v'2 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => Bool -> Tensor v'1 t -> Tensor v'2 output_type -> (Tensor Build Int64, Tensor Build output_type, Tensor Build Int64)
|
|
denseCountSparseOutput' :: 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)
|
|
|
|
denseToCSRSparseMatrix :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor Build Variant
|
|
denseToCSRSparseMatrix' :: 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
|
|
|
|
denseToDenseSetOperation :: forall v'1 v'2 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
|
|
denseToDenseSetOperation' :: 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)
|
|
|
|
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
|
|
|
|
denseToSparseSetOperation :: forall v'1 v'2 v'3 v'4 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => 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' :: 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)
|
|
|
|
depthToSpace :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor Build t
|
|
depthToSpace' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
|
|
|
|
depthwiseConv2dNative :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
depthwiseConv2dNative' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
depthwiseConv2dNativeBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
depthwiseConv2dNativeBackpropFilter' :: 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
|
|
|
|
depthwiseConv2dNativeBackpropInput :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 Int32 -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
depthwiseConv2dNativeBackpropInput' :: 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
|
|
|
|
dequantize :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Word16, Float] dtype) => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build dtype
|
|
dequantize' :: 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
|
|
|
|
deserializeIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
deserializeIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
deserializeManySparse :: forall v'1 dtype. TensorType dtype => Tensor v'1 ByteString -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
|
|
deserializeManySparse' :: forall v'1 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
|
|
|
|
deserializeSparse :: forall v'1 dtype tserialized. (TensorType dtype, OneOf '[ByteString, Variant] tserialized) => Tensor v'1 tserialized -> (Tensor Build Int64, Tensor Build dtype, Tensor Build Int64)
|
|
deserializeSparse' :: 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)
|
|
|
|
destroyResourceOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
destroyResourceOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
destroyTemporaryVariable :: forall t m'. (MonadBuild m', TensorType t) => ByteString -> Tensor Ref t -> m' (Tensor Value t)
|
|
destroyTemporaryVariable' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Value t)
|
|
|
|
deviceIndex :: Tensor Build Int32
|
|
deviceIndex' :: OpParams -> Tensor Build Int32
|
|
|
|
diag :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
diag' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
diagPart :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
diagPart' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
digamma :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
digamma' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
dilation2D :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
dilation2D' :: 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
|
|
|
|
dilation2DBackpropFilter :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
dilation2DBackpropFilter' :: 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 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
dilation2DBackpropInput' :: 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
|
|
|
|
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
|
|
|
|
div :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
div' :: 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
|
|
|
|
divNoNan :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
divNoNan' :: 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
|
|
|
|
drawBoundingBoxes :: forall v'1 v'2 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
drawBoundingBoxes' :: forall v'1 v'2 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 Float -> Tensor Build t
|
|
|
|
drawBoundingBoxesV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor Build t
|
|
drawBoundingBoxesV2' :: 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
|
|
|
|
dummyIterationCounter :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
dummyIterationCounter' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
dummyMemoryCache :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
dummyMemoryCache' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
dummySeedGenerator :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
dummySeedGenerator' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
dynamicPartition :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
|
|
dynamicPartition' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
|
|
|
|
dynamicStitch :: forall v'1 v'2 t. TensorType t => [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
|
|
dynamicStitch' :: forall v'1 v'2 t. TensorType t => OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
|
|
|
|
eagerPyFunc :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
|
|
eagerPyFunc' :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
|
|
|
|
editDistance :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. TensorType t => 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' :: 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
|
|
|
|
eig :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float, Double, Float] t, OneOf '[Complex Double, Complex Float] tout) => Tensor v'1 t -> (Tensor Build tout, Tensor Build tout)
|
|
eig' :: 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)
|
|
|
|
einsum :: forall v'1 t. TensorType t => ByteString -> [Tensor v'1 t] -> Tensor Build t
|
|
einsum' :: forall v'1 t. TensorType t => OpParams -> ByteString -> [Tensor v'1 t] -> Tensor Build t
|
|
|
|
elu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
elu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
eluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
eluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
empty :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 Int32 -> m' (Tensor Value dtype)
|
|
empty' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value dtype)
|
|
|
|
emptyTensorList :: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type => DataType -> Tensor v'1 shape_type -> Tensor v'2 Int32 -> Tensor Build Variant
|
|
emptyTensorList' :: 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
|
|
|
|
encodeBase64 :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
encodeBase64' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
encodeJpeg :: Tensor v'1 Word8 -> Tensor Build ByteString
|
|
encodeJpeg' :: OpParams -> Tensor v'1 Word8 -> Tensor Build ByteString
|
|
|
|
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
|
|
|
|
encodePng :: forall v'1 t. OneOf '[Word16, Word8] t => Tensor v'1 t -> Tensor Build ByteString
|
|
encodePng' :: forall v'1 t. OneOf '[Word16, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
|
|
|
|
encodeProto :: forall v'1 v'2 tinput_types. TensorTypes tinput_types => ByteString -> Tensor v'1 Int32 -> TensorList v'2 tinput_types -> Tensor Build ByteString
|
|
encodeProto' :: forall v'1 v'2 tinput_types. TensorTypes tinput_types => OpParams -> ByteString -> Tensor v'1 Int32 -> TensorList v'2 tinput_types -> Tensor Build ByteString
|
|
|
|
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
|
|
|
|
enqueueTPUEmbeddingIntegerBatch :: forall v'1 v'2 m'. MonadBuild m' => [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
|
|
enqueueTPUEmbeddingIntegerBatch' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
enqueueTPUEmbeddingRaggedTensorBatch :: 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) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
|
|
enqueueTPUEmbeddingRaggedTensorBatch' :: 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 :: 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) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
|
|
enqueueTPUEmbeddingSparseBatch' :: 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 :: 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) => [Tensor v'1 t1] -> [Tensor v'2 t2] -> [Tensor v'3 t3] -> Tensor v'4 ByteString -> m' ControlNode
|
|
enqueueTPUEmbeddingSparseTensorBatch' :: 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
|
|
|
|
ensureShape :: forall v'1 t. TensorType t => Shape -> Tensor v'1 t -> Tensor Build t
|
|
ensureShape' :: forall v'1 t. TensorType t => OpParams -> Shape -> Tensor v'1 t -> Tensor Build t
|
|
|
|
enter :: forall v'1 t. TensorType t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
enter' :: forall v'1 t. TensorType t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
equal :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
equal' :: 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
|
|
|
|
erf :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
erf' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
erfc :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
erfc' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
erfinv :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
erfinv' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
euclideanNorm :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
euclideanNorm' :: 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
|
|
|
|
exit :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
exit' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
exp :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
exp' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
expandDims :: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim) => Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
|
|
expandDims' :: forall v'1 v'2 t tdim. (TensorType t, OneOf '[Int32, Int64] tdim) => OpParams -> Tensor v'1 t -> Tensor v'2 tdim -> Tensor Build t
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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 '[ByteString, Int32, Int64, Double, Float] output_types) => 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' :: 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)
|
|
|
|
experimentalChooseFastestDataset :: Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
experimentalChooseFastestDataset' :: OpParams -> Int64 -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
|
|
experimentalDatasetCardinality :: Tensor v'1 Variant -> Tensor Build Int64
|
|
experimentalDatasetCardinality' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int64
|
|
|
|
experimentalDatasetToTFRecord :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' ControlNode
|
|
experimentalDatasetToTFRecord' :: 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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
experimentalIgnoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
experimentalIgnoreErrorsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
experimentalIteratorGetDevice :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
experimentalIteratorGetDevice' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
experimentalLMDBDataset :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
experimentalLMDBDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
|
|
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
|
|
|
|
experimentalMatchingFilesDataset :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
experimentalMatchingFilesDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
|
|
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
|
|
|
|
experimentalNonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
experimentalNonSerializableDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
experimentalParseExampleDataset :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
|
|
experimentalParseExampleDataset' :: 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
|
|
|
|
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
|
|
|
|
experimentalRandomDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
|
|
experimentalRandomDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
|
|
|
|
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
|
|
|
|
experimentalSetStatsAggregatorDataset :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> m' (Tensor Value Variant)
|
|
experimentalSetStatsAggregatorDataset' :: 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)
|
|
|
|
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
|
|
|
|
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
|
|
|
|
experimentalSqlDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value Variant)
|
|
experimentalSqlDataset' :: 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)
|
|
|
|
experimentalStatsAggregatorHandle :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
experimentalStatsAggregatorHandle' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
experimentalStatsAggregatorSummary :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
experimentalStatsAggregatorSummary' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
experimentalThreadPoolDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
|
|
experimentalThreadPoolDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
|
|
|
|
experimentalThreadPoolHandle :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
|
|
experimentalThreadPoolHandle' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
|
|
|
|
experimentalUnbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
experimentalUnbatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
experimentalUniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
experimentalUniqueDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
expint :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
expint' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
expm1 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
expm1' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
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
|
|
|
|
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
|
|
|
|
extractImagePatches :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
extractImagePatches' :: 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
|
|
|
|
extractJpegShape :: forall v'1 output_type. OneOf '[Int32, Int64] output_type => Tensor v'1 ByteString -> Tensor Build output_type
|
|
extractJpegShape' :: forall v'1 output_type. OneOf '[Int32, Int64] output_type => OpParams -> Tensor v'1 ByteString -> Tensor Build output_type
|
|
|
|
extractVolumePatches :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
extractVolumePatches' :: 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
|
|
|
|
fFT :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
fFT' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
fFT2D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
fFT2D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
fFT3D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
fFT3D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
fIFOQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
|
|
fIFOQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
|
|
|
|
fIFOQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
|
|
fIFOQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
|
|
|
|
fact :: Tensor Build ByteString
|
|
fact' :: OpParams -> Tensor Build ByteString
|
|
|
|
fakeParam :: forall dtype. TensorType dtype => Shape -> Tensor Build dtype
|
|
fakeParam' :: forall dtype. TensorType dtype => OpParams -> Shape -> Tensor Build dtype
|
|
|
|
fakeQuantWithMinMaxArgs :: Tensor v'1 Float -> Tensor Build Float
|
|
fakeQuantWithMinMaxArgs' :: OpParams -> Tensor v'1 Float -> Tensor Build Float
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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)
|
|
|
|
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
|
|
|
|
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)
|
|
|
|
fakeQueue :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
|
|
fakeQueue' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Ref ByteString)
|
|
|
|
fill :: forall v'1 v'2 t index_type. (TensorType t, OneOf '[Int32, Int64] index_type) => Tensor v'1 index_type -> Tensor v'2 t -> Tensor Build t
|
|
fill' :: 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
|
|
|
|
filterByLastComponentDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
filterByLastComponentDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
fingerprint :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
|
|
fingerprint' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 ByteString -> Tensor Build Word8
|
|
|
|
fixedLengthRecordDataset :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => 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' :: 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)
|
|
|
|
fixedLengthRecordDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => 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' :: 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)
|
|
|
|
fixedLengthRecordReader :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Ref ByteString)
|
|
fixedLengthRecordReader' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Ref ByteString)
|
|
|
|
fixedLengthRecordReaderV2 :: forall m'. MonadBuild m' => Int64 -> m' (Tensor Value ResourceHandle)
|
|
fixedLengthRecordReaderV2' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' (Tensor Value ResourceHandle)
|
|
|
|
fixedUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
fixedUnigramCandidateSampler' :: 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)
|
|
|
|
floor :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
floor' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
floorDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
floorDiv' :: 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
|
|
|
|
floorMod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Word64, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
floorMod' :: 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
|
|
|
|
flushSummaryWriter :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
flushSummaryWriter' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
fractionalAvgPool :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
|
|
fractionalAvgPool' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
|
|
|
|
fractionalAvgPoolGrad :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build t
|
|
fractionalAvgPoolGrad' :: 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
|
|
|
|
fractionalMaxPool :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
|
|
fractionalMaxPool' :: forall v'1 t. OneOf '[Int32, Int64, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int64, Tensor Build Int64)
|
|
|
|
fractionalMaxPoolGrad :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int32, Int64, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor v'5 Int64 -> Tensor Build t
|
|
fractionalMaxPoolGrad' :: 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
|
|
|
|
fresnelCos :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
fresnelCos' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
fresnelSin :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
fresnelSin' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
fusedBatchNorm :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => 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' :: 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 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Float] t => 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' :: 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)
|
|
|
|
fusedBatchNormGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => 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' :: 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)
|
|
|
|
fusedBatchNormGradV3 :: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => 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' :: 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)
|
|
|
|
fusedBatchNormV2 :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => 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' :: 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)
|
|
|
|
fusedBatchNormV3 :: forall v'1 v'2 v'3 v'4 v'5 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => 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' :: 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)
|
|
|
|
fusedPadConv2D :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
fusedPadConv2D' :: 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
|
|
|
|
fusedResizeAndPadConv2D :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 t -> Tensor Build t
|
|
fusedResizeAndPadConv2D' :: 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
|
|
|
|
gRUBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Float] t => 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' :: 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)
|
|
|
|
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 -> 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' :: 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)
|
|
|
|
gather :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
|
|
gather' :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
|
|
|
|
gatherNd :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
|
|
gatherNd' :: forall v'1 v'2 tparams tindices. (TensorType tparams, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor Build tparams
|
|
|
|
gatherV2 :: forall v'1 v'2 v'3 tparams tindices taxis. (TensorType tparams, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] taxis) => Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor v'3 taxis -> Tensor Build tparams
|
|
gatherV2' :: forall v'1 v'2 v'3 tparams tindices taxis. (TensorType tparams, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] taxis) => OpParams -> Tensor v'1 tparams -> Tensor v'2 tindices -> Tensor v'3 taxis -> Tensor Build tparams
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
getSessionHandle :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value ByteString)
|
|
getSessionHandle' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value ByteString)
|
|
|
|
getSessionHandleV2 :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value ResourceHandle)
|
|
getSessionHandleV2' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value ResourceHandle)
|
|
|
|
getSessionTensor :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ByteString -> m' (Tensor Value dtype)
|
|
getSessionTensor' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value dtype)
|
|
|
|
greater :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
greater' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
greaterEqual' :: 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
|
|
|
|
guaranteeConst :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
|
|
guaranteeConst' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
hSVToRGB :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
hSVToRGB' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
hashTable :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
hashTable' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
|
|
hashTableV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
hashTableV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
|
|
histogramFixedWidth :: forall v'1 v'2 v'3 t dtype. (OneOf '[Int32, Int64, Double, Float] t, OneOf '[Int32, Int64] dtype) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build dtype
|
|
histogramFixedWidth' :: 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
|
|
|
|
histogramSummary :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
|
|
histogramSummary' :: 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
|
|
|
|
hostConst :: forall dtype. TensorType dtype => Tensor Build dtype
|
|
hostConst' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
|
|
|
|
iFFT :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
iFFT' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
iFFT2D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
iFFT2D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
iFFT3D :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
iFFT3D' :: forall v'1 tcomplex. OneOf '[Complex Double, Complex Float] tcomplex => OpParams -> Tensor v'1 tcomplex -> Tensor Build tcomplex
|
|
|
|
iRFFT :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
|
|
iRFFT' :: 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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
|
|
iRFFT2D' :: 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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 tcomplex -> Tensor v'2 Int32 -> Tensor Build treal
|
|
iRFFT3D' :: 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
|
|
|
|
identity :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
identity' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
identityN :: forall v'1 t. TensorTypes t => TensorList v'1 t -> TensorList Build t
|
|
identityN' :: forall v'1 t. TensorTypes t => OpParams -> TensorList v'1 t -> TensorList Build t
|
|
|
|
identityReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
identityReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
identityReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
identityReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
igamma :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
igamma' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
igammaGradA :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
igammaGradA' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
igammac :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
igammac' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
ignoreErrorsDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
ignoreErrorsDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
imag :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
|
|
imag' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
|
|
|
|
imageProjectiveTransformV2 :: forall v'1 v'2 v'3 dtype. OneOf '[Int32, Int64, Word16, Word8, Double, Float] dtype => ByteString -> Tensor v'1 dtype -> Tensor v'2 Float -> Tensor v'3 Int32 -> Tensor Build dtype
|
|
imageProjectiveTransformV2' :: 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
|
|
|
|
imageSummary :: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
|
|
imageSummary' :: forall v'1 v'2 t. OneOf '[Word16, Word8, Double, Float] t => OpParams -> Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
|
|
|
|
immutableConst :: forall dtype. TensorType dtype => ByteString -> Shape -> Tensor Build dtype
|
|
immutableConst' :: forall dtype. TensorType dtype => OpParams -> ByteString -> Shape -> Tensor Build dtype
|
|
|
|
importEvent :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
importEvent' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
inTopK :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
|
|
inTopK' :: forall v'1 v'2 t. OneOf '[Int32, Int64] t => OpParams -> Int64 -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build Bool
|
|
|
|
inTopKV2 :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build Bool
|
|
inTopKV2' :: 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
|
|
|
|
infeedDequeue :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
|
|
infeedDequeue' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
|
|
|
|
infeedDequeueTuple :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
|
|
infeedDequeueTuple' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
|
|
|
|
infeedEnqueue :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 dtype -> m' ControlNode
|
|
infeedEnqueue' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 dtype -> m' ControlNode
|
|
|
|
infeedEnqueuePrelinearizedBuffer :: forall v'1 m'. MonadBuild m' => Tensor v'1 Variant -> m' ControlNode
|
|
infeedEnqueuePrelinearizedBuffer' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> m' ControlNode
|
|
|
|
infeedEnqueueTuple :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
|
|
infeedEnqueueTuple' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
|
|
|
|
initializeTable :: forall v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => Tensor Ref ByteString -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
|
|
initializeTable' :: 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
|
|
|
|
initializeTableFromDataset :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
initializeTableFromDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Variant -> m' ControlNode
|
|
|
|
initializeTableFromTextFile :: forall v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
initializeTableFromTextFile' :: forall v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
initializeTableFromTextFileV2 :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
initializeTableFromTextFileV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
initializeTableV2 :: forall v'1 v'2 v'3 tkey tval m'. (MonadBuild m', TensorType tkey, TensorType tval) => Tensor v'1 ResourceHandle -> Tensor v'2 tkey -> Tensor v'3 tval -> m' ControlNode
|
|
initializeTableV2' :: 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
|
|
|
|
inplaceAdd :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
inplaceAdd' :: 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 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
inplaceSub' :: 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 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
inplaceUpdate' :: 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
|
|
|
|
inv :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
inv' :: 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
|
|
|
|
invGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
invGrad' :: 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
|
|
|
|
invert :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor Build t
|
|
invert' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
invertPermutation :: forall v'1 t. OneOf '[Int32, Int64] t => Tensor v'1 t -> Tensor Build t
|
|
invertPermutation' :: forall v'1 t. OneOf '[Int32, Int64] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
isBoostedTreesEnsembleInitialized :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
isBoostedTreesEnsembleInitialized' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
|
|
isBoostedTreesQuantileStreamResourceInitialized :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
isBoostedTreesQuantileStreamResourceInitialized' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
|
|
isFinite :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
|
|
isFinite' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
|
|
|
|
isInf :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
|
|
isInf' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
|
|
|
|
isNan :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build Bool
|
|
isNan' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build Bool
|
|
|
|
isVariableInitialized :: forall dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref dtype -> m' (Tensor Value Bool)
|
|
isVariableInitialized' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref dtype -> m' (Tensor Value Bool)
|
|
|
|
iterator :: forall m'. MonadBuild m' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
iterator' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
iteratorFromStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
iteratorFromStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
iteratorFromStringHandleV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
iteratorFromStringHandleV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
iteratorGetDevice :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
iteratorGetDevice' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
iteratorGetNext :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
|
|
iteratorGetNext' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
|
|
|
|
iteratorGetNextAsOptional :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
iteratorGetNextAsOptional' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
|
|
iteratorGetNextSync :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
|
|
iteratorGetNextSync' :: forall v'1 output_types m'. (MonadBuild m', TensorTypes output_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value output_types)
|
|
|
|
iteratorToStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
iteratorToStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
iteratorV2 :: forall m'. MonadBuild m' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
iteratorV2' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
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
|
|
|
|
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
|
|
|
|
l2Loss :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
l2Loss' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
lMDBDataset :: forall v'1 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
lMDBDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
|
|
lMDBReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
lMDBReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
lRN :: forall v'1 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor Build t
|
|
lRN' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
lRNGrad :: forall v'1 v'2 v'3 t. OneOf '[Word16, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
lRNGrad' :: 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
|
|
|
|
lSTMBlockCell :: forall v'1 v'2 v'3 v'4 v'5 v'6 v'7 v'8 t. OneOf '[Word16, Float] t => 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' :: 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)
|
|
|
|
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 '[Word16, Float] t => 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' :: 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)
|
|
|
|
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
|
|
|
|
leakyRelu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
leakyRelu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
leakyReluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
leakyReluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
learnedUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
learnedUnigramCandidateSampler' :: 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)
|
|
|
|
leftShift :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
leftShift' :: 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
|
|
|
|
less :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
less' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
lessEqual' :: 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
|
|
|
|
lgamma :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
lgamma' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
linSpace :: forall v'1 v'2 v'3 t tidx. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tidx -> Tensor Build t
|
|
linSpace' :: 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
|
|
|
|
listDiff :: forall v'1 v'2 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build out_idx)
|
|
listDiff' :: 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)
|
|
|
|
loadAndRemapMatrix :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => 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' :: 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)
|
|
|
|
loadTPUEmbeddingADAMParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingADAMParameters' :: 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
|
|
|
|
loadTPUEmbeddingADAMParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingADAMParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingAdadeltaParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingAdadeltaParameters' :: 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
|
|
|
|
loadTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingAdadeltaParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingAdagradParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
loadTPUEmbeddingAdagradParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
|
|
loadTPUEmbeddingAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingAdagradParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingCenteredRMSPropParameters :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingCenteredRMSPropParameters' :: 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
|
|
|
|
loadTPUEmbeddingFTRLParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingFTRLParameters' :: 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
|
|
|
|
loadTPUEmbeddingFTRLParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingFTRLParametersGradAccumDebug' :: 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 :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingMDLAdagradLightParameters' :: 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
|
|
|
|
loadTPUEmbeddingMomentumParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
loadTPUEmbeddingMomentumParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
|
|
loadTPUEmbeddingMomentumParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingMomentumParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingProximalAdagradParameters :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
loadTPUEmbeddingProximalAdagradParameters' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
|
|
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: 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 :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingProximalYogiParameters' :: 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
|
|
|
|
loadTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingProximalYogiParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingRMSPropParameters :: forall v'1 v'2 v'3 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> m' ControlNode
|
|
loadTPUEmbeddingRMSPropParameters' :: 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
|
|
|
|
loadTPUEmbeddingRMSPropParametersGradAccumDebug :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> Tensor v'4 Float -> m' ControlNode
|
|
loadTPUEmbeddingRMSPropParametersGradAccumDebug' :: 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
|
|
|
|
loadTPUEmbeddingStochasticGradientDescentParameters :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
|
|
loadTPUEmbeddingStochasticGradientDescentParameters' :: forall v'1 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> m' ControlNode
|
|
|
|
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
loadTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> Tensor v'1 Float -> Tensor v'2 Float -> m' ControlNode
|
|
|
|
log :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
log' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
log1p :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
log1p' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
logMatrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
logMatrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
|
|
logSoftmax :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
logSoftmax' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
logUniformCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
logUniformCandidateSampler' :: 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)
|
|
|
|
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
|
|
|
|
logicalNot :: Tensor v'1 Bool -> Tensor Build Bool
|
|
logicalNot' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
|
|
|
|
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
|
|
|
|
lookupTableExport :: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => Tensor Ref ByteString -> m' (Tensor Value tkeys, Tensor Value tvalues)
|
|
lookupTableExport' :: forall tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => OpParams -> Tensor Ref ByteString -> m' (Tensor Value tkeys, Tensor Value tvalues)
|
|
|
|
lookupTableExportV2 :: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => Tensor v'1 ResourceHandle -> m' (Tensor Value tkeys, Tensor Value tvalues)
|
|
lookupTableExportV2' :: forall v'1 tkeys tvalues m'. (MonadBuild m', TensorType tkeys, TensorType tvalues) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value tkeys, Tensor Value tvalues)
|
|
|
|
lookupTableFind :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
|
|
lookupTableFind' :: 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)
|
|
|
|
lookupTableFindV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' (Tensor Value tout)
|
|
lookupTableFindV2' :: 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)
|
|
|
|
lookupTableImport :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
|
|
lookupTableImport' :: 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' ControlNode
|
|
|
|
lookupTableImportV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
|
|
lookupTableImportV2' :: 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' ControlNode
|
|
|
|
lookupTableInsert :: forall v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor Ref ByteString -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
|
|
lookupTableInsert' :: 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' ControlNode
|
|
|
|
lookupTableInsertV2 :: forall v'1 v'2 v'3 tin tout m'. (MonadBuild m', TensorType tin, TensorType tout) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> Tensor v'3 tout -> m' ControlNode
|
|
lookupTableInsertV2' :: 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' ControlNode
|
|
|
|
lookupTableRemoveV2 :: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin) => Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
|
|
lookupTableRemoveV2' :: forall v'1 v'2 tin m'. (MonadBuild m', TensorType tin) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tin -> m' ControlNode
|
|
|
|
lookupTableSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
lookupTableSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
|
|
lookupTableSizeV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
lookupTableSizeV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
|
|
loopCond :: Tensor v'1 Bool -> Tensor Build Bool
|
|
loopCond' :: OpParams -> Tensor v'1 Bool -> Tensor Build Bool
|
|
|
|
lowerBound :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
|
|
lowerBound' :: 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
|
|
|
|
lu :: forall v'1 t output_idx_type. (OneOf '[Complex Double, Complex Float, Word16, Double, Float] t, OneOf '[Int32, Int64] output_idx_type) => Tensor v'1 t -> (Tensor Build t, Tensor Build output_idx_type)
|
|
lu' :: 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)
|
|
|
|
makeIterator :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' ControlNode
|
|
makeIterator' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' ControlNode
|
|
|
|
mapClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
|
|
mapClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
|
|
|
|
mapIncompleteSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
|
|
mapIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
|
|
|
|
mapPeek :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
mapPeek' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
|
|
mapSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
|
|
mapSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
|
|
|
|
mapStage :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
|
|
mapStage' :: 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
|
|
|
|
mapUnstage :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
mapUnstage' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
|
|
mapUnstageNoKey :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
|
|
mapUnstageNoKey' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
|
|
|
|
matMul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
matMul' :: 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
|
|
|
|
matchingFiles :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
matchingFiles' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
matchingFilesDataset :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
matchingFilesDataset' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Variant)
|
|
|
|
matrixBandPart :: forall v'1 v'2 v'3 t tindex. (TensorType t, OneOf '[Int32, Int64] tindex) => Tensor v'1 t -> Tensor v'2 tindex -> Tensor v'3 tindex -> Tensor Build t
|
|
matrixBandPart' :: 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
|
|
|
|
matrixDeterminant :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
matrixDeterminant' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixDiag :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
matrixDiag' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixDiagPart :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
matrixDiagPart' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixDiagPartV2 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
matrixDiagPartV2' :: 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 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor Build t
|
|
matrixDiagPartV3' :: 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
|
|
|
|
matrixDiagV2 :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
|
|
matrixDiagV2' :: 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 :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 t -> Tensor Build t
|
|
matrixDiagV3' :: 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
|
|
|
|
matrixExponential :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
matrixExponential' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixInverse :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
matrixInverse' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixLogarithm :: forall v'1 t. OneOf '[Complex Double, Complex Float] t => Tensor v'1 t -> Tensor Build t
|
|
matrixLogarithm' :: forall v'1 t. OneOf '[Complex Double, Complex Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixSetDiag :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
matrixSetDiag' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
matrixSetDiagV2 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
|
|
matrixSetDiagV2' :: 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 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor Build t
|
|
matrixSetDiagV3' :: 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
|
|
|
|
matrixSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
matrixSolve' :: 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
|
|
|
|
matrixSolveLs :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Double -> Tensor Build t
|
|
matrixSolveLs' :: 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
|
|
|
|
matrixSquareRoot :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
matrixSquareRoot' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
matrixTriangularSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
matrixTriangularSolve' :: 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
|
|
|
|
max :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
max' :: 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
|
|
|
|
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
|
|
|
|
maxPool :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
maxPool' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
maxPool3D :: forall v'1 t. OneOf '[Word16, Float] t => ByteString -> Tensor v'1 t -> Tensor Build t
|
|
maxPool3D' :: forall v'1 t. OneOf '[Word16, Float] t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
maxPool3DGrad :: forall v'1 v'2 v'3 t tInput. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] tInput) => ByteString -> Tensor v'1 tInput -> Tensor v'2 tInput -> Tensor v'3 t -> Tensor Build t
|
|
maxPool3DGrad' :: 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
|
|
|
|
maxPool3DGradGrad :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
maxPool3DGradGrad' :: 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 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
maxPoolGrad' :: 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 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
maxPoolGradGrad' :: 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
|
|
|
|
maxPoolGradGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => 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' :: 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
|
|
|
|
maxPoolGradGradWithArgmax :: 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) => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
|
|
maxPoolGradGradWithArgmax' :: 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
|
|
|
|
maxPoolGradV2 :: forall v'1 v'2 v'3 v'4 v'5 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => 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' :: 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
|
|
|
|
maxPoolGradWithArgmax :: 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) => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 targmax -> Tensor Build t
|
|
maxPoolGradWithArgmax' :: 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
|
|
|
|
maxPoolV2 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build t
|
|
maxPoolV2' :: 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
|
|
|
|
maxPoolWithArgmax :: forall v'1 targmax t. (OneOf '[Int32, Int64] targmax, OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => ByteString -> Tensor v'1 t -> (Tensor Build t, Tensor Build targmax)
|
|
maxPoolWithArgmax' :: 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)
|
|
|
|
maximum :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
maximum' :: 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
|
|
|
|
mean :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
mean' :: 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
|
|
|
|
merge :: forall v'1 t. TensorType t => [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
|
|
merge' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> (Tensor Build t, Tensor Build Int32)
|
|
|
|
mergeSummary :: [Tensor v'1 ByteString] -> Tensor Build ByteString
|
|
mergeSummary' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
|
|
|
|
mergeV2Checkpoints :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
mergeV2Checkpoints' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
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
|
|
|
|
min :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
min' :: 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
|
|
|
|
minimum :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
minimum' :: 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
|
|
|
|
mirrorPad :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => ByteString -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
mirrorPad' :: 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 :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => ByteString -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
mirrorPadGrad' :: 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
|
|
|
|
mlirPassthroughOp :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
|
|
mlirPassthroughOp' :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => OpParams -> ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
|
|
|
|
mod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
mod' :: 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
|
|
|
|
modelDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
modelDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
mul :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
mul' :: 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
|
|
|
|
mulNoNan :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
mulNoNan' :: 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
|
|
|
|
multiDeviceIterator :: forall m'. MonadBuild m' => ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
multiDeviceIterator' :: forall m'. MonadBuild m' => OpParams -> ByteString -> [DataType] -> ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
multiDeviceIteratorFromStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
multiDeviceIteratorFromStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ResourceHandle)
|
|
|
|
multiDeviceIteratorGetNextFromShard :: forall v'1 v'2 v'3 output_types m'. (MonadBuild m', TensorTypes output_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Int64 -> m' (TensorList Value output_types)
|
|
multiDeviceIteratorGetNextFromShard' :: 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)
|
|
|
|
multiDeviceIteratorInit :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value Int64)
|
|
multiDeviceIteratorInit' :: 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)
|
|
|
|
multiDeviceIteratorToStringHandle :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
multiDeviceIteratorToStringHandle' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
multinomial :: 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) => Tensor v'1 t -> Tensor v'2 Int32 -> m' (Tensor Value output_dtype)
|
|
multinomial' :: 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)
|
|
|
|
mutableDenseHashTable :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
|
|
mutableDenseHashTable' :: forall v'1 key_dtype m'. (MonadBuild m', TensorType key_dtype) => OpParams -> DataType -> Tensor v'1 key_dtype -> m' (Tensor Ref ByteString)
|
|
|
|
mutableDenseHashTableV2 :: forall v'1 v'2 key_dtype m'. (MonadBuild m', TensorType key_dtype) => DataType -> Tensor v'1 key_dtype -> Tensor v'2 key_dtype -> m' (Tensor Value ResourceHandle)
|
|
mutableDenseHashTableV2' :: 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)
|
|
|
|
mutableHashTable :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
mutableHashTable' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
|
|
mutableHashTableOfTensors :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
mutableHashTableOfTensors' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Ref ByteString)
|
|
|
|
mutableHashTableOfTensorsV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
mutableHashTableOfTensorsV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
|
|
mutableHashTableV2 :: forall m'. MonadBuild m' => DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
mutableHashTableV2' :: forall m'. MonadBuild m' => OpParams -> DataType -> DataType -> m' (Tensor Value ResourceHandle)
|
|
|
|
mutexLock :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
mutexLock' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
|
|
mutexV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
mutexV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
ncclAllReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
ncclAllReduce' :: 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)
|
|
|
|
ncclBroadcast :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Shape -> Tensor v'1 t -> m' (Tensor Value t)
|
|
ncclBroadcast' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Shape -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
ncclReduce :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
|
|
ncclReduce' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> ByteString -> [Tensor v'1 t] -> m' (Tensor Value t)
|
|
|
|
ndtri :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
ndtri' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
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)
|
|
|
|
neg :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
neg' :: 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
|
|
|
|
negTrain :: forall v'3 v'4 v'5 m'. MonadBuild m' => Int64 -> Tensor Ref Float -> Tensor Ref Float -> Tensor v'3 Int32 -> Tensor v'4 Int32 -> Tensor v'5 Float -> m' ControlNode
|
|
negTrain' :: 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
|
|
|
|
nextAfter :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
nextAfter' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
nextIteration :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
nextIteration' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
noOp :: forall m'. MonadBuild m' => m' ControlNode
|
|
noOp' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
|
|
|
|
nonDeterministicInts :: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
|
|
nonDeterministicInts' :: forall v'1 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => OpParams -> Tensor v'1 shape_dtype -> m' (Tensor Value dtype)
|
|
|
|
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
|
|
|
|
nonMaxSuppressionV2 :: forall v'1 v'2 v'3 v'4 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Int32 -> Tensor v'4 t_threshold -> Tensor Build Int32
|
|
nonMaxSuppressionV2' :: 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
|
|
|
|
nonMaxSuppressionV3 :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => 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' :: 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
|
|
|
|
nonMaxSuppressionV4 :: forall v'1 v'2 v'3 v'4 v'5 t t_threshold. (OneOf '[Word16, Float] t, OneOf '[Word16, Float] t_threshold) => 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' :: 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)
|
|
|
|
nonMaxSuppressionV5 :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[Word16, Float] t => 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' :: 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)
|
|
|
|
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
|
|
|
|
nonSerializableDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
nonSerializableDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
notEqual :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build Bool
|
|
notEqual' :: 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
|
|
|
|
nthElement :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
nthElement' :: 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
|
|
|
|
oneHot :: forall v'1 v'2 v'3 v'4 t tI. (TensorType t, OneOf '[Int32, Int64, Word8] tI) => Tensor v'1 tI -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
|
|
oneHot' :: 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
|
|
|
|
onesLike :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
onesLike' :: 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
|
|
|
|
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
|
|
|
|
optionalFromValue :: forall v'1 toutput_types. TensorTypes toutput_types => TensorList v'1 toutput_types -> Tensor Build Variant
|
|
optionalFromValue' :: forall v'1 toutput_types. TensorTypes toutput_types => OpParams -> TensorList v'1 toutput_types -> Tensor Build Variant
|
|
|
|
optionalGetValue :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 Variant -> TensorList Build output_types
|
|
optionalGetValue' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 Variant -> TensorList Build output_types
|
|
|
|
optionalHasValue :: Tensor v'1 Variant -> Tensor Build Bool
|
|
optionalHasValue' :: OpParams -> Tensor v'1 Variant -> Tensor Build Bool
|
|
|
|
optionalNone :: Tensor Build Variant
|
|
optionalNone' :: OpParams -> Tensor Build Variant
|
|
|
|
orderedMapClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
|
|
orderedMapClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
|
|
|
|
orderedMapIncompleteSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
|
|
orderedMapIncompleteSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
|
|
|
|
orderedMapPeek :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
orderedMapPeek' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
|
|
orderedMapSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
|
|
orderedMapSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
|
|
|
|
orderedMapStage :: forall v'1 v'2 v'3 fake_dtypes m'. (MonadBuild m', TensorTypes fake_dtypes) => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> TensorList v'3 fake_dtypes -> m' ControlNode
|
|
orderedMapStage' :: 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
|
|
|
|
orderedMapUnstage :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
orderedMapUnstage' :: forall v'1 v'2 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int64 -> Tensor v'2 Int32 -> m' (TensorList Value dtypes)
|
|
|
|
orderedMapUnstageNoKey :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
|
|
orderedMapUnstageNoKey' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (Tensor Value Int64, TensorList Value dtypes)
|
|
|
|
outfeedDequeue :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
|
|
outfeedDequeue' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
|
|
|
|
outfeedDequeueTuple :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
|
|
outfeedDequeueTuple' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
|
|
|
|
outfeedEnqueue :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 dtype -> m' ControlNode
|
|
outfeedEnqueue' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 dtype -> m' ControlNode
|
|
|
|
outfeedEnqueueTuple :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
|
|
outfeedEnqueueTuple' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
|
|
|
|
pack :: forall v'1 t. TensorType t => [Tensor v'1 t] -> Tensor Build t
|
|
pack' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> Tensor Build t
|
|
|
|
pad :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
pad' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
|
|
padV2 :: forall v'1 v'2 v'3 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor v'3 t -> Tensor Build t
|
|
padV2' :: 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
|
|
|
|
paddedBatchDataset :: forall v'1 v'2 v'3 v'4 toutput_types. TensorTypes toutput_types => Tensor v'1 Variant -> Tensor v'2 Int64 -> [Tensor v'3 Int64] -> TensorList v'4 toutput_types -> Tensor Build Variant
|
|
paddedBatchDataset' :: 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
|
|
|
|
paddedBatchDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 toutput_types. TensorTypes toutput_types => 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' :: 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
|
|
|
|
paddingFIFOQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
|
|
paddingFIFOQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
|
|
|
|
paddingFIFOQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
|
|
paddingFIFOQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
|
|
|
|
parallelConcat :: forall v'1 t. TensorType t => Shape -> [Tensor v'1 t] -> Tensor Build t
|
|
parallelConcat' :: forall v'1 t. TensorType t => OpParams -> Shape -> [Tensor v'1 t] -> Tensor Build t
|
|
|
|
parallelDynamicStitch :: forall v'1 v'2 t. TensorType t => [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
|
|
parallelDynamicStitch' :: forall v'1 v'2 t. TensorType t => OpParams -> [Tensor v'1 Int32] -> [Tensor v'2 t] -> Tensor Build t
|
|
|
|
parameterizedTruncatedNormal :: 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) => 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' :: 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)
|
|
|
|
parseExample :: 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) => 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' :: 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)
|
|
|
|
parseExampleDataset :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
|
|
parseExampleDataset' :: 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 :: forall v'1 v'2 v'3 tdense. OneOfs '[ByteString, Int64, Float] tdense => [DataType] -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> TensorList v'3 tdense -> Tensor Build Variant
|
|
parseExampleDatasetV2' :: 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
|
|
|
|
parseExampleV2 :: 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) => 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' :: 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)
|
|
|
|
parseSingleExample :: forall v'1 v'2 sparse_types tdense. (OneOfs '[ByteString, Int64, Float] sparse_types, OneOfs '[ByteString, Int64, Float] tdense) => Int64 -> Tensor v'1 ByteString -> TensorList v'2 tdense -> ([Tensor Build Int64], TensorList Build sparse_types, [Tensor Build Int64], TensorList Build tdense)
|
|
parseSingleExample' :: 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)
|
|
|
|
parseTensor :: forall v'1 out_type. TensorType out_type => Tensor v'1 ByteString -> Tensor Build out_type
|
|
parseTensor' :: forall v'1 out_type. TensorType out_type => OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
|
|
|
|
placeholder :: forall dtype. TensorType dtype => Tensor Build dtype
|
|
placeholder' :: forall dtype. TensorType dtype => OpParams -> Tensor Build dtype
|
|
|
|
placeholderV2 :: forall dtype. TensorType dtype => Shape -> Tensor Build dtype
|
|
placeholderV2' :: forall dtype. TensorType dtype => OpParams -> Shape -> Tensor Build dtype
|
|
|
|
placeholderWithDefault :: forall v'1 dtype. TensorType dtype => Shape -> Tensor v'1 dtype -> Tensor Build dtype
|
|
placeholderWithDefault' :: forall v'1 dtype. TensorType dtype => OpParams -> Shape -> Tensor v'1 dtype -> Tensor Build dtype
|
|
|
|
polygamma :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
polygamma' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
populationCount :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor Build Word8
|
|
populationCount' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => OpParams -> Tensor v'1 t -> Tensor Build Word8
|
|
|
|
pow :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
pow' :: 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
|
|
|
|
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
|
|
|
|
prelinearize :: forall v'1 dtype. TensorType dtype => Tensor v'1 dtype -> Tensor Build Variant
|
|
prelinearize' :: forall v'1 dtype. TensorType dtype => OpParams -> Tensor v'1 dtype -> Tensor Build Variant
|
|
|
|
prelinearizeTuple :: forall v'1 dtypes. TensorTypes dtypes => TensorList v'1 dtypes -> Tensor Build Variant
|
|
prelinearizeTuple' :: forall v'1 dtypes. TensorTypes dtypes => OpParams -> TensorList v'1 dtypes -> Tensor Build Variant
|
|
|
|
preventGradient :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
preventGradient' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
print :: forall v'1 v'2 t u m'. (MonadBuild m', TensorType t, TensorTypes u) => Tensor v'1 t -> TensorList v'2 u -> m' (Tensor Value t)
|
|
print' :: 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)
|
|
|
|
printV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
|
|
printV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
|
|
|
|
priorityQueue :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
priorityQueue' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
priorityQueueV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
priorityQueueV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
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
|
|
|
|
prod :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
prod' :: 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
|
|
|
|
pyFunc :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
|
|
pyFunc' :: forall v'1 tin tout m'. (MonadBuild m', TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> m' (TensorList Value tout)
|
|
|
|
pyFuncStateless :: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout) => ByteString -> TensorList v'1 tin -> TensorList Build tout
|
|
pyFuncStateless' :: forall v'1 tin tout. (TensorTypes tin, TensorTypes tout) => OpParams -> ByteString -> TensorList v'1 tin -> TensorList Build tout
|
|
|
|
qr :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
qr' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
|
|
quantizeAndDequantize :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
quantizeAndDequantize' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
quantizeAndDequantizeV2 :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
quantizeAndDequantizeV2' :: 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
|
|
|
|
quantizeAndDequantizeV3 :: forall v'1 v'2 v'3 v'4 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 Int32 -> Tensor Build t
|
|
quantizeAndDequantizeV3' :: 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
|
|
|
|
quantizeDownAndShrinkRange :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
|
|
quantizeDownAndShrinkRange' :: 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)
|
|
|
|
quantizeV2 :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 Float -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
|
|
quantizeV2' :: 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)
|
|
|
|
quantizedAdd :: 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) => 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' :: 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)
|
|
|
|
quantizedAvgPool :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => ByteString -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
|
|
quantizedAvgPool' :: 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)
|
|
|
|
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 '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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' :: 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)
|
|
|
|
quantizedBiasAdd :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 out_type. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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' :: forall v'1 v'2 v'3 v'4 v'5 v'6 t1 t2 out_type. (OneOf '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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)
|
|
|
|
quantizedConcat :: forall v'1 v'2 v'3 v'4 t. TensorType t => 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' :: 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)
|
|
|
|
quantizedConv2D :: 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) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
quantizedConv2DAndReluAndRequantize :: 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) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
quantizedConv2DPerChannel :: 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) => 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' :: 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)
|
|
|
|
quantizedConv2DWithBias :: 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) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
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 '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
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 '[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) => 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' :: 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)
|
|
|
|
quantizedConv2DWithBiasSumAndRelu :: 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) => 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' :: 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)
|
|
|
|
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 '[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) => 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' :: 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)
|
|
|
|
quantizedDepthwiseConv2D :: 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) => 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' :: 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)
|
|
|
|
quantizedDepthwiseConv2DWithBias :: 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) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
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 '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] tfilter, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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' :: 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)
|
|
|
|
quantizedInstanceNorm :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
|
|
quantizedInstanceNorm' :: 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)
|
|
|
|
quantizedMatMul :: 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) => 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' :: 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)
|
|
|
|
quantizedMatMulWithBias :: 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) => 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' :: 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)
|
|
|
|
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 '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Float] toutput) => 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' :: 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
|
|
|
|
quantizedMatMulWithBiasAndRelu :: 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) => 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' :: 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)
|
|
|
|
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 '[Int16, Int32, Word16, Word8] t1, OneOf '[Int16, Int32, Word16, Word8] t2, OneOf '[Int32, Float] tbias, OneOf '[Int16, Int32, Word16, Word8] toutput) => 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' :: 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 :: 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) => 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' :: 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)
|
|
|
|
quantizedMaxPool :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => ByteString -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build t, Tensor Build Float, Tensor Build Float)
|
|
quantizedMaxPool' :: 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)
|
|
|
|
quantizedMul :: 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) => 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' :: 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)
|
|
|
|
quantizedRelu :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
|
|
quantizedRelu' :: 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 :: forall v'1 v'2 v'3 tinput out_type. (OneOf '[Int16, Int32, Word16, Word8] tinput, OneOf '[Int16, Int32, Word16, Word8] out_type) => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build out_type, Tensor Build Float, Tensor Build Float)
|
|
quantizedRelu6' :: 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)
|
|
|
|
quantizedReluX :: 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) => 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' :: 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)
|
|
|
|
quantizedReshape :: forall v'1 v'2 v'3 v'4 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => 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' :: 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)
|
|
|
|
quantizedResizeBilinear :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Word8, Float] t => 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' :: 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)
|
|
|
|
queueClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
|
|
queueClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
|
|
|
|
queueCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
queueCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
queueDequeue :: forall component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> m' (TensorList Value component_types)
|
|
queueDequeue' :: forall component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> m' (TensorList Value component_types)
|
|
|
|
queueDequeueMany :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
queueDequeueMany' :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
|
|
queueDequeueManyV2 :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
queueDequeueManyV2' :: 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)
|
|
|
|
queueDequeueUpTo :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
queueDequeueUpTo' :: forall v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
|
|
queueDequeueUpToV2 :: forall v'1 v'2 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (TensorList Value component_types)
|
|
queueDequeueUpToV2' :: 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)
|
|
|
|
queueDequeueV2 :: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types) => Tensor v'1 ResourceHandle -> m' (TensorList Value component_types)
|
|
queueDequeueV2' :: forall v'1 component_types m'. (MonadBuild m', TensorTypes component_types) => OpParams -> Tensor v'1 ResourceHandle -> m' (TensorList Value component_types)
|
|
|
|
queueEnqueue :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
|
|
queueEnqueue' :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
|
|
|
|
queueEnqueueMany :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
|
|
queueEnqueueMany' :: forall v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor Ref ByteString -> TensorList v'2 tcomponents -> m' ControlNode
|
|
|
|
queueEnqueueManyV2 :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
|
|
queueEnqueueManyV2' :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
|
|
|
|
queueEnqueueV2 :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
|
|
queueEnqueueV2' :: forall v'1 v'2 tcomponents m'. (MonadBuild m', TensorTypes tcomponents) => OpParams -> Tensor v'1 ResourceHandle -> TensorList v'2 tcomponents -> m' ControlNode
|
|
|
|
queueIsClosed :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Bool)
|
|
queueIsClosed' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Bool)
|
|
|
|
queueIsClosedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
queueIsClosedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
|
|
queueSize :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
queueSize' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int32)
|
|
|
|
queueSizeV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
queueSizeV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
|
|
rFFT :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
|
|
rFFT' :: 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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
|
|
rFFT2D' :: 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 :: forall v'1 v'2 treal tcomplex. (OneOf '[Double, Float] treal, OneOf '[Complex Double, Complex Float] tcomplex) => Tensor v'1 treal -> Tensor v'2 Int32 -> Tensor Build tcomplex
|
|
rFFT3D' :: 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
|
|
|
|
rGBToHSV :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
rGBToHSV' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
raggedBincount :: forall v'1 v'2 v'3 v'4 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor v'4 t -> Tensor Build t
|
|
raggedBincount' :: 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
|
|
|
|
raggedCountSparseOutput :: forall v'1 v'2 v'3 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => 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' :: 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)
|
|
|
|
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 '[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) => 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' :: 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)
|
|
|
|
raggedGather :: forall v'1 v'2 v'3 tvalues tindices tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tsplits) => Int64 -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor v'3 tindices -> ([Tensor Build tsplits], Tensor Build tvalues)
|
|
raggedGather' :: 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)
|
|
|
|
raggedRange :: forall v'1 v'2 v'3 t tsplits. (OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tsplits) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> (Tensor Build tsplits, Tensor Build t)
|
|
raggedRange' :: 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)
|
|
|
|
raggedTensorFromVariant :: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => Int64 -> Int64 -> Tensor v'1 Variant -> ([Tensor Build tsplits], Tensor Build tvalues)
|
|
raggedTensorFromVariant' :: forall v'1 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => OpParams -> Int64 -> Int64 -> Tensor v'1 Variant -> ([Tensor Build tsplits], Tensor Build tvalues)
|
|
|
|
raggedTensorToSparse :: forall v'1 v'2 t tsplits. (TensorType t, OneOf '[Int32, Int64] tsplits) => [Tensor v'1 tsplits] -> Tensor v'2 t -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
|
|
raggedTensorToSparse' :: 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)
|
|
|
|
raggedTensorToTensor :: forall v'1 v'2 v'3 v'4 t tindex tshape. (TensorType t, OneOf '[Int32, Int64] tindex, OneOf '[Int32, Int64] tshape) => Tensor v'1 tshape -> Tensor v'2 t -> Tensor v'3 t -> [Tensor v'4 tindex] -> Tensor Build t
|
|
raggedTensorToTensor' :: 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
|
|
|
|
raggedTensorToVariant :: forall v'1 v'2 tvalues tsplits. (TensorType tvalues, OneOf '[Int32, Int64] tsplits) => Bool -> [Tensor v'1 tsplits] -> Tensor v'2 tvalues -> Tensor Build Variant
|
|
raggedTensorToVariant' :: 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
|
|
|
|
randomCrop :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8, Double, Float] t) => Tensor v'1 t -> Tensor v'2 Int64 -> m' (Tensor Value t)
|
|
randomCrop' :: 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)
|
|
|
|
randomDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
|
|
randomDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> m' (Tensor Value Variant)
|
|
|
|
randomGamma :: forall v'1 v'2 s t m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] t) => Tensor v'1 s -> Tensor v'2 t -> m' (Tensor Value t)
|
|
randomGamma' :: 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)
|
|
|
|
randomGammaGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
randomGammaGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
randomPoisson :: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype) => Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
|
|
randomPoisson' :: forall v'1 v'2 s dtype m'. (MonadBuild m', OneOf '[Int32, Int64] s, OneOf '[Word16, Double, Float] dtype) => OpParams -> Tensor v'1 s -> Tensor v'2 dtype -> m' (Tensor Value dtype)
|
|
|
|
randomPoissonV2 :: 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) => Tensor v'1 s -> Tensor v'2 r -> m' (Tensor Value dtype)
|
|
randomPoissonV2' :: 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)
|
|
|
|
randomShuffle :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 t -> m' (Tensor Value t)
|
|
randomShuffle' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 t -> m' (Tensor Value t)
|
|
|
|
randomShuffleQueue :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Ref ByteString)
|
|
randomShuffleQueue' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Ref ByteString)
|
|
|
|
randomShuffleQueueV2 :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value ResourceHandle)
|
|
randomShuffleQueueV2' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value ResourceHandle)
|
|
|
|
randomStandardNormal :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
|
|
randomStandardNormal' :: 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 :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
|
|
randomUniform' :: 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)
|
|
|
|
randomUniformInt :: forall v'1 v'2 v'3 tout t m'. (MonadBuild m', OneOf '[Int32, Int64] tout, OneOf '[Int32, Int64] t) => Tensor v'1 t -> Tensor v'2 tout -> Tensor v'3 tout -> m' (Tensor Value tout)
|
|
randomUniformInt' :: 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)
|
|
|
|
range :: forall v'1 v'2 v'3 tidx. OneOf '[Int32, Int64, Word16, Double, Float] tidx => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor v'3 tidx -> Tensor Build tidx
|
|
range' :: 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
|
|
|
|
rangeDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Int64 -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
|
|
rangeDataset' :: 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)
|
|
|
|
rank :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Int32
|
|
rank' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Int32
|
|
|
|
readFile :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
readFile' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
readVariableOp :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
|
|
readVariableOp' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value dtype)
|
|
|
|
readerNumRecordsProduced :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
readerNumRecordsProduced' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
|
|
readerNumRecordsProducedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
readerNumRecordsProducedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
|
|
readerNumWorkUnitsCompleted :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
readerNumWorkUnitsCompleted' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value Int64)
|
|
|
|
readerNumWorkUnitsCompletedV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
readerNumWorkUnitsCompletedV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int64)
|
|
|
|
readerRead :: forall m'. MonadBuild m' => Tensor Ref ByteString -> Tensor Ref ByteString -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
readerRead' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor Ref ByteString -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
|
|
readerReadUpTo :: forall v'3 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor Ref ByteString -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
readerReadUpTo' :: forall v'3 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor Ref ByteString -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
|
|
readerReadUpToV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 Int64 -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
readerReadUpToV2' :: 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)
|
|
|
|
readerReadV2 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
readerReadV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' (Tensor Value ByteString, Tensor Value ByteString)
|
|
|
|
readerReset :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
|
|
readerReset' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
|
|
|
|
readerResetV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
readerResetV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
readerRestoreState :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
readerRestoreState' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
readerRestoreStateV2 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
readerRestoreStateV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
readerSerializeState :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' (Tensor Value ByteString)
|
|
readerSerializeState' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' (Tensor Value ByteString)
|
|
|
|
readerSerializeStateV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
readerSerializeStateV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
real :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => Tensor v'1 t -> Tensor Build tout
|
|
real' :: forall v'1 t tout. (OneOf '[Complex Double, Complex Float] t, OneOf '[Double, Float] tout) => OpParams -> Tensor v'1 t -> Tensor Build tout
|
|
|
|
realDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
realDiv' :: 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
|
|
|
|
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
|
|
|
|
reciprocal :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
reciprocal' :: 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
|
|
|
|
reciprocalGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
reciprocalGrad' :: 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
|
|
|
|
recordInput :: forall m'. MonadBuild m' => ByteString -> m' (Tensor Value ByteString)
|
|
recordInput' :: forall m'. MonadBuild m' => OpParams -> ByteString -> m' (Tensor Value ByteString)
|
|
|
|
recv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
recv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
|
|
recvTPUEmbeddingActivations :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' [Tensor Value Float]
|
|
recvTPUEmbeddingActivations' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> m' [Tensor Value Float]
|
|
|
|
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
|
|
|
|
refEnter :: forall t m'. (MonadBuild m', TensorType t) => ByteString -> Tensor Ref t -> m' (Tensor Ref t)
|
|
refEnter' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> Tensor Ref t -> m' (Tensor Ref t)
|
|
|
|
refExit :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
|
|
refExit' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
|
|
|
|
refIdentity :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
|
|
refIdentity' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
|
|
|
|
refMerge :: forall t m'. (MonadBuild m', TensorType t) => [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
|
|
refMerge' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> [Tensor Ref t] -> m' (Tensor Ref t, Tensor Value Int32)
|
|
|
|
refNextIteration :: forall t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> m' (Tensor Ref t)
|
|
refNextIteration' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> m' (Tensor Ref t)
|
|
|
|
refSelect :: forall v'1 t m'. (MonadBuild m', TensorType t) => Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
|
|
refSelect' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 Int32 -> [Tensor Ref t] -> m' (Tensor Ref t)
|
|
|
|
refSwitch :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref t -> Tensor v'2 Bool -> m' (Tensor Ref t, Tensor Ref t)
|
|
refSwitch' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref t -> Tensor v'2 Bool -> m' (Tensor Ref t, Tensor Ref t)
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
relu :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
relu' :: 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 :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
relu6' :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
relu6Grad :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
relu6Grad' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
reluGrad' :: 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
|
|
|
|
remoteFusedGraphExecute :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
|
|
remoteFusedGraphExecute' :: forall v'1 tinputs toutputs. (TensorTypes tinputs, TensorTypes toutputs) => OpParams -> ByteString -> TensorList v'1 tinputs -> TensorList Build toutputs
|
|
|
|
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
|
|
|
|
requantizationRange :: forall v'1 v'2 v'3 tinput. OneOf '[Int16, Int32, Word16, Word8] tinput => Tensor v'1 tinput -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build Float, Tensor Build Float)
|
|
requantizationRange' :: 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)
|
|
|
|
requantizationRangePerChannel :: forall v'1 v'2 v'3 t. OneOf '[Int16, Int32, Word16, Word8] t => Float -> Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> (Tensor Build Float, Tensor Build Float)
|
|
requantizationRangePerChannel' :: 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)
|
|
|
|
requantize :: 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) => 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' :: 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 :: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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' :: forall v'1 v'2 v'3 v'4 v'5 t out_type. (OneOf '[Int16, Int32, Word16, Word8] t, OneOf '[Int16, Int32, Word16, Word8] out_type) => 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)
|
|
|
|
reshape :: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
|
|
reshape' :: forall v'1 v'2 t tshape. (TensorType t, OneOf '[Int32, Int64] tshape) => OpParams -> Tensor v'1 t -> Tensor v'2 tshape -> Tensor Build t
|
|
|
|
resizeArea :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
|
|
resizeArea' :: 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 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
|
|
resizeBicubic' :: 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
|
|
|
|
resizeBicubicGrad :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
|
|
resizeBicubicGrad' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
|
|
|
|
resizeBilinear :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build Float
|
|
resizeBilinear' :: 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
|
|
|
|
resizeBilinearGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
|
|
resizeBilinearGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 Float -> Tensor v'2 t -> Tensor Build t
|
|
|
|
resizeNearestNeighbor :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
resizeNearestNeighbor' :: 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
|
|
|
|
resizeNearestNeighborGrad :: forall v'1 v'2 t. OneOf '[Int32, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
resizeNearestNeighborGrad' :: 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
|
|
|
|
resourceAccumulatorApplyGradient :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceAccumulatorApplyGradient' :: 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
|
|
|
|
resourceAccumulatorNumAccumulated :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
resourceAccumulatorNumAccumulated' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
|
|
resourceAccumulatorSetGlobalStep :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
|
|
resourceAccumulatorSetGlobalStep' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> m' ControlNode
|
|
|
|
resourceAccumulatorTakeGradient :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> m' (Tensor Value dtype)
|
|
resourceAccumulatorTakeGradient' :: 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)
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => 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' :: 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
|
|
|
|
resourceApplyAdadelta :: 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) => 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' :: 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
|
|
|
|
resourceApplyAdagrad :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> m' ControlNode
|
|
resourceApplyAdagrad' :: 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
|
|
|
|
resourceApplyAdagradDA :: 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) => 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' :: 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
|
|
|
|
resourceApplyAdagradV2 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
|
|
resourceApplyAdagradV2' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => 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' :: 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
|
|
|
|
resourceApplyAddSign :: 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) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => 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' :: 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
|
|
|
|
resourceApplyFtrl :: 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) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t) => 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' :: 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
|
|
|
|
resourceApplyGradientDescent :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> m' ControlNode
|
|
resourceApplyGradientDescent' :: 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
|
|
|
|
resourceApplyKerasMomentum :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
|
|
resourceApplyKerasMomentum' :: 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 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
|
|
resourceApplyMomentum' :: 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
|
|
|
|
resourceApplyPowerSign :: 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) => 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' :: 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
|
|
|
|
resourceApplyProximalAdagrad :: 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) => 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' :: 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
|
|
|
|
resourceApplyProximalGradientDescent :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 t -> m' ControlNode
|
|
resourceApplyProximalGradientDescent' :: 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
|
|
|
|
resourceApplyRMSProp :: 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) => 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' :: 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
|
|
|
|
resourceConditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Value ResourceHandle)
|
|
resourceConditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
|
|
|
|
resourceCountUpTo :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
|
|
resourceCountUpTo' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64] t) => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> m' (Tensor Value t)
|
|
|
|
resourceGather :: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> m' (Tensor Value dtype)
|
|
resourceGather' :: 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 :: forall v'1 v'2 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> m' (Tensor Value dtype)
|
|
resourceGatherNd' :: 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)
|
|
|
|
resourceScatterAdd :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterAdd' :: 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 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterDiv' :: 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 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterMax' :: 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 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterMin' :: 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 :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterMul' :: 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
|
|
|
|
resourceScatterNdAdd :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
|
|
resourceScatterNdAdd' :: 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 :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
|
|
resourceScatterNdMax' :: 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 :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
|
|
resourceScatterNdMin' :: 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 :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
|
|
resourceScatterNdSub' :: 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 :: forall v'1 v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 t -> m' ControlNode
|
|
resourceScatterNdUpdate' :: 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
|
|
|
|
resourceScatterSub :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterSub' :: 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
|
|
|
|
resourceScatterUpdate :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
resourceScatterUpdate' :: forall v'1 v'2 v'3 dtype tindices m'. (MonadBuild m', TensorType dtype, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 tindices -> Tensor v'3 dtype -> m' ControlNode
|
|
|
|
resourceSparseApplyAdadelta :: 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) => 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' :: 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
|
|
|
|
resourceSparseApplyAdagrad :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' ControlNode
|
|
resourceSparseApplyAdagrad' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => 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' :: 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
|
|
|
|
resourceSparseApplyAdagradV2 :: 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) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => 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' :: 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
|
|
|
|
resourceSparseApplyKerasMomentum :: 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) => 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' :: 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 :: 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) => 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' :: 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
|
|
|
|
resourceSparseApplyProximalAdagrad :: 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) => 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' :: 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
|
|
|
|
resourceSparseApplyProximalGradientDescent :: 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) => 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' :: 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
|
|
|
|
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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => 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' :: 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
|
|
|
|
resourceStridedSliceAssign :: forall v'1 v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 ResourceHandle -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' ControlNode
|
|
resourceStridedSliceAssign' :: 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
|
|
|
|
restore :: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' (Tensor Value dt)
|
|
restore' :: forall v'1 v'2 dt m'. (MonadBuild m', TensorType dt) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' (Tensor Value dt)
|
|
|
|
restoreSlice :: forall v'1 v'2 v'3 dt m'. (MonadBuild m', TensorType dt) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value dt)
|
|
restoreSlice' :: 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)
|
|
|
|
restoreV2 :: forall v'1 v'2 v'3 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (TensorList Value dtypes)
|
|
restoreV2' :: 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)
|
|
|
|
retrieveTPUEmbeddingADAMParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingADAMParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingADAMParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingADAMParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingAdadeltaParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingAdadeltaParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingAdadeltaParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingAdagradParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingAdagradParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingAdagradParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingAdagradParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingCenteredRMSPropParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingCenteredRMSPropParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingFTRLParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingFTRLParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingFTRLParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingFTRLParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingMDLAdagradLightParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingMDLAdagradLightParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingMomentumParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingMomentumParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingMomentumParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingMomentumParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingProximalAdagradParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingProximalAdagradParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingProximalAdagradParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingProximalYogiParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingProximalYogiParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingProximalYogiParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingRMSPropParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingRMSPropParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingRMSPropParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float, Tensor Value Float, Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingStochasticGradientDescentParameters :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float)
|
|
retrieveTPUEmbeddingStochasticGradientDescentParameters' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float)
|
|
|
|
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug :: forall m'. MonadBuild m' => Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
retrieveTPUEmbeddingStochasticGradientDescentParametersGradAccumDebug' :: forall m'. MonadBuild m' => OpParams -> Int64 -> Int64 -> m' (Tensor Value Float, Tensor Value Float)
|
|
|
|
reverse :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Bool, ByteString, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Bool -> Tensor Build t
|
|
reverse' :: 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
|
|
|
|
reverseSequence :: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
|
|
reverseSequence' :: forall v'1 v'2 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor Build t
|
|
|
|
reverseV2 :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
reverseV2' :: 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
|
|
|
|
rightShift :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
rightShift' :: 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
|
|
|
|
rint :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
rint' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
rngSkip :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> m' ControlNode
|
|
rngSkip' :: 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
|
|
|
|
roll :: forall v'1 v'2 v'3 t tshift taxis. (TensorType t, OneOf '[Int32, Int64] tshift, OneOf '[Int32, Int64] taxis) => Tensor v'1 t -> Tensor v'2 tshift -> Tensor v'3 taxis -> Tensor Build t
|
|
roll' :: forall v'1 v'2 v'3 t tshift taxis. (TensorType t, OneOf '[Int32, Int64] tshift, OneOf '[Int32, Int64] taxis) => OpParams -> Tensor v'1 t -> Tensor v'2 tshift -> Tensor v'3 taxis -> Tensor Build t
|
|
|
|
round :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
round' :: 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
|
|
|
|
rpc :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString)
|
|
rpc' :: 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)
|
|
|
|
rsqrt :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
rsqrt' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
rsqrtGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
rsqrtGrad' :: 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
|
|
|
|
sampleDistortedBoundingBox :: forall v'1 v'2 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => Tensor v'1 t -> Tensor v'2 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
|
|
sampleDistortedBoundingBox' :: 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)
|
|
|
|
sampleDistortedBoundingBoxV2 :: forall v'1 v'2 v'3 t m'. (MonadBuild m', OneOf '[Int16, Int32, Int64, Int8, Word8] t) => Tensor v'1 t -> Tensor v'2 Float -> Tensor v'3 Float -> m' (Tensor Value t, Tensor Value t, Tensor Value Float)
|
|
sampleDistortedBoundingBoxV2' :: 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)
|
|
|
|
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
|
|
|
|
save :: forall v'1 v'2 v'3 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> TensorList v'3 t -> m' ControlNode
|
|
save' :: 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
|
|
|
|
saveSlices :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorTypes t) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 t -> m' ControlNode
|
|
saveSlices' :: 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 :: forall v'1 v'2 v'3 v'4 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 dtypes -> m' ControlNode
|
|
saveV2' :: forall v'1 v'2 v'3 v'4 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> TensorList v'4 dtypes -> m' ControlNode
|
|
|
|
scalarSummary :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor Build ByteString
|
|
scalarSummary' :: 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
|
|
|
|
scaleAndTranslate :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build Float
|
|
scaleAndTranslate' :: 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
|
|
|
|
scaleAndTranslateGrad :: forall v'1 v'2 v'3 v'4 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Float -> Tensor v'4 Float -> Tensor Build t
|
|
scaleAndTranslateGrad' :: 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
|
|
|
|
scatterAdd :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterAdd' :: 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 :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterDiv' :: 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)
|
|
|
|
scatterMax :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterMax' :: 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 :: forall v'2 v'3 t tindices m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterMin' :: 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)
|
|
|
|
scatterMul :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterMul' :: 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)
|
|
|
|
scatterNd :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
|
|
scatterNd' :: 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
|
|
|
|
scatterNdAdd :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterNdAdd' :: 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 :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterNdMax' :: 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 :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterNdMin' :: 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)
|
|
|
|
scatterNdNonAliasingAdd :: 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) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
scatterNdNonAliasingAdd' :: 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
|
|
|
|
scatterNdSub :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterNdSub' :: 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)
|
|
|
|
scatterNdUpdate :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterNdUpdate' :: 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)
|
|
|
|
scatterSub :: 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) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterSub' :: 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)
|
|
|
|
scatterUpdate :: forall v'2 v'3 t tindices m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] tindices) => Tensor Ref t -> Tensor v'2 tindices -> Tensor v'3 t -> m' (Tensor Ref t)
|
|
scatterUpdate' :: 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)
|
|
|
|
sdcaFprint :: Tensor v'1 ByteString -> Tensor Build Int64
|
|
sdcaFprint' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
|
|
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])
|
|
|
|
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])
|
|
|
|
sdcaShrinkL1 :: forall m'. MonadBuild m' => Float -> Float -> [Tensor Ref Float] -> m' ControlNode
|
|
sdcaShrinkL1' :: forall m'. MonadBuild m' => OpParams -> Float -> Float -> [Tensor Ref Float] -> m' ControlNode
|
|
|
|
segmentMax :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
|
|
segmentMax' :: 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
|
|
|
|
segmentMean :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
|
|
segmentMean' :: forall v'1 v'2 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 tindices -> Tensor Build t
|
|
|
|
segmentMin :: forall v'1 v'2 t tindices. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
|
|
segmentMin' :: 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
|
|
|
|
segmentProd :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
|
|
segmentProd' :: forall v'1 v'2 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 tindices -> Tensor Build t
|
|
|
|
segmentSum :: forall v'1 v'2 t tindices. (OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor Build t
|
|
segmentSum' :: forall v'1 v'2 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 tindices -> Tensor Build t
|
|
|
|
select :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
select' :: 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 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 Bool -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build t
|
|
selectV2' :: 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
|
|
|
|
selfAdjointEig :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
selfAdjointEig' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
selfAdjointEigV2 :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
selfAdjointEigV2' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
|
|
selu :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
selu' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
seluGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
seluGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
send :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
send' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
sendTPUEmbeddingGradients :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> [Tensor v'1 Float] -> [Tensor v'2 Float] -> m' ControlNode
|
|
sendTPUEmbeddingGradients' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> [Tensor v'1 Float] -> [Tensor v'2 Float] -> m' ControlNode
|
|
|
|
serializeIterator :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
serializeIterator' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Variant)
|
|
|
|
serializeManySparse :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
|
|
serializeManySparse' :: 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 :: forall v'1 v'2 v'3 t out_type. (TensorType t, OneOf '[ByteString, Variant] out_type) => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build out_type
|
|
serializeSparse' :: 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
|
|
|
|
serializeTensor :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build ByteString
|
|
serializeTensor' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
|
|
|
|
setSize :: forall v'1 v'2 v'3 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Int32
|
|
setSize' :: 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
|
|
|
|
setStatsAggregatorDataset :: forall v'1 v'2 v'3 v'4 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> Tensor v'3 ByteString -> Tensor v'4 ByteString -> m' (Tensor Value Variant)
|
|
setStatsAggregatorDataset' :: 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)
|
|
|
|
shape :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor Build out_type
|
|
shape' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor Build out_type
|
|
|
|
shapeN :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => [Tensor v'1 t] -> [Tensor Build out_type]
|
|
shapeN' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> [Tensor v'1 t] -> [Tensor Build out_type]
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
shuffleAndRepeatDatasetV2 :: forall v'1 v'2 v'3 v'4 v'5 v'6 m'. MonadBuild m' => [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' :: 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)
|
|
|
|
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
|
|
|
|
shuffleDatasetV2 :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 Int64 -> Tensor v'3 ResourceHandle -> m' (Tensor Value Variant)
|
|
shuffleDatasetV2' :: 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)
|
|
|
|
shuffleDatasetV3 :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => [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' :: 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)
|
|
|
|
shutdownDistributedTPU :: forall m'. MonadBuild m' => m' ControlNode
|
|
shutdownDistributedTPU' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
|
|
|
|
sigmoid :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
sigmoid' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
sigmoidGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
sigmoidGrad' :: 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
|
|
|
|
sign :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
sign' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
sin :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
sin' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
sinh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
sinh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
size :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor Build out_type
|
|
size' :: forall v'1 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 t -> Tensor Build out_type
|
|
|
|
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
|
|
|
|
skipgram :: forall m'. MonadBuild m' => 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' :: 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)
|
|
|
|
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
|
|
|
|
slice :: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor Build t
|
|
slice' :: forall v'1 v'2 v'3 t index. (TensorType t, OneOf '[Int32, Int64] index) => OpParams -> Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor Build t
|
|
|
|
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
|
|
|
|
snapshot :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
snapshot' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
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
|
|
|
|
sobolSample :: forall v'1 v'2 v'3 dtype. OneOf '[Double, Float] dtype => Tensor v'1 Int32 -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build dtype
|
|
sobolSample' :: 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
|
|
|
|
softmax :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
softmax' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
softmaxCrossEntropyWithLogits :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
|
|
softmaxCrossEntropyWithLogits' :: 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)
|
|
|
|
softplus :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
softplus' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
softplusGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
softplusGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
softsign :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
softsign' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
softsignGrad :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
softsignGrad' :: forall v'1 v'2 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
spaceToBatch :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
spaceToBatch' :: forall v'1 v'2 t tpaddings. (TensorType t, OneOf '[Int32, Int64] tpaddings) => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 tpaddings -> Tensor Build t
|
|
|
|
spaceToBatchND :: forall v'1 v'2 v'3 t tblock_shape tpaddings. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tpaddings) => Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tpaddings -> Tensor Build t
|
|
spaceToBatchND' :: forall v'1 v'2 v'3 t tblock_shape tpaddings. (TensorType t, OneOf '[Int32, Int64] tblock_shape, OneOf '[Int32, Int64] tpaddings) => OpParams -> Tensor v'1 t -> Tensor v'2 tblock_shape -> Tensor v'3 tpaddings -> Tensor Build t
|
|
|
|
spaceToDepth :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor Build t
|
|
spaceToDepth' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor Build t
|
|
|
|
sparseAccumulatorApplyGradient :: 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) => Bool -> Tensor Ref ByteString -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 dtype -> Tensor v'5 Int64 -> m' ControlNode
|
|
sparseAccumulatorApplyGradient' :: 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
|
|
|
|
sparseAccumulatorTakeGradient :: forall v'2 dtype m'. (MonadBuild m', OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
|
|
sparseAccumulatorTakeGradient' :: 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)
|
|
|
|
sparseAdd :: 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) => 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' :: 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)
|
|
|
|
sparseAddGrad :: 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 => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> (Tensor Build t, Tensor Build t)
|
|
sparseAddGrad' :: 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)
|
|
|
|
sparseApplyAdadelta :: 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) => 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' :: 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)
|
|
|
|
sparseApplyAdagrad :: 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) => Tensor Ref t -> Tensor Ref t -> Tensor v'3 t -> Tensor v'4 t -> Tensor v'5 tindices -> m' (Tensor Ref t)
|
|
sparseApplyAdagrad' :: 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)
|
|
|
|
sparseApplyAdagradDA :: 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) => 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' :: 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)
|
|
|
|
sparseApplyAdagradV2 :: 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) => 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' :: 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)
|
|
|
|
sparseApplyCenteredRMSProp :: 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) => 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' :: 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)
|
|
|
|
sparseApplyFtrl :: 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) => 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' :: 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)
|
|
|
|
sparseApplyFtrlV2 :: 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) => 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' :: 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)
|
|
|
|
sparseApplyMomentum :: 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) => 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' :: 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)
|
|
|
|
sparseApplyProximalAdagrad :: 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) => 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' :: 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)
|
|
|
|
sparseApplyProximalGradientDescent :: 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) => 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' :: 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)
|
|
|
|
sparseApplyRMSProp :: 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) => 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' :: 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)
|
|
|
|
sparseBincount :: forall v'1 v'2 v'3 v'4 v'5 tidx t. (OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64, Double, Float] t) => Tensor v'1 Int64 -> Tensor v'2 tidx -> Tensor v'3 Int64 -> Tensor v'4 tidx -> Tensor v'5 t -> Tensor Build t
|
|
sparseBincount' :: 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
|
|
|
|
sparseConcat :: forall v'1 v'2 v'3 t. TensorType t => Int64 -> [Tensor v'1 Int64] -> [Tensor v'2 t] -> [Tensor v'3 Int64] -> (Tensor Build Int64, Tensor Build t, Tensor Build Int64)
|
|
sparseConcat' :: 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)
|
|
|
|
sparseConditionalAccumulator :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Ref ByteString)
|
|
sparseConditionalAccumulator' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Ref ByteString)
|
|
|
|
sparseCountSparseOutput :: forall v'1 v'2 v'3 v'4 t output_type. (OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Double, Float] output_type) => 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' :: 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)
|
|
|
|
sparseCross :: 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) => 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' :: 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)
|
|
|
|
sparseCrossHashed :: 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) => [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' :: 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)
|
|
|
|
sparseCrossV2 :: 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) => [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' :: 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)
|
|
|
|
sparseDenseCwiseAdd :: 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 => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
|
|
sparseDenseCwiseAdd' :: 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 :: 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 => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
|
|
sparseDenseCwiseDiv' :: 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 :: 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 => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
|
|
sparseDenseCwiseMul' :: 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
|
|
|
|
sparseFillEmptyRows :: forall v'1 v'2 v'3 v'4 t. TensorType t => 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' :: 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)
|
|
|
|
sparseFillEmptyRowsGrad :: forall v'1 v'2 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
|
|
sparseFillEmptyRowsGrad' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Int64 -> Tensor v'2 t -> (Tensor Build t, Tensor Build t)
|
|
|
|
sparseMatMul :: forall v'1 v'2 ta tb. (OneOf '[Word16, Float] ta, OneOf '[Word16, Float] tb) => Tensor v'1 ta -> Tensor v'2 tb -> Tensor Build Float
|
|
sparseMatMul' :: 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
|
|
|
|
sparseMatrixAdd :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 Variant -> Tensor v'2 Variant -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build Variant
|
|
sparseMatrixAdd' :: 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
|
|
|
|
sparseMatrixMatMul :: forall v'1 v'2 t. TensorType t => Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
|
|
sparseMatrixMatMul' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build t
|
|
|
|
sparseMatrixMul :: forall v'1 v'2 t. TensorType t => Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
|
|
sparseMatrixMul' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 Variant -> Tensor v'2 t -> Tensor Build Variant
|
|
|
|
sparseMatrixNNZ :: Tensor v'1 Variant -> Tensor Build Int32
|
|
sparseMatrixNNZ' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
|
|
|
|
sparseMatrixOrderingAMD :: Tensor v'1 Variant -> Tensor Build Int32
|
|
sparseMatrixOrderingAMD' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
|
|
|
|
sparseMatrixSoftmax :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
|
|
sparseMatrixSoftmax' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
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
|
|
|
|
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
|
|
|
|
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
|
|
|
|
sparseMatrixTranspose :: DataType -> Tensor v'1 Variant -> Tensor Build Variant
|
|
sparseMatrixTranspose' :: OpParams -> DataType -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
sparseMatrixZeros :: DataType -> Tensor v'1 Int64 -> Tensor Build Variant
|
|
sparseMatrixZeros' :: OpParams -> DataType -> Tensor v'1 Int64 -> Tensor Build Variant
|
|
|
|
sparseReduceMax :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
|
|
sparseReduceMax' :: 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
|
|
|
|
sparseReduceMaxSparse :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => 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' :: 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)
|
|
|
|
sparseReduceSum :: 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 => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Int32 -> Tensor Build t
|
|
sparseReduceSum' :: 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
|
|
|
|
sparseReduceSumSparse :: 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 => 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' :: 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)
|
|
|
|
sparseReorder :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> (Tensor Build Int64, Tensor Build t)
|
|
sparseReorder' :: 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)
|
|
|
|
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)
|
|
|
|
sparseSegmentMean :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
|
|
sparseSegmentMean' :: 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
|
|
|
|
sparseSegmentMeanGrad :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
|
|
sparseSegmentMeanGrad' :: 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
|
|
|
|
sparseSegmentMeanWithNumSegments :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
|
|
sparseSegmentMeanWithNumSegments' :: 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
|
|
|
|
sparseSegmentSqrtN :: forall v'1 v'2 v'3 t tidx tsegmentids. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
|
|
sparseSegmentSqrtN' :: 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
|
|
|
|
sparseSegmentSqrtNGrad :: forall v'1 v'2 v'3 v'4 t tidx tsegmentids. (OneOf '[Double, Float] t, OneOf '[Int32, Int64] tidx, OneOf '[Int32, Int64] tsegmentids) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 Int32 -> Tensor Build t
|
|
sparseSegmentSqrtNGrad' :: 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
|
|
|
|
sparseSegmentSqrtNWithNumSegments :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
|
|
sparseSegmentSqrtNWithNumSegments' :: 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
|
|
|
|
sparseSegmentSum :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor Build t
|
|
sparseSegmentSum' :: 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
|
|
|
|
sparseSegmentSumWithNumSegments :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor v'3 tsegmentids -> Tensor v'4 tnumsegments -> Tensor Build t
|
|
sparseSegmentSumWithNumSegments' :: 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
|
|
|
|
sparseSlice :: forall v'1 v'2 v'3 v'4 v'5 t. TensorType t => 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' :: 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)
|
|
|
|
sparseSliceGrad :: 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 => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor v'4 Int64 -> Tensor Build t
|
|
sparseSliceGrad' :: 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
|
|
|
|
sparseSoftmax :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build t
|
|
sparseSoftmax' :: 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
|
|
|
|
sparseSoftmaxCrossEntropyWithLogits :: forall v'1 v'2 t tlabels. (OneOf '[Word16, Double, Float] t, OneOf '[Int32, Int64] tlabels) => Tensor v'1 t -> Tensor v'2 tlabels -> (Tensor Build t, Tensor Build t)
|
|
sparseSoftmaxCrossEntropyWithLogits' :: 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)
|
|
|
|
sparseSparseMaximum :: 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 => 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' :: 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)
|
|
|
|
sparseSparseMinimum :: 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 => 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' :: 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)
|
|
|
|
sparseSplit :: forall v'1 v'2 v'3 v'4 t. TensorType t => 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' :: 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])
|
|
|
|
sparseTensorDenseAdd :: 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) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 t -> Tensor Build t
|
|
sparseTensorDenseAdd' :: 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
|
|
|
|
sparseTensorDenseMatMul :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 t -> Tensor Build t
|
|
sparseTensorDenseMatMul' :: 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
|
|
|
|
sparseTensorSliceDataset :: forall v'1 v'2 v'3 tvalues m'. (MonadBuild m', TensorType tvalues) => Tensor v'1 Int64 -> Tensor v'2 tvalues -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
|
|
sparseTensorSliceDataset' :: 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)
|
|
|
|
sparseTensorToCSRSparseMatrix :: forall v'1 v'2 v'3 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 Int64 -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor Build Variant
|
|
sparseTensorToCSRSparseMatrix' :: 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
|
|
|
|
sparseToDense :: forall v'1 v'2 v'3 v'4 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 tindices -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
|
|
sparseToDense' :: 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
|
|
|
|
sparseToSparseSetOperation :: forall v'1 v'2 v'3 v'4 v'5 v'6 t. OneOf '[ByteString, Int16, Int32, Int64, Int8, Word16, Word8] t => 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' :: 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)
|
|
|
|
spence :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
spence' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
split :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
|
|
split' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 Int32 -> Tensor v'2 t -> [Tensor Build t]
|
|
|
|
splitV :: forall v'1 v'2 v'3 t tlen. (TensorType t, OneOf '[Int32, Int64] tlen) => Int64 -> Tensor v'1 t -> Tensor v'2 tlen -> Tensor v'3 Int32 -> [Tensor Build t]
|
|
splitV' :: 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]
|
|
|
|
sqlDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => [DataType] -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value Variant)
|
|
sqlDataset' :: 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)
|
|
|
|
sqrt :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
sqrt' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
sqrtGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
sqrtGrad' :: 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
|
|
|
|
square :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
square' :: 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
|
|
|
|
squaredDifference :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
squaredDifference' :: 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
|
|
|
|
squeeze :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
squeeze' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
stack :: forall m'. MonadBuild m' => DataType -> m' (Tensor Ref ByteString)
|
|
stack' :: forall m'. MonadBuild m' => OpParams -> DataType -> m' (Tensor Ref ByteString)
|
|
|
|
stackClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
|
|
stackClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
|
|
|
|
stackCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
stackCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
stackPop :: forall elem_type m'. (MonadBuild m', TensorType elem_type) => Tensor Ref ByteString -> m' (Tensor Value elem_type)
|
|
stackPop' :: forall elem_type m'. (MonadBuild m', TensorType elem_type) => OpParams -> Tensor Ref ByteString -> m' (Tensor Value elem_type)
|
|
|
|
stackPopV2 :: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type) => Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
|
|
stackPopV2' :: forall v'1 elem_type m'. (MonadBuild m', TensorType elem_type) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value elem_type)
|
|
|
|
stackPush :: forall v'2 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
|
|
stackPush' :: forall v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor Ref ByteString -> Tensor v'2 t -> m' (Tensor Value t)
|
|
|
|
stackPushV2 :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
|
|
stackPushV2' :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 t -> m' (Tensor Value t)
|
|
|
|
stackV2 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
|
|
stackV2' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle)
|
|
|
|
stage :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => TensorList v'1 dtypes -> m' ControlNode
|
|
stage' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> TensorList v'1 dtypes -> m' ControlNode
|
|
|
|
stageClear :: forall m'. MonadBuild m' => [DataType] -> m' ControlNode
|
|
stageClear' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' ControlNode
|
|
|
|
stagePeek :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => Tensor v'1 Int32 -> m' (TensorList Value dtypes)
|
|
stagePeek' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> Tensor v'1 Int32 -> m' (TensorList Value dtypes)
|
|
|
|
stageSize :: forall m'. MonadBuild m' => [DataType] -> m' (Tensor Value Int32)
|
|
stageSize' :: forall m'. MonadBuild m' => OpParams -> [DataType] -> m' (Tensor Value Int32)
|
|
|
|
statefulRandomBinomial :: 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) => 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' :: 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)
|
|
|
|
statefulStandardNormal :: forall v'1 v'2 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 shape_dtype -> m' (Tensor Value dtype)
|
|
statefulStandardNormal' :: 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)
|
|
|
|
statefulStandardNormalV2 :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
|
|
statefulStandardNormalV2' :: 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 :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
|
|
statefulTruncatedNormal' :: 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 :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
|
|
statefulUniform' :: 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 :: forall v'1 v'2 v'3 dtype shape_dtype m'. (MonadBuild m', TensorType dtype, TensorType shape_dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> m' (Tensor Value dtype)
|
|
statefulUniformFullInt' :: 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)
|
|
|
|
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 -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value dtype)
|
|
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 -> Tensor v'2 Int64 -> Tensor v'3 shape_dtype -> Tensor v'4 dtype -> Tensor v'5 dtype -> m' (Tensor Value dtype)
|
|
|
|
statelessMultinomial :: 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) => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor v'3 tseed -> Tensor Build output_dtype
|
|
statelessMultinomial' :: 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
|
|
|
|
statelessParameterizedTruncatedNormal :: 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) => 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' :: 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
|
|
|
|
statelessRandomBinomial :: 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) => Tensor v'1 s -> Tensor v'2 tseed -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build dtype
|
|
statelessRandomBinomial' :: 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
|
|
|
|
statelessRandomGammaV2 :: forall v'1 v'2 v'3 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor Build dtype
|
|
statelessRandomGammaV2' :: 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
|
|
|
|
statelessRandomNormal :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
|
|
statelessRandomNormal' :: 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
|
|
|
|
statelessRandomPoisson :: 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) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 rtype -> Tensor Build dtype
|
|
statelessRandomPoisson' :: 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
|
|
|
|
statelessRandomUniform :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
|
|
statelessRandomUniform' :: 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
|
|
|
|
statelessRandomUniformFullInt :: forall v'1 v'2 dtype t tseed. (OneOf '[Int32, Int64, Word32, Word64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64, Word32, Word64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
|
|
statelessRandomUniformFullInt' :: 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
|
|
|
|
statelessRandomUniformInt :: forall v'1 v'2 v'3 v'4 dtype t tseed. (OneOf '[Int32, Int64] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor v'3 dtype -> Tensor v'4 dtype -> Tensor Build dtype
|
|
statelessRandomUniformInt' :: 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
|
|
|
|
statelessTruncatedNormal :: forall v'1 v'2 dtype t tseed. (OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t, OneOf '[Int32, Int64] tseed) => Tensor v'1 t -> Tensor v'2 tseed -> Tensor Build dtype
|
|
statelessTruncatedNormal' :: 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
|
|
|
|
staticRegexFullMatch :: ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
|
|
staticRegexFullMatch' :: OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor Build Bool
|
|
|
|
staticRegexReplace :: ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
staticRegexReplace' :: OpParams -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
statsAggregatorHandle :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
statsAggregatorHandle' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
statsAggregatorHandleV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
statsAggregatorHandleV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
statsAggregatorSetSummaryWriter :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' ControlNode
|
|
statsAggregatorSetSummaryWriter' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ResourceHandle -> m' ControlNode
|
|
|
|
statsAggregatorSummary :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
statsAggregatorSummary' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
stopGradient :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
stopGradient' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
stridedSlice :: forall v'1 v'2 v'3 v'4 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor Build t
|
|
stridedSlice' :: 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
|
|
|
|
stridedSliceAssign :: forall v'2 v'3 v'4 v'5 t index m'. (MonadBuild m', TensorType t, OneOf '[Int32, Int64] index) => Tensor Ref t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> m' (Tensor Ref t)
|
|
stridedSliceAssign' :: 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)
|
|
|
|
stridedSliceGrad :: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 index -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> Tensor Build t
|
|
stridedSliceGrad' :: 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
|
|
|
|
stringFormat :: forall v'1 t. TensorTypes t => TensorList v'1 t -> Tensor Build ByteString
|
|
stringFormat' :: forall v'1 t. TensorTypes t => OpParams -> TensorList v'1 t -> Tensor Build ByteString
|
|
|
|
stringJoin :: [Tensor v'1 ByteString] -> Tensor Build ByteString
|
|
stringJoin' :: OpParams -> [Tensor v'1 ByteString] -> Tensor Build ByteString
|
|
|
|
stringLength :: Tensor v'1 ByteString -> Tensor Build Int32
|
|
stringLength' :: OpParams -> Tensor v'1 ByteString -> Tensor Build Int32
|
|
|
|
stringLower :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
stringLower' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
stringNGrams :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Int64 -> Bool -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 tsplits -> (Tensor Build ByteString, Tensor Build tsplits)
|
|
stringNGrams' :: 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)
|
|
|
|
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)
|
|
|
|
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)
|
|
|
|
stringStrip :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
stringStrip' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
stringToHashBucket :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
stringToHashBucket' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
|
|
stringToHashBucketFast :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
stringToHashBucketFast' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
|
|
stringToHashBucketStrong :: Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
stringToHashBucketStrong' :: OpParams -> Int64 -> Tensor v'1 ByteString -> Tensor Build Int64
|
|
|
|
stringToNumber :: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type => Tensor v'1 ByteString -> Tensor Build out_type
|
|
stringToNumber' :: forall v'1 out_type. OneOf '[Int32, Int64, Double, Float] out_type => OpParams -> Tensor v'1 ByteString -> Tensor Build out_type
|
|
|
|
stringUpper :: Tensor v'1 ByteString -> Tensor Build ByteString
|
|
stringUpper' :: OpParams -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
sub :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
sub' :: 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
|
|
|
|
substr :: forall v'1 v'2 v'3 t. OneOf '[Int32, Int64] t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 t -> Tensor Build ByteString
|
|
substr' :: 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
|
|
|
|
sum :: 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) => Tensor v'1 t -> Tensor v'2 tidx -> Tensor Build t
|
|
sum' :: 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
|
|
|
|
summaryWriter :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
summaryWriter' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
svd :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
|
|
svd' :: 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)
|
|
|
|
switch :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 Bool -> (Tensor Build t, Tensor Build t)
|
|
switch' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Bool -> (Tensor Build t, Tensor Build t)
|
|
|
|
tFRecordDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
|
|
tFRecordDataset' :: 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)
|
|
|
|
tFRecordReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
tFRecordReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
tFRecordReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
tFRecordReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
tPUCompilationResult :: Tensor Build ByteString
|
|
tPUCompilationResult' :: OpParams -> Tensor Build ByteString
|
|
|
|
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
|
|
|
|
tPUOrdinalSelector :: forall m'. MonadBuild m' => m' (Tensor Value Int32)
|
|
tPUOrdinalSelector' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Int32)
|
|
|
|
tPUReplicateMetadata :: forall m'. MonadBuild m' => Int64 -> m' ControlNode
|
|
tPUReplicateMetadata' :: forall m'. MonadBuild m' => OpParams -> Int64 -> m' ControlNode
|
|
|
|
tPUReplicatedInput :: forall v'1 t. TensorType t => [Tensor v'1 t] -> Tensor Build t
|
|
tPUReplicatedInput' :: forall v'1 t. TensorType t => OpParams -> [Tensor v'1 t] -> Tensor Build t
|
|
|
|
tPUReplicatedOutput :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> [Tensor Build t]
|
|
tPUReplicatedOutput' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
|
|
|
|
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
|
|
|
|
takeManySparseFromTensorsMap :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
|
|
takeManySparseFromTensorsMap' :: forall v'1 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value dtype, Tensor Value Int64)
|
|
|
|
tan :: forall v'1 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
tan' :: 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
|
|
|
|
tanh :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
tanh' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
tanhGrad :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
tanhGrad' :: 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
|
|
|
|
temporaryVariable :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
|
|
temporaryVariable' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
|
|
|
|
tensorArray :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
|
|
tensorArray' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Ref ByteString)
|
|
|
|
tensorArrayClose :: forall m'. MonadBuild m' => Tensor Ref ByteString -> m' ControlNode
|
|
tensorArrayClose' :: forall m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> m' ControlNode
|
|
|
|
tensorArrayCloseV2 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' ControlNode
|
|
tensorArrayCloseV2' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
|
|
|
|
tensorArrayCloseV3 :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' ControlNode
|
|
tensorArrayCloseV3' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' ControlNode
|
|
|
|
tensorArrayConcat :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
|
|
tensorArrayConcat' :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
|
|
|
|
tensorArrayConcatV2 :: forall v'1 v'2 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Float -> (Tensor Build dtype, Tensor Build Int64)
|
|
tensorArrayConcatV2' :: forall v'1 v'2 dtype. TensorType dtype => OpParams -> Tensor v'1 ByteString -> Tensor v'2 Float -> (Tensor Build dtype, Tensor Build Int64)
|
|
|
|
tensorArrayConcatV3 :: forall v'1 v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value dtype, Tensor Value Int64)
|
|
tensorArrayConcatV3' :: 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)
|
|
|
|
tensorArrayGather :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
|
|
tensorArrayGather' :: 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)
|
|
|
|
tensorArrayGatherV2 :: forall v'1 v'2 v'3 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
|
|
tensorArrayGatherV2' :: 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
|
|
|
|
tensorArrayGatherV3 :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
|
|
tensorArrayGatherV3' :: 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)
|
|
|
|
tensorArrayGrad :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
|
|
tensorArrayGrad' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Ref ByteString)
|
|
|
|
tensorArrayGradV2 :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
|
|
tensorArrayGradV2' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> ByteString -> Tensor v'1 ByteString -> Tensor v'2 Float -> m' (Tensor Value ByteString)
|
|
|
|
tensorArrayGradV3 :: forall v'1 v'2 m'. MonadBuild m' => ByteString -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value ResourceHandle, Tensor Value Float)
|
|
tensorArrayGradV3' :: 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)
|
|
|
|
tensorArrayGradWithShape :: forall v'1 v'2 v'3 m'. MonadBuild m' => ByteString -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> Tensor v'3 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
|
|
tensorArrayGradWithShape' :: 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)
|
|
|
|
tensorArrayPack :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype)
|
|
tensorArrayPack' :: forall v'2 dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value dtype)
|
|
|
|
tensorArrayRead :: forall v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
|
|
tensorArrayRead' :: 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)
|
|
|
|
tensorArrayReadV2 :: forall v'1 v'2 v'3 dtype. TensorType dtype => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 Float -> Tensor Build dtype
|
|
tensorArrayReadV2' :: 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
|
|
|
|
tensorArrayReadV3 :: forall v'1 v'2 v'3 dtype m'. (MonadBuild m', TensorType dtype) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 Float -> m' (Tensor Value dtype)
|
|
tensorArrayReadV3' :: 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)
|
|
|
|
tensorArrayScatter :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArrayScatter' :: 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)
|
|
|
|
tensorArrayScatterV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
|
|
tensorArrayScatterV2' :: 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
|
|
|
|
tensorArrayScatterV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArrayScatterV3' :: 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)
|
|
|
|
tensorArraySize :: forall v'2 m'. MonadBuild m' => Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value Int32)
|
|
tensorArraySize' :: forall v'2 m'. MonadBuild m' => OpParams -> Tensor Ref ByteString -> Tensor v'2 Float -> m' (Tensor Value Int32)
|
|
|
|
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
|
|
|
|
tensorArraySizeV3 :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Int32)
|
|
tensorArraySizeV3' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Int32)
|
|
|
|
tensorArraySplit :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArraySplit' :: 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)
|
|
|
|
tensorArraySplitV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> Tensor Build Float
|
|
tensorArraySplitV2' :: 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
|
|
|
|
tensorArraySplitV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 t -> Tensor v'3 Int64 -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArraySplitV3' :: 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)
|
|
|
|
tensorArrayUnpack :: forall v'2 v'3 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 t -> Tensor v'3 Float -> m' (Tensor Value Float)
|
|
tensorArrayUnpack' :: 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)
|
|
|
|
tensorArrayV2 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
|
|
tensorArrayV2' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ByteString)
|
|
|
|
tensorArrayV3 :: forall v'1 m'. MonadBuild m' => DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
|
|
tensorArrayV3' :: forall v'1 m'. MonadBuild m' => OpParams -> DataType -> Tensor v'1 Int32 -> m' (Tensor Value ResourceHandle, Tensor Value Float)
|
|
|
|
tensorArrayWrite :: forall v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor Ref ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArrayWrite' :: 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)
|
|
|
|
tensorArrayWriteV2 :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> Tensor Build Float
|
|
tensorArrayWriteV2' :: 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
|
|
|
|
tensorArrayWriteV3 :: forall v'1 v'2 v'3 v'4 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int32 -> Tensor v'3 t -> Tensor v'4 Float -> m' (Tensor Value Float)
|
|
tensorArrayWriteV3' :: 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)
|
|
|
|
tensorDataset :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => TensorList v'1 toutput_types -> m' (Tensor Value Variant)
|
|
tensorDataset' :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => OpParams -> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
|
|
|
|
tensorForestCreateTreeVariable :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
tensorForestCreateTreeVariable' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
tensorForestTreeDeserialize :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
tensorForestTreeDeserialize' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
tensorForestTreeIsInitializedOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
tensorForestTreeIsInitializedOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
|
|
tensorForestTreePredict :: forall v'1 v'2 m'. MonadBuild m' => Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Float)
|
|
tensorForestTreePredict' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Int64 -> Tensor v'1 ResourceHandle -> Tensor v'2 Float -> m' (Tensor Value Float)
|
|
|
|
tensorForestTreeResourceHandleOp :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
tensorForestTreeResourceHandleOp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
tensorForestTreeSerialize :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
tensorForestTreeSerialize' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value ByteString)
|
|
|
|
tensorForestTreeSize :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
tensorForestTreeSize' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Int32)
|
|
|
|
tensorListConcat :: forall v'1 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> (Tensor Build element_dtype, Tensor Build Int64)
|
|
tensorListConcat' :: forall v'1 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> (Tensor Build element_dtype, Tensor Build Int64)
|
|
|
|
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
|
|
|
|
tensorListConcatV2 :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 Variant -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> (Tensor Build element_dtype, Tensor Build Int64)
|
|
tensorListConcatV2' :: 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)
|
|
|
|
tensorListElementShape :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => Tensor v'1 Variant -> Tensor Build shape_type
|
|
tensorListElementShape' :: forall v'1 shape_type. OneOf '[Int32, Int64] shape_type => OpParams -> Tensor v'1 Variant -> Tensor Build shape_type
|
|
|
|
tensorListFromTensor :: forall v'1 v'2 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor Build Variant
|
|
tensorListFromTensor' :: 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
|
|
|
|
tensorListGather :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build element_dtype
|
|
tensorListGather' :: 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 :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 Int32 -> Tensor Build element_dtype
|
|
tensorListGetItem' :: 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
|
|
|
|
tensorListLength :: Tensor v'1 Variant -> Tensor Build Int32
|
|
tensorListLength' :: OpParams -> Tensor v'1 Variant -> Tensor Build Int32
|
|
|
|
tensorListPopBack :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> (Tensor Build Variant, Tensor Build element_dtype)
|
|
tensorListPopBack' :: 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)
|
|
|
|
tensorListPushBack :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
|
|
tensorListPushBack' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
|
|
|
|
tensorListPushBackBatch :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
|
|
tensorListPushBackBatch' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor Build Variant
|
|
|
|
tensorListReserve :: forall v'1 v'2 shape_type. OneOf '[Int32, Int64] shape_type => DataType -> Tensor v'1 shape_type -> Tensor v'2 Int32 -> Tensor Build Variant
|
|
tensorListReserve' :: 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
|
|
|
|
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
|
|
|
|
tensorListScatter :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor Build Variant
|
|
tensorListScatter' :: 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
|
|
|
|
tensorListScatterIntoExistingList :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 element_dtype -> Tensor v'3 Int32 -> Tensor Build Variant
|
|
tensorListScatterIntoExistingList' :: 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
|
|
|
|
tensorListScatterV2 :: forall v'1 v'2 v'3 v'4 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 Int32 -> Tensor v'3 shape_type -> Tensor v'4 Int32 -> Tensor Build Variant
|
|
tensorListScatterV2' :: 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
|
|
|
|
tensorListSetItem :: forall v'1 v'2 v'3 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor v'3 element_dtype -> Tensor Build Variant
|
|
tensorListSetItem' :: 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
|
|
|
|
tensorListSplit :: forall v'1 v'2 v'3 element_dtype shape_type. (TensorType element_dtype, OneOf '[Int32, Int64] shape_type) => Tensor v'1 element_dtype -> Tensor v'2 shape_type -> Tensor v'3 Int64 -> Tensor Build Variant
|
|
tensorListSplit' :: 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
|
|
|
|
tensorListStack :: forall v'1 v'2 element_dtype. TensorType element_dtype => Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
|
|
tensorListStack' :: forall v'1 v'2 element_dtype. TensorType element_dtype => OpParams -> Tensor v'1 Variant -> Tensor v'2 Int32 -> Tensor Build element_dtype
|
|
|
|
tensorScatterAdd :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
tensorScatterAdd' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
|
|
tensorScatterMax :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
tensorScatterMax' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
|
|
tensorScatterMin :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
tensorScatterMin' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
|
|
tensorScatterSub :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
tensorScatterSub' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
|
|
tensorScatterUpdate :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
tensorScatterUpdate' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 t -> Tensor Build t
|
|
|
|
tensorSliceDataset :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => TensorList v'1 toutput_types -> m' (Tensor Value Variant)
|
|
tensorSliceDataset' :: forall v'1 toutput_types m'. (MonadBuild m', TensorTypes toutput_types) => OpParams -> TensorList v'1 toutput_types -> m' (Tensor Value Variant)
|
|
|
|
tensorStridedSliceUpdate :: forall v'1 v'2 v'3 v'4 v'5 t index. (TensorType t, OneOf '[Int32, Int64] index) => Tensor v'1 t -> Tensor v'2 index -> Tensor v'3 index -> Tensor v'4 index -> Tensor v'5 t -> Tensor Build t
|
|
tensorStridedSliceUpdate' :: 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
|
|
|
|
tensorSummary :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build ByteString
|
|
tensorSummary' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build ByteString
|
|
|
|
tensorSummaryV2 :: forall v'1 v'2 v'3 t. TensorType t => Tensor v'1 ByteString -> Tensor v'2 t -> Tensor v'3 ByteString -> Tensor Build ByteString
|
|
tensorSummaryV2' :: 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
|
|
|
|
textLineDataset :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 Int64 -> m' (Tensor Value Variant)
|
|
textLineDataset' :: 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)
|
|
|
|
textLineReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
textLineReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
textLineReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
textLineReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
threadPoolDataset :: forall v'1 v'2 m'. MonadBuild m' => [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
|
|
threadPoolDataset' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor v'2 ResourceHandle -> m' (Tensor Value Variant)
|
|
|
|
threadPoolHandle :: forall m'. MonadBuild m' => ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
|
|
threadPoolHandle' :: forall m'. MonadBuild m' => OpParams -> ByteString -> Int64 -> m' (Tensor Value ResourceHandle)
|
|
|
|
threadUnsafeUnigramCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
threadUnsafeUnigramCandidateSampler' :: 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)
|
|
|
|
tile :: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples) => Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
|
|
tile' :: forall v'1 v'2 t tmultiples. (TensorType t, OneOf '[Int32, Int64] tmultiples) => OpParams -> Tensor v'1 t -> Tensor v'2 tmultiples -> Tensor Build t
|
|
|
|
tileGrad :: forall v'1 v'2 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
tileGrad' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 Int32 -> Tensor Build t
|
|
|
|
timestamp :: forall m'. MonadBuild m' => m' (Tensor Value Double)
|
|
timestamp' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Double)
|
|
|
|
toBool :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build Bool
|
|
toBool' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build Bool
|
|
|
|
topK :: forall v'1 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build Int32)
|
|
topK' :: 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)
|
|
|
|
topKV2 :: forall v'1 v'2 t. OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 Int32 -> (Tensor Build t, Tensor Build Int32)
|
|
topKV2' :: 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)
|
|
|
|
transpose :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
|
|
transpose' :: forall v'1 v'2 t tperm. (TensorType t, OneOf '[Int32, Int64] tperm) => OpParams -> Tensor v'1 t -> Tensor v'2 tperm -> Tensor Build t
|
|
|
|
tridiagonalMatMul :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 t -> Tensor v'4 t -> Tensor Build t
|
|
tridiagonalMatMul' :: 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
|
|
|
|
tridiagonalSolve :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
tridiagonalSolve' :: 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
|
|
|
|
truncateDiv :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
truncateDiv' :: 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
|
|
|
|
truncateMod :: forall v'1 v'2 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
truncateMod' :: 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
|
|
|
|
truncatedNormal :: forall v'1 dtype t m'. (MonadBuild m', OneOf '[Word16, Double, Float] dtype, OneOf '[Int32, Int64] t) => Tensor v'1 t -> m' (Tensor Value dtype)
|
|
truncatedNormal' :: 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)
|
|
|
|
tryRpc :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> Tensor v'3 ByteString -> m' (Tensor Value ByteString, Tensor Value Int32, Tensor Value ByteString)
|
|
tryRpc' :: 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)
|
|
|
|
unbatch :: forall v'1 v'2 v'3 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 Int64 -> Tensor Build t
|
|
unbatch' :: 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
|
|
|
|
unbatchDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
unbatchDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
unbatchGrad :: forall v'1 v'2 v'3 v'4 t. TensorType t => Tensor v'1 t -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 Int64 -> Tensor Build t
|
|
unbatchGrad' :: 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
|
|
|
|
uncompressElement :: forall v'1 output_types. TensorTypes output_types => Tensor v'1 Variant -> TensorList Build output_types
|
|
uncompressElement' :: forall v'1 output_types. TensorTypes output_types => OpParams -> Tensor v'1 Variant -> TensorList Build output_types
|
|
|
|
unicodeDecode :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32)
|
|
unicodeDecode' :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32)
|
|
|
|
unicodeDecodeWithOffsets :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
|
|
unicodeDecodeWithOffsets' :: forall v'1 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 ByteString -> (Tensor Build tsplits, Tensor Build Int32, Tensor Build Int64)
|
|
|
|
unicodeEncode :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => ByteString -> Tensor v'1 Int32 -> Tensor v'2 tsplits -> Tensor Build ByteString
|
|
unicodeEncode' :: forall v'1 v'2 tsplits. OneOf '[Int32, Int64] tsplits => OpParams -> ByteString -> Tensor v'1 Int32 -> Tensor v'2 tsplits -> Tensor Build ByteString
|
|
|
|
unicodeScript :: Tensor v'1 Int32 -> Tensor Build Int32
|
|
unicodeScript' :: OpParams -> Tensor v'1 Int32 -> Tensor Build Int32
|
|
|
|
unicodeTranscode :: ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
unicodeTranscode' :: OpParams -> ByteString -> ByteString -> Tensor v'1 ByteString -> Tensor Build ByteString
|
|
|
|
uniformCandidateSampler :: forall v'1 m'. MonadBuild m' => Int64 -> Int64 -> Int64 -> Bool -> Tensor v'1 Int64 -> m' (Tensor Value Int64, Tensor Value Float, Tensor Value Float)
|
|
uniformCandidateSampler' :: 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)
|
|
|
|
unique :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx)
|
|
unique' :: 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)
|
|
|
|
uniqueDataset :: [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
uniqueDataset' :: OpParams -> [DataType] -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
uniqueV2 :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx)
|
|
uniqueV2' :: 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)
|
|
|
|
uniqueWithCounts :: forall v'1 t out_idx. (TensorType t, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
|
|
uniqueWithCounts' :: 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)
|
|
|
|
uniqueWithCountsV2 :: forall v'1 v'2 t taxis out_idx. (TensorType t, OneOf '[Int32, Int64] taxis, OneOf '[Int32, Int64] out_idx) => Tensor v'1 t -> Tensor v'2 taxis -> (Tensor Build t, Tensor Build out_idx, Tensor Build out_idx)
|
|
uniqueWithCountsV2' :: 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)
|
|
|
|
unpack :: forall v'1 t. TensorType t => Int64 -> Tensor v'1 t -> [Tensor Build t]
|
|
unpack' :: forall v'1 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> [Tensor Build t]
|
|
|
|
unravelIndex :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
|
|
unravelIndex' :: forall v'1 v'2 tidx. OneOf '[Int32, Int64] tidx => OpParams -> Tensor v'1 tidx -> Tensor v'2 tidx -> Tensor Build tidx
|
|
|
|
unsortedSegmentJoin :: forall v'1 v'2 v'3 tindices tnumsegments. (OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 ByteString -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build ByteString
|
|
unsortedSegmentJoin' :: 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
|
|
|
|
unsortedSegmentMax :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
|
|
unsortedSegmentMax' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[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
|
|
|
|
unsortedSegmentMin :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t, OneOf '[Int32, Int64] tindices, OneOf '[Int32, Int64] tnumsegments) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
|
|
unsortedSegmentMin' :: forall v'1 v'2 v'3 t tindices tnumsegments. (OneOf '[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 :: 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) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
|
|
unsortedSegmentProd' :: 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 :: 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) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tnumsegments -> Tensor Build t
|
|
unsortedSegmentSum' :: 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
|
|
|
|
unstage :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => m' (TensorList Value dtypes)
|
|
unstage' :: forall dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> m' (TensorList Value dtypes)
|
|
|
|
unwrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
|
|
unwrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
upperBound :: forall v'1 v'2 t out_type. (TensorType t, OneOf '[Int32, Int64] out_type) => Tensor v'1 t -> Tensor v'2 t -> Tensor Build out_type
|
|
upperBound' :: 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
|
|
|
|
varHandleOp :: forall m'. MonadBuild m' => DataType -> Shape -> m' (Tensor Value ResourceHandle)
|
|
varHandleOp' :: forall m'. MonadBuild m' => OpParams -> DataType -> Shape -> m' (Tensor Value ResourceHandle)
|
|
|
|
varIsInitializedOp :: forall v'1 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
varIsInitializedOp' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value Bool)
|
|
|
|
variable :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
|
|
variable' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
|
|
|
|
variableShape :: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type) => Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
|
|
variableShape' :: forall v'1 out_type m'. (MonadBuild m', OneOf '[Int32, Int64] out_type) => OpParams -> Tensor v'1 ResourceHandle -> m' (Tensor Value out_type)
|
|
|
|
variableV2 :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Ref dtype)
|
|
variableV2' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Ref dtype)
|
|
|
|
where' :: forall v'1 t. OneOf '[Complex Double, Complex Float, Bool, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Tensor v'1 t -> Tensor Build Int64
|
|
where'' :: 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
|
|
|
|
wholeFileReader :: forall m'. MonadBuild m' => m' (Tensor Ref ByteString)
|
|
wholeFileReader' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Ref ByteString)
|
|
|
|
wholeFileReaderV2 :: forall m'. MonadBuild m' => m' (Tensor Value ResourceHandle)
|
|
wholeFileReaderV2' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value ResourceHandle)
|
|
|
|
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
|
|
|
|
workerHeartbeat :: forall v'1 m'. MonadBuild m' => Tensor v'1 ByteString -> m' (Tensor Value ByteString)
|
|
workerHeartbeat' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value ByteString)
|
|
|
|
wrapDatasetVariant :: Tensor v'1 Variant -> Tensor Build Variant
|
|
wrapDatasetVariant' :: OpParams -> Tensor v'1 Variant -> Tensor Build Variant
|
|
|
|
writeAudioSummary :: forall v'1 v'2 v'3 v'4 v'5 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 Float -> Tensor v'5 Float -> m' ControlNode
|
|
writeAudioSummary' :: 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
|
|
|
|
writeFile :: forall v'1 v'2 m'. MonadBuild m' => Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
writeFile' :: forall v'1 v'2 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> Tensor v'2 ByteString -> m' ControlNode
|
|
|
|
writeGraphSummary :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
|
|
writeGraphSummary' :: 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
|
|
|
|
writeHistogramSummary :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
|
|
writeHistogramSummary' :: 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
|
|
|
|
writeImageSummary :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', OneOf '[Word16, Word8, Float] t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> Tensor v'5 Word8 -> m' ControlNode
|
|
writeImageSummary' :: 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
|
|
|
|
writeRawProtoSummary :: forall v'1 v'2 v'3 m'. MonadBuild m' => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> m' ControlNode
|
|
writeRawProtoSummary' :: 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
|
|
|
|
writeScalarSummary :: 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) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 ByteString -> Tensor v'4 t -> m' ControlNode
|
|
writeScalarSummary' :: 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
|
|
|
|
writeSummary :: forall v'1 v'2 v'3 v'4 v'5 t m'. (MonadBuild m', TensorType t) => Tensor v'1 ResourceHandle -> Tensor v'2 Int64 -> Tensor v'3 t -> Tensor v'4 ByteString -> Tensor v'5 ByteString -> m' ControlNode
|
|
writeSummary' :: 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
|
|
|
|
xdivy :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
xdivy' :: 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
|
|
|
|
-- | Helper operator for performing XLA-style broadcasts
|
|
--
|
|
-- Broadcasts <tt>lhs</tt> and <tt>rhs</tt> to the same rank, by adding
|
|
-- size 1 dimensions to whichever of <tt>lhs</tt> and <tt>rhs</tt> has
|
|
-- the lower rank, using XLA's broadcasting rules for binary operators.
|
|
xlaBroadcastHelper :: 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) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> (Tensor Build t, Tensor Build t)
|
|
xlaBroadcastHelper' :: 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)
|
|
|
|
-- | 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 -> Tensor Build t
|
|
xlaClusterOutput' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
-- | Wraps the XLA ConvGeneralDilated operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#conv_convolution</a>
|
|
-- .
|
|
xlaConv :: 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) => 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' :: 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
|
|
|
|
-- | Takes the packed uint32 input and unpacks the input to uint8 to do
|
|
--
|
|
-- Dequantization on device.
|
|
xlaDequantize :: forall v'1 m'. MonadBuild m' => Float -> Float -> ByteString -> Bool -> Tensor v'1 Word32 -> m' (Tensor Value Word16)
|
|
xlaDequantize' :: forall v'1 m'. MonadBuild m' => OpParams -> Float -> Float -> ByteString -> Bool -> Tensor v'1 Word32 -> m' (Tensor Value Word16)
|
|
|
|
-- | Wraps the XLA DotGeneral operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#dotgeneral</a>
|
|
-- .
|
|
xlaDot :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => ByteString -> ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
xlaDot' :: 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
|
|
|
|
-- | Wraps the XLA DynamicSlice operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#dynamicslice</a>
|
|
-- .
|
|
--
|
|
-- 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 '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
|
|
xlaDynamicSlice' :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => OpParams -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
|
|
|
|
-- | Wraps the XLA DynamicUpdateSlice operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#dynamicupdateslice</a>
|
|
-- .
|
|
--
|
|
-- XlaDynamicUpdateSlice generates a result which is the value of the
|
|
-- <tt>input</tt> operand, with a slice update overwritten at
|
|
-- <tt>indices</tt>. The shape of <tt>update</tt> 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
|
|
-- <tt>input</tt>.
|
|
--
|
|
-- Handling of out-of-bounds slice indices is implementation-defined.
|
|
xlaDynamicUpdateSlice :: forall v'1 v'2 v'3 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor Build t
|
|
xlaDynamicUpdateSlice' :: 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
|
|
|
|
-- | 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 '[Complex Float, Word16, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
xlaEinsum' :: 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
|
|
|
|
-- | Wraps the XLA Gather operator documented at
|
|
--
|
|
-- <a>https://www.tensorflow.org/xla/operation_semantics#gather</a>
|
|
xlaGather :: 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) => ByteString -> Bool -> Tensor v'1 t -> Tensor v'2 tindices -> Tensor v'3 tindices -> Tensor Build t
|
|
xlaGather' :: 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
|
|
|
|
-- | Wraps the XLA Sort operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#sort</a>
|
|
-- .
|
|
--
|
|
-- Sorts a tensor. Currently only sorts in ascending order are supported.
|
|
xlaKeyValueSort :: forall v'1 v'2 k v. (OneOf '[Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] k, TensorType v) => Tensor v'1 k -> Tensor v'2 v -> (Tensor Build k, Tensor Build v)
|
|
xlaKeyValueSort' :: 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)
|
|
|
|
-- | Wraps the XLA Pad operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#pad</a>
|
|
-- .
|
|
xlaPad :: forall v'1 v'2 v'3 v'4 v'5 t tindices. (TensorType t, OneOf '[Int32, Int64] tindices) => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 tindices -> Tensor v'4 tindices -> Tensor v'5 tindices -> Tensor Build t
|
|
xlaPad' :: 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
|
|
|
|
-- | Receives the named tensor from another XLA computation. Wraps the XLA
|
|
-- Recv
|
|
--
|
|
-- operator documented at
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#recv</a>
|
|
-- .
|
|
xlaRecv :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> ByteString -> m' (Tensor Value dtype)
|
|
xlaRecv' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> ByteString -> m' (Tensor Value dtype)
|
|
|
|
-- | Replica ID.
|
|
xlaReplicaId :: Tensor Build Int32
|
|
xlaReplicaId' :: OpParams -> Tensor Build 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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Float -> Bool -> Int64 -> Tensor v'1 t -> (Tensor Build t, Tensor Build t)
|
|
xlaSelfAdjointEig' :: 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)
|
|
|
|
-- | Sends the named tensor to another XLA computation. Wraps the XLA Send
|
|
-- operator
|
|
--
|
|
-- documented at
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#send</a>
|
|
-- .
|
|
xlaSend :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> Tensor v'1 t -> m' ControlNode
|
|
xlaSend' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | An op which shards the input based on the given sharding attribute.
|
|
xlaSharding :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
xlaSharding' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
-- | Wraps the XLA Sort operator, documented at
|
|
--
|
|
--
|
|
-- <a>https://www.tensorflow.org/performance/xla/operation_semantics#sort</a>
|
|
-- .
|
|
--
|
|
-- Sorts a tensor. Currently only sorts in ascending order are supported.
|
|
xlaSort :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
xlaSort' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
-- | 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 -> Tensor v'1 t -> Tensor Build t
|
|
xlaSpmdFullToShardShape' :: forall v'1 t. TensorType t => OpParams -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
-- | 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 -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
xlaSpmdShardToFullShape' :: forall v'1 t. TensorType t => OpParams -> Shape -> ByteString -> Tensor v'1 t -> Tensor Build t
|
|
|
|
-- | 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 '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word32, Word64, Word8, Double, Float] t => Float -> Int64 -> ByteString -> Tensor v'1 t -> (Tensor Build t, Tensor Build t, Tensor Build t)
|
|
xlaSvd' :: 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)
|
|
|
|
xlog1py :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
xlog1py' :: 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 :: forall v'1 v'2 t. OneOf '[Complex Double, Complex Float, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
xlogy' :: 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
|
|
|
|
zerosLike :: forall v'1 t. TensorType t => Tensor v'1 t -> Tensor Build t
|
|
zerosLike' :: forall v'1 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
zeta :: forall v'1 v'2 t. OneOf '[Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
zeta' :: forall v'1 v'2 t. OneOf '[Double, Float] t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
zipDataset :: [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
zipDataset' :: OpParams -> [DataType] -> [Tensor v'1 Variant] -> Tensor Build Variant
|
|
|
|
-- | A graph node which represents an argument to a function.
|
|
_Arg :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> m' (Tensor Value t)
|
|
_Arg' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> m' (Tensor Value t)
|
|
|
|
-- | 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] -> TensorList Build out_types
|
|
_ArrayToList' :: forall v'1 t out_types. (TensorType t, TensorTypes out_types) => OpParams -> [Tensor v'1 t] -> TensorList Build out_types
|
|
|
|
-- | An op that sets up the centralized structures for a distributed TPU
|
|
--
|
|
-- system.
|
|
_ConfigureDistributedTPU :: forall v'1 m'. MonadBuild m' => [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
|
|
_ConfigureDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
|
|
|
|
-- | A graph node which represents an argument to a function.
|
|
_DeviceArg :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> m' (Tensor Value t)
|
|
_DeviceArg' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> m' (Tensor Value t)
|
|
|
|
-- | A graph node which represents a return value of a function.
|
|
_DeviceRetval :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
|
|
_DeviceRetval' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | An op that disconnects the TPUs on a host from a running distributed
|
|
--
|
|
-- TPU system.
|
|
_DisconnectHostFromDistributedTPUSystem :: forall m'. MonadBuild m' => m' (Tensor Value Int32)
|
|
_DisconnectHostFromDistributedTPUSystem' :: forall m'. MonadBuild m' => OpParams -> m' (Tensor Value Int32)
|
|
|
|
-- | <ul>
|
|
-- <li>NOTE*: Do not invoke this operator directly in Python. Grappler
|
|
-- is</li>
|
|
-- </ul>
|
|
--
|
|
-- expected to create these operators.
|
|
_FusedBatchNormEx :: forall v'1 v'2 v'3 v'4 v'5 v'6 t u. (OneOf '[Word16, Float] t, OneOf '[Float] u) => 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' :: 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)
|
|
|
|
-- | Performs a convolution followed by a specified series of operations.
|
|
--
|
|
-- The inputs to the convolution are <tt>input</tt> and <a>filter</a>.
|
|
-- The series of operations that follows is specified by the
|
|
-- <tt>fused_ops</tt> attribute, which is a list of TF op names specified
|
|
-- as strings (e.g. <a>Relu</a>). 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 {<a>BiasAdd</a>,<a>FusedBatchNorm</a>} and A is one of
|
|
-- {<a>Elu</a>,<a>Relu</a>,<a>Relu6</a>}.
|
|
--
|
|
-- <ul>
|
|
-- <li>The first input to op X is the Conv2D result, and the additional
|
|
-- input(s) to X are specified by <tt>args</tt>.</li>
|
|
-- <li>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.</li>
|
|
-- <li>NOTE*: Do not invoke this operator directly in Python. Grappler is
|
|
-- expected to create these operators.</li>
|
|
-- </ul>
|
|
_FusedConv2D :: forall v'1 v'2 v'3 t. OneOf '[Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
|
|
_FusedConv2D' :: 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
|
|
|
|
_FusedDepthwiseConv2dNative :: forall v'1 v'2 v'3 t. OneOf '[Word16, Double, Float] t => ByteString -> Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
|
|
_FusedDepthwiseConv2dNative' :: 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
|
|
|
|
-- | Performs a MatMul followed by a specified series of operations.
|
|
--
|
|
-- The inputs to the MatMul are specified by <tt>a</tt> and <tt>b</tt>.
|
|
-- The series of operations that follows is specified by the
|
|
-- <tt>fused_ops</tt> attribute, which is a list of TF op names specified
|
|
-- as strings (e.g. <a>Relu</a>). 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: [<a>BiasAdd</a>] and
|
|
-- [<a>BiasAdd</a>,A], where A is one of
|
|
-- {<a>Elu</a>,<a>Relu</a>,<a>Relu6</a>}.
|
|
--
|
|
-- <ul>
|
|
-- <li>The first input to BiasAdd is the Conv2D result, and the
|
|
-- additional BiasAdd input is specified by <tt>args</tt>.</li>
|
|
-- <li>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.</li>
|
|
-- <li>NOTE*: Do not invoke this operator directly in Python. Grappler is
|
|
-- expected to create these operators.</li>
|
|
-- </ul>
|
|
_FusedMatMul :: forall v'1 v'2 v'3 t. OneOf '[Float] t => Tensor v'1 t -> Tensor v'2 t -> [Tensor v'3 t] -> Tensor Build t
|
|
_FusedMatMul' :: 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
|
|
|
|
-- | 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 -> Tensor Build dstT
|
|
_HostCast' :: forall v'1 srcT dstT. (TensorType srcT, TensorType dstT) => OpParams -> Tensor v'1 srcT -> Tensor Build dstT
|
|
|
|
-- | 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 -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
_HostRecv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
|
|
-- | 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 -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
_HostSend' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | 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 ByteString -> m' (Tensor Value Int32)
|
|
_InitializeHostForDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' (Tensor Value Int32)
|
|
|
|
-- | Converts a list of tensors to an array of tensors.
|
|
_ListToArray :: forall v'1 tin t. (TensorTypes tin, TensorType t) => Int64 -> TensorList v'1 tin -> [Tensor Build t]
|
|
_ListToArray' :: forall v'1 tin t. (TensorTypes tin, TensorType t) => OpParams -> Int64 -> TensorList v'1 tin -> [Tensor Build t]
|
|
|
|
-- | Returns the max of x and y (i.e. x > y ? x : y) element-wise.
|
|
--
|
|
-- <ul>
|
|
-- <li>NOTE*: <tt>Maximum</tt> supports broadcasting. More about
|
|
-- broadcasting <a>here</a></li>
|
|
-- </ul>
|
|
_MklMaximum :: forall v'1 v'2 v'3 v'4 t. OneOf '[Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
|
|
_MklMaximum' :: 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)
|
|
|
|
-- | Returns x * y element-wise.
|
|
--
|
|
-- <ul>
|
|
-- <li>NOTE*: <tt>Mul</tt> supports broadcasting. More about broadcasting
|
|
-- <a>here</a></li>
|
|
-- </ul>
|
|
_MklMul :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int16, Int32, Int64, Int8, Word16, Word8, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
|
|
_MklMul' :: 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)
|
|
|
|
-- | Returns (x - y)(x - y) element-wise.
|
|
--
|
|
-- <ul>
|
|
-- <li>NOTE*: <tt>SquaredDifference</tt> supports broadcasting. More
|
|
-- about broadcasting <a>here</a></li>
|
|
-- </ul>
|
|
_MklSquaredDifference :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
|
|
_MklSquaredDifference' :: 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)
|
|
|
|
-- | Returns x - y element-wise.
|
|
--
|
|
-- <ul>
|
|
-- <li>NOTE*: <tt>Sub</tt> supports broadcasting. More about broadcasting
|
|
-- <a>here</a></li>
|
|
-- </ul>
|
|
_MklSub :: forall v'1 v'2 v'3 v'4 t. OneOf '[Complex Double, Complex Float, Int32, Int64, Word16, Double, Float] t => Tensor v'1 t -> Tensor v'2 t -> Tensor v'3 Word8 -> Tensor v'4 Word8 -> (Tensor Build t, Tensor Build Word8)
|
|
_MklSub' :: 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)
|
|
|
|
-- | Replacement node for NcclBroadcast.
|
|
--
|
|
-- Sends data of shape <a>shape</a> from the _NcclBroadcastSend op
|
|
-- registered in the same <tt>shared_name</tt>. The graph should be
|
|
-- constructed so that one device runs <a>_NcclBroadcastSend</a> and
|
|
-- `num_devices-1` devices run _NcclBroadcastRecv ops with shared_name
|
|
-- value <tt>c</tt>. Failure to do so will cause the graph execution to
|
|
-- fail to complete.
|
|
_NcclBroadcastRecv :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> Tensor v'1 Int32 -> m' (Tensor Value t)
|
|
_NcclBroadcastRecv' :: 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)
|
|
|
|
-- | Replacement node for NcclBroadcast.
|
|
--
|
|
-- Sends <tt>input</tt> to the _NcclBroadcastRecv ops registered in the
|
|
-- same <tt>shared_name</tt>. The graph should be constructed so that one
|
|
-- device runs <a>_NcclBroadcastSend</a> and `num_devices-1` devices run
|
|
-- _NcclBroadcastRecv ops with shared_name value <tt>c</tt>. Failure to
|
|
-- do so will cause the graph execution to fail to complete.
|
|
_NcclBroadcastSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
_NcclBroadcastSend' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | Replacement node for NcclReduce.
|
|
--
|
|
-- Reduces <tt>input</tt> from this op and the NcclReduceSend ops
|
|
-- registered in the same <tt>shared_name</tt>. The graph should be
|
|
-- constructed so that 'num_devices-1' devices run <a>_NcclReduceSend</a>
|
|
-- and one device runs _NcclReduceRecv op with shared_name value
|
|
-- <tt>c</tt>. Failure to do so will cause the graph execution to fail to
|
|
-- complete.
|
|
_NcclReduceRecv :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' (Tensor Value t)
|
|
_NcclReduceRecv' :: 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)
|
|
|
|
-- | Replacement node for NcclReduce.
|
|
--
|
|
-- Reduces <tt>input</tt> to the NcclReduceRecv op registered in the same
|
|
-- <tt>shared_name</tt>. The graph should be constructed so that
|
|
-- 'num_devices-1' devices run <a>_NcclReduceSend</a> and one device runs
|
|
-- _NcclReduceRecv op with shared_name value <tt>c</tt>. Failure to do so
|
|
-- will cause the graph execution to fail to complete.
|
|
_NcclReduceSend :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
_NcclReduceSend' :: forall v'1 t m'. (MonadBuild m', OneOf '[Int32, Int64, Word16, Double, Float] t) => OpParams -> Int64 -> ByteString -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | Creates an empty Tensor with shape <a>shape</a> and type
|
|
-- <tt>dtype</tt>.
|
|
--
|
|
-- The memory can optionally be initialized. This is usually useful in
|
|
-- conjunction with inplace operations.
|
|
_ParallelConcatStart :: forall dtype m'. (MonadBuild m', TensorType dtype) => Shape -> m' (Tensor Value dtype)
|
|
_ParallelConcatStart' :: forall dtype m'. (MonadBuild m', TensorType dtype) => OpParams -> Shape -> m' (Tensor Value dtype)
|
|
|
|
-- | Updates input <a>value</a> at <tt>loc</tt> with <tt>update</tt>.
|
|
--
|
|
-- 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 => Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
_ParallelConcatUpdate' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
_ReadVariablesOp :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
|
|
_ReadVariablesOp' :: forall v'1 dtypes m'. (MonadBuild m', TensorTypes dtypes) => OpParams -> [Tensor v'1 ResourceHandle] -> m' (TensorList Value dtypes)
|
|
|
|
-- | Receives the named tensor from send_device on recv_device.
|
|
_Recv :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
_Recv' :: forall tensor_type m'. (MonadBuild m', TensorType tensor_type) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> m' (Tensor Value tensor_type)
|
|
|
|
-- | A graph node which represents a return value of a function.
|
|
_Retval :: forall v'1 t m'. (MonadBuild m', TensorType t) => Int64 -> Tensor v'1 t -> m' ControlNode
|
|
_Retval' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | 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.
|
|
--
|
|
-- <tt>shapes</tt> is a list of the shapes of the tensors that are to be
|
|
-- allocated by this ScopedAllocator. <a>shape</a> is the shape of the
|
|
-- output of this Op, i.e. the 1D backing tensor from which the
|
|
-- individual allocated tensors are aliased. <tt>sa_name</tt> is the name
|
|
-- assigned to the Node, for connectivity specification and debugging.
|
|
-- <a>id</a> is a non-negative integer <tt>scope_id</tt> handled by the
|
|
-- ScopedAllocatorMgr. <tt>expected_call_count</tt> is the number of
|
|
-- individual tensors expected to be allocated from the backing tensor.
|
|
_ScopedAllocator :: forall t m'. (MonadBuild m', TensorType t) => Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
|
|
_ScopedAllocator' :: forall t m'. (MonadBuild m', TensorType t) => OpParams -> Int64 -> Int64 -> ByteString -> Shape -> m' (Tensor Value t)
|
|
|
|
-- | 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.
|
|
--
|
|
-- <tt>backing</tt> is the backing tensor, i.e. the output of an upstream
|
|
-- ScopedAllocator. <tt>inputs</tt> 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. <a>shape</a> is the shape of the output, which
|
|
-- will usually be the same shape as the input backing tensor.
|
|
-- <a>reshape</a> is true iff the output shape is to be different from
|
|
-- that of the input backing tensor. <tt>sa_name</tt> is the Node name of
|
|
-- the upstream ScopedAllocator. <a>id</a> is the scope_id identifying
|
|
-- the upstream ScopedAllocator. <tt>N</tt> is the number of nominal
|
|
-- inputs to be concatenated.
|
|
_ScopedAllocatorConcat :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Int64 -> ByteString -> Shape -> Tensor v'1 t -> [Tensor v'2 t] -> m' (Tensor Value t)
|
|
_ScopedAllocatorConcat' :: 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)
|
|
|
|
-- | 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.
|
|
--
|
|
-- <a>concat</a> is the single output produced by an upstream
|
|
-- ScopedAllocatorConcat node. This is actually the backing tensor from a
|
|
-- ScopedAllocator node upstream of the ScopedAllocatorConcat.
|
|
-- <a>split</a> 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.
|
|
-- <tt>sa_name</tt> is the Node name of the upstream ScopedAllocator.
|
|
-- <a>id</a> is the scope_id identifying the upstream ScopedAllocator.
|
|
-- <tt>N</tt> is the number of split tensors. <tt>shapes</tt> is a list
|
|
-- of the split tensor shapes.
|
|
_ScopedAllocatorSplit :: forall v'1 v'2 t m'. (MonadBuild m', TensorType t) => Int64 -> ByteString -> Tensor v'1 t -> [Tensor v'2 t] -> m' [Tensor Value t]
|
|
_ScopedAllocatorSplit' :: 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]
|
|
|
|
-- | Sends the named tensor from send_device to recv_device.
|
|
_Send :: forall v'1 t m'. (MonadBuild m', TensorType t) => ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
_Send' :: forall v'1 t m'. (MonadBuild m', TensorType t) => OpParams -> ByteString -> ByteString -> Int64 -> ByteString -> Tensor v'1 t -> m' ControlNode
|
|
|
|
-- | 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 ByteString -> m' ControlNode
|
|
_SetGlobalTPUArray' :: forall v'1 m'. MonadBuild m' => OpParams -> Tensor v'1 ByteString -> m' ControlNode
|
|
|
|
-- | 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' :: forall m'. MonadBuild m' => OpParams -> m' ControlNode
|
|
|
|
_SwitchN :: forall v'1 v'2 t. TensorType t => Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
|
|
_SwitchN' :: forall v'1 v'2 t. TensorType t => OpParams -> Int64 -> Tensor v'1 t -> Tensor v'2 Int32 -> [Tensor Build t]
|
|
|
|
-- | <ul>
|
|
-- <li>NOTE*: Do not invoke this operator directly in Python. Graph
|
|
-- rewrite pass is</li>
|
|
-- </ul>
|
|
--
|
|
-- expected to create these operators.
|
|
_UnaryOpsComposition :: forall v'1 t. OneOf '[Word16, Double, Float] t => Tensor v'1 t -> Tensor Build t
|
|
_UnaryOpsComposition' :: forall v'1 t. OneOf '[Word16, Double, Float] t => OpParams -> Tensor v'1 t -> Tensor Build t
|
|
|
|
_VarHandlesOp :: forall m'. MonadBuild m' => Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
|
|
_VarHandlesOp' :: forall m'. MonadBuild m' => OpParams -> Int64 -> [DataType] -> m' [Tensor Value ResourceHandle]
|
|
|
|
-- | 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 Int32] -> m' (Tensor Value ByteString)
|
|
_WaitForDistributedTPU' :: forall v'1 m'. MonadBuild m' => OpParams -> [Tensor v'1 Int32] -> m' (Tensor Value ByteString)
|
|
|
|
-- | 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 -> Tensor v'2 t -> Tensor Build t
|
|
_XlaMerge' :: forall v'1 v'2 t. TensorType t => OpParams -> Tensor v'1 t -> Tensor v'2 t -> Tensor Build t
|
|
|
|
-- | A placeholder op to receive values from a running XLA computation.
|
|
_XlaRecvAtHost :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => Int64 -> ByteString -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
|
|
_XlaRecvAtHost' :: forall v'1 toutputs m'. (MonadBuild m', TensorTypes toutputs) => OpParams -> Int64 -> ByteString -> Tensor v'1 ByteString -> m' (TensorList Value toutputs)
|
|
|
|
-- | 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 -> Tensor v'2 ByteString -> m' (TensorList Value tresults)
|
|
_XlaRun' :: 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)
|
|
|
|
-- | A placeholder op to send values to a running XLA computation.
|
|
_XlaSendFromHost :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => Int64 -> ByteString -> TensorList v'1 tinputs -> Tensor v'2 ByteString -> m' ControlNode
|
|
_XlaSendFromHost' :: forall v'1 v'2 tinputs m'. (MonadBuild m', TensorTypes tinputs) => OpParams -> Int64 -> ByteString -> TensorList v'1 tinputs -> Tensor v'2 ByteString -> m' ControlNode
|