OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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