Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for serialize_portable_artifact (0.36 sec)

  1. tensorflow/compiler/mlir/stablehlo/stablehlo_test.py

            return %1 : tensor<i1>
          }
        }
      """
      target = stablehlo.get_current_version()
      artifact = stablehlo.serialize_portable_artifact(assembly, target)
      deserialized = stablehlo.deserialize_portable_artifact(artifact)
      rountrip = stablehlo.serialize_portable_artifact(deserialized, target)
      assert artifact == rountrip
    
    
    if __name__ == "__main__":
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 16 19:48:21 UTC 2023
    - 1.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_xla_call_module_op_to_bfloat16.cc

        return absl::InternalError(
            "Failed to convert StableHLO module to bfloat16.");
      }
    
      std::string bytecode;
      llvm::raw_string_ostream os(bytecode);
      if (failed(mlir::stablehlo::serializePortableArtifact(
              stablehlo_module_op.get(), mlir::stablehlo::getCurrentVersion(),
              os))) {
        return absl::InternalError("Failed to serialize StableHLO module.");
      }
      return bytecode;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 05 08:32:43 UTC 2024
    - 5.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc

    // Serializes the stablehlo module into bytecode.
    FailureOr<std::string> SerializeStablehlo(ModuleOp stablehlo_module) {
      std::string bytecode;
      llvm::raw_string_ostream os(bytecode);
      if (mlir::failed(stablehlo::serializePortableArtifact(
              stablehlo_module, stablehlo::getCurrentVersion(), os))) {
        return stablehlo_module.emitError()
               << "failed to serialize the pruned stablehlo module";
      }
      return bytecode;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/python/integration_test/quantize_model_test_base.py

                # Serialization in VHLO dialect.
                serialized = node_def.attr.get('module').s
                # MLIR bytecode matching StableHLO version.
                mlir_bytecode = stablehlo.deserialize_portable_artifact(serialized)
                stablehlo_module = ir.Module.parse(mlir_bytecode, context=context)
                return str(stablehlo_module)
        raise ValueError('No XlaCallModule found in saved model.')
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 18.2K bytes
    - Viewed (0)
Back to top