Legend:
Library
Module
Module type
Parameter
Class
Class type
Library
Module
Module type
Parameter
Class
Class type
type t = {
f : 'a. self:[< `float | `double ] as 'a Node.t ->
gradient:'a Node.t ->
Node.p Base.option Base.list;
}
val add : Node.Op_name.t -> t -> Base.unit
val find :
Node.Op_name.t ->
(self:Node.p -> gradient:Node.p -> Node.p Base.option Base.list) Base.option
type multi = {
g : 'a. self:[< `float | `double ] as 'a Node.t ->
gradients:'a Node.t Base.Map.M(Base.Int).t ->
Node.p Base.option Base.list;
}
val add_multi : Node.Op_name.t -> multi -> Base.unit
val find_multi :
Node.Op_name.t ->
(self:Node.p ->
gradients:Node.p Base.Map.M(Base.Int).t ->
Node.p Base.option Base.list)
Base.option