package tensorflow

  1. Overview
  2. Docs
module Type : sig ... end
module Tensor : sig ... end
module Dim : sig ... end
module Attr_list : sig ... end
type attr =
  1. | String of Base.string
  2. | Int of Base.int
  3. | Float of Base.float
  4. | Bool of Base.bool
  5. | Type of Type.p
  6. | List of Attr_list.t
  7. | Tensor_float of Base.float Tensor.t
  8. | Tensor_int of Base.int Tensor.t
  9. | Tensor_string of Base.string Tensor.t
  10. | Shape of Dim.t Base.list
type 'a t
type p =
  1. | P : _ t -> p
type input = [
  1. | `single of p
  2. | `multi of p Base.list
]
val create : name:Name.t -> op_name:Op_name.t -> output_type:'a Type.t -> inputs:input Base.list -> control_inputs:p Base.list -> attributes:(Base.string * attr) Base.list -> output_idx:Base.int Base.option -> 'a t
val name : _ t -> Name.t
val op_name : _ t -> Op_name.t
val output_type : 'a t -> 'a Type.t
val inputs : _ t -> input Base.list
val flat_inputs : _ t -> p Base.list
val control_inputs : _ t -> p Base.list
val attributes : _ t -> (Base.string * attr) Base.list
val output_idx : _ t -> Base.int Base.option
val unique_name : _ t -> Base.string
val packed_name : p -> Name.t
val packed_op_name : p -> Op_name.t
val packed_inputs : p -> input Base.list
val packed_flat_inputs : p -> p Base.list
val packed_is_real : p -> Base.bool
val packed_id : p -> Id.t
val packed_output_idx : p -> Base.int Base.option
val get_attr_bool : _ t -> Base.string -> Base.bool Base.option
val get_attr_string : _ t -> Base.string -> Base.string Base.option
val get_attr_int : _ t -> Base.string -> Base.int Base.option
val get_attr_int_list : _ t -> Base.string -> Base.int Base.list Base.option
val get_shape : _ t -> Dim.t Base.list Base.option
val set_output_idx : 'a t -> Base.int Base.option -> 'a t
val set_output_idx_and_output_type : 'b t -> Base.int Base.option -> type_:'a Type.t -> 'a t
val id : _ t -> Id.t
val extract : p -> 'a Type.t -> 'a t Base.option
val extract_exn : p -> 'a Type.t -> 'a t
module Weak_table : sig ... end