- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for IfRegionOp (0.52 sec)
-
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/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)