xla
b8a18531 - Use tensor shape as output shape when XLA compiling.

Commit

7 years ago

Use tensor shape as output shape when XLA compiling.

References

#168 - Generate proper TPU friendly shape in case of 4D/TPU tensors.

Author

dlibenzi

dlibenzi

Committer

asuhan

asuhan

Parents

Loading