package tensorflow

  1. Overview
  2. Docs
type t
type operation
type operation_description
type output
val create : unit -> t
val new_operation : t -> op_name:string -> name:string -> operation_description
val add_control_input : operation_description -> operation -> unit
val add_input : operation_description -> operation -> index:int -> unit
val add_inputs : operation_description -> (operation * int) list -> unit
val create_output : operation -> index:int -> output
val output_op_and_index : t -> output -> operation * int
val set_attr_int : operation_description -> attr_name:string -> int -> unit
val set_attr_int_list : operation_description -> attr_name:string -> int list -> unit
val set_attr_float : operation_description -> attr_name:string -> float -> unit
val set_attr_float_list : operation_description -> attr_name:string -> float list -> unit
val set_attr_bool : operation_description -> attr_name:string -> bool -> unit
val set_attr_bool_list : operation_description -> attr_name:string -> bool list -> unit
val set_attr_string : operation_description -> attr_name:string -> string -> unit
val set_attr_type : operation_description -> attr_name:string -> data_type -> unit
val set_attr_type_list : operation_description -> attr_name:string -> data_type list -> unit
val set_attr_tensor : operation_description -> attr_name:string -> Tensor.p -> unit Status.result
val set_attr_tensor_string : operation_description -> attr_name:string -> shape:int list -> string list -> unit Status.result
val set_attr_tensors : operation_description -> attr_name:string -> Tensor.p list -> unit Status.result
val set_attr_shape : operation_description -> attr_name:string -> int list -> unit
val import : t -> string -> unit Status.result
val find_operation : t -> string -> operation option
val shape : t -> output -> int list Status.result
val add_gradients : t -> output list -> xs:output list -> output list Status.result
val graph : operation -> t
OCaml

Innovation. Community. Security.