Home
last modified time | relevance | path

Searched refs:Fprint128 (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dattr_builder.cc259 inline tensorflow::Fprint128 FingerprintCat128(const tensorflow::Fprint128& a, in FingerprintCat128()
260 const tensorflow::Fprint128& b) { in FingerprintCat128()
265 void CombineUnordered(const tensorflow::Fprint128& a, in CombineUnordered()
266 tensorflow::Fprint128* b) { in CombineUnordered()
271 inline tensorflow::Fprint128 CacheKeyHelper(StringPiece s, in CacheKeyHelper()
272 const tensorflow::Fprint128& b) { in CacheKeyHelper()
273 tensorflow::Fprint128 a = tensorflow::Fingerprint128(s); in CacheKeyHelper()
277 inline tensorflow::Fprint128 CacheKeyHelper(StringPiece s, uint64 b) { in CacheKeyHelper()
283 tensorflow::Fprint128 AttrBuilder::CacheKey(const StringPiece device) { in CacheKey()
292 tensorflow::Fprint128 AttrBuilder::BuildCacheKeyForDevice( in BuildCacheKeyForDevice()
[all …]
Dattr_builder.h150 tensorflow::Fprint128 CacheKey(const StringPiece device);
180 tensorflow::Fprint128 BuildCacheKeyForDevice(const StringPiece device) const;
206 absl::optional<tensorflow::Fprint128> cached_cache_key_;
Dcontext.h245 core::RefCountPtr<KernelAndDevice> GetCachedKernel(Fprint128 cache_key);
247 void AddKernelToCache(Fprint128 cache_key, KernelAndDevice* kernel);
622 std::unique_ptr<std::vector<Fprint128>> cached_kernel_keys;
624 std::unordered_map<Fprint128, core::RefCountPtr<KernelAndDevice>,
Dexecute.cc256 inline tensorflow::Fprint128 FingerprintCat128(const tensorflow::Fprint128& a, in FingerprintCat128()
257 const tensorflow::Fprint128& b) { in FingerprintCat128()
262 inline tensorflow::Fprint128 FingerprintCat128(const tensorflow::Fprint128& a, in FingerprintCat128()
312 Fprint128* fingerprint) { in AppendTensorShapeToFingerprint()
389 Fprint128 cache_key = op->MutableAttrs()->CacheKey(op->DeviceName()); in GetOrCreateKernelAndDevice()
Dattr_builder_test.cc77 tensorflow::Fprint128 cache_key = a.CacheKey("cpu:0"); in TEST()
Dcontext.cc725 absl::make_unique<std::vector<Fprint128>>(); in AddFunctionDef()
841 Fprint128 cache_key) { in GetCachedKernel()
852 void EagerContext::AddKernelToCache(Fprint128 cache_key, in AddKernelToCache()
/external/tensorflow/tensorflow/core/platform/
Dfingerprint.h34 struct Fprint128 { struct
39 inline bool operator==(const Fprint128& lhs, const Fprint128& rhs) { argument
44 size_t operator()(const Fprint128& v) const { in operator()
103 inline Fprint128 Fingerprint128(const StringPiece s) { in Fingerprint128()
Dfingerprint_test.cc33 const Fprint128 fingerprint = Fingerprint128("Hello"); in TEST()
39 const Fprint128 fingerprint = Fingerprint128("World"); in TEST()
47 const std::unordered_set<Fprint128, Fprint128Hasher> map = {{1, 2}, {3, 4}}; in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc65 Fprint128 cache_key = in Run()
/external/tensorflow/tensorflow/core/kernels/
Dsdca_ops.cc320 const Fprint128 fprint = Fingerprint128(in_values(i)); in Compute()