- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for use_nesterov (0.44 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) -
tensorflow/compiler/mlir/tensorflow/tests/decompose_resource_ops.mlir
// CHECK: [[VAR_NEW:%.*]] = "tf.Sub"([[VAR]], [[ACCUM_NEW_LR]]) // CHECK: "tf.AssignVariableOp"([[VAR_HANDLE]], [[VAR_NEW]]) "tf.ResourceApplyMomentum"(%0, %1, %arg0, %arg1, %arg2) {use_locking = false, use_nesterov = false} : (tensor<!tf_type.resource<tensor<f32>>>, tensor<!tf_type.resource<tensor<f32>>>, tensor<f32>, tensor<f32>, tensor<f32>) -> () tf_device.return }) : () -> () func.return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 51.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
TF_Float32Tensor:$epsilon, TF_Float32Tensor:$accumulator, TF_Float32Tensor:$momentum, TF_Float32Tensor:$embedding_table, I64Attr:$feature_width, BoolAttr:$use_nesterov, F32Attr:$beta_2, F32Attr:$exponent ); let results = (outs TF_Float32Tensor:$updated_embedding_table, TF_Float32Tensor:$updated_accumulator,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
DefaultValuedOptionalAttr<BoolAttr, "false">:$use_nesterov ); let results = (outs); TF_DerivedOperandTypeAttr T = TF_DerivedOperandTypeAttr<2>; } def TF_ResourceApplyMomentumOp : TF_Op<"ResourceApplyMomentum", []> { let summary = "Update '*var' according to the momentum scheme."; let description = [{ Set use_nesterov = True if you want to use Nesterov momentum.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)