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