Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 26 for GetBlock (0.59 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_program_key.cc

    };
    
    // Checks if `op` is nested in `block`.
    bool OpInBlock(Operation* op, Block* block) {
      Block* op_block = op->getBlock();
      while (op_block) {
        if (op_block == block) return true;
        if (auto* parent_op = op_block->getParentOp()) {
          op_block = parent_op->getBlock();
        } else {
          break;
        }
      }
      return false;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/merge_control_flow.cc

                     const TF::SideEffectAnalysis::Info& side_effect_analysis) {
      // IfRegion ops must be in the same block.
      if (second_if.getOperation()->getBlock() !=
          first_if.getOperation()->getBlock()) {
        return false;
      }
      assert(first_if.getOperation()->isBeforeInBlock(second_if.getOperation()));
    
      llvm::SmallSetVector<Operation*, 4> destination_ops =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 17 07:31:01 UTC 2023
    - 25.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc

        assert(op && "member operation must be not null");
    
        for (Operation *user : op->getUsers()) {
          // Skip users in other blocks.
          if (user->getBlock() != op->getBlock()) continue;
    
          // Skip users is in the `dst_root` or `src_root` clusters, if we'll merge
          // roots they'll become a single cluster and will not violate the
          // dominance property after that.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 27.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc

    LogicalResult SortTopologically(Block::iterator begin, Block::iterator end) {
      Block* block = begin->getBlock();
      // Either sort from `begin` to end of block or both `begin` and
      // `end` should belong to the same block.
      assert(end == block->end() ||
             end->getBlock() == block && "ops must be in the same block");
    
      // Track the ops that still need to be scheduled in a set.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 27.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/layout_optimization.cc

        if (args_permutation.empty() || res_permutation.empty()) return;
    
        mlir::Operation* op = layout_sensitive_interface.getOperation();
        Location loc = op->getLoc();
        OpBuilder builder = OpBuilder::atBlockEnd(op->getBlock());
    
        auto perm_attr = [&](Permutation permutation) -> DenseIntElementsAttr {
          auto perm_ty = RankedTensorType::get({4}, builder.getIntegerType(64));
          return DenseIntElementsAttr::get(perm_ty, permutation);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc

          if (mlir::dyn_cast_or_null<BlockArgument>(operand)) continue;
          auto operand_op = operand.getDefiningOp();
          if (IsConstant(operand_op)) continue;
          if (operand_op->getBlock() != op->getBlock()) {
            return WalkResult::interrupt();
          }
        }
        return WalkResult::advance();
      });
      return !walk_result.wasInterrupted();
    }
    
    // It reduces the following pattern:
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_cfg.cc

      Value cond_i1 = LowerCondition(loc, op.getCond(), &builder);
      if (!cond_i1) return failure();
    
      // Split the basic block before the 'if'.  The new dest will be our merge
      // point.
      Block* orig_block = op_inst->getBlock();
      Block* merge_block = orig_block->splitBlock(op);
    
      // Add the block arguments to the merge point, and replace all uses of the
      // original operation results with them.
      for (Value value : op_inst->getResults())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jan 13 11:42:59 UTC 2023
    - 12.2K bytes
    - Viewed (0)
  8. platforms/core-execution/persistent-cache/src/main/java/org/gradle/cache/internal/btree/BTreePersistentIndexedCache.java

                    return block.getPos().compareTo(block1.getPos());
                }
            });
    
            for (int i = 0; i < blocks.size() - 1; i++) {
                Block b1 = blocks.get(i).getBlock();
                Block b2 = blocks.get(i + 1).getBlock();
                if (b1.getPos().getPos() + b1.getSize() > b2.getPos().getPos()) {
                    throw new IOException(String.format("%s overlaps with %s", b1, b2));
                }
            }
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Fri Sep 22 09:08:47 UTC 2023
    - 26.5K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc

    }
    
    // Checks if `op` is nested in `block`.
    bool OpInBlock(Operation* op, Block* block) {
      Block* op_block = op->getBlock();
      while (op_block) {
        if (op_block == block) return true;
        if (auto* parent_op = op_block->getParentOp()) {
          op_block = parent_op->getBlock();
        } else {
          break;
        }
      }
      return false;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 20.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/executor_island_coarsening.cc

          Value inner_op_result = std::get<0>(ret_vals);
          Value island_result = std::get<1>(ret_vals);
          for (auto& use : llvm::make_early_inc_range(island_result.getUses())) {
            if (islandBlocks.count(use.getOwner()->getBlock())) {
              // If the use is within our island group, forward the result from
              // inner op.
              use.set(inner_op_result);
            } else if (!result_captured) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 19.7K bytes
    - Viewed (0)
Back to top