Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for IfRegionOp (0.52 sec)

  1. 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)
  2. 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)
Back to top