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