Contextos del dispositivo TensorFlow, flujos y cambio de contexto
En elGPUDevice código, me di cuenta de que unoGPUDeviceContext se hace por secuencia.
Es el propósito de esto para que cada contexto pueda controlar un OpKernelContext y luego, a medida que se necesiten ejecutar varias secuencias, entonces los contextos solo se pueden cambiar, lo que maneja empujar diferentes datos / códigos en la GPU y luego ejecutarlos.
¿Las distintas transmisiones se registran como dispositivos diferentes (es decir, '/ gpu: 0' y '/ gpu: 1')?
Poresta, ThreadPoolDevice no tienen contextos, pero si tuviera que agregar contextos en ThreadPoolDevice, ¿encajarían mejor como una especie de ThreadContext?