- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for DeserializeMlirModule (0.22 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/bridge_logger_test.cc
mlir::MLIRContext mlir_context(mlir_registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module_with_add, mlir_module_with_sub; TF_ASSERT_OK(DeserializeMlirModule(module_with_add, &mlir_context, &mlir_module_with_add)); TF_ASSERT_OK(DeserializeMlirModule(module_with_sub, &mlir_context, &mlir_module_with_sub));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/serialize_mlir_module_utils.h
std::string SerializeMlirModule(mlir::ModuleOp module_op); // Parses a MLIR module from `mlir_module_string` into `mlir_module` with // context `mlir_context`. Status DeserializeMlirModule(llvm::StringRef serialized_mlir_module, mlir::MLIRContext* mlir_context, mlir::OwningOpRef<mlir::ModuleOp>* mlir_module); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 22 14:25:57 UTC 2022 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/test_utils.cc
DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(module_string, context, &mlir_module); if (!status.ok()) { return status; } return mlir_module; } } // namespace test } // namespace mhlo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/serialize_mlir_module_utils.cc
llvm::raw_string_ostream os(serialized_mlir_module); mlir::OpPrintingFlags print_flags; print_flags.enableDebugInfo(); module_op.print(os, print_flags); return std::move(os.str()); } Status DeserializeMlirModule(llvm::StringRef serialized_mlir_module, mlir::MLIRContext* mlir_context, mlir::OwningOpRef<mlir::ModuleOp>* mlir_module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_remaining_ops.cc
StringRef host_module = op.getHostMlirModule(); if (host_module.empty()) return success(); mlir::OwningOpRef<mlir::ModuleOp> module_for_func; tensorflow::Status status = tensorflow::DeserializeMlirModule( host_module.str(), op->getContext(), &module_for_func); if (!status.ok()) { return op.emitError() << "attribute 'host_mlir_module' can not be deserialized. " << status.message();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 14 20:05:58 UTC 2023 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/data_dumper_logger_config_test.cc
mlir::DialectRegistry mlir_registry; mlir::RegisterAllTensorFlowDialects(mlir_registry); mlir::MLIRContext mlir_context(mlir_registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module_with_add; TF_ASSERT_OK(DeserializeMlirModule(module_with_add, &mlir_context, &mlir_module_with_add)); std::unique_ptr<mlir::Pass> partitioning_pass = mlir::TFTPU::CreateTPUResourceReadsWritesPartitioningPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 31 00:41:24 UTC 2023 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/utils/test_metadata_config.cc
mlir::DialectRegistry registry; mlir::RegisterCommonToolingDialects(registry); mlir::MLIRContext context(registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; TF_RETURN_IF_ERROR( DeserializeMlirModule(mlir_module_str, &context, &mlir_module)); TF_RETURN_IF_ERROR(SetupReturnValues(*mlir_module, metadata_proto)); TF_RETURN_IF_ERROR(SetupArguments(*mlir_module, arg_shapes, metadata_proto));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 23:59:33 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_test.cc
mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); OwningOpRef<ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(module_string, context, &mlir_module); if (!status.ok()) { return status; } return mlir_module; } bool BuildAndRunPipeline(absl::string_view module_string,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types_test.cc
DialectRegistry mlir_registry; RegisterCommonToolingDialects(mlir_registry); context_.appendDialectRegistry(mlir_registry); TF_ASSERT_OK( tensorflow::DeserializeMlirModule(module_string, &context_, &module_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>( quant::stablehlo::CreateConvertTFQuantTypesPass()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; auto status = tensorflow::DeserializeMlirModule(string, context, &mlir_module); if (!status.ok()) { return status; } return mlir_module; } TEST(XlaRewriteUtilTest, TestEraseClusterFuncs) { static const char* const module_str = R"(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0)