- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for activation_mode (0.29 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/gpu_fusion.cc
if (!add_op || add_op.getZ().hasOneUse()) { // We fuse the Relu only if the add has a single use, otherwise we only // fuse the add itself. op->setAttr("activation_mode", rewriter.getStringAttr("Relu")); rewriter.replaceOp(relu_op, op->getResult(0)); } if (add_op) { rewriter.replaceOp(add_op, op->getResult(0)); } return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
DefaultValuedOptionalAttr<F32Attr, "0.0001f">:$epsilon, DefaultValuedOptionalAttr<F32Attr, "1.0f">:$exponential_avg_factor, DefaultValuedOptionalAttr<StrAttr, "\"Identity\"">:$activation_mode, DefaultValuedOptionalAttr<TF_ConvnetDataFormatAttr, "\"NHWC\"">:$data_format, DefaultValuedOptionalAttr<BoolAttr, "true">:$is_training ); let results = (outs
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0)