Searched defs:partition_shape (Results 1 – 4 of 4) sorted by relevance
54 TensorShape partition_shape({2, 2}); in TEST_F() local
72 TensorShapeProto partition_shape; in ValidateRowPartitionTypesAndShapes() local
112 xla::Shape partition_shape = ctx->InputXlaShape(1).ConsumeValueOrDie(); in Compile() local
189 auto partition_shape = shape; in MakeNonPaddedShapeForGivenPartition() local