Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for filename_tensor_name (0.2 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc

      const std::string filename_tensor_name = FindFilePrefixTensorName(graph_def);
      const std::string restore_op_name =
          GetNodeName(control_ret_node_names, kTfSavedModelInitializerRestoreType);
      const std::string save_node_name =
          GetNodeName(control_ret_node_names, kTfQuantSaveOpName);
    
      const std::vector<absl::string_view> fields = {
          filename_tensor_name, restore_op_name, save_node_name};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 12.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc

      ASSERT_TRUE(TextFormat::ParseFromString(
          R"pb(filename_tensor_name: "my_file")pb", &saver_def));
    
      const ExportedModel exported_model = CreateExportedModelFromGraphDef(
          GraphDef(), /*init_node_name=*/"", /*checkpoint_dir=*/"", saver_def,
          /*function_aliases=*/{}, /*asset_file_defs=*/{});
      EXPECT_THAT(exported_model.saver_def().filename_tensor_name(), "my_file");
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 19.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h

    // Creates a new `SaverDef` instance, which contains information regarding
    // checkpoint saving and restoring. This function returns a `SaverDef` instance
    // with four fields populated: `version`, `filename_tensor_name`,
    // `restore_op_name` and `save_tensor_name`. For valid quantized `graph_def` and
    // `control_ret_node_names`, it should be able to retrieve the last three fields
    // if there is at lest one variable in the graph.
    //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  4. tensorflow/cc/saved_model/loader.cc

        TF_RETURN_IF_ERROR(RunRestore(run_options, export_dir,
                                      meta_graph.saver_def().restore_op_name(),
                                      meta_graph.saver_def().filename_tensor_name(),
                                      asset_file_defs, session->get()));
      }
      // Record walltime spent in restoring graph from disk, but postpone metric
      // increments until graph init finishes.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 02 04:36:00 UTC 2024
    - 23K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                                               kSavedModelVariablesFilename)));
        variable_and_assets.push_back(
            {input_.meta_graph_def().saver_def().filename_tensor_name(),
             variable_filename_op});
        variable_and_assets.insert(variable_and_assets.end(), assets.begin(),
                                   assets.end());
    
        const auto& restore_op_name =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
Back to top