- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 35 for session_initializer (0.7 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/rename_entrypoint_to_main.cc
entrypoints[op.getSymName()] = op; }); if (auto session_initializer = tf_saved_model::GetSessionInitializerOp(module)) { // clang-format off // Skip session initializer functions which are present in saved models. // For example: // "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> () // NOLINT
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_restore_op.mlir
// RUN: -quant-insert-restore-op | FileCheck %s --check-prefix CHECK-LOC // RestoreV2 op created for a single VarHandleOp. module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {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 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables_test.cc
}; TEST_F(SaveVariablesToCheckpointTest, VariableSavedToCheckpoint) { constexpr absl::string_view kModuleCode = R"mlir( module attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init_func_restore_op]} : () -> () func.func @init_func_restore_op() -> () attributes {tf_saved_model.exported_names = ["restore"], tf_saved_model.initializer_type = "restore_op"} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 16.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_ops.mlir
// RUN: tf-opt %s | tf-opt | FileCheck %s module attributes {tf_saved_model.semantics} { // CHECK: tf_saved_model.session_initializer "tf_saved_model.session_initializer"() { initializers = [@init] } : () -> () // CHECK: tf_saved_model.asset "tf_saved_model.asset"() { filename = "asset_filename", sym_name = "asset_sym_name" } : () -> () // Representation for constants: (immutable) global tensor.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 28 12:06:33 UTC 2022 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_main_function.mlir
"tf_saved_model.session_initializer"() {initializers = [@NoOp]} : () -> () "tf_saved_model.asset"() {filename = "assets/mydata.txt", sym_name = "__tf_saved_model_asset0_mydata.txt"} : () -> () // Session initializer ops and asset ops untouched. // CHECK: "tf_saved_model.session_initializer"() <{initializers = [@NoOp]}> : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:49:35 UTC 2023 - 15.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/unfreeze_constants.cc
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. SessionInitializerOp GetOrCreateSessionInitializerOp(ModuleOp module_op) { SessionInitializerOp session_init_op = GetSessionInitializerOp(module_op);
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/tfrt/tests/tf_to_corert/basic.mlir
// CHECK: tfrt.cost_threshold = 1024 // CHECK-SAME: tfrt.merge_inter_dependent_streams = true module attributes {tf_saved_model.semantics} { // CHECK-NOT: "tf_saved_model.session_initializer" "tf_saved_model.session_initializer"() { initializers = [@func_init] } : () -> () // CHECK-LABEL: _tfrt_resource_init // CHECK: tf.VarHandleOp // CHECK: tf.ReadVariableOp // CHECK: tfrt_fallback_async.set_resource
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 00:18:59 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_asset_sinking.mlir
// RUN: tf-opt %s -split-input-file -tf-saved-model-asset-sinking='saved-model-dir=foo/bar' | FileCheck %s // CHECK-LABEL: module @asset module @asset attributes {tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init]} : () -> () // CHECK-NOT: "tf_saved_model.asset" "tf_saved_model.asset"() {filename = "assets/test0.txt", sym_name = "asset0"} : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_initialize_variables_in_session_init.mlir
} // ----- // Test with an existing session init function. module attributes {tf_saved_model.semantics, tf_saved_model.under_construction} { "tf_saved_model.session_initializer"() {initializers = [@Init]} : () -> () func.func @serving_default(%arg0: tensor<!tf_type.resource<tensor<100x50xf32>>> {tf.resource_name = "dense/kernel"}) -> (tensor<100x50xf32> {tf_saved_model.index_path = ["dense_2"]})
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 19 13:38:14 UTC 2022 - 5.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/tests/lift_hashtable_ops_as_args.mlir
module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1506 : i32}, tf_saved_model.semantics} { "tf_saved_model.session_initializer"() {initializers = [@init_all_tables]} : () -> () func.func @init_all_tables() attributes {tf_saved_model.exported_names = ["__tf_saved_model_session_initializer_init_all_tables"], tf_saved_model.initializer_type = "init_op"} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 15 05:41:44 UTC 2024 - 13.5K bytes - Viewed (0)