Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getThenValue (0.13 sec)

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // This lowering only works on ranked types.
        auto cond_type =
            mlir::dyn_cast<RankedTensorType>(op.getCondition().getType());
        auto then_type =
            mlir::dyn_cast<RankedTensorType>(op.getThenValue().getType());
        auto else_type =
            mlir::dyn_cast<RankedTensorType>(op.getElseValue().getType());
        if (!cond_type || !then_type || !else_type) {
          return failure();
        }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

    //     (c) `cond` is a vector AND `then` and `else` are non-scalar with their
    //         first dimension equal to `cond`.
    LogicalResult SelectOp::verify() {
      SelectOp op = *this;
      auto then_tensor = op.getThenValue().getType().cast<TensorType>();
      auto else_tensor = op.getElseValue().getType().cast<TensorType>();
      // Check (1).
      if (!AreCastCompatible({then_tensor, else_tensor}))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top