Searched refs:XlaCompilationCache (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache_test.cc | 31 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s1, in TEST()
32 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
36 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s2, in TEST()
37 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
41 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s3, in TEST()
42 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
44 std::vector<XlaCompilationCache::Signature> signatures = {s1, s2, s3}; in TEST()
|
| D | xla_compilation_cache.cc | 41 constexpr int64 XlaCompilationCache::kDefaultCompilationThreshold;
43 XlaCompilationCache::XlaCompilationCache(xla::LocalClient* client, in XlaCompilationCache() function in tensorflow::XlaCompilationCache
47 XlaCompilationCache::~XlaCompilationCache() { in ~XlaCompilationCache()
65 string XlaCompilationCache::DebugString() const { in DebugString()
71 string XlaCompilationCache::Signature::HumanString() const { in HumanString()
84 bool XlaCompilationCache::Signature::operator==(const Signature& other) const { in operator ==()
99 uint64 XlaCompilationCache::Signature::Hash::operator()( in operator ()()
100 const XlaCompilationCache::Signature& signature) const { in operator ()()
116 xla::StatusOr<XlaCompilationCache::Signature>
117 XlaCompilationCache::BuildSignature( in BuildSignature()
[all …]
|
| D | xla_compilation_cache.h | 44 class XlaCompilationCache : public ResourceBase {
46 XlaCompilationCache(xla::LocalClient* client, DeviceType device_type);
47 ~XlaCompilationCache() override;
192 TF_DISALLOW_COPY_AND_ASSIGN(XlaCompilationCache);
|
| D | xla_compile_on_demand_op.cc | 171 XlaCompilationCache* cache; in Compile()
172 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in Compile()
174 [&](XlaCompilationCache** cache) { in Compile()
175 *cache = new XlaCompilationCache(metadata.client(), in Compile()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 200 XlaCompilationCache** cache) { in BuildCompilationCache()
202 *cache = new XlaCompilationCache( in BuildCompilationCache()
249 *cache = new XlaCompilationCache( in BuildCompilationCache()
268 XlaCompilationCache* cache; in CompileToLocalExecutable()
269 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in CompileToLocalExecutable()
271 [&](XlaCompilationCache** cache) { in CompileToLocalExecutable()
318 lazy ? XlaCompilationCache::CompileMode::kLazy in CompileToLocalExecutable()
319 : XlaCompilationCache::CompileMode::kStrict, in CompileToLocalExecutable()
|