flax
Add data_transform argument to nn.scan and preserve PartitionSpec attributes.
#1411

Merged

Add data_transform argument to nn.scan and preserve PartitionSpec attributes. #1411

copybara-service merged 1 commit into master from test_383161650

google-cla added cla: yes

copybara-service force pushed 4 years ago

Add data_transform argument to nn.scan and preserve PartitionSpec att…

5ffd0006

copybara-service force pushed to 5ffd0006 4 years ago

copybara-service merged 5ffd0006 into master 4 years ago

copybara-service deleted the test_383161650 branch 4 years ago

Reviewers

No reviews

Assignees

No one assigned

Labels

cla: yes

Milestone

No milestone