HomeSort by relevance Sort by last modified time
    Searched refs:XlaCompilationCache (Results 1 - 5 of 5) sorted by null

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

Completed in 215 milliseconds