- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ResourceHandleAllocatorInterface (0.36 sec)
-
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.h
// a resource instance ID. That means, they will be treated as non-aliasing. // This is correct for all current cases. A problematic case could be if we // had two ops A and B, A has the `ResourceHandleAllocatorInterface` and B has // not, and both ops might return a handle to the same resource (depending on // attributes). In this case, the return value of A would get a different ID
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 7.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_traits.h
template <typename ConcreteType> class UniqueResourceAllocation : public TraitBase<ConcreteType, UniqueResourceAllocation> { public: // Implements method required for `ResourceHandleAllocatorInterface`. llvm::SmallVector<mlir::TF::ResourceHandleValueAndId> GetResourceHandleValueAndIdList( llvm::SmallDenseMap<mlir::TF::ResourceHandle, int64_t>& resource_handle_id_map,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
} }); llvm::SmallDenseMap<ResourceHandle, int64_t> resource_handle_id_map; func_op.walk([&](Operation* op) { if (auto resource_alloc = dyn_cast<ResourceHandleAllocatorInterface>(op)) { llvm::SmallVector<ResourceHandleValueAndId, 4> resources = resource_alloc.GetResourceHandleValueAndIdList( resource_handle_id_map, next_unique_instance_id);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0)