Searched refs:rhs_tf_shape (Results 1 – 1 of 1) sorted by relevance
98 TensorShape rhs_tf_shape; in BroadcastOpsToSame() local100 TF_RETURN_IF_ERROR(XLAShapeToTensorShape(rhs_xla_shape, &rhs_tf_shape)); in BroadcastOpsToSame()101 if (!lhs_tf_shape.IsSameSize(rhs_tf_shape)) { in BroadcastOpsToSame()102 BCast bcast(BCast::FromShape(lhs_tf_shape), BCast::FromShape(rhs_tf_shape)); in BroadcastOpsToSame()