- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for GetBlock (0.59 sec)
-
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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) -
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)