Searched refs:BroadcastOpsToSame (Results 1 – 4 of 4) sorted by relevance
34 Status BroadcastOpsToSame(xla::XlaOp* lhs, xla::XlaOp* rhs);
94 Status BroadcastOpsToSame(xla::XlaOp* lhs, xla::XlaOp* rhs) { in BroadcastOpsToSame() function
36 OP_REQUIRES_OK(ctx, BroadcastOpsToSame(&lhs, &rhs)); in Compile()
182 OP_REQUIRES_OK(ctx, BroadcastOpsToSame(&logits, &labels)); in Compile()