Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for function_aliases_ (0.41 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/quantize_model.cc

          CreateMlirContextForQuantization();
    
      absl::StatusOr<absl::flat_hash_map<FunctionName, FunctionAlias>>
          function_aliases = GetFunctionAliases(saved_model_path, tags);
      if (!function_aliases.ok()) {
        return absl::InternalError(absl::StrCat(
            "Failed to get function alias: ", function_aliases.status().message()));
      }
    
      absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> module =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py

        """Gets the function aliases in the output saved model."""
        loader = loader_impl.SavedModelLoader(output_saved_model_path)
        return loader.get_meta_graph_def_from_tags(
            tags
        ).meta_info_def.function_aliases
    
      def _create_matmul_model(
          self,
          input_shape: Sequence[int],
          weight_shape: Sequence[int],
          saved_model_path: str,
          bias_fn: Optional[ops.Operation] = None,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 18.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/python/py_function_lib.py

          signature_def_map,
          tags,
          init_op_name=exported_model.init_node_name,
          saver_def=_get_saver_def_or_none(exported_model),
          checkpoint_dir=exported_model.checkpoint_dir,
          function_aliases=exported_model.function_aliases,
          asset_file_defs=exported_model.asset_file_defs,
      )
    
      _copy_assets(src_saved_model_path, dst_saved_model_path)
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 31 05:32:11 UTC 2024
    - 27.4K bytes
    - Viewed (0)
Back to top