Handle None dimensions when converting to jax.ShapeDtypeStruct. - #3496
Open
copybara-service[bot] wants to merge 1 commit into
Open
Handle None dimensions when converting to jax.ShapeDtypeStruct.#3496copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into