- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for visitUsedValuesDefinedAbove (0.37 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sink_constant.cc
// a sunk clone of it. This allows for reusing a sunk constant with // multiple uses in the region. llvm::DenseMap<Value, TF::ConstOp> sunk_constant; Region &body = cluster.getBody(); visitUsedValuesDefinedAbove(body, [&](OpOperand *use) { Value constant = use->get(); auto const_op = dyn_cast_or_null<TF::ConstOp>(constant.getDefiningOp()); if (!const_op) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/graph_pruning.cc
llvm::SmallPtrSetImpl<Operation*>* reachable_ops, llvm::SmallVectorImpl<Operation*>* ops_to_visit) { if (auto island = llvm::dyn_cast<IslandOp>(op)) { mlir::visitUsedValuesDefinedAbove( island.getBody(), island.getBody(), [&](OpOperand* operand) { VisitOpOperand(graph, operand->get(), reachable_ops, ops_to_visit); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc
Operation* op) { for (auto operand : op->getOperands()) { if (IsResource(operand)) { SetPotentiallyWritten(operand); } } visitUsedValuesDefinedAbove(op->getRegions(), [&](OpOperand* operand) { if (IsResource(operand->get())) { SetPotentiallyWritten(operand->get()); } }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
// _TPUDeviceOrdinalPlaceholder implicitly depends on the replica. if (llvm::isa<TF::_TPUDeviceOrdinalPlaceholderOp>(op)) return false; bool has_replicate_operands = false; visitUsedValuesDefinedAbove(op->getRegions(), [&](OpOperand* operand) { if (!ancestor_of_replicate(operand->get().getParentRegion())) has_replicate_operands = true; }); return !has_replicate_operands; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0)