package tensorflow

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