Home
last modified time | relevance | path

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

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