- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 304 for ModuleOp (0.16 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/cc/debugger.cc
#include "tensorflow/compiler/mlir/quantization/tensorflow/passes/tf_quant_ops.h" namespace stablehlo::quantization { void DisableDebugging(mlir::ModuleOp module_op) { module_op.walk( [](mlir::TF::DumpTensorOp dump_op) { dump_op.setEnabled(false); }); } void ChangeToQuantizedFilename(mlir::ModuleOp module_op) { module_op.walk([](mlir::TF::DumpTensorOp dump_op) { dump_op.setFileName("quantized_tensor_data.pb"); }); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 00:17:12 UTC 2024 - 1.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.h
mlir::ModuleOp module, const GraphExportConfig& configs); // Converts an MLIR module to TensorFlow graph and FunctionLibraryDefinition. // The "main" function of the module is stored in the graph and the rest of // functions are stored in the library. Control ret nodes are stored separately // in `control_ret_nodes`. Status ConvertMlirToGraph(mlir::ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc
/*RequiresNullTerminator=*/false); llvm::SourceMgr source_mgr; source_mgr.AddNewSourceBuffer(std::move(mem_buffer), llvm::SMLoc()); OwningOpRef<ModuleOp> module_ref = parseSourceFile<ModuleOp>(source_mgr, module.getContext()); // Inline and optimize loaded functions. MLIRContext* context = &getContext(); PassManager pm(context); pm.addPass(createInlinerPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 8.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc
} llvm::SourceMgr source_mgr; source_mgr.AddNewSourceBuffer(std::move(file), llvm::SMLoc()); return OwningOpRef<mlir::ModuleOp>( parseSourceFile<mlir::ModuleOp>(source_mgr, context)); } TEST(ErrorCollectorTest, TessSuccessPass) { std::string input_file = tensorflow::GetDataDependencyFilepath( "tensorflow/compiler/mlir/lite/metrics/testdata/strided_slice.mlir"); MLIRContext context;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 25 01:48:36 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor_test.cc
mlir_module_ = mlir::parseSourceFile<mlir::ModuleOp>(mlir_module_path, &context_); if (!mlir_module_) { return absl::Status( absl::StatusCode::kNotFound, absl::StrCat("Could not find MLIR module at ", mlir_module_path)); } return absl::OkStatus(); } DialectRegistry registry_; MLIRContext context_; OwningOpRef<mlir::ModuleOp> mlir_module_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
absl::StrCat("Failed to open input file: ", error_message)); } if (input_mlir) { source_mgr->AddNewSourceBuffer(std::move(file), llvm::SMLoc()); return OwningOpRef<ModuleOp>( mlir::parseSourceFile<mlir::ModuleOp>(*source_mgr, context)); } // Register extra TF ops passed as OpDef. auto extra_opdefs_status = RegisterExtraTfOpDefs(extra_tf_opdefs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
// function of type "restore_op" to identify the VarHandleOps that should be // saved using the SaveV2 op. class InsertSaveOpPass : public PassWrapper<InsertSaveOpPass, OperationPass<ModuleOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(InsertSaveOpPass) explicit InsertSaveOpPass() = default; // The argument used to refer to the pass in the textual format (e.g. on the
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/debugger.h
namespace stablehlo::quantization { // Disables debugging on `DumpTensor` ops. void DisableDebugging(mlir::ModuleOp module_op); // Changes the filename from `unquantized_tensor_data.pb` to // `quantized_tensor_data.pb`. void ChangeToQuantizedFilename(mlir::ModuleOp module_op); } // namespace stablehlo::quantization
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 00:17:12 UTC 2024 - 1.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/convert_tpu_model_to_cpu.cc
// Convert a TPU model to be compatible on CPU by rewriting/removing TPU ops. class ConvertTpuModelToCpuPass : public PassWrapper<ConvertTpuModelToCpuPass, OperationPass<ModuleOp>> { public: MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ConvertTpuModelToCpuPass) explicit ConvertTpuModelToCpuPass() = default; StringRef getArgument() const final {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf_test.cc
mlir_module_ = mlir::parseSourceFile<mlir::ModuleOp>(mlir_module_path, &context_); if (!mlir_module_) { return absl::Status( absl::StatusCode::kNotFound, absl::StrCat("Could not find MLIR module at ", mlir_module_path)); } return absl::OkStatus(); } DialectRegistry registry_; MLIRContext context_; OwningOpRef<mlir::ModuleOp> mlir_module_; };
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:44:37 UTC 2024 - 4.1K bytes - Viewed (0)