Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for AssetOp (0.1 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc

      builder.create<tf_saved_model::SessionInitializerOp>(
          module->getLoc(), builder.getArrayAttr(func_names));
      // Create AssetOp; this holds the checkpoint_path.
      // TODO(b/318761632): Cleanup usage of string literals, instead use constants.
      auto asset_op = builder.create<tf_saved_model::AssetOp>(
          module->getLoc(),
          /*sym_name=*/
          builder.getStringAttr("__tf_saved_model_variables"),  // Val unimportant.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args.cc

    #include "tensorflow/compiler/mlir/tensorflow/translate/import_model.h"
    #include "tensorflow/core/protobuf/meta_graph.pb.h"
    
    namespace mlir::quant {
    namespace {
    
    using ::mlir::tf_saved_model::AssetOp;
    using ::mlir::tf_saved_model::kTfSavedModelIndexPathAttr;
    using ::mlir::tf_saved_model::LookupBoundInputOfType;
    using ::tensorflow::AssetFileDef;
    
    // Given argument attributes `arg_attrs`, returns a new set of argument
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc

      EXPECT_THAT(index_path, NotNull());
      EXPECT_THAT(index_path, Eq("arg_0:0"));
    }
    
    TEST_F(ConvertAssetArgsTest, ArgsBoundedToGlobalTensorNotModified) {
      // If the argument is not bound to AssetOp, it is not modified.
      OwningOpRef<ModuleOp> module_op = ParseModuleOpString(R"mlir(
        module {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top