Home
last modified time | relevance | path

Searched refs:EnqueueCompactionOnStreamForHbm (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_executor_interface.h64 virtual Status EnqueueCompactionOnStreamForHbm( in EnqueueCompactionOnStreamForHbm() function
H A Dtpu_executor.h161 Status EnqueueCompactionOnStreamForHbm(Stream* compaction_stream) override;
H A Dtpu_executor.cc337 Status TpuExecutor::EnqueueCompactionOnStreamForHbm(Stream* compaction_stream) { in EnqueueCompactionOnStreamForHbm() function in tensorflow::tpu::TpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.cc215 TF_RETURN_IF_ERROR(ds_executor->EnqueueCompactionOnStreamForHbm(stream)); in PerformCompaction()