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