Searched refs:XlaCompilationCache (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compilation_cache_test.cc | 27 using SignatureHash = XlaCompilationCache::Signature::Hash;
37 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s1, in TEST()
38 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
42 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s2, in TEST()
43 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
47 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s3, in TEST()
48 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
50 std::vector<XlaCompilationCache::Signature> signatures = {s1, s2, s3}; in TEST()
82 TF_ASSERT_OK_AND_ASSIGN(XlaCompilationCache::Signature s1, in TEST()
83 XlaCompilationCache::BuildSignature(fn, args)); in TEST()
[all …]
|
| D | xla_compilation_cache.cc | 78 using TensorTypeAndShape = XlaCompilationCache::Signature::TensorTypeAndShape;
148 constexpr int64_t XlaCompilationCache::kDefaultCompilationThreshold;
150 XlaCompilationCache::AsyncCompilationState::kNumCompilerThreads;
152 XlaCompilationCache::AsyncCompilationState::kMaxNumOngoingCompilations;
154 XlaCompilationCache::XlaCompilationCache(Config config, in XlaCompilationCache() function in tensorflow::XlaCompilationCache
163 XlaCompilationCache::~XlaCompilationCache() { in ~XlaCompilationCache()
187 string XlaCompilationCache::DebugString() const { in DebugString()
193 string XlaCompilationCache::Signature::HumanString() const { in HumanString()
201 bool XlaCompilationCache::Signature::operator==(const Signature& other) const { in operator ==()
212 uint64 XlaCompilationCache::Signature::Hash::operator()( in operator ()()
[all …]
|
| D | xla_compilation_cache_disable_test.cc | 49 auto cache = new XlaCompilationCache(XlaCompilationCache::Config(), client, in TEST()
56 XlaCompilationCache::CompileMode::kStrict, in TEST()
65 XlaCompilationCache::CompileMode::kAsync, in TEST()
74 XlaCompilationCache::CompileMode::kLazy, in TEST()
|
| D | xla_platform_info.cc | 48 XlaCompilationCache** cache) { in BuildXlaCompilationCache()
49 XlaCompilationCache::Config cache_config( in BuildXlaCompilationCache()
55 *cache = new XlaCompilationCache( in BuildXlaCompilationCache()
111 *cache = new XlaCompilationCache( in BuildXlaCompilationCache()
168 const XlaCompilationCache& cache, in GenerateCompilerOptions()
|
| D | xla_compile_on_demand_op.cc | 48 XlaCompilationCache* cache, in Run()
112 XlaCompilationCache** cache, ResourceVarsSnapshot* variable_args, in Compile()
130 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in Compile()
132 [&](XlaCompilationCache** write_into_cache) { in Compile()
175 XlaCompilationCache* cache; in Compute()
|
| D | xla_compile_on_demand_op.h | 48 XlaCompilationCache** cache,
52 Status Run(OpKernelContext* ctx, XlaCompilationCache* cache,
|
| D | xla_compilation_cache.h | 53 class XlaCompilationCache : public ResourceBase {
75 XlaCompilationCache(Config config, xla::LocalClient* client,
77 ~XlaCompilationCache() override;
316 TF_DISALLOW_COPY_AND_ASSIGN(XlaCompilationCache);
|
| D | xla_platform_info.h | 93 XlaCompilationCache** cache);
112 const XlaCompilationCache& cache,
|
| D | get_compiler_ir.cc | 129 XlaCompilationCache* cache; in GetCompilerIr()
130 TF_RETURN_IF_ERROR(rmgr->LookupOrCreate<XlaCompilationCache>( in GetCompilerIr()
132 [&](XlaCompilationCache** cache_write_into) { in GetCompilerIr()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 242 XlaCompilationCache::CompileMode compile_mode, in CompileToLocalExecutable()
253 XlaCompilationCache* cache; in CompileToLocalExecutable()
254 TF_RETURN_IF_ERROR(rm->LookupOrCreate<XlaCompilationCache>( in CompileToLocalExecutable()
256 [&](XlaCompilationCache** cache) { in CompileToLocalExecutable()
324 XlaCompilationCache::CompileMode::kStrict, in ComputeAsync()
492 XlaCompilationCache::CompileMode compile_mode = [&] { in Compute()
494 return XlaCompilationCache::CompileMode::kStrict; in Compute()
497 ? XlaCompilationCache::CompileMode::kAsync in Compute()
498 : XlaCompilationCache::CompileMode::kLazy; in Compute()
517 if (compile_mode != XlaCompilationCache::CompileMode::kLazy || in Compute()
|