Searched refs:partition_shape (Results 1 – 2 of 2) sorted by relevance
54 TensorShape partition_shape({2, 2}); in TEST_F() local56 {partition_shape, partition_shape, partition_shape}); in TEST_F()
218 partition_shape = partition.get_shape()219 maybe_partitioned_axis = np.less(partition_shape,