Searched refs:TensorBlockScratchAllocator (Results 1 – 15 of 15) sorted by relevance
59 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;258 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;476 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;597 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;854 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
183 typedef internal::TensorBlockScratchAllocator<DefaultDevice>364 typedef internal::TensorBlockScratchAllocator<ThreadPoolDevice>471 typedef internal::TensorBlockScratchAllocator<ThreadPoolDevice>
124 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
121 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
131 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
107 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
525 class TensorBlockScratchAllocator {527 explicit TensorBlockScratchAllocator(const Device& device) in TensorBlockScratchAllocator() function530 ~TensorBlockScratchAllocator() { in ~TensorBlockScratchAllocator()
137 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;439 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;730 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
114 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
342 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
155 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
112 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch;
129 typedef internal::TensorBlockScratchAllocator<Device> TensorBlockScratch; in VerifyBlockEvaluator()