Skip to content
Commit 6b1d7f51 authored by Logan Chien's avatar Logan Chien Committed by A. Unique TensorFlower
Browse files

Add TensorRankOf for ranked tensor types with specific ranks 

This commit adds `TensorRankOf<types, typeNames, ranks>` to specify ranked
tensor types with the specified types and ranks.  For example,
`TensorRankOf<[I32, F32], ["i32", "F32"], [0, 1]>` matches `tensor<i32>`,
`tensor<?xi32>`, `tensor<f32>`, or `tensor<?xf32>`.

PiperOrigin-RevId: 266461256
parent 14075705
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment