- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for kImportModelDefaultGraphFuncName (0.4 sec)
-
tensorflow/compiler/mlir/quantization/common/func.cc
#include "tensorflow/compiler/mlir/tensorflow/translate/import_model.h" namespace mlir::quant { namespace { using ::tensorflow::kDefaultServingSignatureDefKey; using ::tensorflow::kImportModelDefaultGraphFuncName; // Returns true iff the function's symbol is public. bool IsPublicFuncOp(func::FuncOp func_op) { return SymbolTable::getSymbolVisibility(&*func_op) == SymbolTable::Visibility::Public; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 19 06:55:11 UTC 2024 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_main_function.cc
namespace mlir { namespace quant { namespace { using ::mlir::tf_saved_model::kTfSavedModelExportedNamesAttr; using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr; using ::tensorflow::kImportModelDefaultGraphFuncName; constexpr StringRef kEntryFunctionAttr = "tf.entry_function"; // The ConvertMlirToGraphdef requires the provided input module to have a main
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/tensorflow/passes/merge_save_function_ops_to_main.cc
namespace { using ::mlir::tf_executor::FetchOp; using ::mlir::tf_executor::GraphOp; using ::mlir::tf_executor::IslandOp; using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr; using ::tensorflow::kImportModelDefaultGraphFuncName; class MergeSaveFunctionOpsToMainPass : public PassWrapper<MergeSaveFunctionOpsToMainPass, OperationPass<ModuleOp>> { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tfg-to-tfe.cc
#include "tensorflow/core/util/device_name_utils.h" namespace mlir { namespace TF { namespace { // FIXME: This should be consistent with // tensorflow::kImportModelDefaultGraphFuncName static const char kImportModelDefaultGraphFuncName[] = "main"; // Please refer to the TFG dialect description for the list of used attributes. // Belows are the attributes in TFE. // TFE Arguments and Results (Got from "_Arg",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.h
#include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/framework/graph_debug_info.pb.h" #include "tensorflow/core/graph/graph.h" namespace tensorflow { inline constexpr absl::string_view kImportModelDefaultGraphFuncName = "main"; // Given a GraphDef, returns a MLIR module containing the graph, expressed with // tf_executor dialect. absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ConvertGraphdefToMlir(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
} // Checks if the function is the main or initializer function. bool IsMainOrInitializerFunction(ModuleOp module, func::FuncOp func) { if (func.getSymName() == llvm::StringRef(tensorflow::kImportModelDefaultGraphFuncName) || func.getSymName() == kTfQuantSaveFuncName) { return true; } for (func::FuncOp init_func : tf_saved_model::GetInitializerFunctions(module)) {
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/tensorflow/translate/import_model.cc
} } // Record version info. PopulateTfVersions(module.get(), graph.versions()); const llvm::StringRef& graph_func_name = specs.graph_func_name.empty() ? kImportModelDefaultGraphFuncName : specs.graph_func_name; TF_RETURN_IF_ERROR(importer.ImporterBase::Convert(graph_func_name, func_type, arg_nodes, ret_nodes,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)