Searched refs: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::EigenAllocator98 eigen_allocator_.reset(new EigenAllocator(allocator)); in EigenThreadPoolInfo()111 std::unique_ptr<EigenAllocator> eigen_allocator_;