Searched defs:inner_dim_sizes (Results 1 – 1 of 1) sorted by relevance
85 def __init__(self, partitioned_dim_sizes, inner_dim_sizes): argument216 def inner_dim_sizes(self): member in RaggedTensorDynamicShape