Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for createPrintOpStatsPass (0.55 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.h

    namespace mlir {
    namespace odml {
    
    // Creates a pass which prints out a detailed report of conversion stats with:
    // success or not, % of Ops non-converted, list of non-converted Ops, etc.
    std::unique_ptr<Pass> createPrintOpStatsPass(
        std::vector<std::string> accepted_dialects);
    
    }  // namespace odml
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 24 21:06:11 UTC 2023
    - 1.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/odml_to_stablehlo.cc

      if (verbose) {
        // Print out a detailed report of non-converted stats.
        // Because this pass aborts the pass if there are unconverted ops,
        // we need to locate createPrintOpStatsPass after all optimization.
        pm.addPass(
            mlir::odml::createPrintOpStatsPass(GetAcceptedStableHLODialects()));
      }
    
      if (failed(pm.run(tf_module))) {
        return tensorflow::errors::Aborted("Lowering to StableHLO failed.");
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:16:49 UTC 2024
    - 14.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/op_stat_pass.cc

            delim = ", ";
          }
          *os_ << ")";
        }
        *os_ << "\n";
      }
    }
    
    }  // namespace odml
    }  // namespace mlir
    
    std::unique_ptr<mlir::Pass> mlir::odml::createPrintOpStatsPass(
        std::vector<std::string> accepted_dialects) {
      return std::make_unique<mlir::odml::PrintOpStatsPass>(&llvm::errs(),
                                                            accepted_dialects);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

                                         /*smuggle_disallowed_ops=*/true);
      // Print out a detailed report of non-converted stats.
      pass_manager.addPass(mlir::odml::createPrintOpStatsPass(
          mlir::odml::GetAcceptedStableHLODialects()));
      mlir::odml::AddStablehloOptimizationPasses(pass_manager);
      if (toco_flags.has_quantization_options()) {
        stablehlo::quantization::AddQuantizationPasses(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
Back to top