- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 218 for getOperands (0.21 sec)
-
tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc
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/gpu_fusion.cc
// Build the newly fused operation to replace the batch norm OperationState state(batch_norm.getLoc(), _FusedBatchNormExOp::getOperationName()); state.addOperands(batch_norm.getOperands()); if (side_input) state.operands.push_back(side_input); state.addTypes(batch_norm.getResultTypes()); state.addAttributes(batch_norm->getAttrs()); Operation *op = rewriter.create(state);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
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>( op->getLoc(), source_op.getToken(), /*input=*/op->getOperand(0), /*controlInputs=*/new_operands); op->replaceAllUsesWith(
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/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/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/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/breakup-islands.cc
Value operand = yield.getOperand(0); auto identity = builder.create<TF::IdentityOp>(island.getLoc(), operand.getType(), operand); yield.setOperand(0, identity.getOutput()); } else { auto identity_n = builder.create<TF::IdentityNOp>( island.getLoc(), yield.getOperandTypes(), yield.getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 11 20:52:36 UTC 2023 - 16.7K 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/tensorflow/translate/split_into_island_per_op_pass.cc
Value operand = yield.getOperand(0); auto identity = builder.create<TF::IdentityOp>(island.getLoc(), operand.getType(), operand); yield.setOperand(0, identity.getOutput()); } else { auto identity_n = builder.create<TF::IdentityNOp>( island.getLoc(), yield.getOperandTypes(), yield.getOperands());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 17 07:31:01 UTC 2023 - 7.3K bytes - Viewed (0)