- Sort Score
- Result 10 results
- Languages All
Results 71 - 80 of 435 for ModuleOp (0.13 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_cluster_util.cc
mlir::LogicalResult WalkReachableFromTpuCluster( ModuleOp module, std::function<WalkResult(Operation*, tf_device::ClusterOp, std::optional<std::string>)> callback) { return WalkReachableFromTpuCluster(true, module, callback); } mlir::LogicalResult WalkReachableFromTpuCluster( ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 06 04:50:13 UTC 2023 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/rewrite_quantized_io.cc
public: StringRef getArgument() const final { return "tfr-rewrite-quantized-io"; } StringRef getDescription() const final { return "Replaces operands and results that has quantized type with their " "storage types."; } void runOnOperation() override; }; void RewriteQuantizedIOPass::runOnOperation() { ModuleOp module = getOperation();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_tf_xla_call_module_to_stablehlo_pass.cc
: OpRewritePattern<TF::XlaCallModuleOp>(context), module_op_(module_op) {} using OpRewritePattern<TF::XlaCallModuleOp>::OpRewritePattern; private: ModuleOp module_op_; LogicalResult matchAndRewrite(TF::XlaCallModuleOp op, PatternRewriter &rewriter) const override { OwningOpRef<ModuleOp> stablehlo_module_op =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jan 25 09:43:18 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_graphdef.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: Wed Jun 12 22:19:26 UTC 2024 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
#include "mlir/IR/OpDefinition.h" // from @llvm-project #include "mlir/Support/LLVM.h" // from @llvm-project #include "tensorflow/core/platform/errors.h" namespace tensorflow { void PopulateTfVersions(mlir::ModuleOp module, const VersionDef& versions) { mlir::Builder b(module.getContext()); auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer())); auto min_consumer = b.getNamedAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
} } return success(); } LogicalResult SessionInitializerOp::verify() { SessionInitializerOp session_initializer = *this; mlir::SymbolTable symbol_table( session_initializer->getParentOfType<ModuleOp>()); for (auto sym_ref : session_initializer.getInitializers()) { auto init_func_op = symbol_table.lookup<mlir::func::FuncOp>( mlir::cast<FlatSymbolRefAttr>(sym_ref).getValue()); if (!init_func_op)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/report.h
// quantization summary from a quantized `ModuleOp` and emit it in a human- and // machine-readable format. class QuantizationReport { public: QuantizationReport() = default; // Initializes `QuantizationReport` by collecting `QuantizationResults` from // `module_op`. explicit QuantizationReport(ModuleOp module_op); // Adds a `QuantizationResult` to the report.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 10:10:34 UTC 2024 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_analyzer_test_pass.cc
} }); } void ResourceAnalyzerTestPass::runOnOperation() { ModuleOp module_op = getOperation(); TF::ResourceAnalyzer resource_analyzer(module_op); PrintAnalysisResults(resource_analyzer, module_op); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateResourceAnalyzerTestPass() { return std::make_unique<ResourceAnalyzerTestPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 00:21:29 UTC 2023 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/passes.h
// Legalizes TF/XLA communication ops (TF dialect) to HLO dialect communication // ops. std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCommunicationPass(); // Legalizes TF/XLA collective ops (TF dialect) to HLO dialect collective // ops. std::unique_ptr<OperationPass<ModuleOp>> CreateLegalizeTFCollectivePass(); // Verifies that the TF/XLA ops have all been lowered to MHLO.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.h
static constexpr absl::string_view kName = "quant_ptq_weight_only"; explicit WeightOnlyPtqComponent(absl::Nonnull<MLIRContext*> ctx); absl::StatusOr<ModuleOp> Run( ModuleOp module_op, const ::stablehlo::quantization::QuantizationConfig& config) override; private: absl::Nonnull<MLIRContext*> ctx_; }; // Runs weight-only quantization on a SavedModel at
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 12:18:22 UTC 2024 - 3.2K bytes - Viewed (0)