xla
0bd6ca06 - [Dynamo] Initialize data from CPU then move to TPU for `DynamoInferenceBasicTest.test_resnet18`

Commit
2 years ago
[Dynamo] Initialize data from CPU then move to TPU for `DynamoInferenceBasicTest.test_resnet18` For `DynamoInferenceBasicTest.test_resnet18`, input data initialized from TPU then copy to CPU has been changed, so change logic to initialize input data from CPU then move to TPU to confirm test with the same input data
Author
Parents
Loading