tensorflow 0.0.11 (latest)

module Session : sig ... end

module Op_name : Base.Identifiable.S

module Name : Base.Identifiable.S

module Id : Base.Identifiable.S

module Type = Tensorflow_core.Operation.Type

module Tensor = Tensorflow_core.Operation.Tensor_attr

module Dim = Tensorflow_core.Operation.Dim

module Attr_list = Tensorflow_core.Operation.Attr_list

type attr = Tensorflow_core.Operation.attr

type output = Tensorflow_core.Wrapper.Graph.output

type 'a t

type p =

| P : _ t -> p

type input = [

| `single of p
| `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 create_gradient : output -> output_type:'a Type.t -> '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 operation : _ t -> Tensorflow_core.Operation.t

val output : _ t -> output

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 packed_operation : p -> Tensorflow_core.Operation.t

val packed_output : p -> output

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 shape : _ t -> Base.int Base.list

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

package tensorflow