Home
last modified time | relevance | path

Searched refs:TpuCompilationCacheKey (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_compilation_cache_interface.h108 const TpuCompilationCacheKey& subgraph_key,
165 std::string ConstructCompilationCacheKey(const TpuCompilationCacheKey& key) { in ConstructCompilationCacheKey()
197 const TpuCompilationCacheKey& subgraph_key,
254 std::string FindCacheKey(const TpuCompilationCacheKey& subgraph_key)
267 const TpuCompilationCacheKey& subgraph_key)
H A Dtpu_compile_op_common.h103 const TpuCompilationCacheKey* key,
123 const tpu::TpuCompilationCacheKey& key,
132 const tpu::TpuCompilationCacheKey& key,
144 const tpu::TpuCompilationCacheKey& key, in LookupPersistentCompilationCacheAndFillCaches()
H A Dtpu_compilation_cache_key.h27 struct TpuCompilationCacheKey { struct
63 explicit TpuCompilationCacheKey() {} in TpuCompilationCacheKey() argument
64 explicit TpuCompilationCacheKey(const std::string& p) : prefix(p) {} in TpuCompilationCacheKey() function
H A Dtpu_util.cc33 xla::StatusOr<TpuCompilationCacheKey> ParseCompilationCacheKey( in ParseCompilationCacheKey()
38 return TpuCompilationCacheKey(key); in ParseCompilationCacheKey()
43 TpuCompilationCacheKey parsed_key(splits.at(0)); in ParseCompilationCacheKey()
H A Dtpu_compilation_cache_interface.cc236 const TpuCompilationCacheKey parsed_key = in RemoveEntry()
352 const TpuCompilationCacheKey parsed_key = in InsertEntry()
366 const TpuCompilationCacheKey& subgraph_key, in CompileIfKeyAbsent()
385 const TpuCompilationCacheKey& subgraph_key) { in FindCacheKey()
404 const TpuCompilationCacheKey& subgraph_key, in CompileIfKeyAbsentHelper()
H A Dtpu_compile_op_common.cc167 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCache()
182 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCacheInternal()
249 const TpuCompilationCacheKey key = CreateCompilationCacheKey( in ComputeInternal()
H A Dtpu_op_util.cc105 TpuCompilationCacheKey CreateCompilationCacheKey( in CreateCompilationCacheKey()
138 TpuCompilationCacheKey key; in CreateCompilationCacheKey()
H A Dtpu_compilation_cache_external.h64 const TpuCompilationCacheKey& subgraph_key)
H A Dtpu_op_util.h35 TpuCompilationCacheKey CreateCompilationCacheKey(
H A Dtpu_util.h42 xla::StatusOr<TpuCompilationCacheKey> ParseCompilationCacheKey(
H A Dtpu_compile_op_impl.h58 const TpuCompilationCacheKey* key,
H A Dtpu_compile_op_impl.cc32 const TpuCompilationCacheKey* key, in Compile()
H A Dtpu_compilation_cache_external.cc72 const TpuCompilationCacheKey& subgraph_key) { in InitializeEntry()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dtpu_compile_ops.cc91 tpu::TpuCompilationCacheKey CompilationCacheKey( in CompilationCacheKey()
100 const tpu::TpuCompilationCacheKey key = CreateCompilationCacheKey( in CompilationCacheKey()
183 const tpu::TpuCompilationCacheKey key = CompilationCacheKey( in Compute()