- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 87 for getOperation (0.47 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/remove_vars_in_session_initializer.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/ir/ConvertConst.cc
newConstOp); return success(); } void ConvertConstPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); auto *context = &getContext(); patterns.add<QuantizedConstRewrite>(context); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } std::unique_ptr<OperationPass<func::FuncOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_device_copy_conversion.cc
}; // Folds tf.IdentityOp and tf.IdentityNOp if op device and the argument devices // from the defining ops match. void TensorDeviceCopyConversionPass::runOnOperation() { func::FuncOp func_op = getOperation(); auto should_fold_op_func = [&func_op](const Value &arg, const StringAttr &op_device) { // In TFRT TPU, tensor transfer is handled specifically by D2H and
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/batchmatmul_to_einsum.cc
: public impl::BatchMatMulToEinsumPassBase<BatchMatMulToEinsumPass> { void runOnOperation() override; }; void BatchMatMulToEinsumPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); patterns.add<ConvertTFBatchMatMulToEinsumOp<TF::BatchMatMulOp>, ConvertTFBatchMatMulToEinsumOp<TF::BatchMatMulV2Op>>( &getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize.cc
enable_per_channel_quantized_weight_ = enable_per_channel_quantized_weight; } private: void runOnOperation() override; }; void QuantizePass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); patterns.add<StableHloQuantization, StableHloQuantizationReverse>(&ctx); PopulateCommonQuantizationPatterns(ctx, patterns,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 07:08:19 UTC 2024 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/verify_quant_legalization.cc
} bool IsMhloUniformQuantizedOp(Operation& op) { return llvm::isa<mhlo::UniformQuantizeOp, mhlo::UniformDequantizeOp>(op); } void VerifyQuantLegalization::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { // Verify all uq and qint types are lowered. if (llvm::any_of(op->getOperandTypes(), IsQuantType) ||
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc
void ReduceTypePrecisionPass::runOnOperation() { RewritePatternSet patterns(&getContext()); patterns.add<CheckRangeAndConvertI8ToI4, SanitizeGatherOpOutputToI4>( &getContext()); (void)applyPatternsAndFoldGreedily(getOperation(), std::move(patterns)); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateReduceTypePrecisionPass() { return std::make_unique<ReduceTypePrecisionPass>(); } } // namespace TFL
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
if (!isa<TF::IdentityOp, TF::ReadVariableOp>(op)) break; v = op->getOperand(0); } return v; } void AnnotateParameterReplicationPass::runOnOperation() { ModuleOp m = getOperation(); OpBuilder builder(m.getContext()); m.walk([&](tf_device::ClusterFuncOp cluster_func) { auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); if (!replicate) return;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_custom_call_to_composite.cc
return op.getCallTargetName() != "stablehlo.composite"; }); RewritePatternSet patterns(context); patterns.add<ReplaceCustomCallWithComposite>(context); if (failed(applyPartialConversion(getOperation(), target, std::move(patterns)))) { return signalPassFailure(); } } }; static PassRegistration<LegalizeStablehloCustomCallToCompositePass>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/analyze_variables.cc
public: explicit AnalyzeVariablesPass() {} void runOnOperation() override; }; void AnalyzeVariablesPass::runOnOperation() { auto* context = &getContext(); auto module = getOperation(); bool legalize_to_tfl = true; module.walk([&](Operation* op) { // Skip ops that are supported natively by TFLite. if (IsSupportedTFLiteResourceOp(op)) return WalkResult::advance();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.3K bytes - Viewed (0)