Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 81 for func_ops (1.09 sec)

  1. tensorflow/compiler/mlir/quantization/common/lift_as_function_call.cc

      return false;
    }
    
    SmallVector<func::FuncOp> GetSortedFunctions(ModuleOp module_op) {
      auto iterator_range = module_op.getOps<func::FuncOp>();
      SmallVector<func::FuncOp> func_ops(iterator_range.begin(),
                                         iterator_range.end());
      absl::c_sort(func_ops, [](func::FuncOp op1, func::FuncOp op2) {
        return op1.getName() < op2.getName();
      });
      return func_ops;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 21.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/insert_calibration_statistics_saver.cc

      // Insert CalibrationStatisticsSaverOp to the end of each region.
      for (auto func_op : module_op.getOps<func::FuncOp>()) {
        int32_t output_file_idx = 0;
        StringRef func_name = func_op.getSymName();
    
        func_op.walk([&output_file_idx, &ctx, &func_name, &aggregator_ops_to_ignore,
                      this](Operation* op) {
          for (Region& region : op->getRegions()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 7.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/initialize_variables_in_session_init.cc

        return failure();
      }
    
      // Fetch all VarHandleOp.
      llvm::StringSet<> variable_names;
      llvm::SmallVector<TF::VarHandleOp, 4> var_ops;
      for (auto func_op : module.getOps<func::FuncOp>()) {
        for (auto var_handle_op : func_op.getOps<TF::VarHandleOp>()) {
          auto variable_name = GetVariableName(var_handle_op);
          if (variable_names.count(variable_name)) continue;
          var_ops.emplace_back(var_handle_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 23 09:05:47 UTC 2024
    - 7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc

      OwningOpRef<ModuleOp> module_op =
          ParseModuleOpString(kArgumentWithUnknownDims);
      ASSERT_TRUE(module_op);
    
      auto func_op = module_op->lookupSymbol<func::FuncOp>("unknown_dims_arg");
      ASSERT_THAT(func_op, NotNull());
      ASSERT_THAT(func_op.getNumArguments(), Eq(1));
    
      EXPECT_TRUE(HasRankOf(func_op.getArgument(0), /*rank=*/2));
    }
    
    TEST_F(AttrsAndConstraintsTest, HasRankOfReturnsFalseForUnknownRank) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 22.9K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc

      }
    
      return success();
    }
    
    void LiftHashTableOpsAsArgsPass::runOnOperation() {
      auto module_op = getOperation();
    
      for (auto func_op : module_op.getOps<func::FuncOp>()) {
        if (failed(LiftHashTableOpsToArguments(module_op, func_op))) {
          signalPassFailure();
          return;
        }
      }
    }
    
    static PassRegistration<LiftHashTableOpsAsArgsPass> pass;
    
    }  // namespace
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc

      // we do not overwrite the function argument mapping here.
      func_op.walk([&](Operation* op) {
        if (op == func_op) return;
        for (Region& region : op->getRegions()) {
          assign_unique_id_to_all(region.getArguments());
        }
      });
    
      llvm::SmallDenseMap<ResourceHandle, int64_t> resource_handle_id_map;
      func_op.walk([&](Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.cc

        return false;  // not a call
      func::FuncOp func_op = dyn_cast<func::FuncOp>(call.resolveCallable(
          &symbol_table_collection_));
      return IsPureFunction(func_op);
    }
    
    bool OpSideEffectCollector::IsPureFunction(func::FuncOp func_op) const {
      auto it = is_pure_function_.find(func_op);
      if (it == is_pure_function_.end()) {
        bool is_pure = true;
        is_pure_function_[func_op] = is_pure;  // prevent infinite recursion
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 41.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc

      }
    };
    
    // Returns true iff func_op has either no Region or the body has no Blocks.
    bool IsFuncOpEmpty(func::FuncOp func_op) {
      return func_op->getNumRegions() == 0 || func_op.getBody().empty();
    }
    
    // Gets the GraphOp from the function op. Returns an empty op iff it doesn't
    // exist.
    GraphOp GetGraphOpFromFuncOp(func::FuncOp func_op) {
      if (IsFuncOpEmpty(func_op)) return {};
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:54:52 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc

      for (func::FuncOp func_op : getOperation().getOps<func::FuncOp>()) {
        if (functions_to_skip.contains(func_op)) {
          OpBuilder builder(func_op);
          // Mark this function as being skipped in island outlining.
          func_op->setAttr(mlir::TF::kSkipIslandOutlining,
                           builder.getBoolAttr(true));
          continue;
        }
    
        func_op.walk([&](GraphOp graph) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 27.6K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/analysis/side_effect_analysis.h

        return alias_analysis_;
      }
    
     private:
      // Runs the analysis and populates `sorted_control_predecessors_` and
      // `sorted_control_successors_` for `func_op`. Clears `control_predecessors_`.
      void AnalyzeFunction(func::FuncOp func_op);
    
      // Runs the analysis and populates `control_predecessors_` for `region`.
      void AnalyzeRegion(Region* region);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 14.8K bytes
    - Viewed (0)
Back to top