- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 753 for module_ (0.53 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/verify_clustering_pass_test.cc
protected: void CreateModule(const char* module_string) { TF_ASSERT_OK_AND_ASSIGN(module_, GetMlirModuleFromString(module_string, &context_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>(CreateVerifyClusteringPass()); } mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: mlir::MLIRContext context_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 17:03:53 UTC 2023 - 2.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/debug/debug_test.cc
return registry; }()) { context_.loadAllAvailableDialects(); mlir::OpBuilder builder(&context_); module_ = builder.create<mlir::ModuleOp>(builder.getUnknownLoc()); builder.setInsertionPointToStart(module_->getBody()); auto func = builder.create<mlir::func::FuncOp>( // builder.getUnknownLoc(), "main", builder.getFunctionType({}, {}));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 11:15:16 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass_test.cc
} mlir::LogicalResult Run() { mlir::OwningOpRef<mlir::ModuleOp> module_before = module_->clone(); LogicalResult run_result = pm_->run(module_.get()); EXPECT_TRUE(ModulesEqual(*module_before, *module_)); return run_result; } private: mlir::MLIRContext context_; mlir::OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 23 19:42:09 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/tac_pass.h
explicit TacPass(const TacModule* module) : OperationPass<T>::OperationPass(mlir::TypeID::get<T>()), module_(module) {} ~TacPass() override = default; const TargetHardware* GetTargetHardware( const std::string& hardware_name) const { return module_ != nullptr ? module_->GetTargetHardware(hardware_name) : mlir::TFL::tac::GetTargetHardware(hardware_name); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 03:08:33 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_targets_test.cc
public: XlaLegalizeTargetsTest() : context_(GetDefaultDialectRegistry()), module_(mlir::ModuleOp::create(mlir::UnknownLoc::get(&context_))), builder_(&module_->getBodyRegion()) { context_.loadAllAvailableDialects(); } protected: mlir::MLIRContext context_; mlir::OwningOpRef<mlir::ModuleOp> module_; mlir::OpBuilder builder_; }; TEST_F(XlaLegalizeTargetsTest, CreatesConversionTargets) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 05:56:39 UTC 2023 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/inference/inference_metrics_pass_test.cc
TF_ASSERT_OK_AND_ASSIGN(module_, GetMlirModuleFromString(module_string, &context_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addPass(CreateInferenceMetricsPass()); } mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: MLIRContext context_; OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/transforms/target_annotation.cc
TargetAnnotationPass(const TargetAnnotationPass& copy) : TacFunctionPass(copy.module_) {} explicit TargetAnnotationPass(llvm::ArrayRef<std::string> device_specs) : TacFunctionPass(nullptr) { device_specs_flag_ = device_specs; } explicit TargetAnnotationPass(const TacModule* module) : TacFunctionPass(module) {} private: void runOnFunction() override; void SetTargetAnnotation(Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 19 19:32:06 UTC 2023 - 5.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass_test.cc
} mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: MLIRContext context_; OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; }; TEST_F(VerifyNoOutsideCompilationMarkersPassTest, PassesValidOps) { static constexpr char kMlirModuleStr[] = R"( module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 268 : i32}} {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_types_test.cc
tensorflow::DeserializeMlirModule(module_string, &context_, &module_)); pm_ = std::make_unique<mlir::PassManager>(&context_); pm_->addNestedPass<mlir::func::FuncOp>( quant::stablehlo::CreateConvertTFQuantTypesPass()); } mlir::LogicalResult Run() { return pm_->run(module_.get()); } private: MLIRContext context_; OwningOpRef<ModuleOp> module_; std::unique_ptr<mlir::PassManager> pm_; };
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/tf2xla/transforms/legalization_op_config_test.cc
} return main; } protected: MLIRContext context_; OwningOpRef<ModuleOp> module_; }; TEST_F(LegalizationOpConfigTest, FailsWithExpectsLegalizationWithMlir) { TF_EXPECT_OK(CreateMlirModule()); EXPECT_FALSE(IsOpLegalizedWithMlir(*module_->getOperation())); } TEST_F(LegalizationOpConfigTest, ExpectsFalseForNonMlirOps) { TF_EXPECT_OK(CreateMlirModule()); TF_ASSERT_OK_AND_ASSIGN(FuncOp main, GetMain());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 30 03:31:01 UTC 2024 - 8.4K bytes - Viewed (0)