Searched refs:DeviceWrapper (Results 1 – 4 of 4) sorted by relevance
12 name: "DeviceWrapper"
1 path: "tensorflow.nn.rnn_cell.DeviceWrapper"3 is_instance: "<class \'tensorflow.python.ops.rnn_cell_impl.DeviceWrapper\'>"
1564 class DeviceWrapper(DeviceWrapperBase, _RNNCellWrapperV1): class1567 super(DeviceWrapper, self).__init__(*args, **kwargs)
2865 [rnn_cell_impl.DeviceWrapper, rnn_cell_impl.DeviceWrapperV2])2900 cell = rnn_cell_impl.DeviceWrapper(rnn_cell_impl.GRUCell(3), gpu_dev)