- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for ConvertAssetArgs (0.37 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
return } } )mlir"); FailureOr<SmallVector<AssetFileDef>> asset_file_defs = ConvertAssetArgs(*module_op); EXPECT_TRUE(succeeded(asset_file_defs)); EXPECT_THAT(*asset_file_defs, SizeIs(1)); const AssetFileDef& asset_file_def = *asset_file_defs->begin();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.h
// `tf_saved_model::AssetOp`. // * Strips the "assets/" prefix of the filename when setting it to // `AssetFileDef`. FailureOr<SmallVector<tensorflow::AssetFileDef>> ConvertAssetArgs( ModuleOp module_op); } // namespace mlir::quant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 01:45:41 UTC 2023 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
ReplaceBoundInputAttrWithIndexPathAttr(arg_attrs, index_path, builder); 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);
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/cc/saved_model_export.cc
PassManager& pm) { AddExportPasses(pm, dup_constants); }, ctx, module_op)); FailureOr<SmallVector<AssetFileDef>> asset_file_defs = quant::ConvertAssetArgs(module_op); if (failed(asset_file_defs)) { return absl::InternalError("Failed to convert asset args."); } return *asset_file_defs; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0)