Searched defs:EigenAllocator (Results 1 – 1 of 1) sorted by relevance
61 class EigenAllocator : public Eigen::Allocator { class63 explicit EigenAllocator(tensorflow::Allocator* a) : allocator_(a) {} in EigenAllocator() function in tensorflow::LocalDevice::EigenThreadPoolInfo::EigenAllocator