- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 435 for ModuleOp (0.61 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/passes.td
// Passes only used for testing purposes. include "mlir/Pass/PassBase.td" def TestPreCalibrationComponentPass : Pass<"stablehlo-test-pre-calibration-component", "mlir::ModuleOp"> { let summary = "Test-only pass to test the PreCalibrationComponent."; let description = [{ Runs the pre calibration passes for post-training quantization with default configuration. }];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 23:21:42 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_test.cc
#include "tsl/platform/statusor.h" namespace tensorflow { namespace { using ::mlir::MLIRContext; using ::mlir::ModuleOp; using ::mlir::OwningOpRef; using ::mlir::PassManager; using ::tensorflow::monitoring::testing::CellReader; absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString( absl::string_view module_string, MLIRContext* context) { mlir::DialectRegistry mlir_registry;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 4.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/pre_calibration.h
// debugging purposes. static constexpr absl::string_view kName = "quant_ptq_pre_calibration"; explicit PreCalibrationComponent(absl::Nonnull<MLIRContext*> ctx); absl::StatusOr<ModuleOp> Run( ModuleOp, const ::stablehlo::quantization::QuantizationConfig& config) override; private: absl::Nonnull<MLIRContext*> ctx_; }; } // namespace mlir::quant::stablehlo
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Mar 11 19:29:56 UTC 2024 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/passes.h
std::unique_ptr<OperationPass<func::FuncOp>> CreateDecomposeTFOpsPass( std::optional<ModuleOp> tfr_module = std::nullopt); // Rewrites quantized operands and results with their storage types. // This pass should be run at module level after decomposition, if there are // quantized operands or results. std::unique_ptr<OperationPass<ModuleOp>> CreateRewriteQuantizedIOPass(); // Raise to TF ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/utils/utils.cc
mlir::func::registerAllExtensions(registry); context->appendDialectRegistry(registry); source_mgr->AddNewSourceBuffer(std::move(buffer), llvm::SMLoc()); return mlir::OwningOpRef<mlir::ModuleOp>( mlir::parseSourceFile<mlir::ModuleOp>(*source_mgr, context)); } mlir::Location loc = mlir::FileLineColLoc::get(context, input_filename, /*line=*/0, /*column=*/0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 03 03:47:03 UTC 2024 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
namespace tensorflow { namespace { absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GetMlirModuleFromString( llvm::StringRef string, mlir::MLIRContext* context) { mlir::DialectRegistry mlir_registry; RegisterAllTensorFlowDialects(mlir_registry); context->appendDialectRegistry(mlir_registry); mlir::OwningOpRef<mlir::ModuleOp> mlir_module; auto status =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/saved_model/saved_model_test.cc
mlir::DialectRegistry registry; mlir::RegisterAllTensorFlowDialects(registry); mlir::MLIRContext context(registry); auto module = mlir::parseSourceFile<mlir::ModuleOp>(saved_model_mlir_path, &context); ASSERT_TRUE(module); std::vector<std::string> inputs; std::vector<std::pair<tensorflow::DataType, tensorflow::PartialTensorShape>> in_specs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 01:17:29 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/post_calibration.h
// debugging purposes. static constexpr absl::string_view kName = "quant_ptq_post_calibration"; explicit PostCalibrationComponent(absl::Nonnull<MLIRContext*> ctx); absl::StatusOr<ModuleOp> Run( ModuleOp module_op, const ::stablehlo::quantization::QuantizationConfig& config) override; void AddPasses( OpPassManager& pm, const ::stablehlo::quantization::QuantizationSpecs& specs,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 12:53:33 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/compile_mlir_util.cc
return absl::OkStatus(); } // Converts arg_shapes to xla::Shape's and store into xla_input_shapes. Status GetXlaInputShapes( mlir::ModuleOp module, llvm::ArrayRef<TensorOrResourceShape> arg_shapes, bool use_tuple_args, const XlaShapeLayoutHelpers::ShapeDeterminationFns shape_determination_fns, std::vector<xla::Shape>* xla_input_shapes) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 17:24:39 UTC 2024 - 45.3K 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)