Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for use_nesterov (0.13 sec)

  1. 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)
Back to top