Searched refs:from_dim_sizes (Results 1 – 2 of 2) sorted by relevance
93 expected = RaggedTensorDynamicShape.from_dim_sizes(expected_dim_sizes)113 shape = RaggedTensorDynamicShape.from_dim_sizes(dim_sizes)114 expected = RaggedTensorDynamicShape.from_dim_sizes(expected_dim_sizes)303 original_shape = RaggedTensorDynamicShape.from_dim_sizes(original_dim_sizes)304 bcast_shape = RaggedTensorDynamicShape.from_dim_sizes(broadcast_dim_sizes)373 x_shape = RaggedTensorDynamicShape.from_dim_sizes(x_dims)374 y_shape = RaggedTensorDynamicShape.from_dim_sizes(y_dims)375 expected = RaggedTensorDynamicShape.from_dim_sizes(expected_dims)382 shape = RaggedTensorDynamicShape.from_dim_sizes([2, (2, 1), 2, 1])421 shape = RaggedTensorDynamicShape.from_dim_sizes(dim_sizes)
132 def from_dim_sizes(dim_sizes): member in RaggedTensorDynamicShape