- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 34 for GetBlock (0.13 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/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) -
platforms/core-execution/persistent-cache/src/main/java/org/gradle/cache/internal/btree/FreeListBlockStore.java
store.close(); } @Override public void clear() { store.clear(); } @Override public void remove(BlockPayload block) { Block container = block.getBlock(); store.remove(block); freeListBlock.add(container.getPos(), container.getSize()); } @Override public <T extends BlockPayload> T readFirst(Class<T> payloadType) {
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Fri Apr 26 16:02:32 UTC 2024 - 8.7K 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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/lower_variable_ops_to_ml_program.cc
if (!initial_value) { initial_value = builder.getArrayAttr({}); } } if (!type) return nullptr; auto globalOp = builder.create<ml_program::GlobalOp>( builder.getBlock()->getParentOp()->getLoc(), name, type, false, initial_value, nullptr); symbol_table.insert(globalOp); return globalOp; } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 7.6K 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_tpuv1_outline_tpu_island.cc
SymbolTable &to) { if (to.lookup<func::FuncOp>(symbol_ref.getValue())) return; func::FuncOp callee = from.lookup<func::FuncOp>(symbol_ref.getValue()); callee.getOperation()->getBlock()->getOperations().remove( callee.getOperation()); to.insert(callee); } void TPUBridgeExecutorIslandOutlining::runOnOperation() { MLIRContext *ctx = &getContext();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)