- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for use_nesterov (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops.td
def DecomposeResourceApplyMomentumOpNonNesterov : Pattern< (TF_ResourceApplyMomentumOp:$src_op $var_resource, $accum_resource, $lr, $grad, $momentum, BoolAttr:$_, ConstBoolAttrFalse:$use_nesterov ), [(TF_AddV2Op:$accum_new (TF_MulOp (CreateTFReadVariableOp $src_op, $grad, $accum_resource), $momentum ), $grad ),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 20.7K bytes - Viewed (0)