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