- Sort Score
- Result 10 results
- Languages All
Results 41 - 50 of 69 for func_20 (0.15 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
}; } // namespace void NchwConvolutionToNhwcPass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); patterns.add<RewriteNchwConvolutionToNhwc>(&ctx); if (failed(applyPatternsAndFoldGreedily(func_op, std::move(patterns)))) { func_op.emitError() << "Failed to run NchwConvolutionToNhwcPass."; signalPassFailure(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
EXPECT_TRUE(module_op_ref); return module_op_ref; } mlir::MLIRContext ctx_{}; }; func::FuncOp GetMainFuncOp(ModuleOp module_op) { for (auto func_op : module_op.getOps<func::FuncOp>()) { if (func_op.getSymName() == "main") { return func_op; } } return {}; } TEST_F(ConvertAssetArgsTest, ConvertsSingleAssetArg) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(R"mlir( module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/fold_constant_transpose.cc
}; void FoldConstantTransposePass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); patterns.add<FoldTransposedConstantOp>(&ctx); if (failed(applyPatternsAndFoldGreedily(func_op, std::move(patterns)))) { func_op.emitError("Failed to fold constant->transpose pattern."); signalPassFailure(); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc
return op.getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } void VerifyNoOutsideCompilationMarkersPass::runOnOperation() { Operation* func_op = getOperation(); auto walk_result = func_op->walk([&](Operation* op) { if (IsDeviceClusterOp(*op) && HasChildLaunchDeviceOp(*op)) { std::string launch_error = absl::StrCat("Node `", op->getName().getStringRef().str(), "` ",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Aug 22 19:52:08 UTC 2023 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/merge_node_with_function.pbtxt
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 14 20:47:36 UTC 2021 - 1K 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) || llvm::any_of(op->getResultTypes(), 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/utils/nms_utils.h
void RewriteFunc(); LogicalResult VerifySignature(); private: func::FuncOp func_; }; // Abstracts the conversion of the SSD post-processing composite function to // TFLite. class ConvertSSDPostProcessFunc { public: explicit ConvertSSDPostProcessFunc(func::FuncOp func, mlir::TF::FuncAttr attr) : func_(func), attr_(attr) {}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 19 00:13:50 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/pre_calibration_test.cc
component.Run(*module_op, quantization_config); EXPECT_THAT(pre_calibration_result, IsOk()); SmallVector<func::FuncOp> func_ops; for (auto func_op : pre_calibration_result->getOps<func::FuncOp>()) { func_ops.push_back(func_op); } ASSERT_THAT(func_ops, SizeIs(2)); EXPECT_THAT(func_ops, Contains(HasSymName("main"))); EXPECT_THAT(func_ops, Contains(HasSymName("composite_dot_general_fn_1")));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 21:41:08 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc
explicit ConvertFuncToBfloat16Pass() = default; private: void runOnOperation() override; }; void ConvertFuncToBfloat16Pass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext* context = func_op.getContext(); RewritePatternSet patterns(context); BFloat16TypeConverter converter; patterns.add<BFloat16TypePattern, BitcastConvertOpPattern>(converter,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_outline_tpu_island.cc
} } } }); } // Remove `kSkipIslandOutlining` attributes. for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) { if (func_op->hasAttr(mlir::TF::kSkipIslandOutlining)) { func_op->removeAttr(mlir::TF::kSkipIslandOutlining); } } } } // namespace std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.7K bytes - Viewed (0)