- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 15 for exported_names (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model_freeze_global_tensors.mlir
// CHECK: func @f() func.func @f(%arg0: tensor<!tf_type.resource<tensor<f32>>> {tf_saved_model.bound_input = @v}) attributes {tf_saved_model.exported_names = ["f"]} { %val = "tf.ReadVariableOp"(%arg0) : (tensor<!tf_type.resource<tensor<f32>>>) -> tensor<f32> // CHECK: "tf.Const"() <{value = dense<1.000000e+00> : tensor<f32>}> func.return } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 03:07:35 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.h
absl::Span<std::string> exported_names, MLIRImportOptions options = {}); // Given a V1 SavedModel, returns a MLIR module containing the functions, // expressed with tf_executor dialect. absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> ConvertSavedModelV1ToMlir( const SavedModelBundle& saved_model, absl::Span<std::string> exported_names, mlir::MLIRContext* context, MLIRImportOptions options = {});
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/lite/tests/insert_call_once_op.mlir
module attributes {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"]} { %cst = arith.constant dense<[1, 2, 3, 4]> : tensor<4xi64> %cst_0 = arith.constant dense<["a", "b", "c", "d"]> : tensor<4x!tf_type.string>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tests/legalize-tf-variables.mlir
%handle_0 = "tf.VarHandleOp"() {container="c", shared_name="a"} : () -> tensor<!tf_type.resource<tensor<1x10xf32>>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 7.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/mlprogram.mlir
// CHECK-LABEL: func @lowers_to_stablehlo func.func @lowers_to_stablehlo(%arg0: tensor<i32> {tf_saved_model.index_path = []}) -> (tensor<i32> {tf_saved_model.index_path = []}) attributes {tf_saved_model.exported_names = ["lowers_to_stablehlo"]} { // CHECK-DAG: [[one:%.*]] = stablehlo.constant dense<1> // CHECK-DAG: [[twenty:%.*]] = stablehlo.constant dense<20> // CHECK-DAG: [[r3:%.*]] = stablehlo.subtract [[twenty]], %arg0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 19:27:16 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.h
absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> SavedModelObjectGraphToMlirImport( absl::string_view saved_model_dir, const std::unordered_set<std::string>& tags, absl::Span<std::string> exported_names, mlir::MLIRContext* context, bool unconditionally_use_set_output_shapes = false); // Converts a TensorFlow V1 SavedModel stored in the directory with the given
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/stablehlo/quantization.cc
const absl::flat_hash_map<std::string, SignatureDef> signature_def_map = GetSignatureDefMapFromBundle(*saved_model_bundle); std::vector<std::string> exported_names; for (const auto& [key, value_unused] : signature_def_map) { exported_names.push_back(key); } if (failed(mlir::tf_saved_model::FreezeVariables( module_op, saved_model_bundle->GetSession()))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 10:49:12 UTC 2024 - 7.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/basic.mlir
// CHECK-SAME: {device = "/device:CPU:0", index = 0 : i64} // CHECK-LABEL: func @init // CHECK-SAME: {tfrt.cost_threshold = 1 : i64} func.func @func_init() attributes {tf_saved_model.exported_names = ["init"]} { func.return } // CHECK-LABEL: func @basic // CHECK-SAME: ([[in_chain:%.*]]: !tfrt.chain // CHECK-SAME: [[arg0:%.*]]: !tfrt_fallback.tf_tensor,
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/lite/stablehlo/tests/odml-to-stablehlo-allow-tf.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 18:33:43 UTC 2024 - 914 bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import.cc
// Copy to eliminate the `const` qualifier so that `absl::MakeSpan` can be // called on it. std::vector<std::string> exported_names = signature_keys; absl::StatusOr<OwningOpRef<ModuleOp>> module_op = SavedModelSignatureDefsToMlirImport(saved_model_path, tags, absl::MakeSpan(exported_names), &ctx, import_options, &bundle); if (!module_op.status().ok()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 6.4K bytes - Viewed (0)