Searched refs:CompilationCacheKey (Results 1 – 3 of 3) sorted by relevance
93 struct CompilationCacheKey { struct94 CompilationCacheKey(std::string ptx, int cc_major, int cc_minor, in CompilationCacheKey() function106 size_t operator()(const CompilationCacheKey& key) const { in operator() argument116 size_t operator()(const CompilationCacheKey& a, in operator()117 const CompilationCacheKey& b) const { in operator()132 absl::node_hash_map<CompilationCacheKey, CompilationCacheValue,
90 tpu::TpuCompilationCacheKey CompilationCacheKey( in CompilationCacheKey() function182 const tpu::TpuCompilationCacheKey key = CompilationCacheKey( in Compute()
109 Status CompilationCacheKey(const xrt::XLAComputation& computation, in CompilationCacheKey() function206 OP_REQUIRES_OK(ctx, CompilationCacheKey(computation_proto, &key)); in Compute()