Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 50 of 435 for ModuleOp (0.32 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/debugging/mlir_dump_test.cc

     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(NoOpPass)
    
      NoOpPass() = default;
    
      llvm::StringRef getArgument() const final { return "no-op-pass"; }
    
      void runOnOperation() override {
        // Noop pass does nothing on the operation.
      }
    };
    
    std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateNoOpPass() {
      return std::make_unique<NoOpPass>();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 14 03:17:14 UTC 2024
    - 6.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

    namespace mlir {
    namespace quant {
    namespace {
    
    constexpr StringRef kSharedNameAttr = "shared_name";
    
    class LiftHashTableOpsAsArgsPass
        : public PassWrapper<LiftHashTableOpsAsArgsPass, OperationPass<ModuleOp>> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftHashTableOpsAsArgsPass)
      explicit LiftHashTableOpsAsArgsPass() = default;
    
      StringRef getArgument() const final {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/passes.td

      let summary = "Insert CallOnce op when tf_saved_model's session initializer is give.";
      let constructor = "CreateInsertCallOnceOpFromSessionInitializerPass()";
      let dependentDialects = ["TFL::TensorFlowLiteDialect"];
    }
    
    def LegalizeHashTablesPass : Pass<"tfl-legalize-hashtables-tf", "mlir::ModuleOp"> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 22.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/integration/tfr_decompose_ctx.cc

      // Load the TFR functions in a mlir::ModuleOp
      auto memory_buffer = llvm::MemoryBuffer::getMemBuffer(
          llvm::StringRef(tfr_raw_text.data(), tfr_raw_text.size()));
      llvm::SourceMgr source_mgr;
      source_mgr.AddNewSourceBuffer(std::move(memory_buffer), llvm::SMLoc());
      mlir::OwningOpRef<mlir::ModuleOp> module =
          mlir::parseSourceFile<mlir::ModuleOp>(source_mgr, mlir_ctx);
      // The MLIRContext owns the module
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 29 02:34:43 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.h

    See the License for the specific language governing permissions and
    limitations under the License.
    ==============================================================================*/
    // Functionalities for exporting MLIR ModuleOp to TensorFlow SavedModel.
    
    #ifndef TENSORFLOW_COMPILER_MLIR_QUANTIZATION_STABLEHLO_CC_SAVED_MODEL_EXPORT_H_
    #define TENSORFLOW_COMPILER_MLIR_QUANTIZATION_STABLEHLO_CC_SAVED_MODEL_EXPORT_H_
    
    #include <optional>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 20 11:11:25 UTC 2024
    - 6.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/utils/cluster_util_test.cc

    namespace {
    
    constexpr StringRef kTestClusterName = "tpu0";
    
    absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString(
        StringRef string, MLIRContext* context) {
      DialectRegistry mlir_registry;
      RegisterAllTensorFlowDialects(mlir_registry);
      context->appendDialectRegistry(mlir_registry);
      OwningOpRef<ModuleOp> mlir_module;
      auto status =
          tensorflow::DeserializeMlirModule(string, context, &mlir_module);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 26 09:37:10 UTC 2024
    - 7.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tf2xla/transforms/test_utils.cc

    using ::mlir::MLIRContext;
    using ::mlir::ModuleOp;
    using ::mlir::OwningOpRef;
    using ::tsl::StatusOr;
    
    absl::StatusOr<OwningOpRef<ModuleOp>> GetMlirModuleFromString(
        absl::string_view module_string, MLIRContext* context) {
      DialectRegistry mlir_registry;
      RegisterCommonToolingDialects(mlir_registry);
      context->appendDialectRegistry(mlir_registry);
    
      OwningOpRef<ModuleOp> mlir_module;
      auto status =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:16:07 UTC 2024
    - 1.8K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h

    // file; otherwise, load from a GraphDef.
    // Setting prune_unused_nodes to true, would prune unreachable nodes if
    // output_arrays is specified.
    absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> LoadFromGraphdefOrMlirSource(
        const std::string& input_filename, bool input_mlir,
        bool use_splatted_constant, const std::vector<std::string>& extra_tf_opdefs,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 08:30:24 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/init_text_file_to_import_test_pass.cc

    }
    
    }  // namespace
    }  // namespace TF
    
    namespace tf_test {
    std::unique_ptr<OperationPass<ModuleOp>> CreateInitTextFileToImportTestPass() {
      return std::make_unique<TF::InitTextFileToImportTestPass>();
    }
    std::unique_ptr<OperationPass<ModuleOp>>
    CreateInitTextFileToImportSavedModelTestPass() {
      return std::make_unique<TF::InitTextFileToImportSavedModelTestPass>();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:19:38 UTC 2022
    - 5.5K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/lift_variables_test_pass.h

    class LiftVariablesTestPass
        : public PassWrapper<LiftVariablesTestPass, OperationPass<ModuleOp>> {
     public:
      LiftVariablesTestPass() { session_ = new FakeSession(); }
    
      ~LiftVariablesTestPass() override { delete session_; }
    
      void runOnOperation() override {
        ModuleOp module = getOperation();
        if (failed(LiftVariables(module, session_))) signalPassFailure();
      }
    
     private:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 21 15:49:06 UTC 2022
    - 5.6K bytes
    - Viewed (0)
Back to top