Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsResourceAllocatingOp (0.14 sec)

  1. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc

    // ResourceAliasAnalysisInfo
    //===----------------------------------------------------------------------===//
    
    namespace {
    
    constexpr char kResourceArgUniqueIdAttr[] = "tf._resource_arg_unique_id";
    
    bool IsResourceAllocatingOp(Operation* op) {
      auto mem_interface = dyn_cast<MemoryEffectOpInterface>(op);
      if (!mem_interface) return false;
    
      for (Value value : 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)
Back to top