- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for MoveBefore (0.12 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/hoist_broadcast_read.cc
if (read->isBeforeInBlock(peer)) continue; if (llvm::isa<ReadVariableOp>(peer)) continue; has_conflicting_write = true; } if (has_conflicting_write) continue; read->moveBefore(replicate); } } // Hoist `ReadVariableOp`s above the `tf_device.replicate`s. void HoistBroadcastRead::runOnOperation() { FuncOp func = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/replicate_invariant_op_hoisting.cc
// defined by tf_device.replicate. if (DescendantUsesVirtualDevice(virtual_device_list, &inner_op)) continue; if (IsOpReplicateInvariant(replicate_region, &inner_op)) inner_op.moveBefore(replicate_op); } } void ReplicateInvariantOpHoistingPass::runOnOperation() { getOperation().walk( [](tf_device::ReplicateOp op) { HoistReplicateInvariantOps(op); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/extract_tpu_copy_with_dynamic_shape_op.cc
tpu_copy_with_dynamic_shape_op->getLoc(), tpu_copy_with_dynamic_shape_op->getResults()) .getOperation(); tpu_copy_with_dynamic_shape_op->moveBefore(return_op); return success(); } // Update all the usage of tf_device.return op with launch op result. void UpdateReturnOpResultWithLaunchOpResult(tf_device::LaunchOp* launch_op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.6K bytes - Viewed (0)