Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 22 of 22 for getParentRegion (0.35 sec)

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

      // `mhlo.while` cond terminator does not need to be rewritten as it always
      // returns a tensor<i1> predicate value.
      if (auto while_parent = dyn_cast_or_null<WhileOp>(terminator->getParentOp()))
        if (terminator->getParentRegion() == &while_parent.getCond()) return;
    
      builder.setInsertionPoint(terminator);
      llvm::SmallDenseMap<Value, Value> rewritten_operands;
      auto new_results =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        RegionBranchPoint point) {
      if (auto whileOp =
              llvm::dyn_cast<WhileRegionOp>(this->getOperation()->getParentOp())) {
        if (&whileOp.getCond() == this->getOperation()->getParentRegion()) {
          // cut off the boolean (the condition itself) at the start
          return MutableOperandRange(
              this->getOperation(), 1,
              this->getOperation()->getOperands().size() - 1);
        }
    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