Copyright | (c) 2016 Tao He |
---|---|
License | MIT |
Maintainer | sighingnow@gmail.com |
Safe Haskell | None |
Language | Haskell2010 |
Functions about Symbol that generated by template haskell.
- zeros_like :: String -> SymbolHandle -> IO SymbolHandle
- trunc :: String -> SymbolHandle -> IO SymbolHandle
- transpose :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axes" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- 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
- tile :: String -> SymbolHandle -> String -> IO SymbolHandle
- tanh :: String -> SymbolHandle -> IO SymbolHandle
- tan :: String -> SymbolHandle -> IO SymbolHandle
- swapaxes :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim1" Int) ((:) ((:=) "dim2" Int) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sum_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- stop_gradient :: String -> SymbolHandle -> IO SymbolHandle
- stack :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle
- square :: String -> SymbolHandle -> IO SymbolHandle
- sqrt :: String -> SymbolHandle -> IO SymbolHandle
- split :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "squeeze_axis" Bool) '[]))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- sort :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "is_ascend" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- softmax_cross_entropy :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- smooth_l1 :: String -> SymbolHandle -> Float -> IO SymbolHandle
- slice_axis :: String -> SymbolHandle -> Int -> Int -> Int -> IO SymbolHandle
- slice :: String -> SymbolHandle -> String -> String -> IO SymbolHandle
- sinh :: String -> SymbolHandle -> IO SymbolHandle
- sin :: String -> SymbolHandle -> IO SymbolHandle
- sign :: String -> SymbolHandle -> IO SymbolHandle
- sigmoid :: String -> SymbolHandle -> IO SymbolHandle
- 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
- 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
- scatter_nd :: String -> SymbolHandle -> SymbolHandle -> String -> IO SymbolHandle
- sample_uniform :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_poisson :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_normal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_multinomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "get_prob" Bool) ((:) ((:=) "dtype" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_generalized_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_gamma :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sample_exponential :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- rsqrt :: String -> SymbolHandle -> IO SymbolHandle
- round :: String -> SymbolHandle -> IO SymbolHandle
- 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
- 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
- rint :: String -> SymbolHandle -> IO SymbolHandle
- reverse :: String -> SymbolHandle -> String -> IO SymbolHandle
- reshape_like :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- 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
- repeat :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- relu :: String -> SymbolHandle -> IO SymbolHandle
- reciprocal :: String -> SymbolHandle -> IO SymbolHandle
- rcbrt :: String -> SymbolHandle -> IO SymbolHandle
- radians :: String -> SymbolHandle -> IO SymbolHandle
- prod :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- pick :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- pad :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "constant_value" Double) '[])) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle
- ones_like :: String -> SymbolHandle -> IO SymbolHandle
- 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
- norm :: String -> SymbolHandle -> IO SymbolHandle
- negative :: String -> SymbolHandle -> IO SymbolHandle
- nansum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- nanprod :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- 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
- 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
- min_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- min :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- mean :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- max_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- max :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- make_loss :: String -> SymbolHandle -> IO SymbolHandle
- log_softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- log2 :: String -> SymbolHandle -> IO SymbolHandle
- log1p :: String -> SymbolHandle -> IO SymbolHandle
- log10 :: String -> SymbolHandle -> IO SymbolHandle
- log :: String -> SymbolHandle -> IO SymbolHandle
- linalg_trsm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- linalg_trmm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- linalg_syrk :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "alpha" Double) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- linalg_sumlogdiag :: String -> SymbolHandle -> IO SymbolHandle
- linalg_potri :: String -> SymbolHandle -> IO SymbolHandle
- linalg_potrf :: String -> SymbolHandle -> IO SymbolHandle
- 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
- 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
- linalg_gelqf :: String -> SymbolHandle -> IO SymbolHandle
- identity :: String -> SymbolHandle -> IO SymbolHandle
- gather_nd :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- gammaln :: String -> SymbolHandle -> IO SymbolHandle
- gamma :: String -> SymbolHandle -> IO SymbolHandle
- 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
- floor :: String -> SymbolHandle -> IO SymbolHandle
- flip :: String -> SymbolHandle -> String -> IO SymbolHandle
- flatten :: String -> SymbolHandle -> IO SymbolHandle
- fix :: String -> SymbolHandle -> IO SymbolHandle
- fill_element_0index :: String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- expm1 :: String -> SymbolHandle -> IO SymbolHandle
- expand_dims :: String -> SymbolHandle -> Int -> IO SymbolHandle
- exp :: String -> SymbolHandle -> IO SymbolHandle
- elemwise_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- elemwise_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- elemwise_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- elemwise_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- degrees :: String -> SymbolHandle -> IO SymbolHandle
- cosh :: String -> SymbolHandle -> IO SymbolHandle
- cos :: String -> SymbolHandle -> IO SymbolHandle
- concat :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle
- clip :: String -> SymbolHandle -> Float -> Float -> IO SymbolHandle
- choose_element_0index :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- ceil :: String -> SymbolHandle -> IO SymbolHandle
- cbrt :: String -> SymbolHandle -> IO SymbolHandle
- cast_storage :: String -> SymbolHandle -> String -> IO SymbolHandle
- broadcast_to :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- broadcast_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_not_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_maximum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_lesser_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_greater_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- broadcast_axis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "size" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- broadcast_axes :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "size" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- broadcast_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- batch_take :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- batch_dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- argsort :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "is_ascend" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- argmin :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- argmax_channel :: String -> SymbolHandle -> IO SymbolHandle
- argmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- arctanh :: String -> SymbolHandle -> IO SymbolHandle
- arctan :: String -> SymbolHandle -> IO SymbolHandle
- arcsinh :: String -> SymbolHandle -> IO SymbolHandle
- arcsin :: String -> SymbolHandle -> IO SymbolHandle
- arccosh :: String -> SymbolHandle -> IO SymbolHandle
- arccos :: String -> SymbolHandle -> IO SymbolHandle
- add_n :: String -> [SymbolHandle] -> IO SymbolHandle
- 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
- abs :: String -> SymbolHandle -> IO SymbolHandle
- _sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _square_sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sparse_zeros_like :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_trunc :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_tanh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_tan :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_sum :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" String) ((:) ((:=) "keepdims" Bool) ((:) ((:=) "exclude" Bool) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sparse_stop_gradient :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_square :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_sqrt :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_slice :: String -> SymbolHandle -> String -> String -> IO SymbolHandle
- _sparse_sinh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_sin :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_sign :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_sigmoid :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_rsqrt :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_round :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_rint :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_retain :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sparse_relu :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_radians :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_negative :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_make_loss :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_log2 :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_log1p :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_log10 :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_log :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_gammaln :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_gamma :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_floor :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_fix :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_expm1 :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_exp :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_elemwise_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sparse_elemwise_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sparse_elemwise_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sparse_elemwise_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sparse_dot :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose_a" Bool) ((:) ((:=) "transpose_b" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sparse_degrees :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_cosh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_cos :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_ceil :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_cast_storage :: String -> SymbolHandle -> String -> IO SymbolHandle
- _sparse_arctanh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_arctan :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_arcsinh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_arcsin :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_arccosh :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_arccos :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_add_n :: String -> [SymbolHandle] -> IO SymbolHandle
- _sparse_abs :: String -> SymbolHandle -> IO SymbolHandle
- _sparse_ElementWiseSum :: String -> [SymbolHandle] -> IO SymbolHandle
- _slice_assign :: String -> SymbolHandle -> SymbolHandle -> String -> String -> IO SymbolHandle
- _set_value :: String -> Float -> IO SymbolHandle
- _scatter_plus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _scatter_minus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _scatter_elemwise_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _sample_uniform :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_poisson :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_normal :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_multinomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "get_prob" Bool) ((:) ((:=) "dtype" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_generalized_negative_binomial :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_gamma :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _sample_exponential :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) ((:) ((:=) "dtype" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _rpower_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _rmod_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _rminus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _rdiv_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _power_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _plus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _onehot_encode :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _not_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _not_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _mul_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _mod_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _minus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _minimum_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _maximum_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _maximum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _linalg_trsm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _linalg_trmm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "rightside" Bool) ((:) ((:=) "alpha" Double) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _linalg_syrk :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "transpose" Bool) ((:) ((:=) "alpha" Double) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _linalg_syevd :: String -> SymbolHandle -> IO SymbolHandle
- _linalg_sumlogdiag :: String -> SymbolHandle -> IO SymbolHandle
- _linalg_potri :: String -> SymbolHandle -> IO SymbolHandle
- _linalg_potrf :: String -> SymbolHandle -> IO SymbolHandle
- _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
- _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
- _linalg_gelqf :: String -> SymbolHandle -> IO SymbolHandle
- _lesser_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _lesser_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _lesser_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _imdecode :: String -> SymbolHandle -> Int -> Int -> Int -> Int -> Int -> Int -> Int -> IO SymbolHandle
- _identity_with_attr_like_rhs :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _hypot_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _greater_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _greater_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _greater_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _grad_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _div_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _cvimresize :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "interp" Int) '[])) => String -> SymbolHandle -> Int -> Int -> HMap kvs -> IO SymbolHandle
- _cvimread :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "flag" Int) ((:) ((:=) "to_rgb" Bool) '[]))) => String -> String -> HMap kvs -> IO SymbolHandle
- _cvimdecode :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "flag" Int) ((:) ((:=) "to_rgb" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _cvcopyMakeBorder :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "type" Int) ((:) ((:=) "value" Double) '[]))) => String -> SymbolHandle -> Int -> Int -> Int -> Int -> [Float] -> HMap kvs -> IO SymbolHandle
- _crop_assign_scalar :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "scalar" Float) '[])) => String -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle
- _crop_assign :: String -> SymbolHandle -> SymbolHandle -> String -> String -> IO SymbolHandle
- _copyto :: String -> SymbolHandle -> IO SymbolHandle
- _copy :: String -> SymbolHandle -> IO SymbolHandle
- _contrib_quantize :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "out_type" String) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _contrib_ifft :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "compute_size" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _contrib_fft :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "compute_size" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- _contrib_dequantize :: String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> IO SymbolHandle
- _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
- _contrib_count_sketch :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "processing_batch_size" Int) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- _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
- _contrib_PSROIPooling :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "group_size" Int) '[])) => String -> SymbolHandle -> SymbolHandle -> Float -> Int -> Int -> HMap kvs -> IO SymbolHandle
- _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
- _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
- _contrib_MultiBoxPrior :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "clip" Bool) '[])) => String -> SymbolHandle -> [Float] -> [Float] -> [Float] -> [Float] -> HMap kvs -> IO SymbolHandle
- _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
- _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
- _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
- _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
- _backward_tanh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_tan :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_square :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_sqrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_softmax :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_smooth_l1 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_sinh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_sin :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_sign :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_sigmoid :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_rsqrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_rpower_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_rmod_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_relu :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_reciprocal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_rdiv_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_rcbrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_radians :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_power_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_mul_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_mod_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_minimum_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_maximum_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_log_softmax :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_log2 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_log1p :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_log10 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_log :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_hypot_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle
- _backward_gammaln :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_gamma :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_expm1 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_degrees :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_cosh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_cos :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_cbrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arctanh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arctan :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arcsinh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arcsin :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arccosh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_arccos :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _backward_abs :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _RPowerScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _RModScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _RMinusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _RDivScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _PowerScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _PlusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _Not_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _NotEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _MulScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _ModScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _MinusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _MinimumScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _MaximumScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Maximum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _Lesser_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _LesserScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _LesserEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _HypotScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _Greater_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _GreaterScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _GreaterEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _EqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- _DivScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle
- _Div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- 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
- swapaxis :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim1" Int) ((:) ((:=) "dim2" Int) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- spatialtransformer :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "target_shape" String) '[])) => String -> SymbolHandle -> SymbolHandle -> String -> String -> HMap kvs -> IO SymbolHandle
- 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
- softmaxactivation :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "mode" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- slicechannel :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "squeeze_axis" Bool) '[]))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- sequencereverse :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sequencemask :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) ((:) ((:=) "value" Float) '[]))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- sequencelast :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "use_sequence_length" Bool) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- svmoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "margin" Float) ((:) ((:=) "regularization_coefficient" Float) ((:) ((:=) "use_linear" Bool) '[])))) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- roipooling :: String -> SymbolHandle -> SymbolHandle -> String -> Float -> IO SymbolHandle
- 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
- 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
- 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
- makeloss :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) ((:) ((:=) "valid_thresh" Float) ((:) ((:=) "normalization" String) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- maeregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- logisticregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- linearregressionoutput :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "grad_scale" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- 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
- lrn :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "alpha" Float) ((:) ((:=) "beta" Float) ((:) ((:=) "knorm" Float) '[])))) => String -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- l2normalization :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Float) ((:) ((:=) "mode" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- instancenorm :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "eps" Float) '[])) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- identityattachklsparsereg :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "sparseness_target" Float) ((:) ((:=) "penalty" Float) ((:) ((:=) "momentum" Float) '[])))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- gridgenerator :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "target_shape" String) '[])) => String -> SymbolHandle -> String -> HMap kvs -> IO SymbolHandle
- fullyconnected :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "no_bias" Bool) ((:) ((:=) "flatten" Bool) '[]))) => String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> Int -> HMap kvs -> IO SymbolHandle
- embedding :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dtype" String) '[])) => String -> SymbolHandle -> SymbolHandle -> Int -> Int -> HMap kvs -> IO SymbolHandle
- elementwisesum :: String -> [SymbolHandle] -> IO SymbolHandle
- dropout :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "p" Float) ((:) ((:=) "mode" String) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle
- 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
- custom :: String -> [SymbolHandle] -> String -> IO SymbolHandle
- crop :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "offset" String) ((:) ((:=) "h_w" String) ((:) ((:=) "center_crop" Bool) '[])))) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle
- 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
- 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
- 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
- cast :: String -> SymbolHandle -> String -> IO SymbolHandle
- blockgrad :: String -> SymbolHandle -> IO SymbolHandle
- bilinearsampler :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle
- 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
- 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
- activation :: String -> SymbolHandle -> String -> IO SymbolHandle
Documentation
zeros_like :: String -> SymbolHandle -> IO SymbolHandle Source #
Register symbol operators.
trunc :: String -> SymbolHandle -> IO SymbolHandle Source #
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 #
tile :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
tanh :: String -> SymbolHandle -> IO SymbolHandle Source #
tan :: String -> SymbolHandle -> 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 #
stop_gradient :: String -> SymbolHandle -> IO SymbolHandle Source #
stack :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle Source #
square :: String -> SymbolHandle -> IO SymbolHandle Source #
sqrt :: String -> SymbolHandle -> 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_cross_entropy :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #
smooth_l1 :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
slice_axis :: String -> SymbolHandle -> Int -> Int -> Int -> IO SymbolHandle Source #
slice :: String -> SymbolHandle -> String -> String -> IO SymbolHandle Source #
sinh :: String -> SymbolHandle -> IO SymbolHandle Source #
sin :: String -> SymbolHandle -> IO SymbolHandle Source #
sign :: String -> SymbolHandle -> IO SymbolHandle Source #
sigmoid :: String -> SymbolHandle -> 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 #
scatter_nd :: String -> SymbolHandle -> SymbolHandle -> String -> 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 #
rsqrt :: String -> SymbolHandle -> IO SymbolHandle Source #
round :: String -> SymbolHandle -> 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 #
rint :: String -> SymbolHandle -> IO SymbolHandle Source #
reverse :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
reshape_like :: String -> SymbolHandle -> SymbolHandle -> 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 #
relu :: String -> SymbolHandle -> IO SymbolHandle Source #
reciprocal :: String -> SymbolHandle -> IO SymbolHandle Source #
rcbrt :: String -> SymbolHandle -> IO SymbolHandle Source #
radians :: String -> SymbolHandle -> 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 #
ones_like :: String -> SymbolHandle -> 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 #
norm :: String -> SymbolHandle -> IO SymbolHandle Source #
negative :: String -> SymbolHandle -> 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 #
make_loss :: String -> SymbolHandle -> IO SymbolHandle Source #
log_softmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #
log2 :: String -> SymbolHandle -> IO SymbolHandle Source #
log1p :: String -> SymbolHandle -> IO SymbolHandle Source #
log10 :: String -> SymbolHandle -> IO SymbolHandle Source #
log :: String -> SymbolHandle -> 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_sumlogdiag :: String -> SymbolHandle -> IO SymbolHandle Source #
linalg_potri :: String -> SymbolHandle -> IO SymbolHandle Source #
linalg_potrf :: String -> SymbolHandle -> 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 #
linalg_gelqf :: String -> SymbolHandle -> IO SymbolHandle Source #
identity :: String -> SymbolHandle -> IO SymbolHandle Source #
gather_nd :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
gammaln :: String -> SymbolHandle -> IO SymbolHandle Source #
gamma :: String -> SymbolHandle -> 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 #
floor :: String -> SymbolHandle -> IO SymbolHandle Source #
flip :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
flatten :: String -> SymbolHandle -> IO SymbolHandle Source #
fix :: String -> SymbolHandle -> IO SymbolHandle Source #
fill_element_0index :: String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
expm1 :: String -> SymbolHandle -> IO SymbolHandle Source #
expand_dims :: String -> SymbolHandle -> Int -> IO SymbolHandle Source #
exp :: String -> SymbolHandle -> IO SymbolHandle Source #
elemwise_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
elemwise_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
elemwise_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
elemwise_add :: String -> SymbolHandle -> SymbolHandle -> 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 #
degrees :: String -> SymbolHandle -> IO SymbolHandle Source #
cosh :: String -> SymbolHandle -> IO SymbolHandle Source #
cos :: String -> SymbolHandle -> IO SymbolHandle Source #
concat :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "dim" Int) '[])) => String -> [SymbolHandle] -> Int -> HMap kvs -> IO SymbolHandle Source #
clip :: String -> SymbolHandle -> Float -> Float -> IO SymbolHandle Source #
choose_element_0index :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
ceil :: String -> SymbolHandle -> IO SymbolHandle Source #
cbrt :: String -> SymbolHandle -> IO SymbolHandle Source #
cast_storage :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
broadcast_to :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "shape" String) '[])) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #
broadcast_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_not_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_maximum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_lesser_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_greater_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
broadcast_div :: String -> SymbolHandle -> SymbolHandle -> 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 #
broadcast_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
batch_take :: String -> SymbolHandle -> SymbolHandle -> 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_channel :: String -> SymbolHandle -> IO SymbolHandle Source #
argmax :: forall (kvs :: [KV Type]). (ShowKV kvs, MatchKVList kvs ((:) ((:=) "axis" Int) ((:) ((:=) "keepdims" Bool) '[]))) => String -> SymbolHandle -> HMap kvs -> IO SymbolHandle Source #
arctanh :: String -> SymbolHandle -> IO SymbolHandle Source #
arctan :: String -> SymbolHandle -> IO SymbolHandle Source #
arcsinh :: String -> SymbolHandle -> IO SymbolHandle Source #
arcsin :: String -> SymbolHandle -> IO SymbolHandle Source #
arccosh :: String -> SymbolHandle -> IO SymbolHandle Source #
arccos :: String -> SymbolHandle -> IO SymbolHandle Source #
add_n :: String -> [SymbolHandle] -> 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 #
abs :: String -> SymbolHandle -> IO SymbolHandle Source #
_sub :: String -> SymbolHandle -> SymbolHandle -> 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_zeros_like :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_trunc :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_tanh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_tan :: String -> SymbolHandle -> 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_stop_gradient :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_square :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_sqrt :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_slice :: String -> SymbolHandle -> String -> String -> IO SymbolHandle Source #
_sparse_sinh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_sin :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_sign :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_sigmoid :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_rsqrt :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_round :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_rint :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_retain :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_sparse_relu :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_radians :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_negative :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_make_loss :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_log2 :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_log1p :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_log10 :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_log :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_gammaln :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_gamma :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_floor :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_fix :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_expm1 :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_exp :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_elemwise_sub :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_sparse_elemwise_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_sparse_elemwise_div :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_sparse_elemwise_add :: String -> SymbolHandle -> SymbolHandle -> 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 #
_sparse_degrees :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_cosh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_cos :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_ceil :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_cast_storage :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
_sparse_arctanh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_arctan :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_arcsinh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_arcsin :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_arccosh :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_arccos :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_add_n :: String -> [SymbolHandle] -> IO SymbolHandle Source #
_sparse_abs :: String -> SymbolHandle -> IO SymbolHandle Source #
_sparse_ElementWiseSum :: String -> [SymbolHandle] -> IO SymbolHandle Source #
_slice_assign :: String -> SymbolHandle -> SymbolHandle -> String -> String -> IO SymbolHandle Source #
_set_value :: String -> Float -> IO SymbolHandle Source #
_scatter_plus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_scatter_minus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_scatter_elemwise_div :: String -> SymbolHandle -> SymbolHandle -> 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 #
_rpower_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_rmod_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_rminus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_rdiv_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_power_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_plus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_onehot_encode :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_not_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_not_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_mul_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_mod_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_minus_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_minimum_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_maximum_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_maximum :: String -> SymbolHandle -> SymbolHandle -> 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_syevd :: String -> SymbolHandle -> IO SymbolHandle Source #
_linalg_sumlogdiag :: String -> SymbolHandle -> IO SymbolHandle Source #
_linalg_potri :: String -> SymbolHandle -> IO SymbolHandle Source #
_linalg_potrf :: String -> SymbolHandle -> 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 #
_linalg_gelqf :: String -> SymbolHandle -> IO SymbolHandle Source #
_lesser_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_lesser_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_lesser_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_imdecode :: String -> SymbolHandle -> Int -> Int -> Int -> Int -> Int -> Int -> Int -> IO SymbolHandle Source #
_hypot_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_greater_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_greater_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_greater_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_grad_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_equal_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_div_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_div :: String -> SymbolHandle -> SymbolHandle -> 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 #
_crop_assign :: String -> SymbolHandle -> SymbolHandle -> String -> String -> IO SymbolHandle Source #
_copyto :: String -> SymbolHandle -> IO SymbolHandle Source #
_copy :: String -> SymbolHandle -> 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_dequantize :: String -> SymbolHandle -> SymbolHandle -> SymbolHandle -> String -> 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 #
_backward_tanh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_tan :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_square :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_sqrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_softmax :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_smooth_l1 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_sinh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_sin :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_sign :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_sigmoid :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_rsqrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_rpower_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_rmod_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_relu :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_reciprocal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_rdiv_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_rcbrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_radians :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_power_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_mul_scalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_mod_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_minimum_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_maximum_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_log_softmax :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_log2 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_log1p :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_log10 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_log :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_hypot_scalar :: String -> SymbolHandle -> SymbolHandle -> Float -> IO SymbolHandle Source #
_backward_gammaln :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_gamma :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_expm1 :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_degrees :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_cosh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_cos :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_cbrt :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arctanh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arctan :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arcsinh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arcsin :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arccosh :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_arccos :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_backward_abs :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_add :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_RPowerScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_RModScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_RMinusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_RDivScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_PowerScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Power :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_PlusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Plus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_Not_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_NotEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_MulScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Mul :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_ModScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Mod :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_MinusScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Minus :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_MinimumScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Minimum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_MaximumScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Maximum :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_Lesser_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_LesserScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_LesserEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Lesser :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_HypotScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Hypot :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_Greater_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_GreaterScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_GreaterEqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Greater :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_EqualScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Equal :: String -> SymbolHandle -> SymbolHandle -> IO SymbolHandle Source #
_DivScalar :: String -> SymbolHandle -> Float -> IO SymbolHandle Source #
_Div :: String -> SymbolHandle -> SymbolHandle -> 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 #
roipooling :: String -> SymbolHandle -> SymbolHandle -> String -> Float -> 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 #
elementwisesum :: String -> [SymbolHandle] -> 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 #
custom :: String -> [SymbolHandle] -> String -> 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 #
cast :: String -> SymbolHandle -> String -> IO SymbolHandle Source #
blockgrad :: String -> SymbolHandle -> IO SymbolHandle Source #
bilinearsampler :: String -> SymbolHandle -> SymbolHandle -> 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 #
activation :: String -> SymbolHandle -> String -> IO SymbolHandle Source #