Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 63 for GetBlock (0.3 sec)

  1. 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)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tf_executor_to_functional.cc

    // order of ops in tf_executor.graph.
    LogicalResult LiftIslandOpInnerOpsFromGraph(tf_executor::GraphOp graph) {
      auto graph_position = graph.getOperation()->getIterator();
      Block* parent_block = graph.getOperation()->getBlock();
      for (Operation& op : graph.GetBody().without_terminator()) {
        auto island_op = llvm::dyn_cast<tf_executor::IslandOp>(op);
        if (!island_op)
          return op.emitOpError()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.cc

      int cluster_idx = -1;
      for (size_t child_idx = 0;
           child_idx < old_parallel_execute.getRegions().size(); ++child_idx) {
        auto& block = old_parallel_execute.GetRegionBlockWithIndex(child_idx);
        if (cluster_func->getBlock() == &block) {
          assert(cluster_idx == -1);
          cluster_idx = child_idx;
        }
      }
      assert(cluster_idx != -1);
    
      // Move children that are preserved.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 13 03:57:18 UTC 2023
    - 4.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/materialize_mlir_passthrough_op.cc

        Block &block = body.front();
        for (const auto &arg_mapping :
             llvm::zip(block.getArguments(), op->getOperands())) {
          std::get<0>(arg_mapping).replaceAllUsesWith(std::get<1>(arg_mapping));
        }
        op->getBlock()->getOperations().splice(op->getIterator(),
                                               block.getOperations(), block.begin(),
                                               std::prev(block.end()));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 03 12:35:38 UTC 2022
    - 4.2K 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/utils/topological_sort.cc

        int incoming_ctrl_edges = 0;
        int incoming_data_edges = 0;
        op.walk([&](Operation* child) {
          ancestor[child] = &op;
          for (Operation* predecessor : ctrlPredecessors(child)) {
            if (predecessor->getBlock() == &block) {
              incoming_ctrl_edges++;
            }
          }
          for (Value v : child->getOperands()) {
            if (v.getParentBlock() == &block) {
              incoming_data_edges++;
            }
          }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 08 17:01:11 UTC 2022
    - 5K bytes
    - Viewed (0)
  8. build-logic/documentation/src/main/groovy/gradlebuild/docs/dsl/docbook/model/ClassDoc.groovy

            }
            return comment[0]
        }
    
        PropertyDoc findProperty(String name) {
            return classProperties.find { it.name == name }
        }
    
        BlockDoc getBlock(String name) {
            def block = classBlocks.find { it.name == name }
            if (block) {
                return block
            }
            for (extensionDoc in classExtensions) {
    Registered: Wed Jun 12 18:38:38 UTC 2024
    - Last Modified: Wed Dec 09 08:14:05 UTC 2020
    - 6.2K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc

        return failure();
    
      // Move all inner ops of the launch to the block containing the launch.
      auto body = launch.GetBody().without_terminator();
      Operation* launch_op = launch.getOperation();
      launch_op->getBlock()->getOperations().splice(
          launch_op->getIterator(), launch.GetBody().getOperations(), body.begin(),
          body.end());
    
      launch.erase();
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.5K bytes
    - Viewed (0)
  10. 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)
Back to top