Search Options

Results per page
Sort
Preferred Languages
Advance

Results 61 - 70 of 229 for getOperation (0.19 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/tflite_legalize_hlo.cc

      // TODO: b/304003568 - Add TF_TransposeOp folding logic to tflite.
      target.addIllegalOp<mhlo::DotGeneralOp, mhlo::DotOp, mhlo::TransposeOp>();
      if (failed(applyPartialConversion(getOperation(), target,
                                        std::move(patterns)))) {
        getOperation().emitError("mhlo to TFLite legalization failed.");
        signalPassFailure();
      }
    }
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/quantize_variables.cc

        var_handle_op.erase();
      }
    }
    
    void QuantizeVariablesPass::runOnOperation() {
      ResourceIdMap resource_id_map;
      ResourceMap resource_map;
    
      // Collect all resource identities.
      getOperation().walk([&](TFL::VarHandleOp var_handle_op) {
        auto identity = std::make_pair(var_handle_op.getContainer().str(),
                                       var_handle_op.getSharedName().str());
        resource_id_map.insert(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/optimize_functional_ops.cc

        if (op.use_empty() &&
            (op.getIsStateless() ||
             (IsSideEffectFree(then_func) && IsSideEffectFree(else_func)))) {
          rewriter.eraseOp(op.getOperation());
          return success();
        }
    
        // Extract the constant cond value.
        DenseElementsAttr cond;
        if (!matchPattern(op.getCond(), m_Constant(&cond))) return failure();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/strip_noinline_attribute.cc

     public:
      // void runOnOperation() override;
      void runOnOperation() override {
        // Strip the "tf._noinline" attribute from top-level functions.
        for (auto func_op : getOperation().getOps<func::FuncOp>())
          func_op->removeAttr("tf._noinline");
      }
    };
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreateStripNoinlineAttributePass() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 1.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables_test_pass.cc

    struct FreezeVariableTestPass
        : public impl::FreezeVariablesTestPassBase<FreezeVariableTestPass> {
      void runOnOperation() override {
        TF::test_util::FakeSession session;
        if (tf_saved_model::FreezeVariables(getOperation(), &session).failed())
          signalPassFailure();
      }
    };
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreateFreezeVariableTestPass() {
      return std::make_unique<FreezeVariableTestPass>();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:19:38 UTC 2022
    - 1.6K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/verify_suitable_for_graph_export_pass.cc

    class VerifySuitableForExportPass
        : public impl::VerifySuitableForExportPassBase<
              VerifySuitableForExportPass> {
     public:
      void runOnOperation() override {
        if (failed(tensorflow::VerifyExportSuitable(getOperation())))
          return signalPassFailure();
      }
    };
    
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreateVerifySuitableForExportPass() {
      return std::make_unique<VerifySuitableForExportPass>();
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 1.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/tf_functional_to_executor.cc

        to_fetch = to_fetch.drop_back();
      }
      builder.create<tf_executor::FetchOp>(loc, to_fetch);
      // Build Island.
      island.getBody().push_back(new Block);
      island.getBody().front().getOperations().splice(
          island.getBody().front().begin(), body.getOperations(),
          copy_range.begin(), copy_range.end());
      builder.setInsertionPointToEnd(&island.getBody().front());
      builder.create<tf_executor::YieldOp>(loc, return_op.getOperands());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Aug 11 20:52:36 UTC 2023
    - 4.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/odml_converter/transforms/shlo_simplify.cc

    class SHLOSimplifyPass : public impl::SHLOSimplifyPassBase<SHLOSimplifyPass> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(SHLOSimplifyPass)
    
      void runOnOperation() override {
        ModuleOp module = getOperation();
        RewritePatternSet patterns(&getContext());
        populateWithGenerated(patterns);
        PopulateFolderPatterns(patterns);
        if (failed(applyPatternsAndFoldGreedily(module, std::move(patterns)))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 03:05:20 UTC 2024
    - 2.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/stablehlo/passes/remove_sharding_custom_call.cc

          RemoveShardingCustomCallPass>::RemoveShardingCustomCallPassBase;
    
     private:
      void runOnOperation() override;
    };
    
    void RemoveShardingCustomCallPass::runOnOperation() {
      func::FuncOp func_op = getOperation();
      MLIRContext& ctx = getContext();
    
      RewritePatternSet patterns(&ctx);
      populateWithGenerated(patterns);
    
      FrozenRewritePatternSet frozen_patterns(std::move(patterns));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Mar 13 07:04:47 UTC 2024
    - 2.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/print.cc

    void PrintPass::runOnOperation() {
      llvm::sys::SmartScopedLock<true> instrumentationLock(mutex_);
      OpPrintingFlags flags =
          OpPrintingFlags().elideLargeElementsAttrs().enableDebugInfo(false);
      getOperation()->print(*os_, flags);
    }
    }  // namespace
    
    std::unique_ptr<OperationPass<ModuleOp>> CreatePrintPass(raw_ostream* os) {
      return std::make_unique<PrintPass>(os);
    }
    
    }  // namespace TF
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 12 23:15:17 UTC 2023
    - 2.1K bytes
    - Viewed (0)
Back to top