package tensorflow

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