Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_conv_algorithm_picker.cc490 int64 scratch_bytes_used = in PickBestAlgorithmNoCacheCuda() local
492 result.set_scratch_bytes(scratch_bytes_used); in PickBestAlgorithmNoCacheCuda()
744 int64 scratch_bytes_used = scratch_allocator.TotalAllocatedBytes(); in PickBestAlgorithmNoCacheRocm() local
745 result.set_scratch_bytes(scratch_bytes_used); in PickBestAlgorithmNoCacheRocm()