package tensorflow

You can search for identifiers within the package.

in-package search v0.2.0

Legend:
Library
Module
Module type
Parameter
Class
Class type

type t = {

f : 'a. self:[< `double | `float ] as 'b Node.t -> gradient:'b Node.t -> Node.p option list;

}

val add : Node.Op_name.t -> t -> unit

val find : 
  Node.Op_name.t ->
  (self:Node.p -> gradient:Node.p -> Node.p option list) option

type multi = {

g : 'a. self:[< `double | `float ] as 'b Node.t -> gradients:'b Node.t Core_kernel.Std.Int.Map.t -> Node.p option list;

}

val add_multi : Node.Op_name.t -> multi -> unit

val find_multi : 
  Node.Op_name.t ->
  (self:Node.p ->
    gradients:Node.p Core_kernel.Std.Int.Map.t ->
    Node.p option list)
    option