- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for asset_file_def (0.17 sec)
-
tensorflow/cc/saved_model/loader_util.cc
} const auto& any_assets = assets_it->second.any_list().value(); for (const auto& any_asset : any_assets) { AssetFileDef asset_file_def; TF_RETURN_IF_ERROR( ParseAny(any_asset, &asset_file_def, "tensorflow.AssetFileDef")); asset_file_defs->push_back(asset_file_def); } return absl::OkStatus(); } } // namespace internal
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 10 10:25:28 UTC 2024 - 3.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc
} else { return filename; } } AssetFileDef CreateAssetFileDef(const StringRef filename, const StringRef tensor_name) { AssetFileDef asset_file_def{}; asset_file_def.set_filename(MaybeStripAssetDirectoryPrefix(filename).str()); tensorflow::TensorInfo tensor_info{}; tensor_info.set_name(tensor_name.str()); *asset_file_def.mutable_tensor_info() = tensor_info;
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/tensorflow/cc/convert_asset_args_test.cc
} } )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(); EXPECT_THAT(asset_file_def.filename(), Eq("file_0.txt")); EXPECT_THAT(asset_file_def.tensor_info().name(), Eq("arg_0:0"));
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/stablehlo/cc/saved_model_export_test.cc
} } )mlir"); ASSERT_TRUE(module_op); AssetFileDef asset_file_def{}; ASSERT_TRUE( TextFormat::ParseFromString(R"pb(filename: "vocab_file.txt", tensor_info { name: "arg_0:0" })pb", &asset_file_def)); const std::vector<AssetFileDef> asset_file_defs = {asset_file_def}; const absl::StatusOr<ExportedModel> exported_model =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 19.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
exported_model.mutable_function_aliases()->insert(function_aliases.begin(), function_aliases.end()); for (const AssetFileDef& asset_file_def : asset_file_defs) { *exported_model.mutable_asset_file_defs()->Add() = asset_file_def; } if (saver_def != std::nullopt) { *exported_model.mutable_saver_def() = *std::move(saver_def); } return exported_model; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/save_model.py
signature_def_map, ops.get_default_graph() ) # Add `AssetFileDef`s to the collection so that correct values are fed to # the tensors that accept asset file paths. for asset_file_def in asset_file_defs: asset_any_proto = Any() asset_any_proto.Pack(asset_file_def) ops.add_to_collection( saved_model_constants.ASSETS_KEY, asset_any_proto,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 01:09:50 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/cc/saved_model/loader.cc
const std::vector<AssetFileDef>& asset_file_defs, std::vector<std::pair<string, Tensor>>* inputs) { if (asset_file_defs.empty()) { return; } for (auto& asset_file_def : asset_file_defs) { Tensor assets_file_path_tensor = CreateStringTensor(io::JoinPath( export_dir, kSavedModelAssetsDirectory, asset_file_def.filename())); inputs->push_back(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 04:36:00 UTC 2024 - 23K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h
const absl::flat_hash_map<std::string, std::string>& function_aliases, const std::vector<tensorflow::AssetFileDef>& asset_file_defs); // Sets up and runs the passes for exporting `module_op`. The behavior of the // exporting passes is controlled by `export_opts`. Returns `AssetFileDef`s that // associate the input arguments of @main and the asset file names. Asset file
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 6.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/calibration/component.cc
TF_ASSIGN_OR_RETURN(const SmallVector<AssetFileDef> asset_file_defs, RunExportPasses(export_opts, *ctx_, *cloned_module_ref)); TF_ASSIGN_OR_RETURN(ExportedModel exported_model, ConvertMlirModuleToExportedModel( *cloned_module_ref, checkpoint_dir, function_aliases_, {asset_file_defs.begin(), asset_file_defs.end()}));
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/quantization/tensorflow/exported_model.proto
// Holds information about the asset files used for the model. It essentially // associates asset file names with the tensors to which the asset file names // should be fed. repeated AssetFileDef asset_file_defs = 8; // SaverDef including the information required for saving and restoring // variables. This field is not set if there are no variables in the exported
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 06:12:59 UTC 2023 - 2.1K bytes - Viewed (0)