mxnet-0.2.0.0: MXNet interface in Haskell.

Copyright(c) 2016 Tao He
LicenseMIT
Maintainersighingnow@gmail.com
Safe HaskellNone
LanguageHaskell2010

MXNet.Core.Base.Internal.TH.Symbol

Description

Functions about Symbol that generated by template haskell.

Synopsis

Documentation

zeros_like :: String -> SymbolHandle -> IO SymbolHandle Source #

Register symbol operators.

transpose :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axes" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

topk :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "k" Int) ((:) ((:=) "ret_typ" String) ((:) ((:=) "is_ascend" Bool) '[]))))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

swapaxes :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim1" Int) ((:) ((:=) "dim2" Int) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sum_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

stack :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle Source #

split :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "squeeze_axis" Bool) '[]))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

sort :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "is_ascend" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sgd_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[])))) => String -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

sgd_mom_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "momentum" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[]))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

sample_uniform :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_poisson :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_normal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_multinomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "get_prob" Bool) ((:) ((:=) "dtype" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_generalized_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_gamma :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sample_exponential :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

rmspropalex_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "gamma1" Float) ((:) ((:=) "gamma2" Float) ((:) ((:=) "epsilon" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) ((:) ((:=) "clip_weights" Float) '[])))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

rmsprop_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "gamma1" Float) ((:) ((:=) "epsilon" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) ((:) ((:=) "clip_weights" Float) '[]))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

reshape :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "reverse" Bool) ((:) ((:=) "target_shape" String) ((:) ((:=) "keep_highest" Bool) '[]))))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

repeat :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

prod :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

pick :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

pad :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "constant_value" Double) '[])) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle Source #

one_hot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "on_value" Double) ((:) ((:=) "off_value" Double) ((:) ((:=) "dtype" String) '[])))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

nansum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

nanprod :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

mp_sgd_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[])))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

mp_sgd_mom_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "momentum" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[]))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

min_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

min :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

mean :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

max_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

max :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

log_softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linalg_trsm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linalg_trmm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linalg_syrk :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "alpha" Double) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linalg_gemm2 :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linalg_gemm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) ((:) ((:=) "alpha" Double) ((:) ((:=) "beta" Double) '[]))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

ftrl_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "lamda1" Float) ((:) ((:=) "beta" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[])))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

concat :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle Source #

broadcast_to :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

broadcast_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "size" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

broadcast_axes :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "size" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

batch_dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

argsort :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "is_ascend" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

argmin :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

argmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

adam_update :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "beta1" Float) ((:) ((:=) "beta2" Float) ((:) ((:=) "epsilon" Float) ((:) ((:=) "wd" Float) ((:) ((:=) "rescale_grad" Float) ((:) ((:=) "clip_gradient" Float) '[]))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> HMap kvs -> IO SymbolHandle Source #

_square_sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sparse_sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sparse_dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_uniform :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_poisson :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_normal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_multinomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "get_prob" Bool) ((:) ((:=) "dtype" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_generalized_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_gamma :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_sample_exponential :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_linalg_trsm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_linalg_trmm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_linalg_syrk :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "alpha" Double) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_linalg_gemm2 :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_linalg_gemm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) ((:) ((:=) "alpha" Double) ((:) ((:=) "beta" Double) '[]))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_cvimresize :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "interp" Int) '[])) => String -> SymbolHandle -> Int -> Int -> HMap kvs -> IO SymbolHandle Source #

_cvimread :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "flag" Int) ((:) ((:=) "to_rgb" Bool) '[]))) => String -> String -> HMap kvs -> IO SymbolHandle Source #

_cvimdecode :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "flag" Int) ((:) ((:=) "to_rgb" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_cvcopyMakeBorder :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "type" Int) ((:) ((:=) "value" Double) '[]))) => String -> SymbolHandle -> Int -> Int -> Int -> Int -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_crop_assign_scalar :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "scalar" Float) '[])) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle Source #

_contrib_quantize :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "out_type" String) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_contrib_ifft :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "compute_size" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_contrib_fft :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "compute_size" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_contrib_ctc_loss :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_data_lengths" Bool) ((:) ((:=) "use_label_lengths" Bool) ((:) ((:=) "blank_label" String) '[])))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

_contrib_count_sketch :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "processing_batch_size" Int) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

_contrib_Proposal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "rpn_pre_nms_top_n" Int) ((:) ((:=) "rpn_post_nms_top_n" Int) ((:) ((:=) "threshold" Float) ((:) ((:=) "rpn_min_size" Int) ((:) ((:=) "feature_stride" Int) ((:) ((:=) "output_score" Bool) ((:) ((:=) "iou_loss" Bool) '[])))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> [Float] -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_contrib_PSROIPooling :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "group_size" Int) '[])) => String -> SymbolHandle -> SymbolHandle -> Float -> Int -> Int -> HMap kvs -> IO SymbolHandle Source #

_contrib_MultiProposal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "rpn_pre_nms_top_n" Int) ((:) ((:=) "rpn_post_nms_top_n" Int) ((:) ((:=) "threshold" Float) ((:) ((:=) "rpn_min_size" Int) ((:) ((:=) "feature_stride" Int) ((:) ((:=) "output_score" Bool) ((:) ((:=) "iou_loss" Bool) '[])))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> [Float] -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_contrib_MultiBoxTarget :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "overlap_threshold" Float) ((:) ((:=) "ignore_label" Float) ((:) ((:=) "negative_mining_ratio" Float) ((:) ((:=) "negative_mining_thresh" Float) ((:) ((:=) "minimum_negative_samples" Int) '[])))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_contrib_MultiBoxPrior :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "clip" Bool) '[])) => String -> SymbolHandle -> [Float] -> [Float] -> [Float] -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_contrib_MultiBoxDetection :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "clip" Bool) ((:) ((:=) "threshold" Float) ((:) ((:=) "background_id" Int) ((:) ((:=) "nms_threshold" Float) ((:) ((:=) "force_suppress" Bool) ((:) ((:=) "nms_topk" Int) '[]))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> [Float] -> HMap kvs -> IO SymbolHandle Source #

_contrib_DeformablePSROIPooling :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "part_size" Int) ((:) ((:=) "sample_per_part" Int) ((:) ((:=) "trans_std" Float) ((:) ((:=) "no_trans" Bool) '[]))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Float -> Int -> Int -> Int -> HMap kvs -> IO SymbolHandle Source #

_contrib_DeformableConvolution :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "stride" String) ((:) ((:=) "dilate" String) ((:) ((:=) "pad" String) ((:) ((:=) "num_group" Int) ((:) ((:=) "num_deformable_group" Int) ((:) ((:=) "workspace" Int) ((:) ((:=) "no_bias" Bool) ((:) ((:=) "layout" String) '[]))))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> Int -> HMap kvs -> IO SymbolHandle Source #

_contrib_CTCLoss :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_data_lengths" Bool) ((:) ((:=) "use_label_lengths" Bool) ((:) ((:=) "blank_label" String) '[])))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

upsampling :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "num_filter" Int) ((:) ((:=) "multi_input_mode" String) ((:) ((:=) "workspace" Int) '[])))) => String -> [SymbolHandle] -> Int -> String -> Int -> HMap kvs -> IO SymbolHandle Source #

swapaxis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim1" Int) ((:) ((:=) "dim2" Int) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

spatialtransformer :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "target_shape" String) '[])) => String -> SymbolHandle -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle Source #

softmaxoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) ((:) ((:=) "ignore_label" Float) ((:) ((:=) "multi_output" Bool) ((:) ((:=) "use_ignore" Bool) ((:) ((:=) "preserve_shape" Bool) ((:) ((:=) "normalization" String) ((:) ((:=) "out_grad" Bool) ((:) ((:=) "smooth_alpha" Float) '[]))))))))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

softmaxactivation :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "mode" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

slicechannel :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "squeeze_axis" Bool) '[]))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

sequencereverse :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sequencemask :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) ((:) ((:=) "value" Float) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

sequencelast :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

svmoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "margin" Float) ((:) ((:=) "regularization_coefficient" Float) ((:) ((:=) "use_linear" Bool) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

rnn :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "bidirectional" Bool) ((:) ((:=) "p" Float) ((:) ((:=) "state_outputs" Bool) '[])))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Int -> Int -> String -> HMap kvs -> IO SymbolHandle Source #

pooling_v1 :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "global_pool" Bool) ((:) ((:=) "pooling_convention" String) ((:) ((:=) "stride" String) ((:) ((:=) "pad" String) '[]))))) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle Source #

pooling :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "global_pool" Bool) ((:) ((:=) "cudnn_off" Bool) ((:) ((:=) "pooling_convention" String) ((:) ((:=) "stride" String) ((:) ((:=) "pad" String) '[])))))) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle Source #

makeloss :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) ((:) ((:=) "valid_thresh" Float) ((:) ((:=) "normalization" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

maeregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

logisticregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

linearregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

leakyrelu :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "act_type" String) ((:) ((:=) "slope" Float) ((:) ((:=) "lower_bound" Float) ((:) ((:=) "upper_bound" Float) '[]))))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

lrn :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "alpha" Float) ((:) ((:=) "beta" Float) ((:) ((:=) "knorm" Float) '[])))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

l2normalization :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Float) ((:) ((:=) "mode" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

instancenorm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

identityattachklsparsereg :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "sparseness_target" Float) ((:) ((:=) "penalty" Float) ((:) ((:=) "momentum" Float) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

gridgenerator :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "target_shape" String) '[])) => String -> SymbolHandle -> String -> HMap kvs -> IO SymbolHandle Source #

fullyconnected :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "no_bias" Bool) ((:) ((:=) "flatten" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle Source #

embedding :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dtype" String) '[])) => String -> SymbolHandle -> SymbolHandle -> Int -> Int -> HMap kvs -> IO SymbolHandle Source #

dropout :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "p" Float) ((:) ((:=) "mode" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

deconvolution :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "stride" String) ((:) ((:=) "dilate" String) ((:) ((:=) "pad" String) ((:) ((:=) "adj" String) ((:) ((:=) "target_shape" String) ((:) ((:=) "num_group" Int) ((:) ((:=) "workspace" Int) ((:) ((:=) "no_bias" Bool) ((:) ((:=) "cudnn_tune" String) ((:) ((:=) "cudnn_off" Bool) ((:) ((:=) "layout" String) '[])))))))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> Int -> HMap kvs -> IO SymbolHandle Source #

crop :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "offset" String) ((:) ((:=) "h_w" String) ((:) ((:=) "center_crop" Bool) '[])))) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle Source #

correlation :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "kernel_size" Int) ((:) ((:=) "max_displacement" Int) ((:) ((:=) "stride1" Int) ((:) ((:=) "stride2" Int) ((:) ((:=) "pad_size" Int) ((:) ((:=) "is_multiply" Bool) '[]))))))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

convolution_v1 :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "stride" String) ((:) ((:=) "dilate" String) ((:) ((:=) "pad" String) ((:) ((:=) "num_group" Int) ((:) ((:=) "workspace" Int) ((:) ((:=) "no_bias" Bool) ((:) ((:=) "cudnn_tune" String) ((:) ((:=) "cudnn_off" Bool) ((:) ((:=) "layout" String) '[])))))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> Int -> HMap kvs -> IO SymbolHandle Source #

convolution :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "stride" String) ((:) ((:=) "dilate" String) ((:) ((:=) "pad" String) ((:) ((:=) "num_group" Int) ((:) ((:=) "workspace" Int) ((:) ((:=) "no_bias" Bool) ((:) ((:=) "cudnn_tune" String) ((:) ((:=) "cudnn_off" Bool) ((:) ((:=) "layout" String) '[])))))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> Int -> HMap kvs -> IO SymbolHandle Source #

batchnorm_v1 :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Float) ((:) ((:=) "momentum" Float) ((:) ((:=) "fix_gamma" Bool) ((:) ((:=) "use_global_stats" Bool) ((:) ((:=) "output_mean_var" Bool) '[])))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #

batchnorm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Double) ((:) ((:=) "momentum" Float) ((:) ((:=) "fix_gamma" Bool) ((:) ((:=) "use_global_stats" Bool) ((:) ((:=) "output_mean_var" Bool) ((:) ((:=) "axis" Int) ((:) ((:=) "cudnn_off" Bool) '[])))))))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #