- Sort Score
- Result 10 results
- Languages All
Results 21 - 26 of 26 for IfRegionOp (0.23 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// function result types. bool InferShapeForIf(IfOp op); // Infers the shape IfRegion outputs based on the shapes of the then and else // yields. bool InferShapeForIfRegion(IfRegionOp op); // Infers the shape CaseOp outputs based on the shapes of branch function // result types. bool InferShapeForCase(CaseOp op); // Infers the shape CaseRegion outputs based on the shapes of the branch
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
vars_initialized, lifted_partitioned_call_callees))) { return failure(); } } else if (isa<TF::IfRegionOp, TF::CaseRegionOp, TF::WhileRegionOp>(op)) { for (Region& region : op.getRegions()) (void)HoistForControlFlow(®ion.front(), module, vars_initialized,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
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)