- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for serialize_portable_artifact (0.36 sec)
-
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) -
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) -
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) -
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)