- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 230 for eraseOp (0.62 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/xla_inline_device_ops.cc
Value old_val, new_val; for (auto it : llvm::zip(cluster_op.getResults(), return_op.getOperands())) { std::tie(old_val, new_val) = it; old_val.replaceAllUsesExcept(new_val, &return_op); } cluster_op.erase(); } void XlaInlineDeviceOpsPass::runOnOperation() { ModuleOp module = getOperation(); llvm::SmallVector<tf_device::ClusterOp, 4> ops; module.walk(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 00:59:46 UTC 2022 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/tfxla_device_specific_transforms.cc
OpBuilder opbuilder(get_alg_op); auto tf_const = opbuilder.create<TF::ConstOp>( get_alg_op->getLoc(), opbuilder.getI32IntegerAttr((int)tensorflow_rng)); get_alg_op->replaceAllUsesWith(tf_const); get_alg_op->erase(); return success(); } void TFXLADeviceSpecificTransforms::runOnOperation() { if (!device_type_.hasValue()) return; auto func_op = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.h
#include "xla/xla_data.pb.h" #include "tensorflow/core/lib/core/status.h" #include "tensorflow/core/platform/statusor.h" #include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { // Erase rewritten ClusterFuncOp(s). If TPUPartitionedInputV2Op / // TPUPartitionedOutputV2Op are present, they must be removed along with the // ClusterFuncOp(s). mlir::LogicalResult EraseClusterFuncs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/modify_io_nodes.cc
return failure(); } new_input_types[i] = arg_type; arg.dropAllUses(); if (quantize_op.use_empty()) { quantize_op.erase(); } } else { // `arg` has multiple uses or the user isn't a quantiz op (so we couldn't // rewrite it to a different type. Make a copy of the `arg` and replace // its use.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.9K bytes - Viewed (0) -
platforms/core-configuration/kotlin-dsl/src/test/kotlin/org/gradle/kotlin/dsl/execution/ProgramParserTest.kt
assertEmptyProgram(" /* before */buildscript { /* within */ }/* after */ ") } @Test fun `empty Stage 1 with non-empty Stage 2 parse to Stage 2 with Stage 1 fragments erased`() { val scriptOnlySource = programSourceWith("println(\"Stage 2\")") assertProgramOf( scriptOnlySource, Program.Script(scriptOnlySource) )
Registered: Wed Jun 12 18:38:38 UTC 2024 - Last Modified: Wed Aug 02 08:06:49 UTC 2023 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_launch_func_to_tf_call.cc
/*executor_type=*/builder.getStringAttr("")); call_op->setAttr("device", launch->getAttrOfType<StringAttr>("device")); launch.replaceAllUsesWith(call_op); launch.erase(); }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateConvertLaunchFuncToTFCallPass() { return std::make_unique<ConvertLaunchFuncToTFCallPass>(); } } // namespace TFDevice
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 21:08:09 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/name_anonymous_iterators.cc
op->getLoc(), op->getResultTypes()[0], name, /*container=*/"", op.getOutputTypes(), op.getOutputShapes()); op->getResults()[0].replaceAllUsesWith(new_op->getResults()[0]); if (op->use_empty()) op->erase(); return count; } void NameAnonymousIteratorsPass::runOnOperation() { int count = 1; getOperation().walk( [&](TF::AnonymousIteratorOp op) { count = replace(op, count); }); getOperation().walk(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc
auto new_op = builder.create<TFR::CastOp>( returned_op->getLoc(), new_type, returned_op.getArg()); returned_value.set(new_op.getResult()); if (returned_op.use_empty()) { returned_op.erase(); } } else { returned_value.get().getDefiningOp()->emitError( "The producer of quantized type result should be a tfr.cast op."); return; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
auto s_const = builder.create<SparseConstOp>( op->getLoc(), cst.getValue(), s_param, new_value); value.replaceAllUsesWith(s_const.getResult()); cst.erase(); } else if (type.getElementType().isF16()) { std::vector<Eigen::half> dense_data; dense_data.reserve(type.getNumElements()); for (const auto& val : attr.getValues<APFloat>())
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
status = ReplacePartitionedOp(num_cores_per_replica, partitioned_output); } if (status.has_value()) { if (failed(*status) || !op->use_empty()) return WalkResult::interrupt(); op->erase(); } return WalkResult::advance(); }); if (result.wasInterrupted()) { signalPassFailure(); return; } } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0)