Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for IsResourceTensor (0.13 sec)

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

      // be future-proof.
      if (llvm::isa<TF::IfOp, TFL::IfOp, TFL::CallOnceOp, TFL::WhileOp>(op))
        return true;
      for (auto operand : op->getOperands()) {
        if (IsResourceTensor(operand)) return true;
      }
      for (auto result : op->getResults()) {
        if (IsResourceTensor(result)) return true;
      }
      return false;
    }
    
    // This transformation pass takes an operation that has or depends on side
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 7K bytes
    - Viewed (0)
Back to top