- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 435 for ModuleOp (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables_test_pass.cc
public: void runOnOperation() override { ModuleOp module = getOperation(); // Pass an invalid session argument, which is a nullptr. if (failed(tf_saved_model::LiftVariables(module, /*session=*/nullptr))) signalPassFailure(); } }; } // namespace } // namespace tf_saved_model namespace tf_test { std::unique_ptr<OperationPass<ModuleOp>> CreateLiftVariablesTestPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 04 09:19:38 UTC 2022 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_serialization.cc
} // Creates a pruned module containing the XlaCallModule's entry function and // other functions transitively called by the entry function. FailureOr<OwningOpRef<ModuleOp>> PruneStablehloModule( SymbolTableCollection& symbol_table, ModuleOp module, XlaCallModuleOp op) { auto entry_func_symbol = op->getAttrOfType<FlatSymbolRefAttr>(kStablehloEntryFunctionAttrName); if (!entry_func_symbol) {
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/lite/quantization/stablehlo/quantization.h
// `module_op` is the input graph to be quantized and it should contain // StableHLO ops. // // Returns a quantized `ModuleOp` in StableHLO, potentially wrapped inside a // XlaCallModuleOp. Returns a non-OK status if quantization fails, or any of // `saved_model_bundle` or `quantization_py_function_lib` is a nullptr. absl::StatusOr<mlir::ModuleOp> RunQuantization( const SavedModelBundle* saved_model_bundle, absl::string_view saved_model_dir,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 02:44:03 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h
std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateExtractOutsideCompilationPass(); // Create a pass that encapsulates StatefulPartitionedCallOp within a cluster. std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateXlaClusterFormationPass(); // Creates a pass that marks unsupported ops in device cluster for outside // compilation. std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateMarkOpsForOutsideCompilationPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_import_test.cc
using UpdateFunctionAliasesTest = ::mlir::quant::QuantizationTestBase; TEST_F(UpdateFunctionAliasesTest, NoAliasesReturnsEmptyMap) { // MLIR @main function corresponds to the TF function "main_original". OwningOpRef<ModuleOp> module_op = ParseModuleOpString(R"mlir( func.func private @main(%arg: tensor<1x2xf32>) -> (tensor<1x2xf32>) attributes {tf._original_func_name = "main_original"} { return %arg : tensor<1x2xf32> } )mlir");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 07 03:47:17 UTC 2024 - 4.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/input_lowering_metrics_pass_test.cc
} bool ModulesEqual(const ModuleOp& module_before, const ModuleOp& module_after) { return mlir::OperationEquivalence::isEquivalentTo( module_before, module_after, mlir::OperationEquivalence::None); } mlir::LogicalResult Run() { mlir::OwningOpRef<mlir::ModuleOp> module_before = module_->clone(); LogicalResult run_result = pm_->run(module_.get());
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/tensorflow/utils/dump_mlir_util_test.cc
mlir::OwningOpRef<mlir::ModuleOp> module_ref = mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); unsetenv("TF_DUMP_GRAPH_PREFIX"); std::string filepath = DumpMlirOpToFile("module", module_ref.get()); EXPECT_EQ(filepath, "(TF_DUMP_GRAPH_PREFIX not specified)"); } TEST(DumpMlirModuleTest, LogInfo) { mlir::MLIRContext context; mlir::OwningOpRef<mlir::ModuleOp> module_ref =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 18 13:40:21 UTC 2024 - 5.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h
std::unique_ptr<OperationPass<ModuleOp>> CreateTransposeCommuteOpsPass(); // Create a pass that legalizes MHLO to TF dialect. std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeHloToTfPass(); // Creates a pass which replaces a splat constant tensor with a BroadcastInDim // op. std::unique_ptr<OperationPass<ModuleOp>> CreateUnfoldSplatConstantPass();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 21:59:06 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/func_test.cc
TEST_F(FindMainFuncOpTest, ReturnsMainFuncOp) { constexpr absl::string_view kModuleWithMainFunc = R"mlir( module { func.func @main() -> () { return } } )mlir"; OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kModuleWithMainFunc); EXPECT_THAT(*module_op, NotNull()); func::FuncOp main_func_op = FindMainFuncOp(*module_op); EXPECT_THAT(main_func_op, NotNull()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 09:05:02 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_savedmodel_passes.td
]; } def OptimizeGlobalTensorsPass : Pass<"tf-saved-model-optimize-global-tensors", "ModuleOp"> { let summary = "Optimize tf_saved_model.global_tensor's."; let constructor = "mlir::tf_saved_model::CreateOptimizeGlobalTensorsPass()"; } def RemoveVariablesInSessionInitializerPass : Pass<"tf-saved-model-remove-vars-in-session-initializer", "ModuleOp"> { let summary = "Remove variables in tf saved model's session initializer.";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 09 19:11:34 UTC 2023 - 7.3K bytes - Viewed (0)