Searched refs:eigen_gpu_allocator (Results 1 – 1 of 1) sorted by relevance
350 Allocator* eigen_gpu_allocator = get_allocator(AllocatorAttributes()); in OpKernelContext() local353 eigen_gpu_allocator); in OpKernelContext()