package tensorflow

  1. Overview
  2. Docs
type t = {
  1. 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 = {
  1. 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