- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 142 for GetOperands (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
tf_device::LaunchOp launch) { // Forward launch inner op results to launch op results. launch.replaceAllUsesWith(launch.GetBody().getTerminator()->getOperands()); // For all inner ops, assign the launch device as a `device` attribute. if (failed(AssignDevicesInRegion(tf_dialect, launch, launch.getBody()))) return failure();
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/tpu_resource_read_for_write.cc
} if (read_operands.empty()) continue; // Update caller and function types with new read operands. auto operands = llvm::to_vector<4>(cluster_func.getOperands()); operands.append(read_operands.begin(), read_operands.end()); auto loc = cluster_func.getLoc(); auto new_cluster_func = builder.create<tf_device::ClusterFuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
OpFoldResult TFR::EqualOp::fold(FoldAdaptor adaptor) { auto operands = adaptor.getOperands(); assert(operands.size() == 2 && "equal op has two operands"); auto ctx = getContext(); if (operands[0] == operands[1]) return BoolAttr::get(ctx, true); return BoolAttr::get(ctx, false); } OpFoldResult ConstOp::fold(FoldAdaptor adaptor) { auto operands = adaptor.getOperands(); (void)operands;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/group_by_dialect.cc
// All operand values in our set not produced as result by some op in our set. llvm::DenseSet<Value> inputs_seen; for (Operation* outer : ops) { outer->walk([&](Operation* op) { for (Value operand : op->getOperands()) { if (!all_internal_results.contains(operand)) { if (!inputs_seen.contains(operand)) { inputs->push_back(operand); inputs_seen.insert(operand); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
bool HasDataDependencyWithUnscheduledOp( Operation& op, Block* block, SmallPtrSet<Operation*, 16>& unscheduled_ops) { WalkResult ready_to_schedule = op.walk([&](Operation* nested_op) { for (Value operand : nested_op->getOperands()) { Operation* defining_op = operand.getDefiningOp(); if (!defining_op) continue; Operation* producer_in_block = block->findAncestorOpInBlock(*defining_op);
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/host_runtime/tpu_rewrite_pass.cc
} for (auto cluster_operand : cluster.getOperands()) { Operation* def = cluster_operand.getDefiningOp(); // This pass assumes that a TPUPartitionedInputV2 is preceeded by // ReadVariable ops, and not vice versa. An earlier pass, // TPUResourceReadsWritesPartitioning, should have ensured this // precondition. if (!def) continue; for (auto operand : def->getOperands()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc
// original If op. if (op_to_inline.hasTrait<OpTrait::IsTerminator>()) { updated_results.reserve(op_to_inline.getNumOperands()); for (Value operand : op_to_inline.getOperands()) updated_results.push_back(mapper.lookup(operand)); break; } // Otherwise, clone the op here. rewriter.clone(op_to_inline, mapper); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
block.walk([&](Operation *op) { if (op->getName().getStringRef() != "tfg.NextIteration") return; mlir::OpBuilder builder(op); llvm::SmallVector<Value, 2> new_operands; FilterOutBlockArgControlDep(op->getOperands().drop_front(), new_operands); auto source_op = builder.create<tf_executor::NextIterationSourceOp>( op->getLoc(), op->getOperand(0).getType()); builder.create<tf_executor::NextIterationSinkOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc
// `operand_idx`. return op.getNumOperands() + operand_idx; } return operand_idx; } // Returns the pointers to operands at `operand_indices` of `op`. llvm::SmallVector<OpOperand*> GetOperands(Operation& op, llvm::ArrayRef<int> operand_indices) { llvm::SmallVector<OpOperand*> operands{}; for (const int operand_idx : operand_indices) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 17.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/partitioned_topological_sort.cc
if (*this_partition == op_partition && !unscheduled_op .walk([&](Operation *op) { return llvm::all_of(op->getOperands(), [&](Value operand) { return is_ready(operand, &unscheduled_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 15:05:28 UTC 2023 - 7.2K bytes - Viewed (0)