Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for MemoryEffects (0.86 sec)

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

        Value value,
        SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>&
            effects) {
      if (value.getType().cast<TensorType>().getElementType().isa<ResourceType>()) {
        effects.emplace_back(MemoryEffects::Read::get(), value,
                             ResourceEffects::Variable::get());
        effects.emplace_back(MemoryEffects::Write::get(), value,
                             ResourceEffects::Variable::get());
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 16 00:33:17 UTC 2023
    - 2.4K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/side_effect_analysis_util.h

    void MarkResourceAsReadAndWrite(
        Value value,
        SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>&
            effect);
    
    void MarkResourceAsReadOnly(
        Value value,
        SmallVectorImpl<SideEffects::EffectInstance<MemoryEffects::Effect>>&
            effect);
    
    }  // namespace TF
    }  // namespace mlir
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Nov 16 00:33:17 UTC 2023
    - 1.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/hoist_loop_invariant.cc

    }
    
    bool OnlyHasReadEffect(Operation *op) {
      auto interface = llvm::dyn_cast<MemoryEffectOpInterface>(op);
      if (!interface) return false;
      return interface.onlyHasEffect<MemoryEffects::Read>();
    }
    
    // Get read only variable handles in `func`.
    llvm::DenseSet<ResourceHandle> GetReadOnlyVariables(func::FuncOp func) {
      llvm::DenseMap<ResourceHandle, llvm::SmallVector<Operation *, 4>> resources;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.3K bytes
    - Viewed (0)
Back to top