Searched refs:TpuCompilationCacheKey (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | tpu_compilation_cache_key.h | 27 struct TpuCompilationCacheKey { struct
57 explicit TpuCompilationCacheKey() {} in TpuCompilationCacheKey() function
58 explicit TpuCompilationCacheKey(const std::string& p) : prefix(p) {} in TpuCompilationCacheKey() function
|
| D | tpu_compilation_cache_interface.h | 108 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)
|
| D | tpu_util.cc | 33 xla::StatusOr<TpuCompilationCacheKey> ParseCompilationCacheKey( in ParseCompilationCacheKey()
38 return TpuCompilationCacheKey(key); in ParseCompilationCacheKey()
43 TpuCompilationCacheKey parsed_key(splits.at(0)); in ParseCompilationCacheKey()
|
| D | tpu_op_util.cc | 104 TpuCompilationCacheKey CreateCompilationCacheKey( in CreateCompilationCacheKey()
142 TpuCompilationCacheKey key; in CreateCompilationCacheKey()
|
| D | tpu_compilation_cache_interface.cc | 232 const TpuCompilationCacheKey parsed_key = in RemoveEntry()
348 const TpuCompilationCacheKey parsed_key = in InsertEntry()
362 const TpuCompilationCacheKey& subgraph_key, in CompileIfKeyAbsent()
381 const TpuCompilationCacheKey& subgraph_key) { in FindCacheKey()
400 const TpuCompilationCacheKey& subgraph_key, in CompileIfKeyAbsentHelper()
|
| D | tpu_op_util.h | 35 TpuCompilationCacheKey CreateCompilationCacheKey(
|
| D | tpu_compilation_cache_external.h | 64 const TpuCompilationCacheKey& subgraph_key)
|
| D | tpu_program_group_interface.h | 55 virtual Status LogCompilationStats(const TpuCompilationCacheKey& key,
|
| D | tpu_compile_op_common.h | 124 const tpu::TpuCompilationCacheKey& key,
136 const tpu::TpuCompilationCacheKey& key, in LookupPersistentCompilationCacheAndFillCaches()
|
| D | tpu_util.h | 42 xla::StatusOr<TpuCompilationCacheKey> ParseCompilationCacheKey(
|
| D | tpu_compilation_cache_external.cc | 72 const TpuCompilationCacheKey& subgraph_key) { in InitializeEntry()
|
| D | tpu_program_group.h | 121 Status LogCompilationStats(const TpuCompilationCacheKey& key,
|
| D | tpu_program_group.cc | 188 Status TpuProgramGroup::LogCompilationStats(const TpuCompilationCacheKey& key, in LogCompilationStats()
|
| D | tpu_compile_op_common.cc | 595 const OpInputList& guaranteed_constants, const TpuCompilationCacheKey& key, in CompileLocallyAndFillHostCache()
652 const TpuCompilationCacheKey key = CreateCompilationCacheKey( in ComputeInternal()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | tpu_compile_ops.cc | 90 tpu::TpuCompilationCacheKey CompilationCacheKey( in CompilationCacheKey()
99 const tpu::TpuCompilationCacheKey key = CreateCompilationCacheKey( in CompilationCacheKey()
182 const tpu::TpuCompilationCacheKey key = CompilationCacheKey( in Compute()
|