Searched defs:partition_shape (Results 1 – 3 of 3) sorted by relevance
54 TensorShape partition_shape({2, 2}); in TEST_F() local
72 TensorShapeProto partition_shape; in ValidateRowPartitionTypesAndShapes() local
189 auto partition_shape = shape; in MakeNonPaddedShapeForGivenPartition() local