Pjrt device attributes #4403
Add GetDeviceAttributes for pjrt
7f36d628
revert mnist test
0f211139
fix lint issues
1343c6b2
Add DeviceAttirbutes data type for cleanness
d9b00a4f
remove comment
ca188bdc
add unit test for device attributes
b54ab8ac
fix spacing
32d9b651
lint fixes
39720c3e
fix lint
d1d0676d
lint fixes
c3aca00c
lint fixes
d704f9c3
Fix type issues and return flat_has_map
2cfab5dd
use StringToPjRtDevice while getting device attributes
0c1b6010
darisoy
merged
b97f05b3
into master 2 years ago
darisoy
deleted the pjrt-device-attributes branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub