Searched refs:all_tensors_cpu (Results 1 – 1 of 1) sorted by relevance
356 bool all_tensors_cpu = true; in wrap_tensor_node() local364 all_tensors_cpu = all_tensors_cpu && tensor_node.children(i).is_cpu(); in wrap_tensor_node()369 if (!(all_tensors_cpu && all_tensors_contiguous && in wrap_tensor_node()382 if (all_tensors_cpu && all_tensors_contiguous && all_tensors_same_dtype) { in wrap_tensor_node()