- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for init_ops (0.08 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
ModuleOp module = getOperation(); if (!HasTfSavedModelSemantics(module)) { return; } auto init_op = GetSessionInitializerOp(module); if (init_op == nullptr || init_op.getInitializers().empty()) { return; } SymbolTable symbol_table(module); for (auto initializer : init_op.getInitializers()) { auto func = symbol_table.lookup<func::FuncOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/cc/saved_model/loader_util.cc
namespace tensorflow { namespace internal { // A SavedModel may store the name of the initialization op to run in the // in the SignatureDef (v2) or a collection (v1). If an init_op collection // exists, then the collection must contain exactly one op. Status GetInitOp(const string& export_dir, const MetaGraphDef& meta_graph_def, string* init_op_name) {
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/tensorflow/ir/tf_saved_model.h
inline constexpr StringRef kTfSavedModelInitializerRestoreType = "restore_op"; // Indicates that the initializer corresponds to the init op. inline constexpr StringRef kTfSavedModelInitializerInitType = "init_op"; class TensorFlowSavedModelDialect : public Dialect { public: explicit TensorFlowSavedModelDialect(MLIRContext *context); LogicalResult verifyRegionArgAttribute(Operation *op, unsigned region_index,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 18 03:21:34 UTC 2024 - 5.1K bytes - Viewed (0)