- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for loadAllAvailableDialects (0.39 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/tf2xla_rewriter_test.cc
} Status CreateMlirModule(std::string module_string = kMlirModuleStr) { TF_ASSIGN_OR_RETURN( module_, test::GetMlirModuleFromString(module_string, &context_)); context_.loadAllAvailableDialects(); return absl::OkStatus(); } Status LegalizeSingleOp(Operation& op) { SourceMgrDiagnosticHandler sourceMgrHandler(source_manager_, &context_); OpBuilder op_builder(&op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 11.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc
registry.insert<mlir::func::FuncDialect, mlir::tf_type::TFTypeDialect, mlir::quant::QuantizationDialect>(); mlir::quant::RegisterOps(); MLIRContext context(registry); context.loadAllAvailableDialects(); llvm::SourceMgr source_mgr; SourceMgrDiagnosticHandler sourceMgrHandler(source_mgr, &context); auto verbose_dir = llvm::sys::path::parent_path(output_path).str();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/c/c_api_unified_experimental_mlir.cc
namespace { void RegisterDialects(mlir::MLIRContext& ctx) { mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); ctx.appendDialectRegistry(registry); ctx.loadAllAvailableDialects(); } Status ConvertDataTypeToTensor(tensorflow::DataType dtype, Builder builder, Type* type) { Status s = tensorflow::ConvertDataType(dtype, builder, type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
dialects.insert<TF::TensorFlowDialect, func::FuncDialect, chlo::ChloDialect, mhlo::MhloDialect, quant::QuantizationDialect>(); ctx_ = std::make_unique<MLIRContext>(dialects); ctx_->loadAllAvailableDialects(); // Create a CPU client with 1 device. TF_ASSERT_OK_AND_ASSIGN( pjrt_client_, xla::GetTfrtCpuClient(/*asynchronous=*/false, /*cpu_device_count=*/1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0)