Searched defs:ResourceApplyMomentum (Results 1 – 2 of 2) sorted by relevance
117 class ResourceApplyMomentum : public XlaOpKernel { class119 explicit ResourceApplyMomentum(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in ResourceApplyMomentum() function in tensorflow::__anon9de8e2990111::ResourceApplyMomentum
37981 func ResourceApplyMomentum(scope *Scope, var_ tf.Output, accum tf.Output, lr tf.Output, grad tf.Out… func