Searched refs:NeedStaging (Results 1 – 1 of 1) sorted by relevance
256 bool NeedStaging(const Tensor* tensor) { in NeedStaging() function352 if (NeedStaging(cpu_tensor)) { in CopyCPUTensorToGPU()