- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 35 for session_initializer (0.56 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/convert_session_initializer_to_function.mlir
// CHECK-LABEL: simple_initializer // CHECK-NOT: tf_saved_model.session_initializer // CHECK: func @session_initializer // CHECK: call @init1 module @simple_initializer attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init1]} : () -> () func.func @init1() attributes {tf_saved_model.exported_names = ["init1"]} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Dec 02 16:24:00 UTC 2022 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/remove_var_init_by_const.mlir
// the initializer function. // CHECK-LABEL: module module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init_func_restore_op]} : () -> () // CHECK: "tf_saved_model.session_initializer"() // CHECK-SAME: initializers = [@init_func_restore_op] func.func @init_func_restore_op() -> () attributes { tf_saved_model.initializer_type = "restore_op",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_session_initializer_to_function.cc
}; void ConvertSessionInitializerToFunctionPass::runOnOperation() { ModuleOp module = getOperation(); auto session_initializer = tf_saved_model::GetSessionInitializerOp(module); if (!session_initializer) return; OpBuilder builder(session_initializer); const char *name = "session_initializer"; // In the (unlikely) case of there already being a session initializer // function, bail out.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_remove_vars_in_session_initializer.mlir
// expected-error@+1 {{'tf_saved_model.session_initializer' op the initializer function does not exist}} "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> () } // ----- module attributes {tf_saved_model.semantics} { // Test case: Invalid multiple blocks in the initializer funcion. "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 00:22:36 UTC 2023 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_save_op.mlir
// SaveV2 op created for a single VarHandleOp. module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init_func_restore_op]} : () -> () // SessionInitializerOp is untouched. // CHECK: "tf_saved_model.session_initializer"() // CHECK-SAME: {{.*initializers = \[@init_func_restore_op\].*}} func.func @init_func_restore_op() -> () attributes {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_ops_invalid.mlir
} } // ----- module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() { initializer = @init } : () -> () // expected-error@+1 {{there must be no more than one session_initializer op}} "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> () func.func private @init() -> tensor<1xf32> {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 19 13:38:14 UTC 2022 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/merge_initializer_function_ops_to_main.mlir
module attributes {tf.versions = {bad_consumers = [], min_consumer = 12 : i32, producer = 1228 : i32}, tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@NoOp]} : () -> () // Check that the initializers list is empty. // CHECK: "tf_saved_model.session_initializer"() // CHECK-SAME: initializers = [] func.func @NoOp()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 29.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/unfreeze_constants.mlir
// Tests a case with one ConstOp and a tf_saved_model.session_initializer with an empty initializers. module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = []} : () -> () // Check that the init function is created & added to the initializers attribute. // CHECK: "tf_saved_model.session_initializer"() // CHECK-SAME: initializers = [@init_func_restore_op]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 17.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} } SymbolTable symbol_table(module); auto session_initializers = module.getOps<SessionInitializerOp>(); if (!session_initializers.empty()) { if (!llvm::hasSingleElement(session_initializers)) { return (*++session_initializers.begin()).emitError() << "there must be no more than one session_initializer op"; } if (failed(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model_test.cc
} TEST_F(TfSavedModelTest, GetInitializerFunctionReturnsNullWhenInitializersEmpty) { constexpr StringRef kModuleOpStr = R"mlir( module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = []} : () -> () } )mlir"; Block block; ModuleOp module_op = ParseModuleOp(kModuleOpStr, block, ctx_); func::FuncOp init_func_op = GetInitializerFunction(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 01 05:03:09 UTC 2023 - 6.4K bytes - Viewed (0)