Module `Make.Graph`

module Optimiser : sig ... end

type graph =
  Make_Nested(Owl_computation_engine.Make_Graph(Owl_computation_cpu_device.Make(A))).Graph.graph

Sourceval shape_or_value : Optimiser.Operator.Symbol.Shape.Type.t -> string

Sourceval graph_to_dot : graph -> string

Sourceval graph_to_trace : graph -> string

Sourceval save_graph : 'a -> string -> unit

Sourceval load_graph : string -> 'a * 'b

Source

val collect_rvs : 
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array ->
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array

Sourceval invalidate_rvs : graph -> unit

Source

val make_graph : 
  input:Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array ->
  output:Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array ->
  string ->
  graph

Source

val get_inputs : 
  graph ->
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array

Source

val get_outputs : 
  graph ->
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array

Source

val get_node_arr_val : 
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node ->
  Optimiser.Operator.Symbol.Shape.Type.Device.A.arr

Source

val get_node_elt_val : 
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node ->
  Optimiser.Operator.Symbol.Shape.Type.Device.A.elt

Source

val set_node_arr_val : 
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node ->
  Optimiser.Operator.Symbol.Shape.Type.Device.value ->
  unit

Source

val set_node_elt_val : 
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node ->
  Optimiser.Operator.Symbol.Shape.Type.Device.value ->
  unit

Sourceval is_iopair_safe : 'a Owl_graph.node -> 'a Owl_graph.node -> bool

Source

val make_iopair : 
  graph ->
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array ->
  Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node array ->
  unit

Sourceval update_iopair : graph -> unit

Source

val remove_unused_iopair : 
  'a Owl_graph.node array ->
  'b array ->
  'a Owl_graph.node array * 'b array

Source

val init_inputs : 
  (Optimiser.Operator.Symbol.Shape.Type.attr Owl_graph.node ->
    Optimiser.Operator.Symbol.Shape.Type.Device.value) ->
  graph ->
  unit

Sourceval optimise : graph -> unit

package owl-base

Module Make.Graph

Module `Make.Graph`