Searched refs:fetch_target (Results 1 – 10 of 10) sorted by relevance
32 CompilationCacheFetchTarget fetch_target) { in Lookup() argument41 s = (*entry)->ToSubEntryRef(fetch_target); in Lookup()43 << CompilationCacheFetchTarget_Name(fetch_target) << " with status " in Lookup()51 CompilationCacheFetchTarget fetch_target) { in Lookup() argument60 s = (*entry)->ToSubEntryRef(fetch_target); in Lookup()62 << CompilationCacheFetchTarget_Name(fetch_target) << " with status " in Lookup()
66 tpu::CompilationCacheFetchTarget fetch_target) { in Lookup() argument76 proto_key, "_", tpu::CompilationCacheFetchTarget_Name(fetch_target)); in Lookup()84 request.set_fetch_target(fetch_target); in Lookup()101 tpu::CompilationCacheFetchTarget fetch_target) { in Lookup() argument118 tpu::CompilationCacheFetchTarget_Name(fetch_target)); in Lookup()128 request.set_fetch_target(fetch_target); in Lookup()
37 CompilationCacheFetchTarget fetch_target) override;41 CompilationCacheFetchTarget fetch_target) override;
48 CompilationCacheFetchTarget fetch_target) = 0;62 CompilationCacheFetchTarget fetch_target) = 0;
113 s = entry->ToSubEntryRef(call->request.fetch_target()); in GetTpuProgram()119 CompilationCacheFetchTarget_Name(call->request.fetch_target())), in GetTpuProgram()127 CHECK_NE(call->request.fetch_target(), in GetTpuProgram()
46 tpu::CompilationCacheFetchTarget fetch_target) override;50 tpu::CompilationCacheFetchTarget fetch_target) override;
72 Status ToSubEntryRef(CompilationCacheFetchTarget fetch_target) override { in ToSubEntryRef() argument
37 CompilationCacheFetchTarget fetch_target = 3; field
85 CompilationCacheFetchTarget fetch_target) { in ToSubEntryRef() argument87 switch (fetch_target) { in ToSubEntryRef()98 return xla::InvalidArgument("Invalid fetch target: %d", fetch_target); in ToSubEntryRef()
75 virtual Status ToSubEntryRef(CompilationCacheFetchTarget fetch_target);