Searched refs:getTensorMemoryByDevice (Results 1 – 1 of 1) sorted by relevance
536 function getTensorMemoryByDevice(pickles) { function554 let sizes = getTensorMemoryByDevice(new Map([