Search Options

Results per page
Sort
Preferred Languages
Advance

Results 81 - 90 of 229 for getOperation (0.19 sec)

  1. tensorflow/compiler/mlir/lite/transforms/insert_call_once_op.cc

          InsertCallOnceOpFromSessionInitializerPass)
    
     private:
      void runOnOperation() override;
    };
    
    void InsertCallOnceOpFromSessionInitializerPass::runOnOperation() {
      ModuleOp module = getOperation();
    
      for (func::FuncOp init_func_op :
           tf_saved_model::GetInitializerFunctions(module)) {
        for (auto func : module.getOps<func::FuncOp>()) {
          auto dict_attr =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 01 05:03:09 UTC 2023
    - 2.5K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/test_pre_calibration_component.cc

      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestPreCalibrationComponentPass)
    
     private:
      void runOnOperation() override;
    };
    
    void TestPreCalibrationComponentPass::runOnOperation() {
      ModuleOp module_op = getOperation();
      MLIRContext& ctx = getContext();
    
      // Simply runs the PreCalibrationComponent with a default configuration.
      PreCalibrationComponent component(&ctx);
      QuantizationConfig quantization_config{};
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 21:41:08 UTC 2024
    - 2.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/passes/xla_call_module_to_call.cc

        rewriter.replaceOpWithNewOp<func::CallOp>(op, entry_func_op, op.getArgs());
        return success();
      }
    };
    
    void XlaCallModuleToCallPass::runOnOperation() {
      ModuleOp module_op = getOperation();
      MLIRContext* ctx = module_op.getContext();
      RewritePatternSet patterns(&getContext());
      patterns.add<XlaCallModuleOpToCallOp>(ctx);
      if (failed(applyPatternsAndFoldGreedily(module_op, std::move(patterns)))) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 04 20:02:00 UTC 2024
    - 3.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/test_cluster_ops_by_policy.cc

            operands.Insert(op->getOperand(0), ValueConstraint::kShape);
        }
    
        return success();
      }
    };
    
    void TestClusteringPolicyPass::runOnOperation() {
      func::FuncOp func = getOperation();
      ValuesConstraintSet constraints;
    
      ClusteringPolicySet policies;
      policies.Add<TestOpsClusteringPolicy>();
    
      // Initialize constraints based on the return type attributes.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:19:38 UTC 2022
    - 3.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc

      }
    
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MockSuccessPass)
    
      explicit MockSuccessPass() = default;
    
     private:
      void runOnOperation() override {
        getOperation().walk([](Operation* nestedOp) {
          nestedOp->emitError()
              << "Error at " << nestedOp->getName().getStringRef().str() << " op";
        });
      };
    };
    
    // MockFailurePass reports errors and fails.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 25 01:48:36 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/deduplicate_bound_input_bindings.cc

    class DedupBoundInputBindingPass
        : public impl::DedupBoundInputBindingPassBase<DedupBoundInputBindingPass> {
      void runOnOperation() final;
    };
    
    void DedupBoundInputBindingPass::runOnOperation() {
      func::FuncOp func = getOperation();
      if (!mlir::tf_saved_model::IsExported(func)) return;
      llvm::SmallDenseMap<Attribute, unsigned, 8> unique_bound_inputs;
      llvm::BitVector arg_indices_to_erase(func.getNumArguments());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 04 09:25:35 UTC 2022
    - 2.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/transforms/verify_no_outside_compilation_markers_pass.cc

    bool HasXlaOutsideCompilationMarker(Operation& op) {
      return op.getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr;
    }
    
    void VerifyNoOutsideCompilationMarkersPass::runOnOperation() {
      Operation* func_op = getOperation();
    
      auto walk_result = func_op->walk([&](Operation* op) {
        if (IsDeviceClusterOp(*op) && HasChildLaunchDeviceOp(*op)) {
          std::string launch_error =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Aug 22 19:52:08 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/transforms/legalize_variables.cc

        : public impl::LegalizeVariablesPassBase<LegalizeVariablesPass> {
     public:
      MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeVariablesPass)
    
      void runOnOperation() override {
        auto module = getOperation();
        // If TFLite variable legalization is not allowed, then we skip this pass.
        if (auto legalize_tfl_variables_attr =
                module->getAttr(kLegalizeTflVariables)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 3.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/tpu_colocate_splits.cc

      }
      return predecessors;
    }
    
    struct TPUColocateSplits
        : public impl::TPUColocateSplitsPassBase<TPUColocateSplits> {
      void runOnOperation() override;
    };
    
    void TPUColocateSplits::runOnOperation() {
      getOperation().walk([&](Operation* op) {
        if (auto split = llvm::dyn_cast<TF::SplitOp>(op)) {
          if (HasDevice(split) || split->getAttrOfType<ArrayAttr>(kClassAttr))
            return WalkResult::advance();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 27 18:44:34 UTC 2023
    - 2.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc

        }
        return mlir::WalkResult::advance();
      });
      return failure(result.wasInterrupted());
    }
    
    void ConvertToLegacyCompileAndReplicateAttributesPass::runOnOperation() {
      func::FuncOp func_op = getOperation();
      if (failed(ConvertToLegacyAttributes(func_op))) return signalPassFailure();
    }
    
    }  // namespace
    
    std::unique_ptr<OperationPass<func::FuncOp>>
    CreateConvertToLegacyCompileAndReplicateAttributesPass() {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 23:50:19 UTC 2022
    - 2.8K bytes
    - Viewed (0)
Back to top