- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 111 for module_op_ (0.51 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc
: OpRewritePattern<TF::XlaCallModuleOp>(context), module_op_(module_op) {} using OpRewritePattern<TF::XlaCallModuleOp>::OpRewritePattern; private: ModuleOp module_op_; LogicalResult matchAndRewrite(TF::XlaCallModuleOp op, PatternRewriter &rewriter) const override { OwningOpRef<ModuleOp> stablehlo_module_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 25 09:43:18 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kConstantOpWithRankFour); ASSERT_TRUE(module_op); ASSERT_FALSE(module_op->getBodyRegion().empty()); ASSERT_FALSE(module_op->getBodyRegion().front().empty()); auto constant_op = dyn_cast_or_null<mlir::stablehlo::ConstantOp>( module_op->getBodyRegion().front().front()); ASSERT_THAT(constant_op, NotNull());
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/tensorflow/python/quantize_model.cc
absl::StatusOr<ExportedModel> ExportDebuggingModel( mlir::ModuleOp module_op, mlir::MLIRContext *context, const QuantizationOptions &quantization_options, const absl::flat_hash_map<std::string, std::string> &function_aliases) { // Clone ModuleOp and function aliases so changes in this pipeline won't // be reflected in the original values. mlir::OwningOpRef<mlir::ModuleOp> cloned_module_ref(module_op.clone());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(IsOpQuantizableStableHloTest, ConstantOpQuantizable) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleConstantAdd); ASSERT_TRUE(module_op); auto test_func = module_op->lookupSymbol<func::FuncOp>("constant_add"); ASSERT_THAT(test_func, NotNull()); auto constant_op =
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/tf2xla/api/v1/compile_mlir_util.cc
TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); if (VLOG_IS_ON(2)) tensorflow::DumpMlirOpToFile("compile_mlir_shape_refiner", module_op); return absl::OkStatus(); } Status BuildHloFromTf(mlir::ModuleOp module_op, xla::XlaBuilder& builder, llvm::ArrayRef<xla::XlaOp> xla_params, std::vector<xla::XlaOp>& returns,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
return %0 : tensor<1x3xf32> } } )mlir"; TEST_F(LiftAsFunctionCallTest, LiftedFunctionSucceeds) { OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleLifted); ASSERT_TRUE(module_op); auto composite_dot_general_fn = module_op->lookupSymbol<func::FuncOp>("composite_dot_general_fn_1"); ASSERT_THAT(composite_dot_general_fn, NotNull());
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/tensorflow/passes/unfreeze_constants.cc
SessionInitializerOp GetOrCreateSessionInitializerOp(ModuleOp module_op) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module_op); // Create one if it doesn't exist. if (!session_init_op) { OpBuilder builder(&module_op.getBodyRegion()); session_init_op = builder.create<SessionInitializerOp>( module_op.getLoc(), /*initializers=*/builder.getArrayAttr({})); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 16 15:04:53 UTC 2023 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tf_xla_mlir_translate.cc
return PopulateResultIOInfo(module_op, arg_shapes, /*use_tuple_args=*/false, /*use_resource_updates_for_aliases=*/false, shape_determination_fns, compilation_result); } static mlir::LogicalResult MlirTfToHloTextTranslateFunctionImpl( mlir::ModuleOp module_op, llvm::raw_ostream& output, bool via_builder) { if (!module_op) return mlir::failure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
return symbol_table.replaceAllSymbolUses( main_func_op, StringAttr::get(module_op.getContext(), new_func_name), module_op); } void InsertMainFunctionPass::runOnOperation() { ModuleOp module_op = getOperation(); if (failed(RenameExistingMainFunction(module_op))) { module_op->emitError("Failed to rename existing function `@main`."); signalPassFailure(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/uniform_quantized_types_test.cc
return %0 : tensor<2x!quant.uniform<i8:f32, 1.000000e+00:0>> } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kFullyQuantizedAdd); ASSERT_TRUE(module_op); auto func_op = module_op->lookupSymbol<func::FuncOp>("fully_quantized_add"); ASSERT_THAT(func_op, NotNull()); auto add_op_itr = func_op.getBody().op_begin<mlir::stablehlo::AddOp>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.8K bytes - Viewed (0)