- Sort Score
- Result 10 results
- Languages All
Results 21 - 23 of 23 for IfRegionOp (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
} }]; } def TF_YieldOp : TF_Op<"Yield", [Terminator, Pure, NativeOpTrait<"ReturnLike", [], "", "">, ParentOneOf<["CaseRegionOp", "IfRegionOp", "WhileRegionOp", "GeneratorDatasetRegionOp"]>, DeclareOpInterfaceMethods<RegionBranchTerminatorOpInterface, ["getMutableSuccessorOperands"]>, ]> { let summary = "Yield operation";
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/lite/stablehlo/transforms/legalize_hlo.cc
LogicalResult matchAndRewrite( mhlo::IfOp op, OpAdaptor adaptor, ConversionPatternRewriter& rewriter) const final { // HLO IfOp currently doesn't support stateless auto new_op = rewriter.create<TF::IfRegionOp>( op.getLoc(), op->getResultTypes(), op.getPred(), /*is_stateless=*/false, /*_then_func_name=*/nullptr, /*_else_func_name=*/nullptr); new_op.getThenBranch().takeBody(op.getTrueBranch());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
ConvertSigmoidGradOpDynamic, ConvertConv2DDynamic, ConvertPadOpDynamic, ConvertGatherNdOpDynamic, LowerControlFlowOp<TF::CaseRegionOp, mhlo::CaseOp>, LowerControlFlowOp<TF::IfRegionOp, mhlo::IfOp>, LowerControlFlowOp<TF::WhileRegionOp, mhlo::WhileOp>, LowerYieldOp>(context); // clang-format on } // LINT.ThenChange(:MlirAlwaysOps) } // end namespace mhlo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)