- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for FindMainFuncOp (0.16 sec)
-
tensorflow/compiler/mlir/quantization/common/func_test.cc
return } } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleWithMainFunc); EXPECT_THAT(*module_op, NotNull()); func::FuncOp main_func_op = FindMainFuncOp(*module_op); EXPECT_THAT(main_func_op, NotNull()); } TEST_F(FindMainFuncOpTest, ReturnsNullWhenMainFuncOpIsPrivate) { constexpr absl::string_view kModuleWithPrivateMainFunc = R"mlir( module {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_driver_test.cc
} TEST_F(ApplyQuantizationParamsPropagationTest, ConstsUsedMultipleTimesAreDuplicated) { const OwningOpRef<ModuleOp> module_op_ref = ParseModuleOpString(kModuleTFLite); func::FuncOp main_fn = FindMainFuncOp(*module_op_ref); auto op_quant_spec_getter = [&](Operation* op) { return GetOpQuantSpec(op, /*disable_per_channel_for_dense_layers=*/false); }; QuantizationDriver quantization_driver(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func.cc
bool IsPublicFuncOp(func::FuncOp func_op) { return SymbolTable::getSymbolVisibility(&*func_op) == SymbolTable::Visibility::Public; } } // namespace func::FuncOp FindMainFuncOp(ModuleOp module_op) { if (const auto main_func_op = module_op.lookupSymbol<func::FuncOp>( kImportModelDefaultGraphFuncName); main_func_op != nullptr && IsPublicFuncOp(main_func_op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/test_base.h
// Returns `failure` iff there is no `@main` or no such operation is found in // `@main`. template <typename OpT> FailureOr<OpT> FindFirstOpFromMainFunc(ModuleOp module_op) { func::FuncOp main_func_op = FindMainFuncOp(module_op); if (main_func_op == nullptr) return failure(); auto ops = main_func_op.getOps<OpT>(); if (ops.empty()) return failure(); return *ops.begin(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
main_func_op.setArgAttrs(arg_idx, new_arg_attrs); } } // namespace FailureOr<SmallVector<AssetFileDef>> ConvertAssetArgs(ModuleOp module_op) { func::FuncOp main_func_op = FindMainFuncOp(module_op); if (!main_func_op) return failure(); SmallVector<StringRef> input_names = GetEntryFunctionInputs(main_func_op); SymbolTable symbol_table(module_op); SmallVector<AssetFileDef> asset_file_defs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0)