Searched defs:isCuda (Results 1 – 3 of 3) sorted by relevance
144 bool types::isCuda(ID Id) { in isCuda() function in types
924 bool isCuda = outputTensors[0][0].device().is_cuda(); in allgather() local1420 bool isCuda = inputTensors[0][0].device().is_cuda(); in reduce_scatter() local
214 void isCuda(Stack& stack) { in isCuda() function