- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 437 for module_op_ (0.32 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
// passed to the target function. LogicalResult LiftHashTableOpsToArguments(ModuleOp module_op, func::FuncOp target_func) { if (!llvm::hasSingleElement(target_func)) return success(); if (!UsedBySupportedOps(module_op, target_func)) return success(); if (IsMainOrInitializerFunction(module_op, target_func)) return success(); llvm::StringMap<int> shared_name_to_arg_idx;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.cc
} // namespace absl::Status RunCalibrationPasses( mlir::ModuleOp module_op, MLIRContext& ctx, absl::string_view calibration_data_dir, const bool force_regenerate_calibration_data) { // Disable DumpTensor ops when running calibration. DisableDebugging(module_op); std::vector<std::string> skipping_aggregator_ops; if (!force_regenerate_calibration_data) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 06:31:57 UTC 2024 - 9.2K 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/tensorflow/ir/tf_saved_model.h
SessionInitializerOp GetSessionInitializerOp(ModuleOp module_op); // Returns the exported name for the session initializer function. SmallVector<StringRef, 2> GetSessionInitializerExportedName(ModuleOp module_op); // Returns initializer function ops. These functions' symbols are in the // "initializers" attribute of the session initializer op. SmallVector<func::FuncOp, 2> GetInitializerFunctions(ModuleOp module_op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K 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/lite/quantization/stablehlo/quantization.h
// `module_op` is the input graph to be quantized and it should contain // StableHLO ops. // // Returns a quantized `ModuleOp` in StableHLO, potentially wrapped inside a // XlaCallModuleOp. Returns a non-OK status if quantization fails, or any of // `saved_model_bundle` or `quantization_py_function_lib` is a nullptr. absl::StatusOr<mlir::ModuleOp> RunQuantization(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 02:44:03 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/debugging/mlir_dump_test.cc
mlir::OpBuilder builder(&ctx); auto module_op = builder.create<mlir::ModuleOp>(builder.getUnknownLoc()); // Destroy by calling destroy() to avoid memory leak since it is allocated // with malloc(). const absl::Cleanup module_op_cleanup = [module_op] { module_op->destroy(); }; const mlir::LogicalResult result = pm.run(module_op); EXPECT_FALSE(failed(result));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 03:17:14 UTC 2024 - 6.4K 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/stablehlo/cc/debugger.cc
namespace stablehlo::quantization { void DisableDebugging(mlir::ModuleOp module_op) { module_op.walk( [](mlir::TF::DumpTensorOp dump_op) { dump_op.setEnabled(false); }); } void ChangeToQuantizedFilename(mlir::ModuleOp module_op) { module_op.walk([](mlir::TF::DumpTensorOp dump_op) { dump_op.setFileName("quantized_tensor_data.pb"); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 00:17:12 UTC 2024 - 1.3K bytes - Viewed (0)