Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 19 for getNumRegions (0.16 sec)

  1. tensorflow/compiler/mlir/tensorflow/utils/call_graph_util.cc

                 << "Cannot find function " << sym.getRootReference();
        }
        callees.push_back(callee);
      }
      return success();
    }
    
    bool HasSingleBlock(func::FuncOp func) {
      return func->getNumRegions() == 1 && func.getBody().hasOneBlock();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 2.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h

                                   output_types, candidate_op->getAttrs());
          for (int i = 0; i < candidate_op->getNumRegions(); ++i) {
            new_state.addRegion();
          }
          Operation* quantized_op = rewriter.create(new_state);
          if (candidate_op->getNumRegions() != 0) {
            for (const auto& indexed_regions :
                 llvm::enumerate(candidate_op->getRegions())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_arguments.cc

    // This is a lengthy bit of code, since it has to recreate the operation.
    // TODO(kramm): Move this under utils/ somewhere.
    void EraseResults(Operation* op, llvm::BitVector erase) {
      assert(!op->getNumRegions());
      std::vector<Type> result_types;
      for (auto result : op->getResults()) {
        if (!erase[result.getResultNumber()]) {
          result_types.push_back(result.getType());
        }
      }
      OpBuilder builder(op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/transforms/constant_fold_utils.cc

      // could be folded should have a custom folder instead of relying on the
      // TensorFlow folding hook.
      if (inst == nullptr || inst->getNumResults() == 0 ||
          inst->hasTrait<OpTrait::TF::NoConstantFold>() ||
          inst->getNumRegions() != 0 || !isMemoryEffectFree(inst)) {
        return false;
      }
    
      // If any of the result types are variants, don't try to constant fold them.
      // This creates opaque variant constants which lose information and would
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_save_function_ops_to_main.cc

      }
    
      void runOnOperation() override;
    };
    
    // 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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 10.7K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize.cc

                                   output_types, quantizing_op->getAttrs());
          for (int i = 0; i < quantizing_op->getNumRegions(); ++i) {
            new_state.addRegion();
          }
          Operation* quantized_op = rewriter.create(new_state);
          if (quantizing_op->getNumRegions() != 0) {
            for (const auto& indexed_regions :
                 llvm::enumerate(quantizing_op->getRegions())) {
              IRMapping mapping;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 23.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc

            }
          }
        } else if (isa<tf_device::LaunchOp, tf_device::ClusterOp,
                       tf_executor::IslandOp, tf_executor::GraphOp>(op) &&
                   op->getNumRegions() == 1) {
          Region& region = op->getRegion(0);
          const auto& body_info = backtrack_analysis.GetAnalysisForRegion(region);
          for (auto result : filter_resources(op->getResults())) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 28.2K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc

      }
    
      num_new_results_ = op_->getNumResults();
    
      for (auto resource : all_resources) {
        ResourceInfo info;
        info.data_type = GetResourceSubtype(resource);
        llvm::BitVector written_regions(op_->getNumRegions());
        bool unsupported_use = false;
        for (OpOperand& use : resource.getUses()) {
          Operation* user = use.getOwner();
          // If the user is not in one of the regions, we are not interested in it.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 55.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc

                    tf_saved_model::TensorFlowSavedModelDialect>();
      }
    };
    
    // 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) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sun May 12 12:54:52 UTC 2024
    - 15.3K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc

      builder.setInsertionPoint(region_if);
    
      if (!region_idx) {
        RewriteRegionIfOp(builder, region_if, ops_to_visit, token);
        return true;
      }
    
      if (*region_idx < region_if.getNumRegions()) {
        // For the region-blocks of If op, we create a dummy token argument. Later
        // we replace that block-argument's uses with the same (implicitly captured)
        // token 'token', used for If op, and erase the argument.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 40.5K bytes
    - Viewed (0)
Back to top