Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 31 for tf_saved_model (0.27 sec)

  1. tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc

      }
    
      return success();
    }
    
    }  // namespace tf_saved_model
    }  // namespace mlir
    
    #define GET_OP_CLASSES
    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc.inc"
    
    namespace mlir {
    namespace tf_saved_model {
    
    //===----------------------------------------------------------------------===//
    // TensorFlowSavedModelDialect Dialect
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export_test.cc

        module attributes {tf_saved_model.semantics} {
          "tf_saved_model.session_initializer"() <{initializers = []}> : () -> ()
          "tf_saved_model.asset"() <{filename = "assets/vocab_file.txt", sym_name = "__tf_saved_model_asset0_vocab_file.txt"}> : () -> ()
    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/tfrt/tests/hoist_invariant_ops.mlir

    // CHECK: func @test_hoist_hash_table
    func.func @hoist_hash_table(%arg: tensor<?x!tf_type.string> {tf_saved_model.index_path = ["input"]}, %default: tensor<i64> {tf_saved_model.index_path = ["default"]}) -> (tensor<i64> {tf_saved_model.index_path = ["r"]}, tensor<*xi64> {tf_saved_model.index_path = ["r1"]})
      attributes {tf_saved_model.exported_names = ["test_hoist_hash_table"]} {
      // CHECK-NOT: tf.HashTableV2
      // CHECK-NOT: tf.LookupTableSizeV2
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 01 23:54:14 UTC 2024
    - 18.3K bytes
    - Viewed (0)
  4. 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)
  5. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/replace_stablehlo_ops_in_main_function_with_xla_call_module_ops.mlir

      func.func @serving_default(%arg0: tensor<1024x1024xf32> {tf_saved_model.index_path = ["input_tensor"]}) -> (tensor<1024x3xf32> {tf_saved_model.index_path = ["output1"]}, tensor<1024x3xf32> {tf_saved_model.index_path = ["output2"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input_tensor:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 39.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc

    //   2) The `tf_saved_model` dialect requires that the arguments with
    //   `tf_saved_model.index_path` attributes should precede those with
    //   `tf_saved_model.bound_input` attributes. The init function of type
    //   `kTfSavedModelInitializerRestoreType` usually has an argument with
    //   `tf_saved_model.index_path`, whereas the init function of type
    //   `kTfSavedModelInitializerInitType` may have arguments with
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:54:52 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc

    #include "tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.h"
    #include "tensorflow/compiler/mlir/tensorflow/translate/import_model.h"
    
    namespace mlir {
    namespace quant {
    namespace {
    
    using ::mlir::tf_executor::FetchOp;
    using ::mlir::tf_executor::GraphOp;
    using ::mlir::tf_executor::IslandOp;
    using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr;
    using ::tensorflow::kImportModelDefaultGraphFuncName;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.7K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/tests/passes/insert_calibration_statistics_saver.mlir

      func.func @main(%arg0: tensor<10x1x1024xf32> {tf_saved_model.index_path = ["input_tensor"]}) -> (tensor<10x1x3xf32> {tf_saved_model.index_path = ["output"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_input_tensor:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 01:09:50 UTC 2024
    - 24.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/tests/insert_custom_aggregation_ops.mlir

      func.func @serving_default(%arg0: tensor<1x4xf32> {tf_saved_model.index_path = ["x"]}) -> (tensor<1x3xf32> {tf_saved_model.index_path = ["output"]}) attributes {tf.entry_function = {control_outputs = "", inputs = "serving_default_x:0", outputs = "PartitionedCall:0"}, tf_saved_model.exported_names = ["serving_default"]} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 32.1K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc

    #include "tsl/platform/errors.h"
    #include "tsl/platform/statusor.h"
    
    namespace mlir::quant::stablehlo {
    namespace {
    
    using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr;
    using ::mlir::tf_saved_model::kTfSavedModelInitializerInitType;
    using ::mlir::tf_saved_model::kTfSavedModelInitializerRestoreType;
    using ::stablehlo::quantization::QuantizationConfig;
    using ::stablehlo::quantization::io::GetLocalTmpFileName;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 12.3K bytes
    - Viewed (0)
Back to top