- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for setInitializersAttr (0.63 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
llvm::SmallVector<Attribute> initializers_attrs{prev_initializers.begin(), prev_initializers.end()}; initializers_attrs.emplace_back(symbol); session_init_op.setInitializersAttr( ArrayAttr::get(session_init_op.getContext(), initializers_attrs)); } // Returns the session_initializer op in the module if exists. Otherwise, // creates a new session_initializer op and returns it.
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/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
symbol_table.erase(init_func_op); } // Empties the "initializers" attribute from the `SessionInitializerOp` since // all ops of the initializer ops are cloned into @main. session_init_op.setInitializersAttr(ArrayAttr::get(ctx, {})); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateMergeInitializerFunctionOpsToMainPass() { return std::make_unique<MergeInitializerFunctionOpsToMainPass>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0)