Searched refs:print_flat_alloc (Results 1 – 2 of 2) sorted by relevance
186 void CudaPrinter::print_flat_alloc(const AllocatePtr& alloc) { in print_flat_alloc() function in torch::jit::tensorexpr::CudaPrinter207 print_flat_alloc(v); in visit()213 print_flat_alloc(v); in visit()
192 void print_flat_alloc(const AllocatePtr& alloc);