- Sort Score
- Result 10 results
- Languages All
Results 121 - 130 of 271 for getOperation (3.01 sec)
-
tensorflow/compiler/mlir/lite/quantization/ir/ConvertSimQuant.cc
expressedType, fqOp.getIsSigned()); } }; void ConvertSimulatedQuantPass::runOnOperation() { bool hadFailure = false; auto func = getOperation(); RewritePatternSet patterns(func.getContext()); auto *ctx = func.getContext(); patterns.add<ConstFakeQuantRewrite, ConstFakeQuantPerAxisRewrite>( ctx, &hadFailure);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc
} } } return false; } void TPUResourceReadForWritePass::runOnOperation() { SmallVector<tf_device::ClusterFuncOp, 4> cluster_funcs; getOperation().walk([&](tf_device::ClusterFuncOp cluster_func) { cluster_funcs.push_back(cluster_func); }); OpBuilder builder(&getContext()); // Add resource reads for resource writes from TPU cluster where for such
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/hoist_loop_invariant.cc
}, [&](Operation *op, Region *) { loopLike.moveOutOfLoop(op); }); } void HoistLoopInvariantPass::runOnOperation() { func::FuncOp func = getOperation(); // Skip the pass if the function inputs contain any resource. for (const auto &type : func.getArgumentTypes()) { if (mlir::isa<ResourceType>(getElementTypeOrSelf(type))) return; }
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/tf2xla/internal/passes/hoist_broadcast_read.cc
if (has_conflicting_write) continue; read->moveBefore(replicate); } } // Hoist `ReadVariableOp`s above the `tf_device.replicate`s. void HoistBroadcastRead::runOnOperation() { FuncOp func = getOperation(); auto result = func.walk([&](ReplicateOp replicate) { llvm::SmallVector<ReadVariableOp, 4> reads; if (failed(GetReads(func, replicate, reads))) return WalkResult::interrupt();
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/tf2xla/transforms/verify_tfxla_legalization.cc
emitError(op->getLoc()) << "Could not legalize op: " << op->getName(); return false; } return true; } void VerifyTFXLALegalization::runOnOperation() { Operation* func_op = getOperation(); ConversionTarget default_conversion_target = GetDefaultLegalConversionTargets(getContext(), legalize_chlo_); bool has_invalid_ops = false; func_op->walk([&](Operation* op) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_traits.h
StorageTypeMin, StorageTypeMax, Sign>::Impl> { public: QuantizedType GetResultQuantizedType(int index) { auto op = this->getOperation(); const auto result_type = op->getResult(index).getType().template cast<ShapedType>(); if (!result_type.getElementType().template isa<FloatType>()) return {}; Builder builder(op->getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/merge_fusion_with_dequantize.cc
} return_op->setOperand(0, new_result); rewriter.eraseOp(req_op); return success(); } }; void MergeFusionWithDequantizePass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = module_op.getContext(); RewritePatternSet patterns(ctx); patterns.add<MergeFusionWithUniformDequantizePattern>(ctx); if (failed(applyPatternsAndFoldGreedily(module_op, std::move(patterns)))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_graph_optimization_pass.cc
// operation list from module_out to module_in. auto& module_in_ops = module_in.getBody()->getOperations(); module_in_ops.clear(); module_in_ops.splice(module_in_ops.end(), module_out->getBody()->getOperations()); } // Returns a vector of passes from their names. If a pass is not found, then the // corresponding return entry is null.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 22:19:26 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tf_stablehlo_pass.cc
*this, "skip-partitioned-calls", ::llvm::cl::desc( "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")}; }; void TFToMhloPass::runOnOperation() { auto func = getOperation(); MLIRContext *context = func->getContext(); RewritePatternSet patterns(context); mhlo::PopulateLegalizeTfPatterns(context, &patterns); TF::PopulateTFLoweringBeforeHLOPatterns(context, &patterns);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_tf_ops_pass.cc
// the results of the other tf_device.remote_run operations using the // `mapping` as appropriate. Operation *cloned_remote_run_op = builder.clone(*remote_run_op.getOperation(), mapping); remote_run_op.erase(); // Replaces usages of the results of the original operations with the // results of the tf_device.remote_run operations.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0)