Home
last modified time | relevance | path

Searched refs:GetComputationCacheEntry (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.h38 Status GetComputationCacheEntry(
H A Dtpu_execute_op.cc68 Status GetComputationCacheEntry( in GetComputationCacheEntry() function
639 GetComputationCacheEntry(context, &rendezvous_key_base, &entry_ref)); in DoWork()
H A Dtpu_reshard_variables_op.cc144 TF_RETURN_IF_ERROR(reshard_util::GetComputationCacheEntry( in DoTpuExecute()
H A Dtpu_reshard_variables_op_util.cc78 Status GetComputationCacheEntry( in GetComputationCacheEntry() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/
H A Dtpu_execute_op.cc61 Status GetComputationCacheEntry( in GetComputationCacheEntry() function
311 TF_RETURN_IF_ERROR(GetComputationCacheEntry(rm, compilation_handle, in DoWork()