Home
last modified time | relevance | path

Searched refs:CreateTpuCompilationCache (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_configuration_ops.cc96 Status CreateTpuCompilationCache( in CreateTpuCompilationCache() function
143 OP_REQUIRES_OK(ctx, CreateTpuCompilationCache(rmgr, &compilation_cache)); in Compute()
307 OP_REQUIRES_OK(ctx, CreateTpuCompilationCache(rmgr, &compilation_cache)); in Compute()
H A Dtpu_configuration_ops.h28 Status CreateTpuCompilationCache(
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_tpu_kernels.cc136 TF_RETURN_IF_ERROR(CreateTpuCompilationCache(rmgr, &compilation_cache)); in InitializeInternal()