- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for FindMainFuncOp (0.97 sec)
-
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
)mlir"; TEST_F(AttrsAndConstraintsTest, HasStaticShapeSucceedsWithStaticShapes) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleStatic); ASSERT_TRUE(module_op); func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); Value dot_general_result = FindOperationOfType<DotGeneralOp>(main_fn)->getResult(0); EXPECT_TRUE(HasStaticShape(dot_general_result));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0) -
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/lift_as_function_call_test.cc
} )mlir"; TEST_F(LiftAsFunctionCallTest, FunctionLiftedAsXlaCallModuleOp) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleStableHlo); ASSERT_TRUE(module_op); func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); auto dot_general_op = FindOperationOfType<mlir::stablehlo::DotGeneralOp>(main_fn); const SmallVector<NamedAttribute>& attributes = {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K 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) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
func.return %2 : tensor<2x3x2x2xf32> } } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleGather); ASSERT_TRUE(module_op); func::FuncOp main_fn = FindMainFuncOp(*module_op); ASSERT_THAT(main_fn, NotNull()); Operation* gather_op = FindOperationOfType<GatherOp>(main_fn); const auto spec = GetStableHloQuantConstraints(gather_op); EXPECT_THAT(spec, NotNull());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
} void MergeInitializerFunctionOpsToMainPass::runOnOperation() { ModuleOp module_op = getOperation(); MLIRContext* ctx = module_op.getContext(); func::FuncOp main_func_op = FindMainFuncOp(module_op); if (!main_func_op) { module_op.emitError("Main function op not found."); return signalPassFailure(); } GraphOp main_graph_op = GetGraphOpFromFuncOp(main_func_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0)