Searched refs:BuildPartitionId (Results 1 – 3 of 3) sorted by relevance
1244 xla::internal::XlaBuilderFriend::BuildPartitionId(ctx.builder, shape); in ExportXlaOp()
71 static XlaOp BuildPartitionId(XlaBuilder* builder, const Shape& shape);
152 XlaOp XlaBuilderFriend::BuildPartitionId(XlaBuilder* builder, in BuildPartitionId() function in xla::internal::XlaBuilderFriend