Module Nx_nativeSource

Sourcetype ('a, 'b) buffer = ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t
Sourcetype context
Sourcetype ('a, 'b) t = {
  1. context : context;
  2. dtype : ('a, 'b) Nx_core.Dtype.t;
  3. buffer : ('a, 'b) buffer;
  4. view : Nx_core.View.t;
}
Sourceval view : ('a, 'b) t -> Nx_core.View.t
Sourceval dtype : ('a, 'b) t -> ('a, 'b) Nx_core.Dtype.t
Sourceval data : ('a, 'b) t -> ('a, 'b) buffer
Sourceval context : ('a, 'b) t -> context
Sourceval with_view : ('a, 'b) t -> Nx_core.View.t -> ('a, 'b) t
Sourceval create_context : unit -> Nx_native__.Internal.context
Sourceval op_buffer : context -> ('a, 'b) Nx_core.Dtype.t -> int -> ('a, 'b) t
Sourceval op_const_scalar : context -> 'a -> ('a, 'b) Nx_core.Dtype.t -> ('a, 'b) t
Sourceval op_const_array : context -> ('a, 'b, Bigarray.c_layout) Bigarray.Array1.t -> ('a, 'b) t
Sourceval op_add : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_mul : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_idiv : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_fdiv : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_max : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_mod : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_pow : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_cmplt : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> (int, Nx_core.Dtype.uint8_elt) t
Sourceval op_cmpne : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> (int, Nx_core.Dtype.uint8_elt) t
Sourceval op_xor : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_or : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_and : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval op_neg : ('a, 'b) t -> ('a, 'b) t
Sourceval op_log2 : ('a, 'b) t -> ('a, 'b) t
Sourceval op_exp2 : ('a, 'b) t -> ('a, 'b) t
Sourceval op_sin : ('a, 'b) t -> ('a, 'b) t
Sourceval op_sqrt : ('a, 'b) t -> ('a, 'b) t
Sourceval op_recip : ('a, 'b) t -> ('a, 'b) t
Sourceval op_where : (int, Nx_core.Dtype.uint8_elt) t -> ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) t
Sourceval fill_buffer_with_identity : ('a, 'b, 'c) Bigarray.Array1.t -> int -> 'a -> unit
Sourceval op_reduce_sum : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reduce_max : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reduce_prod : axes:int array -> keepdims:bool -> ('a, 'b) t -> ('a, 'b) t
Sourceval op_reshape : ('a, 'b) t -> int array -> ('a, 'b) t
Sourceval op_expand : ('a, 'b) t -> int array -> ('a, 'b) t
Sourceval op_permute : ('a, 'b) t -> int array -> ('a, 'b) t
Sourceval op_pad : ('a, 'b) t -> (int * int) array -> 'a -> ('a, 'b) t
Sourceval op_shrink : ('a, 'b) t -> (int * int) array -> ('a, 'b) t
Sourceval op_flip : ('a, 'b) t -> bool array -> ('a, 'b) t
Sourceval op_cat : ('a, 'b) Nx_native__.Internal.t list -> int -> ('a, 'b) t
Sourceval op_cast : ('a, 'b) t -> ('c, 'd) Nx_core.Dtype.t -> ('c, 'd) t
Sourceval op_contiguous : ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) Nx_native__.Internal.t
Sourceval op_copy : ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) Nx_native__.Internal.t
Sourceval op_assign : ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) Nx_native__.Internal.t -> unit
Sourceval op_threefry : (int32, Nx_core.Dtype.int32_elt) t -> (int32, Nx_core.Dtype.int32_elt) Nx_native__.Internal.t -> (int32, Nx_core.Dtype.int32_elt) t
Sourceval op_gather : ('a, 'b) t -> (int32, 'c) Nx_native__.Internal.t -> int -> ('a, 'b) t
Sourceval op_scatter : ?mode:[< `Add | `Set Set ] -> ?unique_indices:bool -> ('a, 'b) Nx_native__.Internal.t -> (int32, Nx_core.Dtype.int32_elt) Nx_native__.Internal.t -> ('a, 'b) Nx_native__.Internal.t -> int -> ('a, 'b) Nx_native__.Internal.t
Sourceval op_unfold : ('a, 'b) t -> kernel_size:int array -> stride:int array -> dilation:int array -> padding:(int * int) array -> ('a, 'b) Nx_native__.Internal.t
Sourceval op_fold : ('a, 'b) t -> output_size:int array -> kernel_size:int array -> stride:int array -> dilation:int array -> padding:(int * int) array -> ('a, 'b) Nx_native__.Internal.t
Sourceval op_matmul : ('a, 'b) t -> ('a, 'b) Nx_native__.Internal.t -> ('a, 'b) Nx_native__.Internal.t