- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for FindMainFuncOp (0.14 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/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/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) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.cc
} } } void ReplaceStablehloOpsInMainFunctionWithXlaCallModuleOpsPass:: runOnOperation() { ModuleOp module_op = getOperation(); func::FuncOp main_func = FindMainFuncOp(module_op); if (!main_func) return; // In case the model has tf.StatefulPartitionedCallOp or tf.PartitionedCallOp, // we recursively find called functions and process StableHLO ops in them.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21K bytes - Viewed (0)