- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for MemoryEffects (0.86 sec)
-
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) -
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) -
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)